Your IP : 216.73.216.85
U
i�f � @ 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 sN dD ]D}|d }|d }| � t�||�� | � t�|�d�|�d��� qd S )N)�aZbcdZxyz123�d �utf-8)�
assertTrue�secrets�compare_digest�encode��self�sr �b� r �6/opt/alt/python38/lib64/python3.8/test/test_secrets.py�
test_equal s
z Compare_Digest_Tests.test_equalc C sz | � t�dd�� | � t�dd�� dD ]L}|d d }|d d }| � t�||�� | � t�|�d �|�d ��� q(d S )
N�abcZabcds abcs abcd)�xZmnZa1b2c3r �q�kr )Z assertFalser r r r
r r r � test_unequal s z!Compare_Digest_Tests.test_unequalc C s6 d}|� d�}| �ttj||� | �ttj||� d S )NZabcder )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 sV d}dD ]H}t d�D ]:}t�|�}| �d| ko<d| k n |||f � qqd S )Nzrandbits(%d) returned %d)� � � � r � )�ranger Zrandbitsr )r �errmsgZnumbits�i�nr r r �
test_randbits6 s
zRandom_Tests.test_randbitsc C s8 dddddddg}t d�D ]}| �t�|�|k� qd S ) N� r% � � � � �@ �
)r& r r Zchoice)r �itemsr( r r r � test_choice> s zRandom_Tests.test_choicec C sJ t dd�D ]}| �t�|�t |�� q
| �ttjd� | �ttjd� d S )Nr% r1 r ���)r& ZassertInr Z randbelowr �
ValueError)r r( r r r �test_randbelowD s zRandom_Tests.test_randbelowN)r r r r r* r3 r6 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� t jt jt jfD ]�}| j|d��j |j}z
|� W n" tk
rT | �d| � Y nX z |d � W n" tk
r� | �d| � Y nX W 5 Q R X qt 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 r8 �name�sizer r r �test_token_defaultsO s �
z Token_Tests.test_token_defaultsc
C sN dD ]D}| j |d��. | �t�|�t� | �tt�|��|� W 5 Q R X qd S )N)r+ r- � r �r) )r< r r r9 �bytesr= r> )r r) r r r �test_token_bytesa s z Token_Tests.test_token_bytesc
C sh dD ]^}| j |d��H t�|�}| �|t� | �t|�d| � | �tdd� |D ��� W 5 Q R X qd S )N)r+ r" � �Z rC 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 �strr= r> r �all�r r) r r r r �test_token_hexh s
zToken_Tests.test_token_hexc
sh t jt j d � dD ]N}| j|d��8 t�|�}| �|t� | �t � fdd�|D ��� W 5 Q R X qd S )Nz-_)r+ � � �L rC c 3 s | ]}|� kV qd S rH r rJ �Zlegalr r rM x s z1Token_Tests.test_token_urlsafe.<locals>.<genexpr>)
rI Z
ascii_lettersZdigitsr< r r; r rN r rO rP r rU r �test_token_urlsafeq s
z Token_Tests.test_token_urlsafeN)r r r r rA rE rQ rV r r r r r7 L s
r7 �__main__)
r r ZunittestrI ZTestCaser r r7 r �mainr r r r �<module> s $/