File: //usr/local/lib/python3.7/test/__pycache__/test_secrets.cpython-37.pyc
B
��g � @ sf d Z ddlZddlZddlZG dd� dej�ZG dd� dej�ZG dd� dej�Zed krbe� � 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 }| � t�||�� | � t�|�d�|�d��� qW d S )N)�aZbcdZxyz123�d zutf-8)�
assertTrue�secrets�compare_digest�encode)�self�sr �b� r �-/usr/local/lib/python3.7/test/test_secrets.py�
test_equal s
z Compare_Digest_Tests.test_equalc C s~ | � t�dd�� | � t�dd�� xTdD ]L}|d d }|d d }| � t�||�� | � t�|�d �|�d ��� q*W d S )
N�abcZabcds abcs abcd)�xZmnZa1b2c3r �q�kzutf-8)Z assertFalser r r )r r
r r r r r
� test_unequal s
z!Compare_Digest_Tests.test_unequalc C sR d}|� d�}t|t�s t�t|t�s*t�| �ttj||� | �ttj||� d S )NZabcdezutf-8) r �
isinstance�str�AssertionError�bytes� assertRaises� TypeErrorr r )r r r r r r
�test_bad_types$ s
z#Compare_Digest_Tests.test_bad_typesc C s, | � t�dd�t� | � t�dd�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 ]:}t�|�}| �d| ko@d| k n |||f � qW q
W d S )Nzrandbits(%d) returned %d)� � � � r � )�ranger Zrandbitsr )r �errmsgZnumbits�i�nr r r
�
test_randbits6 s
zRandom_Tests.test_randbitsc C s< dddddddg}x$t d�D ]}| �t�|�|k� q W d S ) N� r' � � � � �@ �
)r( r r �choice)r �itemsr* r r r
� test_choice> s zRandom_Tests.test_choicec C sN x(t dd�D ]}| �t�|�t |�� q W | �ttjd� | �ttjd� d S )Nr' r3 r ���)r( ZassertInr Z randbelowr �
ValueError)r r* r r r
�test_randbelowD s zRandom_Tests.test_randbelowN)r r r r! r, r6 r9 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 | �d| � Y nX y |d � W n" tk
r� | �d| � Y nX W d Q R X qW t j}| � t
t �d ��|� | � t
t �d ��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��. | �t�|�t� | �tt�|��|� W d Q R X qW d S )N)r- r/ � r )r+ )r? r r r< r r@ rA )r r+ r r r
�test_token_bytesa s
z Token_Tests.test_token_bytesc
C sl xfdD ]^}| j |d��H t�|�}| �|t� | �t|�d| � | �tdd� |D ��� W d Q R X qW d S )N)r- 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 r= r r r@ rA r �all)r r+ r
r r r
�test_token_hexh s
zToken_Tests.test_token_hexc
sl t jt j d � xVdD ]N}| j|d��8 t�|�}| �|t� | �t � 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>)
rI Z
ascii_letters�digitsr? r r> r r r rM )r r+ r
r )rR r
�test_token_urlsafeq s
z Token_Tests.test_token_urlsafeN)r r r r! rD rF rN rT r r r r
r: L s
r: �__main__)
r! r ZunittestrI ZTestCaser r" r: r �mainr r r r
�<module> s $/