File: //usr/local/lib/python3.7/test/__pycache__/test_secrets.cpython-37.opt-1.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 s6 d}|� d�}| �ttj||� | �ttj||� 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, | � 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# r/ r ���)r$ ZassertInr Z randbelowr �
ValueError)r r& r r r
�test_randbelowD s zRandom_Tests.test_randbelowN)r r r r r( r2 r5 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 r7 �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 r8 �bytesr< r= )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 r9 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 � 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 )rG rH )�legalr r
rI x s z1Token_Tests.test_token_urlsafe.<locals>.<genexpr>)
rF Z
ascii_letters�digitsr; r r: r rJ r rK )r r' r
r )rP r
�test_token_urlsafeq s
z Token_Tests.test_token_urlsafeN)r r r r r@ rC rL rR r r r r
r6 L s
r6 �__main__)
r r ZunittestrF ZTestCaser r r6 r �mainr r r r
�<module> s $/