|
Server : Apache/2.4.41 (Ubuntu) System : Linux vmi1525618.contaboserver.net 5.4.0-105-generic #119-Ubuntu SMP Mon Mar 7 18:49:24 UTC 2022 x86_64 User : www-data ( 33) PHP Version : 8.2.12 Disable Function : NONE Directory : /lib/python3/dist-packages/certbot/tests/__pycache__/ |
Upload File : |
U
�]�~ � @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlm m
Z ddl m
Z
ddl mZ ddlmZ ddlmZ ddl m
Z
e �d �Ze �d
�ZG d d � d e j�ZG d
d� de j�ZG dd� de j�ZG dd� de�ZG dd� de�ZG dd� dej�Z e dk�re� � dS )zTests for certbot.client.� N)�
interfaces)�account)�errors)�os)�
filesystem)�utilzrsa512_key.pemzcsr-san_512.pemc @ sT e Zd ZdZdd� Zej�ej ddi�dd� �Z
ej�ej i �dd � �Z d
d � Z d S )
�DetermineUserAgentTestz.Tests for certbot.client.determine_user_agent.c C s ddl m} || j�S )Nr )�determine_user_agent)�certbot.clientr �config)�selfr � r
�;/usr/lib/python3/dist-packages/certbot/tests/client_test.py�_call s z DetermineUserAgentTest._callZ CERTBOT_DOCS�1c C s | j dd� d S )NT��expect_doc_values��_test�r r
r
r �test_docs_value s z&DetermineUserAgentTest.test_docs_valuec C s | j dd� d S )NFr r r r
r
r �test_real_values! s z'DetermineUserAgentTest.test_real_valuesc C sd | � � }|r| j}| j}n | j}| j}|d|� |d|� |d|� |t�� |� |t�� |� d S )Nzcertbot(-auto)zOS_NAME OS_VERSIONzmajor.minor.patchlevel)r ZassertInZ assertNotInr Zget_os_info_ua�platformZpython_version)r r ZuaZdoc_value_checkZreal_value_checkr
r
r r % s
z DetermineUserAgentTest._testN)
�__name__�
__module__� __qualname__�__doc__r �mock�patch�dictr �environr r r r
r
r
r r s
r c s� e Zd ZdZ� fdd�Zdd� Zedd� �Zedd � �Zed
d � �Z ed d
� �Z
dd� Z dd� Z e
�d�e
�d�dd� ��Ze
�d�dd� �Zdd� Ze
�d�dd � �Ze
�d�e
�d�d d � ��Zd d � Zd!d"� Zd#d$� Zd%d&� Z� ZS )'� RegisterTestz"Tests for certbot.client.register.c s4 t t| ��� d| j_d| j_d| j_t�� | _ d S )Ni Fzalias@example.com)
�superr! �setUpr � rsa_key_size� register_unsafely_without_email�emailr ZAccountMemoryStorage�account_storager �� __class__r
r r# 9 s
zRegisterTest.setUpc C s$ ddl m} t�� }|| j| j|�S )Nr )�register)r
r* r � MagicMockr r' )r r* Ztos_cbr
r
r r @ s zRegisterTest._callc C s t jtjd�} d| j_| S )Nr( z{"a": 1})r �Mockr ZJSONDeSerializableZto_partial_json� return_value)�mr
r
r �_public_key_mockE s z RegisterTest._public_key_mockc C s dS )Nz/acme/new-accountr
r
r
r
r �_new_acct_dir_mockK s z RegisterTest._new_acct_dir_mockc C s dS �NTr
r
r
r
r �
_true_mockO s zRegisterTest._true_mockc C s dS �NFr
r
r
r
r � _false_mockS s zRegisterTest._false_mockc C s� t �d���}d|�� _| j|� j_t �d��b}t �d��L tj|� j_| � tj| j
� | � |j � d |� j_| �
� | �
|j � W 5 Q R X W 5 Q R X W 5 Q R X d S )N�6certbot.client.acme_client.BackwardsCompatibleClientV2z
http://tos� certbot.eff.handle_subscription�"certbot.account.report_new_account)r r �new_account_and_tosZterms_of_servicer4 �external_account_required� side_effectr �Error� assertRaisesr � assertFalse�called�
assertTrue)r � mock_client� mock_handler
r
r � test_no_tosW s
zRegisterTest.test_no_tosc C sZ t �d��F}| j|� j_t �d��$ t �d�� | �� W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr5 r7 r6 )r r r4 r9 r: r �r r@ r
r
r �test_ite s
zRegisterTest.test_itr7 z$certbot.client.display_ops.get_emailc
C s� ddl m} d| j_d}|jjd|d�}t�d��\}| j|� j _
t�d��:}|t� � g|� j _
| �
� | �|jd � | �|j� W 5 Q R X W 5 Q R X d S )
Nr ��messagesF�3DNS problem: NXDOMAIN looking up MX for example.com�invalidContact��detailr5 r6 � )�acmerF r �noninteractive_moder; � with_coder r r4 r9 r: r+ r8 r � assertEqual�
call_countr? r> )r �_rep�mock_get_emailrF �msg�mx_errr@ rA r
r
r �test_email_retryl s z RegisterTest.test_email_retryc
C s� ddl m} d| j_d}|jjd|d�}t�d��J}| j|� j _
t�d��( |t� � g|� j _
| �
tj| j� W 5 Q R X W 5 Q R X d S ) Nr rE TrG rH rI r5 r6 )rL rF r rM r; rN r r r4 r9 r: r+ r8 r< r r )r rQ rF rS rT r@ r
r
r �!test_email_invalid_noninteractive{ s z.RegisterTest.test_email_invalid_noninteractivec C s d | j _| �tj| j� d S �N)r r& r< r r; r r r
r
r �test_needs_email� s z RegisterTest.test_needs_email�certbot.client.loggerc C s� t �d��x}t �d��b}| j|� j_t �d��@ d | j_d| j_d| j_| � � |j
� t j � | �
|j� W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr6 r5 r7 TF)r r r4 r9 r: r r& r% �dry_runr �info�assert_called_once_with�ANYr? r> )r � mock_loggerrA Z mock_clntr
r
r �test_without_email� s z RegisterTest.test_without_emailc C s� t �d��t}| j|� j_t �d��R t �d��< d| j_| �� | �|j � | �|� j
j d d j � W 5 Q R X W 5 Q R X W 5 Q R X dS )zJTests dry-run for no staging account, expect account created with no emailr5 r6 r7 Tr N)
r r r4 r9 r: r rZ r r= r> r8 � call_argsZcontact)r rQ rR r@ r
r
r � test_dry_run_no_staging_account� s z,RegisterTest.test_dry_run_no_staging_accountc C s� t �d��|}t j| jd�|� jj_| j|� j_ t �d��D d}t �|��*}d| j
_ d| j
_ | �
� | �|j� W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr5 �r: r6 �8certbot.client.messages.ExternalAccountBinding.from_dataztest-kidz+J2OAqW4MHXsrHVa_PVg0Y-L_R4SYw0_aL1le6mfblbE)r r r, r0 �client� directory� __getitem__r4 r9 r: r �eab_kid� eab_hmac_keyr r? r> �r r@ �targetZmock_eab_from_datar
r
r �test_with_eab_arguments� s � z$RegisterTest.test_with_eab_argumentsc C sz t �d��f}| j|� j_t �d��D d}t �|��*}d | j_d | j_| �� | � |j
� W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr5 r6 rc ) r r r4 r9 r: r rg rh r r= r> ri r
r
r �test_without_eab_arguments� s z'RegisterTest.test_without_eab_argumentsc C s� t �d��v}t j| jd�|� jjj_| j|� j _
t �d��< t �d��& d | j _ d | j _
| �tj| j� W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr5 rb r6 rc )r r r, r/ rd �net�keyZ
public_keyr2 r9 r: r rg rh r< r r; r rC r
r
r �4test_external_account_required_without_eab_arguments� s zARegisterTest.test_external_account_required_without_eab_argumentsc
C s� ddl m} d}|j|ddd�}t�d��`}tj| jd�|� jj_ | j
|� j _ t�d ��(}|t�
� g|� j_ | �|j| j� W 5 Q R X W 5 Q R X | �|j� d S )
Nr rE ZTestZ malformed�title)rJ �typrp r5 rb r6 )rL rF r; r r r, r0 rd re rf r4 r9 r: r+ r8 r< r r= r> )r rF rS rT r@ rA r
r
r �test_unsupported_error� s � $z#RegisterTest.test_unsupported_error)r r r r r# r � staticmethodr/ r0 r2 r4 rB rD r r rU rV rX r_ ra rk rl ro rr �
__classcell__r
r
r( r r! 6 s8
r! c s e Zd ZdZ� fdd�Z� ZS )�ClientTestCommonz2Common base class for certbot.client.Client tests.c s� t t| ��� d| j_d| j_tjf dti�| _ ddl
m } t� d��2}|| _
t�� | _|_|| j| j d d d�| _W 5 Q R X d S )NFzkey.pemr )�Clientr5 )r Zaccount_Zauth� installer)r" ru r# r Z
no_verify_ssl�allow_subset_of_namesr r+ �KEYr r
rv r � acme_clientrL r- rd )r rv rL r( r
r r# � s �zClientTestCommon.setUp)r r r r r# rt r
r
r( r ru � s ru c sN e Zd ZdZ� fdd�Zdd� Zdd� Zd1d d
�Ze� d �e� d �e
� � d
d� ���Z e� d �dd� �Z
e� d �e� d�dd� ��Ze� d �e� d�dd� ��Ze� d �e� d �e� d�dd� ���Zdd� Zdd � Zd2d d �Ze� d �e� d!�d"d#� ��Ze� d$�d%d&� �Zd'd(� Zd)d*� Zd+d,� Ze
� � d-d.� �Ze
� � d/d0� �Z� ZS )3�
ClientTestz Tests for certbot.client.Client.c sB t t| ��� d| j_d| j_ddg| _tjd gtj j
d�| _ d S )NF� example.com�www.example.com)�authorizations�csr_pem) r" r{ r# r rx rZ �
eg_domainsr r+ �sentinelr �eg_orderr r( r
r r# � s
�zClientTest.setUpc C s | j jd d }| �|j� d S )Nr )rz r` r? Z
verify_ssl)r rm r
r
r �test_init_acme_verify_ssl� s z$ClientTest.test_init_acme_verify_sslc C sR t �� | j_d g| jjj_g g f| jjj_| j| jj _| j| jj
_| j| jj _d S rW ) r r+ rd � auth_handler�handle_authorizationsr- � deactivate_valid_authorizationsr� rL �finalize_order� new_order�updater r
r
r �_mock_obtain_certificate s z#ClientTest._mock_obtain_certificaterK c C sN |dkr"| j jj�| j| jj� n| �| j jjj|� | j j
�| jt j � d S �NrK )
rd r� r� r\ r� r rx rO rP rL r� r r] )r �
auth_countr
r
r �_check_obtain_certificate s
� �z$ClientTest._check_obtain_certificatezcertbot.client.crypto_utilrY c C s� | � � tjdd td�}| jj}| �|j� | j� |j
�}|� |d� | � t
jjt
jjf| jj||d�� | �� | � t
jjt
jjf| jj|d d�� |j �| jd� d | j_| �tj| jj|� |j�t
j� d S )N�pem�Zform�file�dataF)�orderr)r� r �CSR�CSR_SANrd r� �_set_mock_from_fullchain� cert_and_chain_from_fullchainrL r� r� r� rO r r� �cert�chainZobtain_certificate_from_csrr� �assert_called_withr� r< r r; �warningr\ r] )r Zunused_mock_get_utilityr^ �mock_crypto_utilZtest_csrr� r� r
r
r � test_obtain_certificate_from_csr s: �����z+ClientTest.test_obtain_certificate_from_csrc C s� t jdd td�}||j_tjj|j_| � |j
� | � tjj|� |j� | j
j| j
j� |j� tjj| j| j
j� |j
� | jj� d S )Nr� r� )r r� r� �
init_save_csrr- r r� rn �
init_save_keyr� r� � _test_obtain_certificate_commonr\ r r$ Zkey_dirr� Zcsr_dirr� Z
fullchain_pem)r r� �csrr
r
r �test_obtain_certificate7 s" � ��z"ClientTest.test_obtain_certificatezcertbot.compat.os.removec C s� t jdtjjtd�}t jdtjjtd�}||j_||j _| �
|j � | � dg�}d| j
_| j|||dd� | �|j jd� | �|jjd� | �|jd� | �|j jd� d S )Nr� r� r| T� )�
authzr_retr� rK )r r� r r� Zcsr_filer� Zkey_filer� r- r� r� r� �_authzr_from_domainsr rx r� rO rP )r Z mock_remover� r� rn �authzrr
r
r �'test_obtain_certificate_partial_successG s z2ClientTest.test_obtain_certificate_partial_successz certbot.client.acme_crypto_utilc C s� t jdd td�}t|j_tjj|j_t j d tjjd�}| �
|j � d| j j
_| �||� |j�| j
j� |j�tjj| j| j
j� |j�� |j�� | �|j jd� d S )Nr� r� �r� r� TrK )r r� r� �make_csrr- r r� �key_pem�make_key�Keyr� r� rd r rZ r� r\ r$ r� Z must_stapler� �assert_not_calledr� rO rP )r �mock_acme_crypto� mock_cryptor� rn r
r
r � test_obtain_certificate_dry_runY s
�
z*ClientTest.test_obtain_certificate_dry_runc C s4 ddl m} tjdd td�}t|j_tjj |j
_tj d tjj d�}| � |j
� | �� d| jj_| �ddg�}|D ]}|j|j_qr| jj} |d g|d gf| j_|| j_|| jjj_t� � � | j� | j �}
W 5 Q R X | � |
tjj tjj!||f� | �"d � | � | jj#d d
� | � | j j$j#d d � |j%�&d
� d S )Nr rE r� r� r� Tr| r} rK z*Deactivate authorizations should be calledr� zHOrder should be recreated due to successfully deactivated authorizationszCertbot was unable to obtain fresh authorizations for every domain. The dry run will continue, but results may not be accurate.)'rL rF r r� r� r� r- r r� r� r� r� r� r� r� rd r rZ r� Z STATUS_VALID�bodyZstatusr� r� r� r~ r� � test_util�patch_get_utility�obtain_certificater� rO r� r� r� rP r� r� r� ) r r� r� �mock_logrF r� rn Zauthzrsr� r� �resultr
r
r �:test_obtain_certificate_dry_run_authz_deactivations_failedl s4
��zEClientTest.test_obtain_certificate_dry_run_authz_deactivations_failedc C s6 t �� }t jj|j_t �� }t jj|j_||f|_d S rW )r r, r� r� �encoder- r� )r Zmock_from_fullchainZ mock_certZ
mock_chainr
r
r r� � s
z#ClientTest._set_mock_from_fullchainc C s8 g }t |�D ]&}|�tjtjtj|d�d�d�� q |S )N)�value)Z
identifier)r� )�reversed�appendr r+ )r �domainsr� �domainr
r
r r� � s ����z ClientTest._authzr_from_domainsNc C sx | � � |p| �| j�}|| j_|| jjj_t �
� � | j� | j�}W 5 Q R X | � |t
jjt
jj||f� | �|� d S rW )r� r� r� r� r~ rd r� r� r- r� r� r� rO r r� r� r� r� )r rn r� r� r� r� r� r
r
r r� � s
�z*ClientTest._test_obtain_certificate_commonz(certbot.client.Client.obtain_certificatez)certbot.storage.RenewableCert.new_lineagec C s� ddg}t �� t �� t �� d f|_d| jj_| �| j�|d�� | �| j�|d �� | �| j�|dd � d �� d| jj_| �| j�|d �� dd� |j D �}| �
|dddg� d S ) Nz
*.example.comr| FZ example_certrK Tc S s g | ]}|d d �qS )r r
)�.0Zcallr
r
r �
<listcomp>� s zAClientTest.test_obtain_and_enroll_certificate.<locals>.<listcomp>) r r+ r- rd r rZ r? Z obtain_and_enroll_certificater= Zcall_args_listrO )r Z mock_storageZmock_obtain_certificater� �namesr
r
r �"test_obtain_and_enroll_certificate� s �
z-ClientTest.test_obtain_and_enroll_certificatezcertbot.cli.helpful_parserc C s� ddg}t �� }t�|d� t�|d �}t�|d �t�|d � }tj�|dd�}tj�|dd�}tj�|dd �}d
|_ d |d |d
|g|_
| j � |||||�\} }
} | �
tj�| �tj�|�� | �
tj�|
�tj�|�� | �
tj�| �tj�|�� t| d��} | �� }
W 5 Q R X | �
|
t�|d �� t|
d��}|�� }W 5 Q R X | �
|t�|d �t�|d � � t�|� d S )Nz cert_512.pemzcert-san_512.pemi� r rK �certsZchainsz chain.pemz
fullchain.pemZcertonlyz --cert-pathz --chain-pathz--fullchain-path�rb)�tempfileZmkdtempr �chmodr� � load_vectorr �path�joinZverb�argsrd Zsave_certificaterO �dirname�open�read�shutilZrmtree)r Z mock_parserr� Ztmp_pathZcert_pemZ chain_pemZcandidate_cert_pathZcandidate_chain_pathZcandidate_fullchain_path� cert_path�
chain_path�fullchain_pathZ cert_fileZ
cert_contentsZ
chain_fileZchain_contentsr
r
r �test_save_certificate� sP � �
�
�
� �z ClientTest.test_save_certificatec C s� | � tj| jjdgdddd� t�� }|| j_| j�dgdddd� |jj t
j � d�t
j � d�ddt
j � d�d� | �
|jjd� |j� � d S )N�foo.barrn r� r� � fullchain)r� r� r� r� Zkey_pathr� )r< r r; rd �deploy_certificater r+ rw � deploy_certr\ r r� �abspathrO �saverP �restart�r rw r
r
r � test_deploy_certificate_success� s2 � �
�z*ClientTest.test_deploy_certificate_successc C sF t �� }|| j_tj|j_| �tj| jj dgdddd� |j
� � d S �Nr� rn r� r� r� ) r r+ rd rw r � PluginErrorr� r: r< r� �recovery_routiner\ r� r
r
r � test_deploy_certificate_failure s
�z*ClientTest.test_deploy_certificate_failurec C sF t �� }|| j_tj|j_| �tj| jj dgdddd� |j
� � d S r� ) r r+ rd rw r r� r� r: r< r� r� r\ r� r
r
r �$test_deploy_certificate_save_failure s
�z/ClientTest.test_deploy_certificate_save_failurec C sl t �� }tjd g|j_|| j_| �tj| jj dgdddd� | �
|� j j d� |j
�� | �
|jj d� d S )Nr� rn r� r� r� rK r� )r r+ r r� r� r: rd rw r< r� rO � add_messagerP �rollback_checkpointsr\ �r Zmock_get_utilityrw r
r
r �'test_deploy_certificate_restart_failure s �
z2ClientTest.test_deploy_certificate_restart_failurec C sr t �� }tj|j_tj|j_|| j_ | �
tj| jj dgdddd� | � |� j
jd� |j�� | � |jjd� d S )Nr� rn r� r� r� rK )r r+ r r� r� r: �
ReverterErrorr� rd rw r< r� rO r� rP r\ r� r
r
r �(test_deploy_certificate_restart_failure2' s
�
z3ClientTest.test_deploy_certificate_restart_failure2)rK )NrK ) r r r r r# r� r� r� r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rt r
r
r( r r{ � sH
!
&
&
r{ c s� e Zd ZdZ� fdd�Zdd� Ze�d�dd� �Ze�d �d
d � �Z e�d �d d
� �Z
e�d �dd� �Z e�d�e�d �dd� ��Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Ze�d�d#d$� �Zd%d&� Zd'd(� Zd)d*� Zd+d,� Z� ZS )-�EnhanceConfigTestz/Tests for certbot.client.Client.enhance_config.c s8 t t| ��� d| j_d| j_d| j_d| j_d| _d S )NFz example.org) r" r� r# r �hsts�redirect�staple�uirr� r r( r
r r# 8 s zEnhanceConfigTest.setUpc C s | � tj| jj| jgd � d S rW )r< r r; rd �enhance_configr� r r
r
r �test_no_installerA s �z#EnhanceConfigTest.test_no_installerzcertbot.client.enhancementsc C s| t �� | j_g | jjj_d | j_d| j_t � d��}| j�
| j gd � W 5 Q R X | � |j
jd� | jjj�� |j�� d S )NTrY rK )r r+ rd rw �supported_enhancementsr- r r� r� r r� r� rO r� rP �enhancer� Zask)r Zmock_enhancementsr^ r
r
r �test_unsupportedE s z"EnhanceConfigTest.test_unsupportedrY c C s: d| j _| �� | �|jj� | �|jjd d d� d S )NTr rK �Strict-Transport-Security)r r� �_test_with_already_existingr? r� r> rO r` �r r� r
r
r �test_already_exists_headerR s �z,EnhanceConfigTest.test_already_exists_headerc C s: d| j _| �� | �|jj� | �|jjd d d� d S )NTr rK r� )r r� r� r? r� r> rO r` r� r
r
r � test_already_exists_redirectZ s �z.EnhanceConfigTest.test_already_exists_redirectc C s" d| j _| �� | �|jj� d S r3 )r r� r� r= r� r> r� r
r
r �#test_config_set_no_warning_redirectb s z5EnhanceConfigTest.test_config_set_no_warning_redirectz certbot.client.enhancements.askc C sB d | j _d|_| �� | �|jj� | �d|jjd d k� d S )NF�disabler )r r� r- r� r? r� r> r` )r r� �mock_askr
r
r �test_warn_redirecth s
z$EnhanceConfigTest.test_warn_redirectc C s* d| j _| �� | jjj�| jdd� d S )NT�ensure-http-headerr� )r r� �_test_with_all_supportedrd rw r� r� r� r r
r
r �test_no_ask_hstsq s
�z"EnhanceConfigTest.test_no_ask_hstsc C s* d| j _| �� | jjj�| jdd � d S )NTr� )r r� r� rd rw r� r� r� r r
r
r �test_no_ask_redirectw s
�z&EnhanceConfigTest.test_no_ask_redirectc C s* d| j _| �� | jjj�| jdd � d S )NT� staple-ocsp)r r� r� rd rw r� r� r� r r
r
r �test_no_ask_staple} s
�z$EnhanceConfigTest.test_no_ask_staplec C s* d| j _| �� | jjj�| jdd� d S )NTr� zUpgrade-Insecure-Requests)r r� r� rd rw r� r� r� r r
r
r �test_no_ask_uir� s
�z!EnhanceConfigTest.test_no_ask_uirc C s4 t �� | j_tj| jjj_| �� | jjj �
� d S rW ) r r+ rd rw r r� r� r: � _test_errorr� r\ r r
r
r �test_enhance_failure� s z&EnhanceConfigTest.test_enhance_failurec C sF t �� | j_tj| jjj_| �� | jjj �
� | jjj�
t j � d S rW ) r r+ rd rw r r� r� r: r� r� r\ r] r r
r
r �test_save_failure� s
z#EnhanceConfigTest.test_save_failurec C s* t �� | j_tjd g| jjj_| �� d S rW ) r r+ rd rw r r� r� r: �_test_error_with_rollbackr r
r
r �test_restart_failure� s z&EnhanceConfigTest.test_restart_failurec C s0 t �� }tj|j_tj|j_|| j_ | �
� d S rW ) r r+ r r� r� r: r� r� rd rw r r� r
r
r �test_restart_failure2� s
z'EnhanceConfigTest.test_restart_failure2c C s d | j _d|_| �� d S r1 )r r� r- r� )r r� r
r
r �test_ask� s zEnhanceConfigTest.test_askc C s | � � | �| jjjj� d S rW )r� r? rd rw r� r> r r
r
r r � s z+EnhanceConfigTest._test_error_with_rollbackc C sB d| j _t�� �}| �tj| j� W 5 Q R X | �|� j j
d� d S )NTrK ) r r� r� r� r< r r� r� rO r� rP )r Zmock_gur
r
r r� � s
�z EnhanceConfigTest._test_errorc C sh | j jd krt�� | j _dddg| j jj_| j �| jgd � | �| j jj j
d� | �| j jj j
d� d S )Nr� r� r� rK ) rd rw r r+ r� r- r� r� rO r� rP r� r r
r
r r� � s � z*EnhanceConfigTest._test_with_all_supportedc C sD t �� | j_dddg| jjj_t�� | jjj_ | j�
| j gd � d S )Nr� r� r� ) r r+ rd rw r� r- r Z PluginEnhancementAlreadyPresentr� r: r� r� r r
r
r r� � s � z-EnhanceConfigTest._test_with_already_existing)r r r r r# r� r r r� r� r� r� r� r� r� r� r� r r r r r r r� r� r� rt r
r
r( r r� 5 s8
r� c @ s4 e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd
S ) � RollbackTestz"Tests for certbot.client.rollback.c C s t �� | _d S rW )r r+ � m_installr r
r
r r# � s zRollbackTest.setUpc C s> ddl m} t�d�� }||_|d |i t�� � W 5 Q R X d S )Nr )�rollbackz.certbot.client.plugin_selection.pick_installer)r
r r r r: r+ )�clsZ checkpointsr: r Zmpir
r
r r � s zRollbackTest._callc C s: | � d| j� | �| �� jjd� | �| �� jjd� d S r� )r r rO r� rP r� r r
r
r �test_no_problems� s z RollbackTest.test_no_problemsc C s | � dd � d S r� )r r r
r
r r� � s z RollbackTest.test_no_installerN) r r r r r# � classmethodr r
r� r
r
r
r r � s
r �__main__) r r r� r� Zunittestr Zjosepyr Zcertbot.tests.utilZtestsr r� Zcertbotr r Zcertbot.compatr r r� ry r� ZConfigTestCaser r! ru r{ r� ZTestCaser r �mainr
r
r
r �<module> s4
' H