← Back
Editing: main_test.cpython-38.pyc
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 �call�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 rB r r r! �test_config_changes� s z MainTest.test_config_changeszcertbot.main.logger.warningc C s6 | � dg� | �|j� |jd d }| �d|� d S )NrD 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- rH rI r r r! �test_certificates s zMainTest.test_certificatesr� c C s | � dg� | �d|j� d S )N�deleter- rH rI 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 )rP r� r Zmoves�range�lenrQ �listr r rS 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_dirrZ rO )rI rJ r r r r6 ra r[ ri rQ r\ rS r] r r r^ r ) r r_ r` r] rO re r� Zmock_set_up_core_dirr ra 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 )NrZ rO rN r- )rI rJ r r ra r[ rQ r\ rS r] r �initrt �verifyr r^ r ) r r_ r` r] rO r� r ra �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 )NrZ rO rN rO r- )rI rJ r r ra r[ rQ r\ rS r] r rg rt rh Zprepare� availabler r^ r ) r r_ r` r] rO r� r ra ri rk 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� rm r� rn 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.35r3 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 ry r r� �reprrr rt 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}rx 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� rn 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� rn r� r' Fz isnot.orgZpem_key)Zpemr� Zcsrc s � � | � dS )zWrite message to stdout.N)�write)rl rN r� rZ 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 rq r8 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 �printrr � 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 rZ 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�-tvvr9 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� rn 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� rn 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 ry 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 rq r8 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 ry 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 rs 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$ r0 r2 r4 r7 r? rC rE rG rJ rK rM rY rb rf rj rl rp rv rw rz 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 )rR �k�vr r r! rU � 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 ro �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 ro 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)rd )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 )Nr5 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 r1 � --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 ) Nr2 r3 rp r4 r z$certbot.main.plug_sel.logger.warningr1 rq r� r5 z make senser ) r r� r r6 rQ rs r| r� �enhance_config)r r6 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 )Nr2 r3 rp r r1 r5 �--hstsZredirectZhstsZuirc s g | ]}t � j|��qS r ��getattrrP �rR ru �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 r9 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 ) Nr2 r3 rp r r1 r5 r; z--non-interactive) r r� r r6 rQ rs r9 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 r1 r5 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 )Nr1 rq r )r[ r r: 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 r4 r1 r; )r r rs ri rQ rk r9 r| )r r# r] r7 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.tldr2 r3 r1 � --auto-hstsr r- ) r0 r r r� rQ rs Zenable_autohstsr| r r� �r r r# r6 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 rG r2 r3 r1 rH ) r rY rP r r r� r[ r r\ rQ rI r r r! �%test_enhancement_enable_not_supported� s �z1EnhanceTest.test_enhancement_enable_not_supportedc C s | � tj| jdddg� d S )Nr1 rH 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 r8 r: rC rD rE rF r ra rb rJ rK rL 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 rM r4 r r� r r/ r0 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 r6 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 )NTr2 )r0 r r rI rJ rP rZ r� r( r� r[ r rv r r rN r r r! �$test_install_enhancement_no_certname� s �z0InstallTest.test_install_enhancement_no_certname) r# r$ r% r& r4 r r6 rO rP r_ r r r@ r! rM � s rM �__main__)Er& Z __future__r r� rP 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� ry r� ZJWKRSAr� Z load_vectorr� r� r� ZTestCaser ZConfigTestCaser' r` r ZTempDirTestCaser� r� r� r� r r* r, rM r# r r r r! �<module> sz @TdkD x>
Save File
Cancel