Your IP : 216.73.216.172
U
�]b � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZ ddl m
Z
ddl m Z ddl m
Z
ddl mZ G d d
� d
ej�Zed kr�e�� dS ) z Tests for certbot.plugins.manual� N)�
challenges)�errors)�os)�
filesystem)� acme_util)�utilc s� e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Ze �
� d d
� �Z e �
� dd� �Z dd� Z
e �
� dd� �Zdd� Z� ZS )�AuthenticatorTestz/Tests for certbot.plugins.manual.Authenticator.c s� t t| ��� tj| _tj| _tj| _ | j| j| j g| _
dD ]}t � t
j�| j|�� q<tjdd d dddt
j�| jd�t
j�| jd�t
j�| jd�t
j�| jd�t
j�| jd�d � | _dd
lm} || jd d �| _d S )
N)�
config_dir�work_dirZ in_progressr Fr r
�
backup_dir�temp_checkpoint_dirZ
in_progess) Z http01_port�manual_auth_hook�manual_cleanup_hook�manual_public_ip_logging_ok�noninteractive_mode�validate_hooksr r
r r Zin_progress_dir)�
AuthenticatorZmanual)�name)�superr �setUpr ZHTTP01_A� http_achallZDNS01_A�
dns_achallZ DNS01_A_2Z dns_achall_2�achallsr �mkdirr �path�joinZtempdir�mockZ MagicMock�configZcertbot.plugins.manualr �auth)�self�dr �� __class__� �=/usr/lib/python3/dist-packages/certbot/plugins/manual_test.pyr s2 �� zAuthenticatorTest.setUpc C s d| j _| �tj| jj� d S �NT)r r � assertRaisesr � PluginErrorr �prepare�r r# r# r$ �#test_prepare_no_hook_noninteractive- s z5AuthenticatorTest.test_prepare_no_hook_noninteractivec C s0 t j�t j�| j_d| j_| �tj | j
j � d S r% ) r r �abspath�sepr r
r r& r ZHookCommandNotFoundr r( r) r# r# r$ �test_prepare_bad_hook1 s z'AuthenticatorTest.test_prepare_bad_hookc C s | � t| j�� tj�� d S )N)�
assertTrue�
isinstancer Z more_info�sixZ string_typesr) r# r# r$ �test_more_info6 s z AuthenticatorTest.test_more_infoc C s | � | j�d�tjtjg� d S )Nz example.org)� assertEqualr Zget_chall_prefr �HTTP01�DNS01r) r# r# r$ �test_get_chall_pref9 s
�z%AuthenticatorTest.test_get_chall_prefc C s" d|� j _| �tj| jjg � d S )NF)�yesno� return_valuer& r r' r �perform�r �mock_get_utilityr# r# r$ �test_ip_logging_not_ok= s
z(AuthenticatorTest.test_ip_logging_not_okc C s( d|� j _| j�g � | �| jj� d S r% )r6 r7 r r8 r. r r r9 r# r# r$ �test_ip_logging_okB s
z$AuthenticatorTest.test_ip_logging_okc C s� d| j _d�tj�| j _d�| jjd| j�| jj ��}d�| j
j| j
j � d�| j
�| j
j ��}| �
| j�| j�dd� | jD �� | �
| jj| j d |� | �
| jj| j
d |� d S ) NTz�{0} -c "from __future__ import print_function;from certbot.compat import os; print(os.environ.get('CERTBOT_DOMAIN'));print(os.environ.get('CERTBOT_TOKEN', 'notoken'));print(os.environ.get('CERTBOT_VALIDATION', 'novalidation'));"z {0}
{1}
{2}Znotoken�tokenc S s g | ]}|� |j��qS r# �Zresponse� account_key��.0�achallr# r# r$ �
<listcomp>Y s z9AuthenticatorTest.test_script_perform.<locals>.<listcomp>�CERTBOT_AUTH_OUTPUT)r r �format�sys�
executabler
r �domain�
validationr? r �chall�encoder2 r r8 r �env)r Z dns_expectedZ
http_expectedr# r# r$ �test_script_performH s6 �� � � ���z%AuthenticatorTest.test_script_performc C s| d| j _| �| j�| j�dd� | jD �� t|� jj�D ]>\}\}}| j| }| � |�
|j �|d k� | � |d � q8d S )NTc S s g | ]}|� |j��qS r# r> r@ r# r# r$ rC f s z9AuthenticatorTest.test_manual_perform.<locals>.<listcomp>r Zwrap)
r r r2 r r8 r � enumerateZ notificationZcall_args_listr. rI r? � assertFalse)r r: �i�args�kwargsrB r# r# r$ �test_manual_performa s �
�z%AuthenticatorTest.test_manual_performc C s� d| j _d�tj�| j _d| j _| j�| j � | j D ]�}| j�
|g� | � t j
d d� | � t j
d |j� t|jtjtjf�r�| � t j
d |�|j�� t|jtj�r�| � t j
d |j�d �� q4| �dt j
k� q4d S )
NTz,{0} -c "import sys; sys.stdout.write('foo')"z # cleanuprD ZfooZCERTBOT_DOMAINZCERTBOT_VALIDATIONZ
CERTBOT_TOKENr= )r r rE rF rG r
r r r8 r Zcleanupr2 r �environrH r/ rJ r r3 r4 rI r? rK rO )r rB r# r# r$ � test_cleanupm s* �
�
�z AuthenticatorTest.test_cleanup)�__name__�
__module__� __qualname__�__doc__r r* r- r1 r5 � test_utilZpatch_get_utilityr; r< rM rS rU �
__classcell__r# r# r! r$ r s
r �__main__)rY ZunittestrF r r0 Zacmer Zcertbotr Zcertbot.compatr r Z
certbot.testsr r rZ ZTempDirTestCaser rV �mainr# r# r# r$ �<module> s s