File: //var/opt/nydus/ops/shortuuid/__pycache__/test_shortuuid.cpython-39.pyc
a
�,�h' � @ sL d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dlm
Z
d dl m Z d dl
mZ d dl
mZ d d l
mZ d d
l
mZ d d l
mZ d d l
mZ d d
l
mZ ej�d e j�ed �� G dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� dej�Z G dd� dej�Z G dd� dej�Z e dk�rHe� � dS ) � N)� defaultdict)�patch)�UUID)�uuid4)�cli)�decode)�encode)� get_alphabet)�random)� set_alphabet�� ShortUUID)�uuidz/../..c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �LegacyShortUUIDTestc C s� | � dtt� � k o dk n � | � dttd�� k oBdk n � | � dttd�� k ohdk n � | � dttd�� k o�dk n � d S �N� � zhttp://www.example.com/zHTTP://www.example.com/z example.com/)�
assertTrue�lenr )�self� r �I/opt/nydus/tmp/pip-target-wkfpz8uv/lib/python/shortuuid/test_shortuuid.py�test_generation s $&&z#LegacyShortUUIDTest.test_generationc C s t d�}| �t|�d� d S �Nz&{3b1f8b40-222c-4a6e-b77e-779d5a94e21c}�CXc85b4rqinB7s5J52TRYb)r � assertEqualr �r �ur r r �
test_encoding s z!LegacyShortUUIDTest.test_encodingc C s t d�}| �td�|� d S r )r r r r r r r �
test_decoding! s z!LegacyShortUUIDTest.test_decodingc C s� t � }d}t|� | �|t � � td� | �|t � � | �tt� �td�� | �dtt� � k ohdk n � t� }| �|tt |��� t� }| �|t t|��� | �
t td� | �
t td� t|� | �
t dd� d � d S )
N�01�01010101010101�t � �1�1111111c S s t | �S )Nr )�xr r r �<lambda>= � z3LegacyShortUUIDTest.test_alphabet.<locals>.<lambda>�0) r r r �setr r r r r r � assertRaises�
ValueError)r Zbackup_alphabet�alphabetr r r r �
test_alphabet% s $z!LegacyShortUUIDTest.test_alphabetc C s: | � tt� �d� tdd�D ]}| � tt|��|� q d S )N� � �d )r r r
�range)r �ir r r � test_random? s z LegacyShortUUIDTest.test_randomN)�__name__�
__module__� __qualname__r r r r. r4 r r r r r s
r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�ClassShortUUIDTestc C s� t � }| �dt|�� � k o$dk n � | �dt|�d�� k oLdk n � | �dt|�d�� k otdk n � | �dt|�d�� k o�dk n � d S r )r
r r r �r �sur r r r F s
&((z"ClassShortUUIDTest.test_generationc C s$ t � }td�}| �|�|�d� d S r )r
r r r �r r: r r r r r M s z ClassShortUUIDTest.test_encodingc C s$ t � }td�}| �|�d�|� d S r )r
r r r r; r r r r R s z ClassShortUUIDTest.test_decodingc C sR t � }td�D ]}| �t|�� �d� qtdd�D ]}| �t|�|��|� q2d S )N�� r/ r0 r1 )r
r2 r r r
)r r: r3 r r r r4 W s
z ClassShortUUIDTest.test_randomc C s d}t |�}t � }| �||�� � |�d� | �||�� � | �t|�� �td�� | �dt|�� � k ordk n � | �dt|�� � k o�dk n � t� }| �||� |�
|��� |�� }| �||�
|� |��� | � t |jd� | � t |jd� d S ) Nr r! r" r# r r r$ r% )
r
r r r r* r r r r r r r+ r, )r r- �su1�su2r r r r r. _ s
&&z ClassShortUUIDTest.test_alphabetc C s~ t � }| �|�� d� tjtj tj d }t |�}| �|�� d� d}t |�}| �|�� d� t � }| �|jdd�d� d S )Nr/ z+/r � � )� num_bytes� )r
r Zencoded_length�string�ascii_uppercase�ascii_lowercase�digits)r r= Zbase64_alphabetr> Zbinary_alphabetZsu3Zsu4r r r �test_encoded_lengthv s �z&ClassShortUUIDTest.test_encoded_lengthN) r5 r6 r7 r r r r4 r. rG r r r r r8 E s r8 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ShortUUIDPaddingTestc C sB t � }t� }tdd�}|�|�}|�|�}| �t|�t|�� d S �Nr )�int)r
r r r r r �r r: �
random_uidZ smallest_uid�encoded_randomZ
encoded_smallr r r � test_padding� s
z!ShortUUIDPaddingTest.test_paddingc C sR t � }t� }tdd�}|�|�}|�|�}| �|�|�|� | �|�|�|� d S rI )r
r r r r r rK r r r r � s
z"ShortUUIDPaddingTest.test_decodingc C s� t � }d}tt�}t|�D ]>}t� }|�|�}|t|� d7 < |�|�}| �||� q| �t|�d� t t
|� � ��}| �|| |� d S )Nr< r0 ) r
r rJ r2 r r r r r �next�iter�keys) r r: Znum_iterationsZ uid_lengths�countrL rM Zdecoded_randomZ
uid_lengthr r r �test_consistency� s
z%ShortUUIDPaddingTest.test_consistencyN)r5 r6 r7 rN r rS r r r r rH � s
rH c @ s e Zd Zdd� ZdS )�EncodingEdgeCasesTestc C sZ t � }| �t|jg � | �t|ji � | �t|jd� | �t|jd� | �t|jd� d S �N)� �* g E@)r
r+ r, r r r9 r r r �test_decode_dict� s z&EncodingEdgeCasesTest.test_decode_dictN�r5 r6 r7 rX r r r r rT � s rT c @ s e Zd Zdd� ZdS )�DecodingEdgeCasesTestc C sZ t � }| �t|jg � | �t|ji � | �t|jd� | �t|jd� | �t|jd� d S rU )r
r+ r, r r9 r r r rX � s z&DecodingEdgeCasesTest.test_decode_dictNrY r r r r rZ � s rZ c @ s< e Zd Zed�dd� �Zed�dd� �Zed�dd� �ZdS ) �CliTestzshortuuid.cli.printc C s2 t g � |�� |jd d }| �t|�d� d S )Nr r/ )r Z
assert_called� call_argsr r �r Z
mock_printZterminal_outputr r r �$test_shortuuid_command_produces_uuid� s z,CliTest.test_shortuuid_command_produces_uuidc C s* t ddg� |jd d }| �|d� d S )Nr �$3b1f8b40-222c-4a6e-b77e-779d5a94e21cr r �r r\ r r] r r r �test_encode_command� s zCliTest.test_encode_commandc C s* t ddg� |jd d }| �|d� d S )Nr r r r_ r` r] r r r �test_decode_command� s zCliTest.test_decode_commandN)r5 r6 r7 r r^ ra rb r r r r r[ � s
r[ �__main__)!�osrC �sys�unittest� collectionsr Z
unittest.mockr r r r Z
shortuuid.clir Zshortuuid.mainr r r r
r r
�path�insert�abspath�__file__ZTestCaser r8 rH rT rZ r[ r5 �mainr r r r �<module> s0 /D)