File: //usr/local/lib/python3.10/test/__pycache__/test_secrets.cpython-310.opt-1.pyc
o
�i � @ sj d Z ddlZddlZddlZG dd� dej�ZG dd� dej�ZG dd� dej�Zed kr3e� � dS 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 @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �Compare_Digest_Testsz%Test secrets.compare_digest function.c C sN dD ]"}|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.10/test/test_secrets.py�
test_equal s �z Compare_Digest_Tests.test_equalc C sz | � t�dd�� | � t�dd�� dD ]&}|d d }|d d }| � t�||�� | � t�|�d �|�d ��� qd 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 ]$}t d�D ] }t�|�}| �d| ko d| k n |||f � q
qd 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& r2 r ���)r' ZassertInr � randbelowr �
ValueError)r r) r r r �test_randbelowD s zRandom_Tests.test_randbelowN)r r r r r+ r5 r9 r r r r r! 3 s
r! c @ r ) � Token_TestszTest token functions.c
C s� t jt jt jfD ]H}| j|d��8 |j}z|� W n ty) | �d| � Y nw z|d � W n ty@ | �d| � Y nw W d � n1 sKw 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* �
� �� ��
z Token_Tests.test_token_defaultsc C sb dD ],}| j |d�� | �t�|�t� | �tt�|��|� W d � n1 s)w Y qd S )N)r, r. � r �r* )r? r r r< �bytesr@ rA )r r* r r r �test_token_bytesa s ���z Token_Tests.test_token_bytesc C s| dD ]9}| j |d��) t�|�}| �|t� | �t|�d| � | �tdd� |D ��� W d � n1 s6w Y qd S )N)r, r# � �Z rF r& c s s � | ]}|t jv V qd S �N)�string� hexdigits��.0�cr r r � <genexpr>o s � z-Token_Tests.test_token_hex.<locals>.<genexpr>) r? r r= r �strr@ rA r �all�r r* r
r r r �test_token_hexh s
���zToken_Tests.test_token_hexc s| t jt j d � dD ]1}| j|d��! t�|�}| �|t� | �t � fdd�|D ��� W d � n1 s6w Y q
d S )Nz-_)r, � � �L rF c 3 s � | ]}|� v V qd S rK r rN �Zlegalr r rQ x s � z1Token_Tests.test_token_urlsafe.<locals>.<genexpr>)
rL �
ascii_letters�digitsr? r r> r rR r rS rT r rY r �test_token_urlsafeq s
���z Token_Tests.test_token_urlsafeN)r r r r rD rH rU r\ r r r r r: L s r: �__main__)
r r ZunittestrL ZTestCaser r! r: r �mainr r r r �<module> s $/ �