File: //usr/local/lib/python3.7/test/__pycache__/test_secrets.cpython-37.opt-2.pyc
B
��g � @ sb d dl Z d dlZd dlZG dd� dej�ZG dd� dej�ZG dd� dej�Zedkr^e�� dS ) � Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�Compare_Digest_Testsc 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__r r r r r r r r
r s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� Random_Testsc 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' r1 r4 r r r r
r 3 s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� Token_Testsc
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 r6 �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 r7 �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 r8 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 )rF rG )�legalr r
rH x s z1Token_Tests.test_token_urlsafe.<locals>.<genexpr>)
rE Z
ascii_letters�digitsr: r r9 r rI r rJ )r r& r
r )rO r
�test_token_urlsafeq s
z Token_Tests.test_token_urlsafeN)r r r r? rB rK rQ r r r r
r5 L s r5 �__main__) r ZunittestrE ZTestCaser r r5 r �mainr r r r
�<module> s $/