|
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
�]K � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl
Z
ddlZddlZddlmZ ddlmZ ddlm mZ ddlmZ ddlmZ ddlmZ dd lm Z dd
lm Z dd lm Z dd lm Z dd
lm Z ddlm!Z! ddlmZ ddl"m#Z# ddl"m$Z$ ddl%m&Z& ddl%m'Z' ddl%m(Z( ddl%m)Z) e�*d�Z+e�*d�Z,e�*d�Z-e�*d�Z.e j/�0e�1d��Z2e�*d�Z3e�*d�Z4G d d � d e
j5�Z6G d d � d ej7�Z8G d d!� d!e
j5�Z9G d"d#� d#e
j5�Z:G d$d%� d%ej;�Z<G d&d'� d'ej7�Z=G d(d)� d)ej7�Z>G d*d+� d+ej7�Z?G d,d-� d-e
j5�Z@G d.d/� d/ej7�ZAG d0d1� d1ej7�ZBG d2d3� d3ej7�ZCeDd4k�r�e
� � dS )5zTests for certbot.main.� )�print_functionN)�
reload_module)�List)�account)�cli)�
configuration)� constants)� crypto_util��errors)�
interfaces)�main)�updater)�util)�os)�
filesystem)�disco)� enhancements)�manual)�null� cert_512.pemz csr_512.derzrsa256_key.pemzrsa512_key.pemzrsa2048_key.pemz
cert_2048.pemc @ s e Zd ZdZdd� ZdS )�TestHandleIdenticalCertsz4Test for certbot.main._handle_identical_cert_requestc C s4 t �� }d|j_t�t �� |�}| �|d|f� d S )NF� reinstall)�mock�MockZensure_deployed� return_valuer
Z _handle_identical_cert_request� assertEqual)�self� mock_lineage�ret� r �9/usr/lib/python3/dist-packages/certbot/tests/main_test.py�*test_handle_identical_cert_request_pending4 s zCTestHandleIdenticalCerts.test_handle_identical_cert_request_pendingN)�__name__�
__module__� __qualname__�__doc__r" r r r r! r 2 s r c sZ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
� d�dd� �Z � Z
S )�RunTestzTests for certbot.main.run.c s� t t| ��� d| _t�d�t�d�t�d�t�d�t�d�t�d�t�d�g| _| jd �� | _| jd
�� | _ | jd �� | _
| jd �� | _ | jd
�� | _ | jd �� | _
| jd �� | _d S )Nz example.org� certbot.main._get_and_save_certz-certbot.main.display_ops.success_installationz(certbot.main.display_ops.success_renewal� certbot.main._init_le_client�-certbot.main._suggest_donation_if_appropriate� certbot.main._report_new_cert�certbot.main._find_certr � � � � � � )�superr' �setUp�domainr �patch�patches�start� mock_auth�mock_success_installation�mock_success_renewal� mock_initZmock_suggest_donation�mock_report_cert�mock_find_cert�r �� __class__r r! r4 ? s" � z
RunTest.setUpc C s | j D ] }|�� qd S �N)r7 �stop�r r6 r r r! �tearDownS s
zRunTest.tearDownc C sF d� | j��� }tj�� }t�t� ||��}ddl
m } |||� d S )Nz-a webroot -i null -d {0}r )�run) �formatr5 �splitr �PluginsRegistry�find_allr �NamespaceConfigr �prepare_and_parse_args� certbot.mainrF )r �args�plugins�configrF r r r! �_callW s
� z
RunTest._callc C s0 t �� | j_d| j_| �� | j�| jg� d S )N)TN� r r r9 r r> rQ r: �assert_called_once_withr5 r? r r r! �test_newcert_success` s z RunTest.test_newcert_successc C s8 t �� | j_dt �� f| j_| �� | j�| jg� d S �NFrR r? r r r! �test_reinstall_successf s z RunTest.test_reinstall_successc C s8 t �� | j_dt �� f| j_| �� | j�| jg� d S )NT) r r r9 r r> rQ r; rS r5 r? r r r! �test_renewal_successl s z RunTest.test_renewal_success�1certbot.main.plug_sel.choose_configurator_pluginsc C s@ t �| jd�d f|_tj�� }d| j_| �t j
t j | j|� d S )Nr T)
r � InstallerrP r r rI rJ � auto_hsts� assertRaisesr �NotSupportedErrorr
rF )r � mock_chooserO r r r! �"test_run_enhancement_not_supportedr s
�z*RunTest.test_run_enhancement_not_supported)r# r$ r% r& r4 rE rQ rT rV rW r r6 r^ �
__classcell__r r r@ r! r' < s r' c @ s� e Zd ZdZdd� Zdd� Zdd� Ze�d�e�d �e�d
�d d � ���Z ddd�Z
e�d�e�d�e�d�e�d
�dd� ����Z e�d�e�d�e�d�e�d
�dd� ����Z dS )� CertonlyTestz Tests for certbot.main.certonly.c C s t �� | _| j�� | _d S rB )� test_util�patch_get_utility�get_utility_patchr8 �mock_get_utilityr? r r r! r4 s
zCertonlyTest.setUpc C s | j �� d S rB �rc rC r? r r r! rE � s zCertonlyTest.tearDownc
C sZ t j�� }t�t�||��}t�d��(}t�d�� t �
||� W 5 Q R X W 5 Q R X |� S )Nr) r* ) r rI rJ r rK r rL r r6 r
�certonly)r rN rO rP r< r r r! rQ � s
� zCertonlyTest._callr, r( r+ c C s4 | � � j}| j|_t�� |_d|_| �d�� � d S )N)FNz!certonly --webroot -d example.com) rd � notification�_assert_no_pause� side_effectr r r rQ rH )r Z
unused_reportr9 r> Zmock_notificationr r r! � test_no_reinstall_text_pause� s
z)CertonlyTest.test_no_reinstall_text_pauseTc C s | � |� d S rB )� assertFalse)r �message�pauser r r! rh � s z CertonlyTest._assert_no_pause�)certbot.cert_manager.lineage_for_certname�)certbot.cert_manager.domains_for_certnamezcertbot.renewal.renew_certc C s� ddg}||_ ||j_ | �d�� � | �|jdk� | �|jdk� | �|jdk� | �|jdk� | �d�� � | �|jdk� | �|jdk� | �|jdk� | �|jdk� d| �� j_ | �t j
| jd�� � d S )N� example.comztest.orgzEcertonly --webroot -d example.com -d test.org --cert-name example.comr- �Ecertonly --webroot -d example.com -d test.com --cert-name example.comr. F) r �namesrQ rH �
assertTrue�
call_countrd �yesnor[ r �ConfigurationError)r r= �mock_renew_cert� mock_domainsr �domainsr r r! �*test_find_lineage_for_domains_and_certname� s" �z7CertonlyTest.test_find_lineage_for_domains_and_certname� certbot.display.ops.choose_namesc C s` d |_ | �d�� � | �|jdk� | �|jdk� dg|_ d |_ | �d�� � | �|j� d S )Nrq r- Zsomenamez*certonly --webroot --cert-name example.com)r rQ rH rs rt �called)r r= r �mock_choose_namesZmock_domains_for_certnamer r r! �*test_find_lineage_for_domains_new_certname� s z7CertonlyTest.test_find_lineage_for_domains_new_certnameN)T)
r# r$ r% r& r4 rE rQ r r6 rj rh rz r~ r r r r! r` | s&
r` c @ sF e Zd ZdZe�d�dd� �Ze�d�dd� �Ze�d�dd � �Zd
S ) �FindDomainsOrCertnameTestz1Tests for certbot.main._find_domains_or_certname.r{ c C s, t jd d d�}d|_| �t�|d �d� d S )N�ry �certname�
domainname)r� N�r r r r r
�_find_domains_or_certname�r r} � mock_configr r r! �test_display_ops� s
�z*FindDomainsOrCertnameTest.test_display_opsc C s, t jd d d�}g |_| �tjtj|d � d S )Nr� )r r r r[ r �Errorr
r� r� r r r! �test_no_results� s z)FindDomainsOrCertnameTest.test_no_resultsro c C s8 t jd dd�}ddg|_| �t�|d �ddgdf� d S )Nzone.comr� ztwo.comr� )r rx r� r r r! �test_grab_domains� s
�z+FindDomainsOrCertnameTest.test_grab_domainsN) r# r$ r% r& r r6 r� r� r� r r r r! r � s
r c s� e Zd ZdZ� fdd�Z� fdd�Zddd�Ze�d �e�d
�d d � ��Z e�d �e�d
�dd� ��Z
e�d �dd� �Z dd� Z e�d �e�d�e
�� dd� ���Z� ZS )�
RevokeTestz Tests for certbot.main.revoke.c s" t t| ��� t�t| j� tj� tj�
| jd��| _ t | j d��}| j |�
� f| _W 5 Q R X t�d�t�d�t�d�t�d�g| _| jd �� | _| jd �� | jd �� | _| jd
�� | _dd lm} t�� | _|jd tjd
dddddt j d�d�| _ || jt | j �| _ | j d f| j_!d S )Nr �rz'acme.client.BackwardsCompatibleClientV2zcertbot.client.Client� certbot.main._determine_accountz+certbot.main.display_ops.success_revocationr r- r. r/ )�Accountztest.certbot.orgi� � r0 � �
)Ztzinfo)Z
creation_hostZ creation_dt)"r3 r� r4 �shutil�copy� CERT_PATHZtempdirr �path�abspath�join�
tmp_cert_path�open�read�tmp_certr r6 r7 r8 �mock_acme_client�mock_determine_account�mock_success_revokeZcertbot.accountr� � MagicMock�regrZMeta�datetime�pytzZUTC�meta�JWK�accr )r �fr� r@ r r! r4 � s@ ��
��zRevokeTest.setUpc s& t t| ��� | jD ] }|�� qd S rB )r3 r� rE r7 rC rD r@ r r! rE
s
zRevokeTest.tearDownNc C sN |sd}|� | j��� }tj�� }t�t� ||��}ddl
m } |||� d S )Nzrevoke --cert-path={0} r )�revoke) rG r� rH r rI rJ r rK r rL rM r� )r rN rO rP r� r r r! rQ s
� zRevokeTest._call�#certbot.main._delete_if_appropriate� certbot.main.client.acme_clientc C s� d|_ |�� j}g }tj�� D ]l\}}d�| j|��� }| � |� |�
t � t j
|�� d�| j|�� ��� }| � |� |�
t � t j
|�� q | �||j� d S )NFz#revoke --cert-path={0} --reason {1})r �BackwardsCompatibleClientV2r� r ZREVOCATION_REASONS�itemsrG r� rH rQ �appendr Zcall�ANY�upperr �call_args_list)r r� �mock_delete_if_appropriate� mock_revokeZexpected�reason�coderN r r r! �test_revoke_with_reason s
�
z"RevokeTest.test_revoke_with_reasonz'certbot.storage.cert_path_for_cert_namec C s2 d� � }| j|_d|_| �|� | j�| j� d S )Nz revoke --cert-name=example.comF)rH r� r rQ r� rS r� )r Z mock_cert_path_for_cert_namer� rN r r r! �test_revoke_by_certname/ s
z"RevokeTest.test_revoke_by_certnamec C s | � � d|_| j�| j� d S rU )rQ r r� rS r� )r r� r r r! �test_revocation_success9 s z"RevokeTest.test_revocation_successc C s6 ddl m} |�� | j_| �|j| j� | j�� d S )Nr r
) �acmer Z ClientErrorr� ri r[ rQ r� �assert_not_called)r Z acme_errorsr r r! �test_revocation_error? s z RevokeTest.test_revocation_error�certbot.cert_manager.deletec C s( d|� j _d|_| �� | �|j� d S rU )ru r rQ rk r| )r rd � mock_deleter� r r r! �test_revocation_with_promptE s
z&RevokeTest.test_revocation_with_prompt)N)r# r$ r% r& r4 rE rQ r r6 r� r� r� r� ra rb r� r_ r r r@ r! r� � s"
r� c @ sH e Zd ZdZdd� Zdd� Ze�� dd� �Ze�� dd � �Z e
� d
�e
� d �e
� d �e
� d
�e
� d�e�� dd� ������Z e
� d
�e
� d �e
� d
�e
� d �e
� d�e�� dd� ������Z
e
� d
�e
� d �e
� d
�e
� d�e
� d �e�� dd� ������Ze
� d
�e
� d �e
� d
�e
� d�e
� d �e�� dd� ������ZdS )�DeleteIfAppropriateTestz.Tests for certbot.main._delete_if_appropriate c C s ddl m} ||� d S )Nr )�_delete_if_appropriate)rM r� )r r� r� r r r! rQ R s z DeleteIfAppropriateTest._callc C s> t �d��}| �| j� W 5 Q R X |�� | �|� jj� d S )Nr� )r r6 rQ rP r� rs � add_messager| )r rd r� r r r! �_test_delete_opt_out_commonV s z3DeleteIfAppropriateTest._test_delete_opt_out_commonc C s d| j _| �|� d S rU )rP �delete_after_revoker� �r rd r r r! �test_delete_flag_opt_out\ s z0DeleteIfAppropriateTest.test_delete_flag_opt_outc C s |� }d|j _| �|� d S rU )ru r r� )r rd � util_mockr r r! �test_delete_prompt_opt_outa s z2DeleteIfAppropriateTest.test_delete_prompt_opt_outz)certbot.storage.renewal_file_for_certnamer� z-certbot.cert_manager.match_and_check_overlapsz!certbot.storage.full_archive_pathz)certbot.cert_manager.cert_path_to_lineagec C s8 | j }d|_d|_d|_t�� |_| �|� |�� d S )N�/some/reasonable/path� rp ) rP � cert_pathr� r r ZOverlappingMatchFoundri rQ r� )r rd �mock_cert_path_to_lineage� mock_archive� mock_match_and_check_overlapsr� � mock_renewal_file_for_certnamerP r r r! � test_overlapping_archive_dirsh s
z5DeleteIfAppropriateTest.test_overlapping_archive_dirsc C s: | j }d|_d|_d|_d|_| �|� | �|jd� d S )Nr� r� rp Fr- )rP r� r� r rQ r rt )r rd r� r� r� Z mock_overlapping_archive_dirsr� rP r r r! �test_cert_path_only| s
z+DeleteIfAppropriateTest.test_cert_path_onlyc C sH | j }d|j_d|_d|_d|_d|_d|_| �|� | �|jd� d S �NTr� r� rp r- ) rP � namespaceZnoninteractive_moder� r� r rQ r rt �r rd r� r� Zmock_full_archive_dirr� r� rP r r r! � test_noninteractive_deletion� s
z4DeleteIfAppropriateTest.test_noninteractive_deletionc C sX | j }d|j_d|_d|_d|_d|_d|_| �|� | �|jd� | � |� j
j � d S r� ) rP r� r� r� r� r rQ r rt rk ru r| r� r r r! �test_opt_in_deletion� s
z,DeleteIfAppropriateTest.test_opt_in_deletionN)r# r$ r% r& rQ r� ra rb r� r� r r6 r� r� r� r� r r r r! r� O sF
r� c sd e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Ze� d
�d d � �Z
e� d
�dd� �Z dd� Z � Z
S )�DetermineAccountTestz*Tests for certbot.main._determine_account.c sb t t| ��� d | j_d | j_d| j_tjdd�tjdd�g| _ t�
� | _ t�� | _ d| j j
_d S )NF�x)�id�y�hi)r3 r� r4 rP r �emailZ register_unsafely_without_emailr r� �accsZAccountMemoryStorage�account_storage� mock_client� directoryZ new_authzr? r@ r r! r4 � s
zDetermineAccountTest.setUpc
C s@ ddl m} t�d�� }| j|_|| j�W 5 Q R � S Q R X d S )Nr )�_determine_accountz'certbot.main.account.AccountFileStorage)rM r� r r6 r� r rP )r r� Z mock_storager r r! rQ � s zDetermineAccountTest._callc C sn | j �| jd | j� | jd j| j_| �| jd d f| �� � | �| jd j| jj� | � | jj
d k� d S )Nr- ) r� �saver� r� r� rP r r rQ rs r� r? r r r! �test_args_account_set� s
z*DetermineAccountTest.test_args_account_setc C s^ | j �| jd | j� | �| jd d f| �� � | �| jd j| jj� | � | jj
d k� d S )Nr ) r� r� r� r� r rQ r� rP r rs r� r? r r r! �test_single_account� s z(DetermineAccountTest.test_single_accountz)certbot.client.display_ops.choose_accountc C s� | j D ]}| j�|| j� q| j d |_| �| j d d f| �� � | �t|jd d �t| j �� | �| j d j | j
j � | � | j
j
d k� d S )Nr- r )r� r� r� r� r r rQ �set� call_argsr� rP r rs r� )r Zmock_choose_accountsr� r r r! �test_multiple_accounts� s
�z+DetermineAccountTest.test_multiple_accountsz$certbot.client.display_ops.get_emailc C s� d|_ t�d��:}| jd tjjf|j_ | �| jd tjjf| �� � W 5 Q R X |jj | j
| j tj d� | �| jd j
| j
j� | �d| j
j� d S )Nz foo@bar.baz�certbot.main.clientr )Ztos_cb)r r r6 r� �sentinelr� �registerr rQ rS rP r� r� r� r r� )r Zmock_get_email�clientr r r! �test_no_accounts_no_email� s �( �z.DetermineAccountTest.test_no_accounts_no_emailc C sh d| j _t�d��$}| jd tjjf|j_| � � W 5 Q R X | �
| jd j | j j � | �
d| j j� d S )Nz other emailr� r- )
rP r� r r6 r� r� r� r� r rQ r r� r )r r� r r r! �test_no_accounts_email� s z+DetermineAccountTest.test_no_accounts_email)r# r$ r% r& r4 rQ r� r� r r6 r� r� r� r_ r r r@ r! r� � s
r� c sD e Zd ZdZ� fdd�Z� fdd�Zd�dd �Zd�d
d �Zd d
� Zdd� Z dd� Z
e � d�dd� �Z
e � d�e � d�e � d�e � d�e � d�e � d�dd� ������Ze � d �e � d �d d � ��Ze � d �e � d �e � d �d!d"� ���Ze � d�e � d �e � d �e � d �d#d$� ����Ze � d �e � d �d%d&� ��Ze � d �e � d �e � d'�e � d �d(d)� ����Ze � d�e � d�e � d*�d+d,� ���Ze � d�d-d.� �Zd/d0� Ze � d1�d2d3� �Ze � d4�d5d6� �Ze � d7�d8d9� �Ze � d:�d;d<� �Ze � d�d=d>� �Ze � d?�e � d@�dAdB� ��Z e � d?�e � d@�dCdD� ��Z e � d?�e � d@�dEdF� ��Z e � d?�e � d@�dGdH� ��Z dIdJ� Z dKdL� Z!dMdN� Z"dOdP� Z#dQdR� Z$dSdT� Z%dUdV� Z&d�dWdX�Z'e(�)� dYdZ� �Z*e � d[�e(�)� d\d]� ��Z+d^d_� Z,ddd`dde-j-�.� dfdadb�Z/e � d[�dcdd� �Z0e � de�e � d[�dfdg� ��Z1dhdi� Z2djdk� Z3dldm� Z4e � dn�dodp� �Z5e � dq�drds� �Z6e � dq�dtdu� �Z7e � dv�dwdx� �Z8e � d�dydz� �Z9d{d|� Z:d}d~� Z;dd�� Z<d�d�� Z=d�d�� Z>d�d�� Z?d�d�d��Z@d�d�� ZAd�d�� ZBd�d�� ZCd�d�� ZDd�d�� ZEe � d��d�d�� �ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLe(�)� e � d��e � d��d�d�� ���ZMd�d�d��ZNd�d�� ZOd�d�� ZPe � d��e � d��d�d�� ��ZQd�d�� ZRe � d��e � d�d�d�� ��ZSe � d�d�d�� �ZTd�d�� ZUd�d�� ZVd�d�� ZWe � d��e(�)� d�d�� ��ZXe � d��e(�)� d�d�� ��ZYe � d��e � d��d�dÄ ��ZZ� Z[S )��MainTestz Tests for different commands.c sP t t| ��� t�| jj� d| jjd| jjd| jjdg| _ t
� d�� � | _
d S )Nz --config-dirz
--work-dirz
--logs-dirz--textz
time.sleep)r3 r� r4 r �mkdirrP �logs_dir�
config_dir�work_dir�
standard_argsr r6 r8 �
mock_sleepr? r@ r r! r4 s �zMainTest.setUpc s t t� tt| ��� d S rB )r r r3 r� rE r? r@ r r! rE s zMainTest.tearDownNFc s� |r~t jj� � fdd�}t�d��T}||_t�d��8}| �||�\}}}||||fW 5 Q R � W 5 Q R � S Q R X W 5 Q R X n<t�d��,}| �||�\}}}||||fW 5 Q R � S Q R X dS )zaRun the cli with output streams, actual client and optionally
os.path.isfile() mocked outc s* | � d�s | � d�s | � d�r"dS � | �S )zMock os.path.isfile()�cert�chain�privkeyT)�endswith)�fnrN �kwargs�Z orig_openr r! � mock_isfile s
��z#MainTest._call.<locals>.mock_isfilez certbot.compat.os.path.isfiler� N)r r� �isfiler r6 ri �_call_no_clientmock) r rN �stdout�
mockisfiler� Zmock_ifr� r �stderrr r� r! rQ s 6 zMainTest._callc C s| | j | }|r|nt�� }tjd|d��D t�d��.}t�d�� t�|dd� �}W 5 Q R X W 5 Q R X W 5 Q R X |||fS )z-Run the client with output streams mocked out�certbot.main.sys.stdout��new�certbot.main.sys.stderrzcertbot.util.atexitN)r� �six�StringIOr r6 r
)r rN r� Z
toy_stdoutr� r r r r! r� * s
0z MainTest._call_no_clientmockc C s2 t �d�� }| �g � | �d|j� W 5 Q R X d S )Nzcertbot.main.runr- )r r6 rQ r rt )r �mock_runr r r! �
test_no_flags5 s
zMainTest.test_no_flagsc C s� t �� }t �� }tjd|d��t tjd|d��Z z*zt�dg� W n tk
rV Y nX W 5 |�� pj|�� }| �d|kd�|�� X W 5 Q R X W 5 Q R X d S )Nr� r r �certbotz
Output is {0}z --version) r r r r6 �getvaluers rG r
�
SystemExit)r Ztoy_outZtoy_err�outputr r r! � test_version_string_program_name: s
z)MainTest.test_version_string_program_namec
C s� d}z2t �d�� t�| j|dd� � W 5 Q R X W n: tjk
rp } z|}| �|t|�k� W 5 d}~X Y nX | �|dk � dS )zQEnsure that a particular error raises a missing cli flag error containing messageNr )r r6 r
r� r ZMissingCommandlineFlagrs �str)r rN rl �excZexc_r r r! �_cli_missing_flagG s &$zMainTest._cli_missing_flagz certbot.log.post_arg_parse_setupc C s4 ddg}| � |d� |�dddg� | � |d� d S )Nz-nrf zspecify a plugin� --standalone�-dzeg.iszregister before running)r �extend�r �_rN r r r! �test_noninteractiveR s z MainTest.test_noninteractiver+ z&certbot.main.client.acme_client.Clientr� z8certbot.main.client.Client.obtain_and_enroll_certificater( c
C s� dddddddg| j }t�� d f|_t�� |_t�d��b}| �|� t�� } |jd d
}
| � | |
k� d d l
} | �
� } d | � � kr�| � t�� |
k� W 5 Q R X t�d��6}d
}
|d|
g7 }| �|� |j tj
tj
d|
d� W 5 Q R X d S )Nr rf z-mz
none@none.comr rp z --agree-tosz-certbot.main.client.acme_client.ClientNetworkr- �
user_agentr ZlinuxZ bandersnatchz --user-agentT)r Z
verify_sslr )r� r r� r r6 r� r Zget_os_info_uar� rs �platform�lowerrS r� )
r ZgscZ_obtZdetZ_clientr �__rN Zacme_netZos_verZuar Zplatr r r! �test_user_agentY s2 ��
�zMainTest.test_user_agent�+certbot.main.plug_sel.record_chosen_plugins�$certbot.main.plug_sel.pick_installerc
C s2 | j ddddddddd g d
d � | �|jd � d S )
N�installz --domains�foo.bar� --cert-pathr� �
--key-pathr� � --chain-pathr� T�r� r- )rQ r rt )r Zmock_pick_installer�_recr r r! �test_installer_selectionx s ��z!MainTest.test_installer_selectionzcertbot.main._install_certc C s� t jt�d�t�d�t�d�t�d�d�}t �d��j}||_| jdddgd d
� |jd d }| �|j t�d�� | �|j
t�d�� | �|j t�d�� W 5 Q R X d S ) Nr� r� r� �r� �
chain_path�fullchain_path�key_pathrn r � --cert-name�whateverTr! r ) r r� ra � temp_joinr6 r rQ r� r r� r& r' )r �_instr" � mock_installr � mock_getlin� call_configr r r! �test_installer_certname s � z MainTest.test_installer_certnamec C s6 t jt�d�t�d�t�d�t�d�d�}t �d���}||_| jdddd t�d
�gd d � |jd
d
}| �|j t�d�� | �|j
t�d�� | �|j t�d�� | �|j t�d
�� |�
� | jddddt�d�gd d � |jd
d
}| �|j t�d�� | �|j
t�d�� | �|j t�d�� W 5 Q R X d S )Nr� r� r� r$ rn r r( r) r Zoverriding_privkeyTr! r r Zoverriding_cert)r r� ra r* r6 r rQ r� r r� r& r% r' �reset)r r+ r" r, r r r- r. r r r! � test_installer_param_override� s: �
��
��z&MainTest.test_installer_param_overridec C s | � tj| jdddddg� d S )Nr r( Znotfoundr Zinvalid�r[ r rv rQ )r r+ r" r r r! �test_installer_param_error� s ��z#MainTest.test_installer_param_error�"certbot.cert_manager.get_certnamesc C sr t jt�d�t�d�t�d�t�d�d�}t �d�� }||_| jdgdd� W 5 Q R X | �|j� | �|j� d S ) Nr� r� r� r$ rn r Tr! ) r r� ra r* r6 r rQ rs r| )r � mock_instZ mock_getcertr+ r" r r- r r r! �test_installer_select_cert� s � z#MainTest.test_installer_select_certzcertbot.util.exe_existsc
C s$ d|_ tj�� }dddg}dddt�d�d t�d�d
d d d
dg }d|krv| �|�\}}}}| �d|k� | �d|k� | �dgd� t �
d��Z}t �
d��D}t � � |_ | �ddd dg� |j d \} }
} | �t
|
tj�� W 5 Q R X W 5 Q R X t �
d��"} | �ddg� | �d | j� W 5 Q R X d S ) NTz--apachez--authenticator�
standaloner z--nginxr Zblahr z--nginx-server-rootz/nonexistent/thingr rp �--debugZnginxz The nginx plugin is not working�MisconfigurationErrorr z(With the standalone plugin, you probablyr) r( rf z--manualr r �certbot.main.certonly�authr- )r r rI rJ ra r* rQ rs r r r6 r� r� �
isinstancer Z
Authenticatorr rt )
r Zmock_exe_existsr r Z real_pluginsrN r r< Zmock_gscZ
unused_configr; Zunused_installer�
mock_certonlyr r r! �test_configurator_selection� s6
�
& z$MainTest.test_configurator_selectionc C sX | � dg�\}}}}| �d|jj� | � dddg�\}}}}|j�tjdtjtj� d S )N�rollbackr- z
--checkpointsZ123�{ )rQ r r? rt rS r r� �r r r� r r r! �
test_rollback� s �zMainTest.test_rollbackc C s( | � dg�\}}}}| �d|jj� d S )N�config_changesr- )rQ r Zview_config_changesrt rA r r r! �test_config_changes� s z MainTest.test_config_changeszcertbot.main.logger.warningc C s6 | � dg� | �|j� |jd d }| �d|� d S )NrC r z-config_changes subcommand has been deprecated)rQ rs r| r� ZassertIn)r Z mock_warning�msgr r r! � test_config_changes_deprecation� s z(MainTest.test_config_changes_deprecationz)certbot.cert_manager.update_live_symlinksc C s | � dg� | �d|j� d S )NZupdate_symlinksr- �r� r rt �r Zmock_cert_managerr r r! �test_update_symlinks� s z MainTest.test_update_symlinksz!certbot.cert_manager.certificatesc C s | � dg� | �d|j� d S )NZ certificatesr- rG rH r r r! �test_certificates s zMainTest.test_certificatesr� c C s | � dg� | �d|j� d S )N�deleter- rG rH r r r! � test_delete s zMainTest.test_deletec sN ddddg� t j� fdd�tj�t� ��D �� D ]}| �dgt|� � q0d S )N�--init� --preparez--authenticatorsz --installersc 3 s | ]}t �� |�V qd S rB )� itertools� combinations)�.0r� ��flagsr r! � <genexpr> s �z(MainTest.test_plugins.<locals>.<genexpr>rO )rO r� r Zmoves�range�lenrQ �listr r rR r! � test_plugins s
��zMainTest.test_pluginszcertbot.main.plugins_discoz<certbot.main.cli.HelpfulArgumentParser.determine_help_topicsc C s� g }|j �� }t�� }tj|d�� | �dg|�\}}}}W 5 Q R X |j�� |�� j �|� |�� � � }| �
|� � � � t
|�� d S )N�r� rO )rI rJ r r ra � patch_get_utility_with_stdoutrQ �visiblerS �ifacesr r �stripr )r �_det�
mock_discor\ rO r� r �filteredr r r! �test_plugins_no_args s
z MainTest.test_plugins_no_argsc
C s� g }|j �� }dd� }t�� }t�d��:}tj|d��" ||_| � dg|�\}}}}W 5 Q R X W 5 Q R X |j
� � |�
� j � |� |�
� � � } | �
|�� �� t| �� d S )Nc S s | || }}}t �� �dS )zRaises error.Error.N)r r� )r� �mode�strictr r r r! � throw_error+ s z?MainTest.test_plugins_no_args_unprivileged.<locals>.throw_errorz certbot.util.set_up_core_dirrY rO )rI rJ r r r r6 ra rZ ri rQ r[ rS r\ r r r] r )
r r^ r_ r\ rO rd r� Zmock_set_up_core_dirr r` r r r! �!test_plugins_no_args_unprivileged% s
*
z*MainTest.test_plugins_no_args_unprivilegedc C s� g }|j �� }t�� }tj|d�� | �ddg|�\}}}}W 5 Q R X |j�� |�� j �|� |�� � � }| �
|j j d� |j
�|� |�
� }| �
|�� �� t|�� d S )NrY rO rM r- )rI rJ r r ra rZ rQ r[ rS r\ r �initrt �verifyr r] r ) r r^ r_ r\ rO r� r r` �verifiedr r r! �test_plugins_init; s
"
zMainTest.test_plugins_initc
C s� g }|j �� }t�� }tj|d�� | �dddg|�\}}}}W 5 Q R X |j�� |�� j �|� |�� � � }| �
|j j d� |j
�|� |�
� }|j�� |j�� |�� } | �
|�� �� t| �� d S )NrY rO rM rN r- )rI rJ r r ra rZ rQ r[ rS r\ r rf rt rg Zprepare� availabler r] r )
r r^ r_ r\ rO r� r r` rh rj r r r! �test_plugins_prepareM s
$
z MainTest.test_plugins_preparec C s� d}d}d}d}t �d��"}| �dd|ddd dd
dg � W 5 Q R X |jd \}}| �|jtj�|�� | �|j tj�|�� | �|j
tj�|�� | �|j tj�|�� d S ) Nr� �keyr� � fullchainr: rf r r r z--fullchain-pathr ) r r6 rQ r� r r� r r� r� r' r% r& )r r� rl r� rm r= rP �unused_pluginsr r r! �test_certonly_abspathb s �z MainTest.test_certonly_abspathc
C s\ z | � dddg� ds td��W n6 tjk
rV } z| �dt|�k� W 5 d }~X Y nX d S )N�-aZbad_authrf Fz!Exception should have been raisedz-The requested bad_auth plugin does not appear)rQ �AssertionErrorr �PluginSelectionErrorrs r �r �er r r! �test_certonly_bad_argss s
z MainTest.test_certonly_bad_argsc C sF | � tj| jddg� | � tj| jddg� | � tj| jddg� d S )Nr Z@aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa� aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa.z
204.11.231.35r2 r? r r r! � test_check_config_sanity_domainz s ���z(MainTest.test_check_config_sanity_domainc C s | � tj| jd�t��� � d S )Nz*certonly --csr {0} --allow-subset-of-names�r[ r r� rQ rG �CSRrH r? r r r! �test_csr_with_besteffort� s
�z!MainTest.test_csr_with_besteffortc
C s^ z| � dtg� W n: tjk
rL } zdt|�ks6t�W Y �d S d }~X Y nX dsZtd��d S )Nz--csrzPlease try the certonlyFz2Expected supplying --csr to fail with default verb)rQ rx r r� �reprrq rs r r r! �test_run_with_csr� s zMainTest.test_run_with_csrc C s& | � tj| jd�t�d���� � d S )Nzcertonly --csr {0}zcsr-nonames_512.pem)r[ r r� rQ rG ra � vector_pathrH r? r r r! �test_csr_with_no_domains� s ��z!MainTest.test_csr_with_no_domainsc C s | � tj| jd�t��� � d S )Nz!certonly -d example.org --csr {0}rw r? r r r! �"test_csr_with_inconsistent_domains� s
�z+MainTest.test_csr_with_inconsistent_domainsc
C s^ t �d��J}d|_t �d��.}||_|d kr0g }|d�� 7 }| �|� W 5 Q R X W 5 Q R X d S )N�3certbot.main._find_lineage_for_domains_and_certname)ZnewcertNr) z!-d foo.bar -a standalone certonly)r r6 r rH rQ )r r� rN � mock_renewalr< r r r! � _certonly_new_request_common� s z%MainTest._certonly_new_request_commonc C s` t �� }d |j_| �|dg� | �|jjd� | �d|� jj d d k� | �|� jjd� d S �N� --dry-runr- �dry runr )
r r� � obtain_and_enroll_certificater r� r rt rs r� r� )r rd r� r r r! �)test_certonly_dry_run_new_request_success� s ��z2MainTest.test_certonly_dry_run_new_request_successz certbot.crypto_util.notAfterc C s� t j�t j�| jjd��}t j�t j�| jjd��}d}||� j_tj ||||d�}t� � }||j
_| � |� | � |j
j
d� |� jjd d d }| �||k� | �||k� | �||k� | �d|� jjd d k� d S )Nz live/foo.barz live/baz.quxz
1970-01-01)r� rm r& r' r- r �donate)r r� �normpathr� rP r� �dater r r� r� r� r rt r� r� rs r� ) r rd Z
mock_notAfterr� r' r� r r� �cert_msgr r r! �!test_certonly_new_request_success� s,
�
��z*MainTest.test_certonly_new_request_successc C s& t �� }d|j_| �tj| j|� d S rU )r r� r� r r[ r r� r� )r r� r r r! �!test_certonly_new_request_failure� s �z*MainTest.test_certonly_new_request_failureTc
s� t �d�}
tj�tj�| jjd��} tj |
| |
| d�} || j
_ d| j _ dg| j
_ t� � }
tj dd�}t� � }t�� � |
d|d f|j_ � fd
d �}�z��z�t�d
���"}| d f|_ t�d��� }||_ t �� ��}|s�||� j_t�d���}t� � }d|j_ ||jj_ t�d���}||j _ |�s*dddddg}|�r8||7 }z>| � |� �\}� }}|�rft d|� t |��|�rtt d��W n" |�s�t dt �!� ��Y nX W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X |�r| �r |j�"dgtj�tj�| jjd��� n|j�"dgd � n| �#|jj$d� W n | �%� � Y nX W 5 |�rzttj�| jjd ���}| �||�� k� W 5 Q R X X | |� fS )Nr zlive/foo.bar/fullchain.pem)r� rm r� r& Fz isnot.orgZpem_key)Zpemr� Zcsrc s � � | � dS )zWrite message to stdout.N)�write)rl rN r� rY r r! � write_msg� s z0MainTest._test_renewal_common.<locals>.write_msg�letsencrypt.logz+certbot.cert_manager.find_duplicative_certsr) z certbot.main.renewal.OpenSSLz Fake fakez certbot.main.renewal.crypto_utilr rp r7 rf ZReturnedzrenewal should have erroredzUnexpected renewal error:
z live/sample-renewal/privkey.pemr )&ra r| r r� r� r� rP r� r r� Zshould_autorenewr Zhas_pending_deploymentrr r r �obtain_certificater� r� rs r� r6 rb rg ri Z
get_issuerZcryptoZload_certificateZnotAfterrQ �printrq � traceback�
format_excrS r rt � _dump_log)r �due_for_renewal�
extra_args�log_outrN � should_renew�error_expected�
quiet_mode� expiry_date� reuse_keyr� r% r Z
mock_certrZmock_keyr� r� �lfZmock_fdcr< rd Zmock_sslZ mock_latestZmock_crypto_utilr r r rY r! �_test_renewal_common� s�
� �
�
��<
�� z MainTest._test_renewal_commonc C sv | � dg �\}}}| �|jjd� |j�|�� � |� jjd d d }| � d|k� | � d|� jj
d d k� d S )NTr- r �
fullchain.pemr� ) r� r Zsave_successorrt Zupdate_all_links_torS Zlatest_common_versionr� r� rs r� )r r Zlineage� get_utilityr� r r r! �test_certonly_renewal s �z MainTest.test_certonly_renewalz;certbot.log.logging.handlers.RotatingFileHandler.doRolloverc C s� | j dddgdd�\}}}| �|� jjd� | �d|� jjd d k� | j dd d
d gd d� | �|� jjd� | j dd
d dgd
dd� d S )NFr� z--keepzsimulating renewal)r� r- r� r z--renew-by-default�-tvvr8 zAuto-renewal forcedz not yet due)r� r� )r� r r� rt rs r� )r r r r� r r r! � test_certonly_renewal_triggers( s � � �z'MainTest.test_certonly_renewal_triggersc C sJ t d� tj�| jjd�}tj�|�rFt|��}t |�� � W 5 Q R X d S )NzLogs:r� ) r� r r� r� rP r� �existsr� r� )r Zlog_pathr� r r r! r� 8 s
zMainTest._dump_logc C s0 t �| jjd� dddg}| jdg |dd� d S )N�sample-renewal.conf�renewr� r� T�rN r� �ra � make_lineagerP r� r� �r rN r r r! �test_renew_verb? s
zMainTest.test_renew_verbc C s2 t �| jjd� dddg}| jdg |ddd� d S )Nr� r� r� � --reuse-keyT�rN r� r� r� r� r r r! �test_reuse_keyD s
zMainTest.test_reuse_keyz,certbot.storage.RenewableCert.save_successorc C s0 t �| jjd� ddg}| jdg |ddd� d S )Nr� r� r� Tr� r� )r Zunused_save_successorrN r r r! �test_reuse_key_no_dry_runI s z"MainTest.test_reuse_key_no_dry_runz sys.stdinc C sv d|j _t�| jjd� dddg}| jdg |dd� | �| jj d� | jj
d d }| � d| kojd
kn � d S ) NFr� r� r� r� Tr� r- r i� ) �isattyr ra r� rP r� r� r r� rt r� rs )r �stdinrN Zsleep_call_argr r r! �!test_noninteractive_renewal_delayO s
z*MainTest.test_noninteractive_renewal_delayc C sH d|j _t�| jjd� dddg}| jdg |dd� | �| jj d� d S )NTr� r� r� r� r� r )
r� r ra r� rP r� r� r r� rt )r r� rN r r r! �!test_interactive_no_renewal_delay[ s
z*MainTest.test_interactive_no_renewal_delayz certbot.renewal.should_renewc C sp d|_ t�| jjd� tj�� tjdd� }| jdd ddg|d�\}}}| � d|�
� k� | � d|�
� k� d S ) NFr� �Z )Zdaysr� )r� r� rN r� zNo renewals were attempted.z0The following certs are not due for renewal yet:) r ra r� rP r� r� �nowZ timedeltar� rs r )r r� Zexpiryr r� r r r! � test_renew_skips_recent_certsc s
� z&MainTest.test_renew_skips_recent_certsc C s� t �| jjd� ddg}| jdg |dd�\}}}|�� }| �d|k� dddg}| jdg |ddd�\}}}|�� }| �d|� d S ) Nr� r� r� Tr� z-q)rN r� r� r� )ra r� rP r� r� r rs r )r r rN r� �outr r r! �test_quiet_renewm s
� zMainTest.test_quiet_renewc C s2 t �| jjd� dddg}| jdg |ddd� d S )Nr� r� r� �--post-hook=no-such-commandTF�rN r� r� r� r� r r r! �test_renew_hook_validation{ s
�z#MainTest.test_renew_hook_validationc C sJ t �| jjd� ddddg}t�d�� | jdg |ddd � W 5 Q R X d S )
Nr� r� r� r� z--disable-hook-validationzcertbot.hooks.post_hookTFr� )ra r� rP r� r r6 r� r� r r r! � test_renew_no_hook_validation� s � �z&MainTest.test_renew_no_hook_validationc C sj t j�| jjd�}t j�|�s(t�|� tt j�|d�d�� W 5 Q R X dddg}| j dg |ddd � d S )
N�renewalz
empty.conf�wr� r� r� FTr� )
r r� r� rP r� r� r �makedirsr� r� )r ZrdrN r r r! � test_renew_verb_empty_config� s
z%MainTest.test_renew_verb_empty_configc C s. t �| jjd� | jdg dddddgd� d S )Nr� Tr� r� r( �sample-renewal)r� rN r� r? r r r! �test_renew_with_certname� s
�z!MainTest.test_renew_with_certnamec C s | j dg dddddgdd� d S )NTFr� r� r( r� )r� rN r� )r� r? r r r! � test_renew_with_bad_certname� s
�z%MainTest.test_renew_with_bad_certnamec C sJ t j�| jjd�}t�|� tt j�|d�d��}|�d� W 5 Q R X d S )Nr� z test.confr� zMy contents don't matter) r r� r� rP r� r r� r� r� )r Zrenewer_configs_dirr� r r r! �_make_dummy_renewal_config� s
z#MainTest._make_dummy_renewal_configc
K s� | � � t�d���}t�� }d|_|d k r4d|i|_|d k rD||j_||_t�d��(}|�ddg� | j d
dd i|�� W 5 Q R X |d k r�|r�| �
|j � n | � |j � W 5 Q R X d S ) N� certbot.storage.RenewableCertzsomepath/fullchain.pem�
renewalparams�certbot.main.renew_certrN r� Tr� F)TN)
r� r r6 r� rm r rr r �
setdefaultr� rs r| rk )r r� rr �assert_oc_calledr� �mock_rcr rw r r r! �_test_renew_common� s
zMainTest._test_renew_commonc C s | j ddd� d S )NFT�r� r� �r� r? r r r! �test_renew_no_renewalparams� s z$MainTest.test_renew_no_renewalparamsc C s | j i ddd� d S )NFT)r� r� r� r� r? r r r! �test_renew_no_authenticator� s �z$MainTest.test_renew_no_authenticatorc C s ddd�}| j |ddd� d S )N�webrootz over 9000)�
authenticatorZ rsa_key_sizeTF)r� r� r� r� �r r� r r r! �test_renew_with_bad_int� s ��z MainTest.test_renew_with_bad_intc C s ddd�}| j |dd� d S )Nr� �None)r� Z http01_portT�r� r� r� r� r r r! � test_renew_with_nonetype_http01� s ��z(MainTest.test_renew_with_nonetype_http01c C s$ ddi}dg}| j |d|dd� d S )Nr� r� u
uniçodé.comTF)r� r� rr r� r� )r r� rr r r r! �test_renew_with_bad_domain� s �z#MainTest.test_renew_with_bad_domainz5certbot.plugins.selection.choose_configurator_pluginsc C s2 t �� t �� f|_ddi}| j|dd�� d� d S )Nr� r� Tzrenew --configurator apache�r� r� rN )r r� r r� rH )r Zmock_selr� r r r! � test_renew_with_configurator� s �z%MainTest.test_renew_with_configuratorc C s dddd�}| j |dd� d S )Nr� r� Z42)r� Z webroot_pathZwebroot_imaginary_flagTr� r� r� r r r! �$test_renew_plugin_config_restoration� s ��z-MainTest.test_renew_plugin_config_restorationc
C s0 ddi}| j |dddt�dt�� i�gd� d S )Nr� r� Tr� z
--webroot-maprp r� )r� �json�dumps�tempfileZ
gettempdirr� r r r! �test_renew_with_webroot_map� s �z$MainTest.test_renew_with_webroot_mapc C s. t �d��}t|_| jddd� W 5 Q R X d S )Nz"certbot.main.renewal._reconstituteFTr� )r r6 � Exceptionri r� )r Zmock_reconstituter r r! � test_renew_reconstitute_error� s z&MainTest.test_renew_reconstitute_errorc
C sv | � � t�d��Z}t�� }d|_||_dddii|_t�d��"}t|_| j dd ddgd d
� W 5 Q R X W 5 Q R X d S ) Nr� zsomewhere/fullchain.pemr� r� r� r� Tr� F)r� rN r� )
r� r r6 r� rm r r r� ri r� )r r� r rw r r r! � test_renew_obtain_cert_error� s �
�z%MainTest.test_renew_obtain_cert_errorc C s: | j dd d�� ddd� | j dd d�t��� ddd� d S )NTzrenew -d example.comFr� zrenew --csr {0})r� rH rG rx r? r r r! � test_renew_with_bad_cli_args� s � �z%MainTest.test_renew_with_bad_cli_argsc C s< | j dd dddd�tj�gd�\}}}| �d|�� k� d S )NFr� z --post-hookzE{0} -c "from __future__ import print_function; print('hello world');")r� r� r� rN zNo hooks were run.)r� rG �sys�
executablers r )r r r� r r r! �test_no_renewal_with_hooks� s ��� z#MainTest.test_no_renewal_with_hooksr r) c C sb dt �� f|_t �� |_}| �dddddg� | �|jj� | �|jj� | �|� j j
d� d S )Nr r r rp r7 rf r ) r r� r rQ rk r� r| r� r r� rt )r r< r� rd r� r r r! �test_certonly_reinstall s z MainTest.test_certonly_reinstallc C s d}d}t �� }||f|j_tj�tj�| jj d��}tj�tj�| jj d��}|d |f|j
_t � d��z}||_t �
� �`}tj�tj�| jj d��} d�t|| |��� }
|r�|
|7 }
t � d�� | �|
� W 5 Q R X W 5 Q R X W 5 Q R X d |
k�r| �|j
j� n|j
�|||| |� |S )
N�certrr� z live/example.com/cert_512.pemz live/example.com/fullchain.pemr) zlive/example.com/chain.pemzV-a standalone certonly --csr {0} --cert-path {1} --chain-path {2} --fullchain-path {3}zcertbot.main.crypto_utilr� )r r� Zobtain_certificate_from_csrr r r� r� r� rP r� Zsave_certificater6 ra rb rG rx rH rQ rk r| rS ) r r� r� r� r� r� � full_pathr< rd r% rN r r r! �_test_certonly_csr_common sP � �
� � (
�z"MainTest._test_certonly_csr_commonc C sZ | � � }|� jjd d d }| �d|k� | �d|k� | �d|� jjd d k� d S )Nr r� z Your key file has been saved atr� )r� r� r� rs rk r� )r rd r� r r r! �test_certonly_csr4 s �zMainTest.test_certonly_csrc C s> | � dg�}| �|� jjd� | �d|� jjd d k� d S r� )r� r r� rt rs r� r� r r r! �test_certonly_csr_dry_run< s
�z"MainTest.test_certonly_csr_dry_runr� r� c C s� d|_ d}| �dtdtd|dg� ttd��$}|j�tjt j
� |� � �|� W 5 Q R X ttd��6}t
�|� � �d }|�� j}|�t �|�tj� W 5 Q R X d S ) NFr r r �--serverr� �rbr )r r� � SS_CERT_PATH�RSA2048_KEY_PATHr� r� rS r r� �joser� �loadr� r �pyopenssl_load_certificater� �ComparableX509)r r� r� �serverr� r� r� r r r! �test_revoke_with_keyB s( � �
�z MainTest.test_revoke_with_keyc C s( d}| � tj| jdtdtd|dg� d S )Nr r r r� r� )r[ r r� r� �CERT�KEY)r r� r r r! � test_revoke_with_key_mismatchT s ��z&MainTest.test_revoke_with_key_mismatchc C st d|_ t�� d f|_ | �dtdg�\}}}}tt��6}t�|�� �d }|� � j
}|� t �
|�tj� W 5 Q R X d S )NFr r� r )r r r� rQ r� r� r r� r� Zacme_from_config_keyr� rS r� r� r� )r r� r� r r� r� r� r� r r r! �test_revoke_without_keyZ s
�z MainTest.test_revoke_without_keyc
C s� t �d���}t �� }d|_|df|j_| �dddg� t �d��B}t �� }||j_dg|j_| �dddg�}| � d |d
k� W 5 Q R X W 5 Q R X d S ) Nr� Zimaginary_accountZworkedr� �--email�user@example.org�certbot.main.account�
an accountz There is an existing accountr )
r r6 r� r� r� r r� �AccountFileStoragerJ rs )r r �
mocked_clientr� �mocked_account�mocked_storager� r r r! �
test_registerh s
zMainTest.test_registerc C sT t �d��@}t �� }||j_g |j_| �dddg�}| �d|d k� W 5 Q R X d S )Nr� �update_accountr� r� �"Could not find an existing accountr �r r6 r� r� r rJ r� rs �r r r r� r r r! �(test_update_account_no_existing_accountsy s ��z1MainTest.test_update_account_no_existing_accountsc C sV t �d��B}t �� }||j_g |j_| �ddddg�}| �d|d k� W 5 Q R X d S )Nr� r� �--update-registrationr� r� r r r r r r r! �8test_update_registration_no_existing_accounts_deprecated� s ��zAMainTest.test_update_registration_no_existing_accounts_deprecatedc C sT t �d��@}t �� }||j_dg|j_| �d�� �}| �d|d k� W 5 Q R X d S )Nr� r� z@register --update-registration --register-unsafely-without-emailz!--register-unsafely-without-emailr ) r r6 r� r� r rJ r� rH rs r r r r! �,test_update_registration_unsafely_deprecated� s
�z5MainTest.test_update_registration_unsafely_deprecatedz"certbot.main.display_ops.get_emailc C s d}||_ t�d���}t�d���}t�d���}t�d���}t�� }||j_ dg|j_ t�� df|_ t�� } | |j_ | �dg�}
| �|
d d k� | �| j j
j � | �|j j � | �||� j
jd d k� | �|j � W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X d S )
N�user@example.com� certbot.eff.handle_subscriptionr� r� r� r� �foor r )r r r6 r� r� rJ �Clientr� rs r� �update_registrationr| � save_regrr� r� ) r � mock_utility�
mock_emailr� � mock_handle�
mocked_detr r� r � cb_clientr� r r r! � test_update_account_with_email� s0
���z'MainTest.test_update_account_with_emailc C s" d}||_ t�d���}t�d���}t�d���}t�d���}t�� }||j_ dg|j_ t�� } | j}
| df|_ t�� } | |j_ | �dd g�} | � | d
d k� | j
j j d
d
}
| �
|
|
�� � | � |jj� | � ||� jj d
d
k� | � |j� W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X d S ) Nr r r� r� r� r� r
r� r r )r r r6 r� r� rJ r� r r� rs r� r r� r �updater r| r� )r r r r� r r r r� r Zmock_accZ mock_regrr r� Zreg_argr r r! �.test_update_registration_with_email_deprecated� s2
��z7MainTest.test_update_registration_with_email_deprecatedz certbot.updater._run_updatersc C sF t j|_| �t jtjd d d � d| j_t� | jd d � | �
|j � d S rU ) r rr ri r[ r
Z
renew_certrP Zdry_runr Zrun_generic_updatersrk r| )r r r] r r r! �test_plugin_selection_error� s �z$MainTest.test_plugin_selection_error)NF)N)N)NNN)N)\r# r$ r% r& r4 rE rQ r� r r r r r6 r r r# r/ r1 r3 r6 r> rB rD rF rI rJ rL rX ra re ri rk ro ru rv ry r{ r} r~ r� ra rb 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� 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! r� � s8
%
�
K
�
"
r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �UnregisterTestc C sD t �d�t �d�t �d�t�� d�| _tdd� | j�� D ��| _d S )Nr� r� r� )r� r r� r� c s s | ]\}}||� � fV qd S rB )r8 )rQ �k�vr r r! rT � s z'UnregisterTest.setUp.<locals>.<genexpr>)r r6 ra rb �patchers�dictr� �mocksr? r r r! r4 � s �zUnregisterTest.setUpc C s | j �� D ] }|�� q
d S rB )r �valuesrC rD r r r! rE � s zUnregisterTest.tearDownc C sR t �� | jd j_| jd � }d|j_t �� }t �� }t�||�}| �|d� d S )Nr r� FzDeactivation aborted.) r r r r� r ru r
�
unregisterr )r r� rP rn �resr r r! �test_abort_unregister� s z$UnregisterTest.test_abort_unregisterc C s� t �� }dg|j_|| jd j_t �� df| jd _t �� }|| jd j_t �� }t �� }t�||�}| � |d k� | � |j
j j � d}| � || jd � j
jd d k� d S ) Nr� r r
r� r� zAccount deactivated.r� r )r r� rJ r r r� r r
r! rs r� �deactivate_registrationr| r� r� �r r r rP rn r" �mr r r! �test_unregister s
z UnregisterTest.test_unregisterc C st t �� }g |j_|| jd j_t �� }|| jd j_t �� }t �� }t�||�}d}| � ||� | �
|j j j
� d S )Nr r� z.Could not find existing account to deactivate.)r r� rJ r r r� r r
r! r rk r� r$ r| r% r r r! �test_unregister_no_account s z)UnregisterTest.test_unregister_no_accountN)r# r$ r% r4 rE r# r' r( r r r r! r � s
r c @ s" e Zd ZdZe�d�dd� �ZdS )�MakeOrVerifyNeededDirsz2Tests for certbot.main.make_or_verify_needed_dirs.zcertbot.main.utilc C sp t �| j� | jj| jjfD ]}|j�|tj| jj � q | jj
| jj | jj f}|D ]}|j
j|| jj d� qRd S )N)rc )r
Zmake_or_verify_needed_dirsrP r� r� Zset_up_core_dirZassert_any_callr ZCONFIG_DIRS_MODEZstrict_permissionsZrenewal_pre_hooks_dirZrenewal_deploy_hooks_dirZrenewal_post_hooks_dirZmake_or_verify_dir)r Z mock_utilZcore_dirZ hook_dirsZhook_dirr r r! �test_it0 s �� �z MakeOrVerifyNeededDirs.test_itN)r# r$ r% r& r r6 r* r r r r! r) - s r) c s� e Zd ZdZ� fdd�Zdd� Zdd� Ze�d�e�d �e�d
�e�d �d d
� ����Z e�d�e�d �e�d
�e�d �dd� ����Z
e�d �e�d
�e�d�dd� ���Z e�d �e�d
�e�d�dd� ���Z e�d
�e�d�dd� ��Z
dd� Ze�d�e�d
�e�d�dd� ���Ze�d �e�d
�e�d�e�d�e�� d d � �����Ze�d �e�d
�e�d�e�d�e�� d d � �����Zd d!� Z� ZS )"� EnhanceTestz Tests for certbot.main.enhance.c s8 t t| ��� t�� | _| j�� | _tj t
j d�| _ d S �N)�spec)
r3 r+ r4 ra rb rc r8 rd r r� r �AutoHSTSEnhancement�
mockinstallerr? r@ r r! r4 E s
zEnhanceTest.setUpc C s | j �� d S rB re r? r r r! rE K s zEnhanceTest.tearDownc C s� t j�� }t�t�||��}t�d���}dg|_ t�d��h}dg|_ t�d��J}t�
� }||_ ||_ t �
||� |W 5 Q R � W 5 Q R � W 5 Q R � S Q R X W 5 Q R X W 5 Q R X d S )Nr4 rp ro r) )r rI rJ r rK r rL r r6 r r� rP r
�enhance)r rN rO rP Z
mock_certsZmock_domr< r� r r r! rQ N s
� zEnhanceTest._callr rn z&certbot.main.display_ops.choose_valuesz&certbot.main._find_domains_or_certnamec C sh t jdd�|_dg|_d|_t �d��8}| �ddg� | �|j� | �d|jd d
k� W 5 Q R X d S ) N�/tmp/nonexistent�r% rp �NNr r0 �
--redirectr r r/ )r r� r r6 rQ rs r| r� )r � mock_findr] r r" � mock_pickr r r! �test_selection_question^ s z#EnhanceTest.test_selection_questionc
C s� t jdd�|_dg|_d|_t �d��` t �d��J}| �ddd d
g�}| �|j� | �d |jd d k� | �|jj� W 5 Q R X W 5 Q R X d S )
Nr1 r2 rp r3 r z$certbot.main.plug_sel.logger.warningr0 rp r� r4 z
make senser ) r r� r r6 rQ rs r| r� �enhance_config)r r5 r] r r" Zmock_logr� r r r! �test_selection_auth_warningl s z'EnhanceTest.test_selection_auth_warningc s� t jdd�|_dg|_t �d��� | �dddg�� dd g}d
g}| �� jj� | �t� fd d �|D ��� | � t
� fd
d �|D ��� | �d� jj d d k� W 5 Q R X d S )Nr1 r2 rp r r0 r4 �--hstsZredirectZhstsZuirc s g | ]}t � j|��qS r ��getattrrP �rQ rt �r� r r! �
<listcomp>� s z8EnhanceTest.test_enhance_config_call.<locals>.<listcomp>c s g | ]}t � j|��qS r r; r= r> r r! r? � s r ) r r� r r6 rQ rs r8 r| �allrk �anyr� )r r" r] r Zreq_enhZ not_req_enhr r> r! �test_enhance_config_call{ s ���z$EnhanceTest.test_enhance_config_callc C s\ t jdd�|_dg|_t �d��2 | �ddddg�}| �|jj� | �|j� W 5 Q R X d S ) Nr1 r2 rp r r0 r4 r: z--non-interactive) r r� r r6 rQ rs r8 r| rk )r r" r] r r� r r r! �test_enhance_noninteractive� s � �z'EnhanceTest.test_enhance_noninteractivec C s8 g |_ t�d�� | �tj| jdddg� W 5 Q R X d S )Nr r0 r4 r: )r r r6 r[ r r� rQ )r r" r] r r r! �test_user_abort_domains� s �z#EnhanceTest.test_user_abort_domainsc C s | � tj| jdddg� d S )Nr0 rp r )r[ r r9 rQ r? r r r! � test_no_enhancements_defined� s �z(EnhanceTest.test_no_enhancements_definedrX c C s8 dg|_ d|_ t�� |_| �ddg�}| �|jj� d S )Nrp r3 r0 r: )r r rr ri rQ rk r8 r| )r r" r] r6 r� r r r! r � s
z'EnhanceTest.test_plugin_selection_errorr c C s` | j |_ddg|_tjdd�|_| �ddg� | �| j jj� | �| j jj d d ddg� d S ) Nrp � another.tldr1 r2 r0 � --auto-hstsr r- )
r/ r r r� rQ rs Zenable_autohstsr| r r� �r r r" r5 r] r r r r! �test_enhancement_enable� s
�z#EnhanceTest.test_enhancement_enablec C sB t �| jd�|_ddg|_tjdd�|_| �tj| j ddg� d S )Nr rp rF r1 r2 r0 rG )
r rY rP r r r� r[ r r\ rQ rH r r r! �%test_enhancement_enable_not_supported� s
�z1EnhanceTest.test_enhancement_enable_not_supportedc C s | � tj| jdddg� d S )Nr0 rG r: )r[ r r� rQ r? r r r! � test_enhancement_enable_conflict� s
�z,EnhanceTest.test_enhancement_enable_conflict)r# r$ r% r& r4 rE rQ r r6 r7 r9 rB rC rD rE r ra rb rI rJ rK r_ r r r@ r! r+ B sV
r+ c sX e Zd ZdZ� fdd�Ze�d�e�d�dd� ��Ze�d�e�d�dd � ��Z� Z S )
� InstallTestz Tests for certbot.main.install.c s" t t| ��� tjtjd�| _d S r, )r3 rL r4 r r� r r. r/ r? r@ r r! r4 � s zInstallTest.setUpr r c C sD t �| jd�|_tj�� }d| j_d| j_| � t
j t j
| j|� d S )Nr TZ nonexistent)r rY rP r r rI rJ rZ r� r[ r r\ r
r �r r5 r" rO r r r! �&test_install_enhancement_not_supported� s
�z2InstallTest.test_install_enhancement_not_supportedc C sL | j |_tj�� }d| j_d | j_d| j_d| j_ | �
t j t
j| j|� d S )NTr1 )r/ r r rI rJ rP rZ r� r' r� r[ r rv r
r rM r r r! �$test_install_enhancement_no_certname� s
�z0InstallTest.test_install_enhancement_no_certname)
r# r$ r% r& r4 r r6 rN rO r_ r r r@ r! rL � s rL �__main__)Er& Z
__future__r r� rO r� r� r� r� r� ZunittestZjosepyr� r r� r Z six.movesr Zacme.magic_typingr Zcertbot.tests.utilZtestsr ra r r r r r r r r r
r Zcertbot.compatr r Zcertbot.pluginsr r r r r| r� r� rx r� ZJWKRSAr� Z load_vectorr� r� r� ZTestCaser ZConfigTestCaser' r` r ZTempDirTestCaser� r� r� r� r r) r+ rL r# r r r r! �<module> sz
@TdkD x>