Your IP : 216.73.216.172
U
�]p4 � @ s d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z
ddl Z ddl Z ddl
mZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ e
j� e� d ��Z G d d
� d
ej �Z G dd� dej �Z!e"dkr�e�#� dS )z"Tests for certbot.plugins.webroot.� )�print_functionN)�
challenges)� achallenges)�errors)�os)�
filesystem)�util)� acme_utilzrsa512_key.pemc @ s� e Zd ZdZejejded�Z dd� Z
dd� Z dd � Z d
d � Z
d d
� Ze�� dd� �Ze�� dd� �Ze�� dd� �Ze�� dd� �Zdd� Zdd� Ze�d�dd � �Ze�� d d � �Zd d � Zd!d"� Z d#d$� Z d%d&� Z d'd(� Z e�d)�d*d+� �Z d,S )-�AuthenticatorTestz0Tests for certbot.plugins.webroot.Authenticator.� thing.com�Zchallb�domainZ account_keyc C s� ddl m} t�� | _tj�| jd�| _t� | j� tj�| jd�| _
tj�| jdd�| _ tj�| j d�| _ t
j| jd| jid�| _|| jd�| _d S ) Nr ��
Authenticator�webrootz .well-knownzacme-challengeZ:ZXZhR3hmQURzNnBTUmIyTEF2OUlaZjE3RHQzanV4R0orUEN0OTJ3citvQQr )� webroot_path� webroot_map)�certbot.plugins.webrootr �tempfile�mkdtempZ workspacer �path�joinr �mkdir�partial_root_challenge_path�root_challenge_path�validation_path�mock� MagicMock�config�auth��selfr � r" �>/usr/lib/python3/dist-packages/certbot/plugins/webroot_test.py�setUp# s*
� ���zAuthenticatorTest.setUpc C s t �| j� d S �N)�shutilZrmtreer �r! r" r" r# �tearDown7 s zAuthenticatorTest.tearDownc C s0 | j �� }| �t|tj�� | �| j|k� d S r% )r � more_info�
assertTrue�
isinstance�sixZ string_typesr )r! r) r" r" r# �test_more_info: s
z AuthenticatorTest.test_more_infoc C s& t �� }| j�|� | �d|j� d S )N� )r r r Zadd_parser_arguments� assertEqualZ
call_count)r! �addr" r" r# �test_add_parser_arguments? s z+AuthenticatorTest.test_add_parser_argumentsc C s | j �� d S r% )r �preparer' r" r" r# � test_prepareD s z AuthenticatorTest.test_preparec s� g | j _d| ji| j _|� }tjdf|j_| j� | j
g� | � |jj � |jj
D ]D� | � | j
j� d d k� | � t� fdd�t�| j j�D ��� qP| �| j j| j
j | j� d S )N�otherthing.com� r c 3 s | ]}|� d d kV qdS �r r5 Nr" ��.0r ��callr" r# � <genexpr>R s �z;AuthenticatorTest.test_webroot_from_list.<locals>.<genexpr>)r r r r � display_util�OK�menu� return_valuer �perform�achallr* �called�call_args_listr
�allr, �
itervaluesr/ �r! �mock_get_utility� mock_displayr" r9 r# �test_webroot_from_listG s � �z(AuthenticatorTest.test_webroot_from_listc s� g | j _d| ji| j _|� }tjdff|j_| �t j
| j j | j
g� | �|jj� |jjD ]D� | �| j
j� d d k� | �t� fdd�t�| j j�D ��� qZd S )Nr4 ���r c 3 s | ]}|� d d kV qdS r6 r" r7 r9 r" r# r; c s �zKAuthenticatorTest.test_webroot_from_list_help_and_cancel.<locals>.<genexpr>)r r r r r< �CANCELr> � side_effect� assertRaisesr � PluginErrorr r@ rA r* rB rC r
rD r, rE rF r" r9 r# �&test_webroot_from_list_help_and_cancelX s �z8AuthenticatorTest.test_webroot_from_list_help_and_cancelc C s� g | j _d| ji| j _|� }tjdf|j_t� d��.}tj
dftj| jff|_ | j �
| jg� W 5 Q R X | �| j j| jj | j� d S )N�
something.comr �'certbot.display.ops.validated_directoryrJ )r r r r r< r= r> r? r �patchrK rL r r@ rA r/ r
�r! rG rH �mr" r" r# �test_new_webrootg s
�z"AuthenticatorTest.test_new_webrootc C sb g | j _i | j _|� }tjdf|j_t�d��*}tj df|_| �
t j | j
j| jg� W 5 Q R X d S )Nr rQ rJ )r r r r< r= r> r? r rR rK rM r rN r r@ rA rS r" r" r# �!test_new_webroot_empty_map_cancelv s �z3AuthenticatorTest.test_new_webroot_empty_map_cancelc C s( d | j _i | j _| �tj| jjg � d S r% )r r r rM r rN r r@ r' r" r" r# �test_perform_missing_root� s z+AuthenticatorTest.test_perform_missing_rootc C s� t j�| jd�| j_t j�| jd�}t|d��}|�d� W 5 Q R X t�| jd� zt|d� t d� W n( t
k
r� | � t j
| jjg � Y nX t�| jd� d S ) NZnullZrnd�wZthingimyr �rz9Warning, running tests as root skips permissions tests...i� )r r r r � full_path�open�writer �chmod�print�IOErrorrM r rN r@ )r! Zpermission_canary�fr" r" r# �"test_perform_reraises_other_errors� s
z4AuthenticatorTest.test_perform_reraises_other_errorsz@certbot.plugins.webroot.filesystem.copy_ownership_and_apply_modec C s" t tjd�|_| j�| jg� d S )N�msg)�OSError�errno�EACCESrL r r@ rA )r! Zmock_ownershipr" r" r# �test_failed_chown� s z#AuthenticatorTest.test_failed_chownc C s� t �� }g | j_d| ji| j_|� }tjdftj|ff|j_ t
j t j
dtd�}t�d�� }tj|f|_| j�|g� W 5 Q R X | �| jj|j |� d S )Nz whatever.comr rP r rQ )r r r r r r r< r= r> rL r �"KeyAuthorizationAnnotatedChallenger �HTTP01_P�KEYr rR r? r r@ r/ r
)r! rG Z new_webrootrH rA rT r" r" r# �#test_perform_new_webroot_not_in_map� s � � z5AuthenticatorTest.test_perform_new_webroot_not_in_mapc C s� | j �� | j �| jg� | �t�| jd�� t� | j
�D ]4\}}}|D ]$}tj
� ||�}| �t�|d�� qHq:| �t� | j| j
�� d S )Ni� i� )
r r2 r@ rA r* r Z
check_moder r �walkr r Zhas_same_ownership)r! �dirpathZdirnames�_Z directoryrZ r" r" r# �test_perform_permissions� s
z*AuthenticatorTest.test_perform_permissionsc C s� | j �� | j �| jg�}| �dt|�� | �tj� | j
�� t | j
��}|� � }W 5 Q R X | �t
j|d��| jjt�� �� | j �| jg� | �tj� | j
�� | �tj� | j�� | �tj� | j�� d S )Nr5 )Zkey_authorization)r r2 r@ rA r/ �lenr* r r �existsr r[ �readr Z!KeyAuthorizationChallengeResponseZverifyZchallri Z
public_key�cleanup� assertFalser r )r! Z responsesZ validation_fZ
validationr" r" r# �test_perform_cleanup� s$
� ��z&AuthenticatorTest.test_perform_cleanupc C sb t �| j� | j�� | j�| jg� | j�| jg� | �t j
� | j �� | �t j
� | j
�� d S r% )r r r r r2 r@ rA rr rs r r rp r r r' r" r" r# �"test_perform_cleanup_existing_dirs� s
z4AuthenticatorTest.test_perform_cleanup_existing_dirsc C s� t jt�tjdd�d�dtd�}d}t�| j � | j
� � | j
� || j
g� | j
�| j
g� | �tj�|�� | �tj�| j�� | j
�|g� | �tj�| j�� | �tj�| j�� d S )Ns bingo)�token�pendingr r ZYmluZ28)r rg r Zchall_to_challbr ZHTTP01ri r r r r r2 r@ rA rr rs r r rp r* r r )r! Z bingo_achallZbingo_validation_pathr" r" r# �(test_perform_cleanup_multiple_challenges� s$
� �
z:AuthenticatorTest.test_perform_cleanup_multiple_challengesc C sz | j �� | j �| jg� tj�| jd�}t� |� | j �
| jg� | � tj� | j
�� | �tj� | j�� t�|� d S )NZleftover)r r2 r@ rA r r r r r r rr rs rp r r* �rmdir)r! Z
leftover_pathr" r" r# �test_cleanup_leftovers� s
z(AuthenticatorTest.test_cleanup_leftoverszcertbot.compat.os.rmdirc C sj | j �� | j �| jg� t� }tj|_||_| j �| jg� | � t
j � | j
�� | �t
j � | j�� d S r% )r r2 r@ rA rc rd re rL rr rs r r rp r r* r )r! Z
mock_rmdirZos_errorr" r" r# �test_cleanup_failure� s
z&AuthenticatorTest.test_cleanup_failureN)!�__name__�
__module__� __qualname__�__doc__r rg r rh ri rA r$ r( r- r1 r3 � test_utilZpatch_get_utilityrI rO rU rV rW ra r rR rf rj rn rt ru rx rz r{ r" r" r" r# r
s@ �
r
c @ sZ e Zd ZdZejejded�Z dd� Z
dd� Z dd � Z d
d � Z
d d
� Zdd� Zdd� ZdS )�WebrootActionTestz#Tests for webroot argparse actions.r r c C sF ddl m} t�� | _t�� | _| jjdddg d� |� | jd� d S )Nr r z-dz --domains�append)�action�defaultr )
r r r r r �argparse�ArgumentParser�parser� add_argumentZinject_parser_optionsr r" r" r# r$ s
�zWebrootActionTest.setUpc C s4 | j �dt�d| ji�g�}| �|jd | j� d S )Nz
--webroot-mapr )r� �
parse_args�json�dumpsr r/ r )r! �argsr" r" r# �test_webroot_map_action s �z)WebrootActionTest.test_webroot_map_actionc C sD | j �d�| jj| j��� �}| �|�}| �|j | jj | j� d S )Nz
-d {0} -w {1})
r� r� �formatrA r
r �split�_get_config_after_performr/ r �r! r� r r" r" r# �test_domain_before_webroot s
�
z,WebrootActionTest.test_domain_before_webrootc C s4 | � tj| jjd�� � | � tj| jjd�� � d S )Nz-d foo -w bar -w bazz-d foo -w bar -d baz -w qux)rM r rN r� r� r� r' r" r" r# � test_domain_before_webroot_error s ��z2WebrootActionTest.test_domain_before_webroot_errorc C sb | j �d�| j| jjt�� ��� �}| � |j
| jj | j� | � |�}| � |j
| jj | j� d S �Nz-w {0} -d {1} -w {2} -d bar) r� r� r� r rA r
r r r� r/ r r� r� r" r" r# �test_multiwebroot! s
�
�z#WebrootActionTest.test_multiwebrootc C sX t �� }| j�d�| j| jj|��� �}| � |j
| jj| ji� | � |j | j|g� d S r� ) r r r� r� r� r rA r
r� r/ r r )r! Zother_webroot_pathr� r" r" r# �(test_webroot_map_partial_without_perform) s
�
z:WebrootActionTest.test_webroot_map_partial_without_performc C s* ddl m} ||d�}|�| jg� |jS )Nr r r )r r r@ rA r )r! r r r r" r" r# r� 6 s
z+WebrootActionTest._get_config_after_performN)r| r} r~ r r rg r rh ri rA r$ r� r� r� r� r� r� r" r" r" r# r� s �
r� �__main__)$r Z
__future__r r� rd r� r& r ZunittestZjosepyZjoser r, Zacmer Zcertbotr r Zcertbot.compatr r Zcertbot.displayr r< Z
certbot.testsr r� ZJWKRSA�loadZ load_vectorri ZTestCaser
r� r| �mainr" r" r" r# �<module> s0 f;