U
    Mf	                     @   sV   d Z ddlZddlmZ G dd dejZi fddZedkrRd	d
 Zej	d
d dS )z)Self-tests for Cryptodome.IO._PBES module    N)PBES2c                   @   sd   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd ZdS )	TestPBES2c                 C   s   d| _ d| _d S )Ns	   Test datas
   Passphrase)ref
passphrase)self r   D/tmp/pip-unpacked-wheel-l_0d1exj/Cryptodome/SelfTest/IO/test_PBES.pysetUp+   s    zTestPBES2.setUpc                 C   s2   t | j| jd}t || j}| | j| d S )Nz"PBKDF2WithHMAC-SHA1AndDES-EDE3-CBCr   Zencryptr   r   ZdecryptassertEqualr   ctptr   r   r   test1/   s
    zTestPBES2.test1c                 C   s2   t | j| jd}t || j}| | j| d S )Nz"PBKDF2WithHMAC-SHA224AndAES128-CBCr
   r   r   r   r   test25   s
    zTestPBES2.test2c                 C   s2   t | j| jd}t || j}| | j| d S )Nz"PBKDF2WithHMAC-SHA256AndAES192-CBCr
   r   r   r   r   test3;   s
    zTestPBES2.test3c                 C   s2   t | j| jd}t || j}| | j| d S )Nz"PBKDF2WithHMAC-SHA384AndAES256-CBCr
   r   r   r   r   test4A   s
    zTestPBES2.test4c                 C   s2   t | j| jd}t || j}| | j| d S )Nz"PBKDF2WithHMAC-SHA512AndAES128-GCMr
   r   r   r   r   test5G   s
    zTestPBES2.test5c                 C   s2   t | j| jd}t || j}| | j| d S )Nz&PBKDF2WithHMAC-SHA512-224AndAES192-GCMr
   r   r   r   r   test6M   s
    zTestPBES2.test6c                 C   s2   t | j| jd}t || j}| | j| d S )Nz$PBKDF2WithHMAC-SHA3-256AndAES256-GCMr
   r   r   r   r   test7S   s
    zTestPBES2.test7c                 C   s2   t | j| jd}t || j}| | j| d S )NzscryptAndAES128-CBCr
   r   r   r   r   test8Y   s
    zTestPBES2.test8c                 C   s2   t | j| jd}t || j}| | j| d S )NzscryptAndAES192-CBCr
   r   r   r   r   test9_   s
    zTestPBES2.test9c                 C   s2   t | j| jd}t || j}| | j| d S )NzscryptAndAES256-CBCr
   r   r   r   r   test10e   s
    zTestPBES2.test10N)__name__
__module____qualname__r	   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   )   s   r   c                 C   s    ddl m} g }||t7 }|S )Nr   )list_test_cases)ZCryptodome.SelfTest.st_commonr   r   )configr   Z	listTestsr   r   r   	get_testsl   s    r   __main__c                   C   s   t t S )N)unittestZ	TestSuiter   r   r   r   r   suitet   s    r!   )ZdefaultTest)
__doc__r    ZCryptodome.IO._PBESr   ZTestCaser   r   r   r!   mainr   r   r   r   <module>"   s   C