← Back
Editing: webroot_test.cpython-38.pyc
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;
Save File
Cancel