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