Sindbad~EG File Manager
a
��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 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 �-/usr/local/lib/python3.9/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 sR d}|� d�}t|t�s J �t|t�s*J �| �ttj||� | �ttj||� d S )NZabcder )r �
isinstance�str�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 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 s. g d�}t d�D ]}| �t�|�|v � qd S )N)� r( � � � � �@ �
)r) r r �choice)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( r4 r ���)r) ZassertInr � randbelowr �
ValueError)r r+ r r r �test_randbelowD s zRandom_Tests.test_randbelowN)r r r! r" r- r7 r; 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��p |j}z
|� W n tyR | �d| � Y n0 z |d � W n ty� | �d| � Y n0 W d � q1 s�0 Y 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 r= �name�sizer r r �test_token_defaultsO s �
4z Token_Tests.test_token_defaultsc C sb dD ]X}| j |d��8 | �t�|�t� | �tt�|��|� W d � q1 sR0 Y qd S )N)r. r0 � r �r, )rA r r r> r rB rC )r r, r r r �test_token_bytesa s z Token_Tests.test_token_bytesc C s| dD ]r}| j |d��R t�|�}| �|t� | �t|�d| � | �tdd� |D ��� W d � q1 sl0 Y qd S )N)r. r% � �Z rH r( c s s | ]}|t jv V qd S �N)�string� hexdigits��.0�cr r r � <genexpr>o � z-Token_Tests.test_token_hex.<locals>.<genexpr>) rA r r? r r rB rC r �all�r r, r r r r �test_token_hexh s
zToken_Tests.test_token_hexc s| t jt j d � dD ]b}| j|d��B t�|�}| �|t� | �t � fdd�|D ��� W d � q1 sl0 Y qd S )Nz-_)r. � � �L rH c 3 s | ]}|� v V qd S rL r rO �Zlegalr r rR x rS z1Token_Tests.test_token_urlsafe.<locals>.<genexpr>)
rM �
ascii_letters�digitsrA r r@ r r r rT rU r rZ r �test_token_urlsafeq s
z Token_Tests.test_token_urlsafeN)r r r! r" rF rI rV r] r r r r r< L s
r< �__main__)
r" r ZunittestrM ZTestCaser r# r< r �mainr r r r �<module> s $/
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists