Your IP : 216.73.216.60
3
� f � @ sf d Z ddlZddlZddlZG dd� dej�ZG dd� dej�ZG dd� dej�Zed krbej � dS )
z�Test the secrets module.
As most of the functions in secrets are thin wrappers around functions
defined elsewhere, we don't need to test them exhaustively.
� Nc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �Compare_Digest_Testsz%Test secrets.compare_digest function.c C sR xLdD ]D}|d }|d }| j tj||�� | j tj|jd�|jd��� qW d S )N�a�bcd�xyz123�d zutf-8)r r r )�
assertTrue�secrets�compare_digest�encode)�self�sr �b� r �6/opt/alt/python36/lib64/python3.6/test/test_secrets.py�
test_equal s
z Compare_Digest_Tests.test_equalc C s~ | j tjdd�� | j tjdd�� xTd D ]L}|d d }|d d
}| j tj||�� | j tj|jd �|jd ��� q*W d S )
N�abcZabcds abcs abcd�x�mn�a1b2c3r �q�kzutf-8)r r r )Z assertFalser r r
)r r r r
r r r � test_unequal s
z!Compare_Digest_Tests.test_unequalc C s6 d}|j d�}| jttj||� | jttj||� d S )NZabcdezutf-8)r
� assertRaises� TypeErrorr r )r r r
r r r �test_bad_types$ s
z#Compare_Digest_Tests.test_bad_typesc C s, | j tjdd�t� | j tjdd�t� d S )Nr Zxyz)�assertIsInstancer r �bool)r r r r � test_bool- s z Compare_Digest_Tests.test_boolN)�__name__�
__module__� __qualname__�__doc__r r r r r r r r r s
r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � Random_Testsz*Test wrappers around SystemRandom methods.c C s^ d}xTdD ]L}xFt d�D ]:}tj|�}| jd| ko@d| k n |||f � qW q
W d S ) Nzrandbits(%d) returned %d� � � � r � )r# r$ r% )�ranger Zrandbitsr )r �errmsgZnumbits�i�nr r r �
test_randbits6 s
zRandom_Tests.test_randbitsc C s< dddddddg}x$t d�D ]}| jtj|�|k� q W d S ) N� r' � � � � �@ �
)r( r r Zchoice)r �itemsr* r r r � test_choice> s zRandom_Tests.test_choicec C sN x(t dd�D ]}| jtj|�t |�� q W | jttjd� | jttjd� d S )Nr' r3 r r- ���)r( ZassertInr Z randbelowr �
ValueError)r r* r r r �test_randbelowD s zRandom_Tests.test_randbelowN)r r r r! r, r5 r8 r r r r r" 3 s r" c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � Token_TestszTest token functions.c C s� x�t jt jt jfD ]�}| j|d��j |j}y
|� W n" tk
rV | jd| � Y nX y |d � W n" tk
r� | jd| � Y nX W d Q R X qW t j}| j t
t jd ��|� | j t
t jd ��d| � d S )N)�funcz$%s cannot be called with no argumentz %s cannot be called with Noner' ) r � token_bytes� token_hex�
token_urlsafe�subTestr r ZfailZDEFAULT_ENTROPY� assertEqual�len)r r: �name�sizer r r �test_token_defaultsO s
"z Token_Tests.test_token_defaultsc C sR xLdD ]D}| j |d��. | jtj|�t� | jttj|��|� W d Q R X qW d S )Nr- r/ � r )r+ )r- r/ rD r )r> r r r; �bytesr? r@ )r r+ r r r �test_token_bytesa s
z Token_Tests.test_token_bytesc C sl xfd D ]^}| j |d��H tj|�}| j|t� | jt|�d| � | jtdd� |D ��� W d Q R X qW d S )
Nr- r$ � �Z )r+ r' c s s | ]}|t jkV qd S )N)�stringZ hexdigits)�.0�cr r r � <genexpr>o s z-Token_Tests.test_token_hex.<locals>.<genexpr>)r- r$ rG rH ) r> r r< r �strr? r@ r �all)r r+ r r r r �test_token_hexh s
zToken_Tests.test_token_hexc
sl t jt j d � xVd D ]N}| j|d��8 tj|�}| j|t� | jt � fdd�|D ��� W d Q R X qW d S )
Nz-_r- � � �L )r+ c 3 s | ]}|� kV qd S )Nr )rJ rK )�legalr r rL x s z1Token_Tests.test_token_urlsafe.<locals>.<genexpr>)r- rP rQ rR )
rI Z
ascii_lettersZdigitsr> r r= r rM r rN )r r+ r r )rS r �test_token_urlsafeq s
z Token_Tests.test_token_urlsafeN)r r r r! rC rF rO rT r r r r r9 L s
r9 �__main__)
r! r ZunittestrI ZTestCaser r" r9 r �mainr r r r �<module> s $/