Current File : //lib/python3.9/site-packages/passlib/tests/__pycache__/test_apache.cpython-39.opt-1.pyc
a
f�Wc�r � @ s d Z ddlmZ ddlmZ ddlZddlZddlmZm Z ddl
m Z ddl m
Z
ddlmZ dd lmZmZmZmZ dd
l mZ dd lmZ dd lmZ ee�Zd dd�Z ej � d�p�dZ d dd�Z dd� Z!dd� Z"e"� \Z#Z$e�%e#d�Z&G dd� de�Z'G dd � d e�Z(dS ) z@tests for passlib.apache -- (c) Assurance Technologies 2008-2011� )�with_statement)� getLoggerN)�apache�registry)�MissingBackendError)�irange)�unittest)�TestCase�get_file�set_file�ensure_mtime_changed)�u)�to_bytes)�to_unicode_for_identify�
c C s0 t j�| �}t j�| �| }t �| ||f� dS )z)backdate file's mtime by specified amountN)�os�path�getatime�getmtime�utime)r �offsetZatime�mtime� r �=/usr/lib/python3.9/site-packages/passlib/tests/test_apache.py�backdate_file_mtime s r ZPASSLIB_TEST_HTPASSWD_PATHZhtpasswdc C sb |dur|� d�}tjtg| tjtj|r0tjndd�}|�|�\}}|�� }t|pVd�}||fS )z$
helper to run htpasswd cmd
N�utf-8)�stdout�stderr�stdin� ) �encode�
subprocess�Popen�
htpasswd_path�PIPEZSTDOUTZ communicate�waitr )�argsr �proc�out�err�rcr r r �_call_htpasswd+ s
� r+ c C s t d| |g|�\}}| S )z%
wrapper for htpasswd verify
z-vi)r+ )r �user�passwordr( r* r r r �_call_htpasswd_verify9 s r. c C sD zt g �\} }W n ty$ Y dS 0 |s4t�d� d| v }d|fS )z0
helper to check if htpasswd is present
)FFz htpasswd test returned with rc=0z -B T)r+ �OSError�logZwarning)r( r* Z have_bcryptr r r �_detect_htpasswdA s
r1 z requires `htpasswd` cmdline toolc @ s� e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
d � Z d d
� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Z d,d-� Z e d.d/� �Z e e �!e"�#d0�d1�d2d3� ��Z$d4S )5�HtpasswdFileTestztest HtpasswdFile class� HtpasswdFilest user2:2CHkkwa2AtqGs
user3:{SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=
user4:pass4
user1:$apr1$t4tc7jTh$GPIWVUo8sQKJlUdV8V5vu0
s4 user3:{SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=
user4:pass4
sy user2:pass2x
user3:{SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=
user4:pass4
user1:$apr1$t4tc7jTh$GPIWVUo8sQKJlUdV8V5vu0
user5:pass5
s useræ:2CHkkwa2AtqGs
s user�:2CHkkwa2AtqGs
s user1:pass1
user1:pass2
s user2:2CHkkwa2AtqGs
user3:{SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=
user4:pass4
user1:$apr1$t4tc7jTh$GPIWVUo8sQKJlUdV8V5vu0
user5:$2a$12$yktDxraxijBZ360orOyCOePFGhuis/umyPNJoL5EbsLk.s6SWdrRO
user6:$5$rounds=110000$cCRp/xUUGVgwR4aP$p0.QKFS5qLNRqw1/47lXYiAcgIjJK.WjCO8nrEKuUK.
c C s* | � � }t|| j� t�|�}| �|�� | j� | �|j|� | �|j � |d |_| �|j|d � | �
|j � tj|dd�}| �|�� d� | �|j|� | �
|j � | � d�� tj|dd�}W d � n1 s�0 Y | �|�� d� | �|j|� | �
|j � t �
|� | �ttj|� dS ) �test constructor autoload�xT��new� z ``autoload=False`` is deprecatedF)ZautoloadN)�mktempr � sample_01r r3 � assertEqual� to_stringr �
assertTruer � assertFalse�assertWarningListr �remove� assertRaises�IOError��selfr �htr r r � test_00_constructor_autoload} s(
,
z-HtpasswdFileTest.test_00_constructor_autoloadc C sP | � � }t|| j� tj�|�}| �|�� | j� | �|jd � | � |j
� d S �N) r9 r r: r r3 Z from_pathr; r< r r> r rC r r r �test_00_from_path� s z"HtpasswdFileTest.test_00_from_pathc C sd t j�| j�}| �|�d�� | �|�d�� | �|�d�� | �|�� | j � | �
t |jd� dS )�
test delete()�user1�user2�user5�user:N) r r3 � from_stringr: r= �deleter> r; r< � sample_02rA �
ValueError�rD rE r r r �test_01_delete� s z HtpasswdFileTest.test_01_deletec C sf | � � }d}t||� t�|�}|�d� | �t|�|� tj|dd�}|�d� | �t|�d� d S )Ns user1:pass1
user2:pass2
rJ T)�autosaves user2:pass2
)r9 r r r3 rO r; r
�rD r ZsamplerE r r r �test_01_delete_autosave� s
z(HtpasswdFileTest.test_01_delete_autosavec C s t jj| jdd�}| �|�dd�� | �|�dd�� | �|�� | j � | �
d��" t jj| jdd�}W d � n1 sz0 Y | �|�dd�� | �|�dd�� | �|�� | j � | � t |jd
d � | �
d �� |�
dd
� W d � n1 s�0 Y | �|�dd
�� d S )ztest set_password()� plaintext��default_schemerK �pass2xrL �pass5z``default`` is deprecated)�defaultNrM �pass�update\(\) is deprecated�test)r r3 rN r: r= � set_passwordr> r; r< � sample_03r? rA rQ �update�check_passwordrR r r r �test_02_set_password� s � 0 *z%HtpasswdFileTest.test_02_set_passwordc C sl | � � }d}t||� t�|�}|�dd� | �t|�|� tj|ddd�}|�dd� | �t|�d� d S )N� user1:pass1
rJ �pass2rW T)rY rT s user1:pass2
)r9 r r r3 r` r; r
rU r r r � test_02_set_password_autosave� s
z.HtpasswdFileTest.test_02_set_password_autosavec C s� dd� }| � |d�d� | � |d�d� | �t|d� | � |d�tjd � | � |d�tjd � | � |d�tjd � | � |d �tjd � d S )
z%test set_password() -- default_schemec S s* t j| d�}|�dd� |j�|�d��S )NrX rJ �pass1)r r3 r` �contextZidentify�get_hash)�schemerE r r r �check� s zCHtpasswdFileTest.test_02_set_password_default_scheme.<locals>.checkZ sha256_cryptZ des_cryptZxxxZportableZportable_apache_22�host_apache_22N)r; rA �KeyErrorr �htpasswd_defaults)rD rl r r r �#test_02_set_password_default_scheme� s z4HtpasswdFileTest.test_02_set_password_default_schemec C sL t j�| j�}|�dd� |�d� |�dd� | �t|�� �g d�� dS )� test users()rL r[ �user3�pass3�rJ rK rr �user4rL N) r r3 rN r: r` rO r; �sorted�usersrR r r r �
test_03_users� s
z HtpasswdFileTest.test_03_usersc C s t j�| j�}| �t|jdd� | �|�dd�du � tdd�D ]f}t |�}z8| �|�d| d| �� | �|�d| d�du � W q@ t
y� |d kr�Y q@� Y q@0 q@| �t |jd
d� | � d gd ��4 | �|�
d
d�� | �|�
d
d�� W d � n1 �s0 Y dS )�test check_password()� Zpass9Zuser9N� r, r] F�5rM �verify\(\) is deprecated� rJ rh rf )r r3 rN � sample_05rA � TypeErrorrc r= r �strr rQ r? �verifyr> �rD rE �ir r r �test_04_check_password s
z'HtpasswdFileTest.test_04_check_passwordc C s | � � }t|d� t|d� tj|dd�}| �|�� d� |�dd� |�� | �|�� d� t|| j � |�� | �|�� | j � |�d d
� |�
� | �|�� | j � t�� }| � t |j
� | � t |j� t|| j
� t�� }|�
|� | �|�dd�� d S ) � test load()r � rW rX r8 rJ rh re rL r[ N)r9 r r r r3 r; r< r` �load_if_changedr: �loadrA � RuntimeError�
sample_dupr= rc �rD r Zha�hbZhcr r r � test_05_load s*
z HtpasswdFileTest.test_05_loadc C s� | � � }t|| j� t�|�}|�d� |�d� |�� | �t|�| j � tjdd�}|�
dd� | � t |j� |�|� | �t|�d� dS )� test save()rJ rK rW rX rh re N)
r9 r r: r r3 rO �saver; r
rP r` rA r� �rD r rE r� r r r � test_06_saveA s
z HtpasswdFileTest.test_06_savec C s� | j ttjdd� tjj| jddd�}| �|�� td�g� | � d��" tjj| jdd�}W d � n1 sn0 Y | �|�� d g� tjj| j
d
dd�}| �|�� td�g� dS ) ztest 'encoding' kwd�utf-16��encodingr T�r� Zreturn_unicode� useræz ``encoding=None`` is deprecatedNs useræ�latin-1) rA rQ r r3 rN �sample_04_utf8r; rw r
r? �sample_04_latin1rR r r r �test_07_encodingsW s � 0
�z"HtpasswdFileTest.test_07_encodingsc C s� t j�| j�}| �|�d�d� | �|�d�d� | �|�d�d� | �d��" | �|�d�d� W d � n1 sv0 Y dS )�test get_hash()rr s! {SHA}3ipNV1GrBtxPmHFC21fCbVCSXIo=ru s pass4rL N�find\(\) is deprecated)r r3 rN r: r; rj r? �findrR r r r �test_08_get_hashk s z!HtpasswdFileTest.test_08_get_hashc C s< t j�| j�}| �|�� | j� t �� }| �|�� d� dS )ztest to_stringr8 N)r r3 rN r: r; r< rR r r r �test_09_to_stringu s z"HtpasswdFileTest.test_09_to_stringc C s t jddddd�}t|� d S )NZfakepathTr� )rT r7 r� )r r3 �reprrR r r r � test_10_repr� s z HtpasswdFileTest.test_10_reprc C s( | � ttjjd� | � ttjjd� d S )Ns realm:user1:pass1
s pass1
)rA rQ r r3 rN �rD r r r �test_11_malformed� s � �z"HtpasswdFileTest.test_11_malformedc C s | j ttjjdd d� d S )Nr8 )r )rA r� r r3 rN r� r r r �test_12_from_string� s �z$HtpasswdFileTest.test_12_from_stringc C s� t d�}tj�|�}| �t|�� �g d�� |�dd� | �t|�� �g d�� |�dd� | �t|�� �g d�� |�d� | �t|�� �g d �� t d
�}| �|� � |� d S ) z whitespace & comment handlingz`
user2:pass2
user4:pass4
user7:pass7
user1:pass1
# legacy users
#user6:pass6
user5:pass5
)rJ rK ru rL �user7ru Zalthash4�user6Zalthash6)rJ rK ru rL r� r� r� )rJ rK ru rL r� zd
user2:pass2
user4:althash4
user1:pass1
# legacy users
#user6:pass6
user5:pass5
user6:althash6
N)
r r r3 rN r; rv rw �set_hashrO r< )rD �sourcerE �targetr r r �test_13_whitespace� s �
� z#HtpasswdFileTest.test_13_whitespacec s | � � }tj|dd�� � fdd�}� �d|dd�� tjd }� �d |d|�� tjd
}� �d |d|�� � �d d
� � �� | �t|dd�� | �t|d d�� | �t|d d�� | �t|d d�� | �t|dd�� | �t|d d�� | �t|d d�� dS )z;
verify "htpasswd" command can read output
T�r r7 c s � j �|��| �S rG �ri Zhandler�hash��pwdrk �rE r r � hash_scheme� s z>HtpasswdFileTest.test_htpasswd_cmd_verify.<locals>.hash_schemerJ r- Z
apr_md5_cryptrm rK �hostrr ru z$xxx$foo$bar$baz�wrongN) r9 r r3 r� ro r� r> r. r= )rD r r� Zhost_no_bcryptZ host_bestr r� r �test_htpasswd_cmd_verify� s"
z)HtpasswdFileTest.test_htpasswd_cmd_verify�bcryptzbcrypt support requiredc s| | � � }tj|dd�� � fdd�}� �d|dd�� � �� | �t|dd�� trf| �t|dd�� n| �t|dd�� d S )
z�
verify "htpasswd" command can read bcrypt format
this tests for regression of issue 95, where we output "$2b$" instead of "$2y$";
fixed in v1.7.2.
Tr� c s � j �|��| �S rG r� r� r� r r r� � s zEHtpasswdFileTest.test_htpasswd_cmd_verify_bcrypt.<locals>.hash_schemerJ r- r� r� N) r9 r r3 r� r� r> r. �HAVE_HTPASSWD_BCRYPTr= )rD r r� r r� r � test_htpasswd_cmd_verify_bcrypt� s z0HtpasswdFileTest.test_htpasswd_cmd_verify_bcryptN)%�__name__�
__module__� __qualname__�__doc__�descriptionPrefixr: rP ra r� r� r� r rF rH rS rV rd rg rp rx r� r� r� r� r� r� r� r� r� r� �requires_htpasswd_cmdr� r �
skipUnlessr Z has_backendr� r r r r r2 Z s@ "
%
/
$r2 c @ s� e Zd ZdZdZdZdZdZdZdZ dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )#�HtdigestFileTestztest HtdigestFile class� HtdigestFiles� user2:realm:549d2a5f4659ab39a80dac99e159ab19
user3:realm:a500bb8c02f6a9170ae46af10c898744
user4:realm:ab7b5d5f28ccc7666315f508c7358519
user1:realm:2a6cf53e7d8f8cf39d946dc880b14128
sZ user3:realm:a500bb8c02f6a9170ae46af10c898744
user4:realm:ab7b5d5f28ccc7666315f508c7358519
s� user2:realm:5ba6d8328943c23c64b50f8b29566059
user3:realm:a500bb8c02f6a9170ae46af10c898744
user4:realm:ab7b5d5f28ccc7666315f508c7358519
user1:realm:2a6cf53e7d8f8cf39d946dc880b14128
user5:realm:03c55fdc6bf71552356ad401bdb9af19
s0 useræ:realmæ:549d2a5f4659ab39a80dac99e159ab19
s. user�:realm�:549d2a5f4659ab39a80dac99e159ab19
c C sl | � � }t|| j� t�|�}| �|�� | j� tj|dd�}| �|�� d� t�|� | � t
tj|� dS )r4 Tr6 r8 N) r9 r r: r r� r; r< r r@ rA rB rC r r r rF s
z-HtdigestFileTest.test_00_constructor_autoloadc C s� t j�| j�}| �|�dd�� | �|�dd�� | �|�dd�� | �|�dd�� | �|�� | j � | �
t |jdd� | �
t |jdd � d
S ) rI rJ �realmrK rL rr �realm5rM r, �realm:N) r r� rN r: r= rO r> r; r< rP rA rQ rR r r r rS + s z HtdigestFileTest.test_01_deletec C s� | � � }t|| j� t�|�}| �|�dd�� | �|�dd�� | �|�dd�� | �t |�| j� d|_
| �|�dd�� | �t |�| j � d S )NrJ r� rr r� rL TrK ) r9 r r: r r� r= rO r> r; r
rT rP rC r r r rV : s
z(HtdigestFileTest.test_01_delete_autosavec C s t j�| j�}| �|�ddd�� | �|�ddd�� | �|�� | j � | �
t |jdd� d|_ |�dd� |�
ddd� | �
t|jddd � | �
t|jd
dd � | �
t|jd d d � | �
t|jd d
d � | �d�� |�ddd� W d � n1 s�0 Y | �|�
dd�� dS )z
test update()rK r� rZ rL r[ rs Zrealm2rM r] A uuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuur, r� A rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr^ r_ N)r r� rN r: r= r` r> r; r< ra rA r� �
default_realmrc rQ r? rb rR r r r rd H s ,z%HtdigestFileTest.test_02_set_passwordc C sd t j�| j�}|�ddd� |�dd� |�ddd� | �t|�d��g d�� | � t
|jd� dS ) rq rL r� r[ rr rs rt rz N) r r� rN r: r` rO r; rv rw rA r� rR r r r rx d s z HtdigestFileTest.test_03_usersc C sJ t j�| j�}| �t|jddd� | �t|jddd� | �|�ddd�d� tdd�D ]B}t |�}| �
|�d| dd| �� | �|�d| dd�d � qV| �t|jdd� d|_ | �
|�d
d �� | �|�dd�d� | � d gd
��8 | �
|�
d
dd �� | �|�
d
dd�� W d � n1 �s(0 Y | �t|jddd� dS )ry rz r� r[ r, rL Nr� r] FrJ rh r} r~ rf rM )r r� rN r: rA r� rc ZassertIsr r� r= r� r? r� r> rQ r� r r r r� n s 4z'HtdigestFileTest.test_04_check_passwordc C sP | � � }t|d� t|d� t�|�}| �|�� d� |�ddd� |�� | �|�� d� t|| j � |�� | �|�� | j � |�ddd � |�
� | �|�� | j � t�� }| � t |j
� | � t |j� t�� }|�
|� | �|�� | j � t
|� t|d� | �d
�� |j
d d � W d
� n1 �s20 Y | �|�� d� d
S )r� r r� r8 rJ r� rh s- user1:realm:2a6cf53e7d8f8cf39d946dc880b14128
rL r[ z!load\(force=False\) is deprecatedF)�forceN)r9 r r r r� r; r< r` r� r: r� rA r� r r? r� r r r r� � s2
,z HtdigestFileTest.test_05_loadc C s� | � � }t|| j� t�|�}|�dd� |�dd� |�� | �t|�| j � t�� }|�
ddd� | � t |j� |�|� | �t|�|�
� � dS )r� rJ r� rK rh N)r9 r r: r r� rO r� r; r
rP r` rA r� r< r� r r r r� � s
z HtdigestFileTest.test_06_savec C sh t j�| j�}| �|�d�d� | �|�� dg� | �|�d�d� | �|�� g � | �|�� d� dS )z test realms() & delete_realm()r5 r r� � r8 N)r r� rN r: r; Z delete_realm�realmsr< rR r r r �test_07_realms� s z HtdigestFileTest.test_07_realmsc C s� t j�| j�}| �|�dd�d� | �|�dd�d� | �|�dd�d� | �d��$ | �|�dd�d� W d � n1 s~0 Y dS ) r� rr r� Z a500bb8c02f6a9170ae46af10c898744ru Z ab7b5d5f28ccc7666315f508c7358519rL Nr� )r r� rN r: r; rj r? r� rR r r r r� � s z!HtdigestFileTest.test_08_get_hashc C s� | j ttjdd� tjj| jddd�}| �|�� td�g� | �|� td��td�g� tjj| j
ddd�}| �|�� td�g� | �|� td��td�g� d S )
ztest encoding parameterr� r� r Tr� u realmær� r� N) rA rQ r r� rN r� r; r� r
rw r� rR r r r �test_09_encodings� s z"HtdigestFileTest.test_09_encodingsc C s< t j�| j�}| �|�� | j� t �� }| �|�� d� dS )ztest to_string()r8 N)r r� rN r: r; r< rR r r r �test_10_to_string� s z"HtdigestFileTest.test_10_to_stringc C s( | � ttjjd� | � ttjjd� d S )Ns realm:user1:pass1:other
re )rA rQ r r� rN r� r r r r� � s � �z"HtdigestFileTest.test_11_malformedN)r� r� r� r� r� r: rP ra r� r� rF rS rV rd rx r� r� r� r� r� r� r� r� r r r r r� s(
)
r� )r )N))r� Z
__future__r Zloggingr r r! Zpasslibr r Z passlib.excr Zpasslib.utils.compatr Zpasslib.tests.backportsr Zpasslib.tests.utilsr r
r r r
Z
passlib.utilsr Zpasslib.utils.handlersr r� r0 r �environ�getr# r+ r. r1 Z
HAVE_HTPASSWDr� r� r� r2 r� r r r r �<module> s2
)
Mini Shell