File: //proc/thread-self/root/usr/local/lib/python3.7/test/__pycache__/test_crypt.cpython-37.opt-2.pyc
B
��g � @ sL d dl Z d dlmZ d dlZe�d�ZG dd� dej�ZedkrHe� � dS )� N)�support�cryptc @ sx e Zd Zdd� Zdd� Zdd� Zdd� Ze�e j
e j kpBe j e j kd �d
d � �Z
e�e je j kd �d
d� �Zdd� ZdS )�
CryptTestCasec C sR t � d�}t � d|�}| �||� t � dd�}|d k rNt � d|�}| �||� d S )N�
mypasswordZab)r � assertEqual)�self�cr�cr2� r
�+/usr/local/lib/python3.7/test/test_crypt.py�
test_crypt s
zCryptTestCase.test_cryptc
C sx | � ttj�d� x`tjD ]V}t�|�}| �t|�|j ddddddh� |jr| �|j|d t|�|j � � qW d S )N�@ r � � � � � ) r �lenr Z
_saltchars�methods�mksalt�assertIn�
salt_charsZident)r �method�saltr
r
r � test_salt s
"zCryptTestCase.test_saltc C sn xht jD ]^}t � d|�}| �t|�|j� t � d|�}| �||� t � dt �|��}| �t|�|j� qW d S )NZassword)r r r r Z
total_sizer )r r r r r
r
r �test_saltedcrypt s z CryptTestCase.test_saltedcryptc C sL | � ttj�dk� tj�d�r4| �tjtjg� n| �tjd tj � d S )Nr Zopenbsd���)
�
assertTruer r r �sys�platform�
startswithr �METHOD_BLOWFISH� METHOD_CRYPT)r r
r
r � test_methods# s zCryptTestCase.test_methodszrequires support of SHA-2c C s� x�t jt jfD ]�}xzdD ]r}t j||d�}| �d| |� | �t|�|j dtt|�� � t � d|�}| � |� t � d|�}| �||� qW qW d S )N)i� i' i�� )�roundsz $rounds=%d$� r )
r �
METHOD_SHA256�
METHOD_SHA512r r r r r �strr )r r r$ r r r r
r
r �test_sha2_rounds* s
z CryptTestCase.test_sha2_roundsz requires support of Blowfishc C s� x�t dd�D ]r}tjtjd|> d�}| �d| |� | �t|�tjj ddh� t�d|�}| �|� t�d|�}| �||� q W d S ) Nr r% r )r$ z$%02d$r r r ) �ranger r r! r r r r r )r Z
log_roundsr r r r
r
r �test_blowfish_rounds9 s
z"CryptTestCase.test_blowfish_roundsc C s x�t jt jt jfD ]�}| �t�� t j|dd� W d Q R X | �t�� t j|dd� W d Q R X x<ddddd> fD ](}| �t�� t j||d� W d Q R X qpW qW | �t�� t jt jdd� W d Q R X x:t jt j fD ]*}| �
td �� t j|d
d� W d Q R X q�W d S ) NZ4096)r$ g �@r r r i� i� r i ) r r& r' r! Z assertRaises� TypeErrorr �
ValueErrorr" Z
METHOD_MD5ZassertRaisesRegex)r r r$ r
r
r �test_invalid_roundsE s
z!CryptTestCase.test_invalid_roundsN)�__name__�
__module__� __qualname__r r r r# �unittestZ
skipUnlessr r& r r' r) r! r+ r. r
r
r
r r s
r �__main__)
r Ztestr r2 �
import_moduler ZTestCaser r/ �mainr
r
r
r �<module> s
O