관리-도구
편집 파일: test_apache.cpython-39.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 )