shell bypass 403
o
a b � @ 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/python-3.10/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 sR d}|� d�}t|t�sJ �t|t�sJ �| �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 ]$}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) r5 r ���)r* ZassertInr � randbelowr �
ValueError)r r, r r r �test_randbelowD s zRandom_Tests.test_randbelowN)r r! r" r# r. r8 r< 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/ r1 � r �r- )rB r r r? r rC rD )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 rI 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>) rB r r@ r r rC rD 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 rI c 3 s � | ]}|� v V qd S rM r rP �Zlegalr r rS x s � z1Token_Tests.test_token_urlsafe.<locals>.<genexpr>)
rN �
ascii_letters�digitsrB r rA r r r rT rU r rZ r �test_token_urlsafeq s
���z Token_Tests.test_token_urlsafeN)r r! r" r# rG rJ rV r] r r r r r= L s r= �__main__)
r# r ZunittestrN ZTestCaser r$ r= r �mainr r r r �<module> s $/ �