JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__/client_test.cpython-39.pyc
a
}|�g@� � @ sl d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl
Z
ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ ddlm mZ e� d �Z e� d
�Z G dd� dej �Z G dd� dej �Z!G dd� dej �Z"G dd� de"�Z#G dd� de"�Z$G dd� dej%�Z&e'dk�rhe�(e
�)ej*dd� e+g �� dS ) z#Tests for certbot._internal.client.� N)�mock)� MagicMock)�
interfaces)�errors)�util)�account)� constants)�obj)�oszrsa512_key.pemzcsr-san_512.pemc @ sT e Zd ZdZdd� Zej�ej ddi�dd� �Z
ej�ej i �dd � �Z d
d � Z d S )
�DetermineUserAgentTestz8Tests for certbot._internal.client.determine_user_agent.c C s ddl m} || j�S )Nr )�determine_user_agent)�certbot._internal.clientr �config)�selfr � r �G/usr/lib/python3.9/site-packages/certbot/_internal/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 sZ | � � }|r| j}| j}n | j}| j}|d|� |d|� |t�� |� |t�� |� d S )NzOS_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 r �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
ee j dd� ��Z
dd� Ze�d�dd� �Ze�� dd� �Ze�d�dd� �Zdd� Zd d � Ze�d �d d � �Ze�d�d!d"� �Ze�� d#d$� �Ze�� d%d&� �Zd'd(� Z d)d*� Z � Z S )+� RegisterTestz,Tests for certbot._internal.client.register.c sF t � �� d| j_d| j_d| j_t�� | _t �
� | _ t �
t
� � d S )Ni Fzalias@example.com)�super�setUpr � rsa_key_size� register_unsafely_without_email�emailr ZAccountMemoryStorage�account_storager r �tos_cb� display_objZ set_displayr �� __class__r r r% ? s
zRegisterTest.setUpc C s ddl m} || j| j| j�S )Nr )�register)r
r. r r) r* )r r. r r r r H 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_mockL s z RegisterTest._public_key_mockc C s dS )Nz/acme/new-accountr r r r r �_new_acct_dir_mockR s z RegisterTest._new_acct_dir_mockc C s dS �NTr r r r r �
_true_mockV s zRegisterTest._true_mockc C s dS �NFr r r r r � _false_mockZ s zRegisterTest._false_mockc c s6 t �d��} | jV W d � n1 s(0 Y d S )N�$certbot._internal.client.acme_client)r r �ClientV2)Zmock_acme_clientr r r �_patched_acme_client^ s z!RegisterTest._patched_acme_clientc
C s� | � � ��}d|�� _| j|� j_t�d��~}tj |� j_t
� tj �� | � � W d � n1 sb0 Y |j
du szJ �d |� j_| � � |j
du s�J �W d � n1 s�0 Y W d � n1 s�0 Y d S )Nz
http://tos�*certbot._internal.eff.prepare_subscriptionFT)r: � new_accountZterms_of_servicer7 �external_account_required� side_effectr r r �Error�pytest�raisesr �called)r � mock_client� mock_preparer r r � test_no_tosd s
&
zRegisterTest.test_no_tosr; c C sj | � � �N}ddlm} |�i �|� _| j|� j_| �� | j j
du sHJ �W d � n1 s\0 Y d S )Nr )� DirectoryF) r: Z
acme.messagesrF Z from_json� directoryr7 r= r> r r* rB )r Zunused_mock_preparerC rF r r r �test_empty_metar s
z RegisterTest.test_empty_metac C sz | � � �^}| j|� j_t�d�� | �� W d � n1 s>0 Y | jjdu sXJ �W d � n1 sl0 Y d S )N�)certbot._internal.eff.handle_subscriptionT) r: r7 r= r> r r r r* rB )r �unused_mock_get_utilityrC r r r �test_it s
&zRegisterTest.test_itz.certbot._internal.client.display_ops.get_emailc C s� ddl m} d| j_d}|jjd|d�}| �� �|}| j|� j_ t
� d��F}|t
� � g|� j
_ | �� |jdksrJ �|jd u s�J �W d � n1 s�0 Y W d � n1 s�0 Y d S )
Nr ��messagesF�3DNS problem: NXDOMAIN looking up MX for example.com�invalidContact��detailr; � T)�acmerM r �noninteractive_moder? � with_coder: r7 r= r> r r r r< r �
call_countrB )r �mock_get_emailrM �msg�mx_errrC rD r r r �test_email_retry� s
z RegisterTest.test_email_retryc
C s� ddl m} d| j_d}|jjd|d�}| �� ��}| j|� j_ t
� d��V |t
� � g|� j
_ t�tj�� | �� W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y d S )Nr rL TrN rO rP rI )rS rM r rT r? rU r: r7 r= r> r r r r< r@ rA r r )r rM rX rY rC r r r �!test_email_invalid_noninteractive� s
z.RegisterTest.test_email_invalid_noninteractivec C s@ d | j _t�tj�� | �� W d � n1 s20 Y d S �N)r r( r@ rA r r? r r r r r �test_needs_email� s z RegisterTest.test_needs_email� certbot._internal.client.loggerc C s� t �d���}| �� �X}| j|� j_d | j_d| j_d| j_ | �
� |j � t j
� |jdu s^J �W d � n1 sr0 Y W d � n1 s�0 Y d S )Nr; TF)r r r: r7 r= r> r r( r' �dry_runr �debug�assert_called_once_with�ANYrB )r � mock_loggerrD rC r r r �test_without_email� s
z RegisterTest.test_without_emailc C s� | � � �|}| j|� j_t�d��F d| j_| �� |j du s@J �|� j
j d d j rXJ �W d � n1 sl0 Y W d � n1 s�0 Y dS )zJTests dry-run for no staging account, expect account created with no emailrI TFr N)
r: r7 r= r> r r r r_ r rB r< � call_argsZcontact)r rW rC r r r � test_dry_run_no_staging_account� s
z,RegisterTest.test_dry_run_no_staging_accountc
C s� | � � ��}tj| jd�|� jj_| j|� j_ t�
d��d d}t�
|��6}d| j _ d| j _
| �� |jdu snJ �W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y d S )N�r> rI �Bcertbot._internal.client.messages.ExternalAccountBinding.from_dataztest-kidz+J2OAqW4MHXsrHVa_PVg0Y-L_R4SYw0_aL1le6mfblbET)r: r r/ r3 �clientrG � __getitem__r7 r= r> r r �eab_kid� eab_hmac_keyr rB �r rJ rC �targetZmock_eab_from_datar r r �test_with_eab_arguments� s
� z$RegisterTest.test_with_eab_argumentsc
C s� | � � ��}| j|� j_t�d��d d}t�|��6}d | j_d | j_| � � |j
du sXJ �W d � n1 sl0 Y W d � n1 s�0 Y W d � n1 s�0 Y d S )NrI rh F) r: r7 r= r> r r r rk rl r rB rm r r r �test_without_eab_arguments� s
z'RegisterTest.test_without_eab_argumentsc C s� | � � ��}tj| jd�|� jjj_| j|� j _
t� d��~ t� d��T d | j _
d | j _t�tj�� | �� W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y d S )Nrg rI rh )r: r r/ r2 ri �net�keyZ
public_keyr5 r= r> r r rk rl r@ rA 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}|jjd|dd�}| �� ��}tj| jd�|� jj _
| j |� j _
t�d��V}|t�� g|� j_
t�|j�� | �� W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y |jd u s�J �d S )
Nr rL ZTest� malformed�title�rQ ru rg rI F)rS rM r? rU r: r r/ r3 ri rG rj r7 r= r> r r r< r@ rA r rB )r rM rX rY rC Z mock_handler r r �test_unsupported_error� s
� bz#RegisterTest.test_unsupported_error) r r r r r% r � staticmethodr2 r3 r5 r7 �
contextlib�contextmanagerr: rE r r rH � test_util�patch_display_utilrK rZ r[ r] rd rf ro rp rs rw �
__classcell__r r r, r r# < sB
r# c s e Zd ZdZ� fdd�Z� ZS )�ClientTestCommonz<Common base class for certbot._internal.client.Client tests.c s� t � �� d| j_d| j_tjf i dti��| _ddl m
} t� d��H}|j | _
t�� | _| j
_|j| _|| j| jd d d�| _W d � n1 s�0 Y d S )NFzkey.pemr )�Clientr8 )r Zaccount_Zauth� installer)r$ r% r Z
no_verify_ssl�allow_subset_of_namesr r �KEYr r
r r r9 Z acme_clientrS r0 Z
ClientNetwork�client_networkri )r r rS r, r r r% � s
�zClientTestCommon.setUp)r r r r r% r} r r r, r r~ � s r~ c s� e Zd ZdZ� fdd�Zdd� Zdd� Zd?d d
�Ze� d �e� d �d
d� ��Z
e� d �dd� �Z e� d �dd� �Z e� d �dd� �Z
e� d �dd� �Ze� d �dd� �Ze� d �dd� �Ze� d �dd � �Ze� d �d d � �Ze� d �e� d �d d!� ��Ze� d �e� d �e� d �d"d#� ���Zd$d%� Zd&d'� Zd@d)d*�Ze� d+�e� d,�d-d.� ��Ze� d/�d0d1� �Ze�� d2d3� �Z e� d4�e�� d5d6� ��Z e�� d7d8� �Z e� d4�e�� d9d:� ��Z e� d �e�� d;d<� ��Z d=d>� Z!� Z"S )A�
ClientTestz*Tests for certbot._internal.client.Client.c sF t � �� d| j_d| j_d| j_ddg| _tjd gtj j
d�| _ d S )NFT� example.com�www.example.com)�authorizations�csr_pem) r$ r% r r� r_ �strict_permissions�
eg_domainsr r �sentinelr� �eg_orderr r, r r r% s
�zClientTest.setUpc C s | j jd d du sJ �d S )NrR Z
verify_sslT)r� re r 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 r\ ) r r ri � auth_handler�handle_authorizationsr0 � deactivate_valid_authorizationsr� rS �finalize_order� new_order�updater r r r �_mock_obtain_certificate s z#ClientTest._mock_obtain_certificaterR c C s^ |dkr&| j jj�| j| j| jj� n| j jjj|ks:J �| jj j| jt
j | jj d ud� d S )NrR )Zfetch_alternative_chains)
ri r� r� ra r� r r� rV rS r� r rb �preferred_chain)r �
auth_countr r r �_check_obtain_certificate% s
�
�z$ClientTest._check_obtain_certificatez$certbot._internal.client.crypto_utilr^ c C s | � � tjdd td�}| jj}| �|j� | j� |j
�}|� || j d� t
jjt
jjf| jj||d�ksnJ �|j�� | �� d| j _t
jjt
jjf| jj||d�ks�J �|j�|jg|j dd� d | j _tj�� tjt j d d� }| jj||d� | jjj j \\}}}t || �tjd d�k�s,J �tj�� tjd
d� }d
| j _!| jj||d� | jjj j \\}}}t || �tjd d�k�s�J �t
jjt
jjf| jj|d d�k�s�J �|j �"| j#| j d� d | j_t$�%t&j'�� | j�|� W d � n1 �s�0 Y |j(�t
j)� d S ) N�pem�Zform�file�dataF)�orderrz some issuerT�issuance_timeout)ZsecondsrR i, )*r� r �CSR�CSR_SANri r� �_set_mock_from_fullchain� cert_and_chain_from_fullchainrS r� r� r� r r r� �cert�chainZobtain_certificate_from_csrZfind_chain_with_issuer�assert_not_calledr� r� ra �
fullchain_pemZalternative_fullchains_pem�datetimeZnowZ timedeltar Z CLI_DEFAULTSr� re �absr� �assert_called_withr� r@ rA r r? �errorrb ) r rc �mock_crypto_utilZtest_csrr� r� Zexpected_deadline�_Zdeadliner r r � test_obtain_certificate_from_csr2 sd ��
�� � �� � ��
,z+ClientTest.test_obtain_certificate_from_csrc C s� t jdd td�}||j_tjj|j_| � |j
� | � tjj|� |jj | j
jd | j
jddd� |j� tjj| jd dd� |j
� | jj� d S )Nr� r� � secp256r1T)Zkey_sizeZkey_dir�key_type�elliptic_curver� F)r r� r� � generate_csrr0 r r� rr � generate_keyr� r� � _test_obtain_certificate_commonra r r& r� r� r� r� )r r� �csrr r r �test_obtain_certificatel s$ ���z"ClientTest.test_obtain_certificatec C s� t jdtjjtd�}t jdtjjtd�}||j_||j _| �
|j � | � dg�}d| j
_| j|||dd� |j jdkszJ �|jjdks�J �|j jdks�J �d S )Nr� r� r� T� )�
authzr_retr� rR )r r� r r� �csr_filer� �key_filer� r0 r� r� r� �_authzr_from_domainsr r� r� rV )r r� r� rr �authzrr r r �'test_obtain_certificate_partial_success� s z2ClientTest.test_obtain_certificate_partial_successc
C s� ddl m} tjdtjjtd�}tjdtjjtd�}||j _
||j _
| � |j
� | �� | �| j�}|| j_|| jjj_
|j|jdd�}|jjddd |d
�}|jjd d d |gd
�}|tjg| jj j_ d| j _ t � � � | j�!| j�} W d � n1 s�0 Y | tjj"tjj#||fk�s J �| jjjj$dk�s6J �| j jj$dk�sJJ �dd� | jD �}
|j j$dk�slJ �|j �%t�&|| jd | j j'| j j(�t�&||
d | j j'| j j(�g� |j
j$dk�s�J �d S )Nr rL r� r� r� ��typ�value�caa�barru �rQ ru �
identifierrt �foo�rQ ru Z subproblemsTr� c S s g | ]}|d kr|�qS �r� r ��.0�dr r r �
<listcomp>� � zUClientTest.test_obtain_certificate_finalize_order_partial_success.<locals>.<listcomp>rR ))rS rM r r� r r� r� r� r� r� r0 r� r� r� r� r� r� r� r� ri r� r� �
Identifier�IDENTIFIER_FQDNr? rU �DEFAULTr� r> r r� r{ r| �obtain_certificater� r� rV �assert_has_calls�call� must_stapler� � r r� rM r� rr r� r� Z
subproblem�error_with_subproblems�resultZsuccessful_domainsr r r �6test_obtain_certificate_finalize_order_partial_success� s:
,�
�zAClientTest.test_obtain_certificate_finalize_order_partial_successc C sz ddl m} tjdtjjtd�}tjdtjjtd�}||j _
||j _
| � |j
� | �� | �| j�}|| j_|| jjj_
|j|jdd�}|j|jdd�}|jjdd d
|d �}|jjdd d
|d �} |jjd d
d
|| gd�}
|
| jj j_d| j _ t � |j�� | j� | j� W d � n1 �s 0 Y | jjjj!dk�s>J �| j jj!dk�sRJ �|j j!dk�sdJ �|j
j!dk�svJ �d S �Nr rL r� r� r� r� r� r� r� ru r� rt r� r� TrR )"rS rM r r� r r� r� r� r� r� r0 r� r� r� r� r� r� r� r� ri r� r� r� r� r? rU r� r> r r� r@ rA r� rV � r r� rM r� rr r� Z identifier1Z identifier2Z subproblem1Z subproblem2r� r r r �;test_obtain_certificate_finalize_order_no_retryable_domains� s. .zFClientTest.test_obtain_certificate_finalize_order_no_retryable_domainsc C s& ddl m} tjdtjjtd�}tjdtjjtd�}||j _
||j _
| � |j
� | �� | �| j�}|| j_|| jjj_
|jjdddd�}|| jj j_d | j_t � |j�� | j� | j� W d � n1 s�0 Y | jjjj d
ks�J �| j jj d
ks�J �|j j d
k�sJ �|j
j dk�s"J �d S � Nr rL r� r� r� r� ru rv TrR ) rS rM r r� r r� r� r� r� r� r0 r� r� r� r� r� r� r� r� ri r� r� r? rU r� r> r r� r@ rA r� rV �r r� rM r� rr r� r� r r r �Itest_obtain_certificate_finalize_order_rejected_identifier_no_subproblems� s& ,zTClientTest.test_obtain_certificate_finalize_order_rejected_identifier_no_subproblemsc
C s� ddl m} tjdtjjtd�}tjdtjjtd�}||j _
||j _
| � |j
� | �� | �| j�}|| j_|| jjj_
|j|jdd�}|jjddd |d
�}|jjd d d |gd
�}|tjg| jj j_ d| j _ t � � � | j�!| j�} W d � n1 s�0 Y | tjj"tjj#||fk�s J �| jjjj$dk�s6J �| j jj$dk�sJJ �dd� | jD �}
|j j$dk�slJ �|j �%t�&|| jd | j j'| j j(�t�&||
d | j j'| j j(�g� |j
j$dk�s�J �d S )Nr rL r� r� r� r� r� r� ru r� rt r� r� TrR r� c S s g | ]}|d kr|�qS r� r r� r r r r�
r� zPClientTest.test_obtain_certificate_get_order_partial_success.<locals>.<listcomp>))rS rM r r� r r� r� r� r� r� r0 r� r� r� r� r� r� r� r� ri r� r� r� r� r? rU r� r� r> r r� r{ r| r� r� r� rV r� r� r� r� r� r r r �1test_obtain_certificate_get_order_partial_success� s:
,�
�z<ClientTest.test_obtain_certificate_get_order_partial_successc C sz ddl m} tjdtjjtd�}tjdtjjtd�}||j _
||j _
| � |j
� | �� | �| j�}|| j_|| jjj_
|j|jdd�}|j|jdd�}|jjdd d
|d �}|jjdd d
|d �} |jjd d
d
|| gd�}
|
| jj j_d| j _ t � |j�� | j� | j� W d � n1 �s 0 Y | jjjj!dk�s>J �| j jj!dk�sRJ �|j j!dk�sdJ �|j
j!dk�svJ �d S r� )"rS rM r r� r r� r� r� r� r� r0 r� r� r� r� r� r� r� r� ri r� r� r� r� r? rU r� r> r r� r@ rA r� rV r� r r r �6test_obtain_certificate_get_order_no_retryable_domains s. .zAClientTest.test_obtain_certificate_get_order_no_retryable_domainsc C s& ddl m} tjdtjjtd�}tjdtjjtd�}||j _
||j _
| � |j
� | �� | �| j�}|| j_|| jjj_
|jjdddd�}|| jj j_d | j_t � |j�� | j� | j� W d � n1 s�0 Y | jjjj dks�J �| j jj d
ks�J �|j j d
k�sJ �|j
j dk�s"J �d S r� ) rS rM r r� r r� r� r� r� r� r0 r� r� r� r� r� r� r� r� ri r� r� r? rU r� r> r r� r@ rA r� rV r� r r r �Dtest_obtain_certificate_get_order_rejected_identifier_no_subproblems/ s& ,zOClientTest.test_obtain_certificate_get_order_rejected_identifier_no_subproblemsz)certbot._internal.client.acme_crypto_utilc C s� t jdd td�}t|j_tjj|j_t j d tjjd�}| �
|j � d| j j
_| �||� |jj| j
jd| j
jd� |j�tjj| j| j
j� |j�� |j�� |j jdks�J �d S )Nr� r� �r� r� Tr� )�bitsr� r� rR )r r� r� �make_csrr0 r r� �key_pem�make_key�Keyr� r� ri r r_ r� ra r& r� r� r� r� r� r� rV )r �mock_acme_crypto� mock_cryptor� rr r r r � test_obtain_certificate_dry_runI s$
��
z*ClientTest.test_obtain_certificate_dry_runc C sR 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 d � n1 s�0 Y |
tjj tjj ||fk�s
J �| �!d � | jj"d k�s*J d
��| j j#j"d k�sBJ d ��|j$�%d
� d S )Nr rL r� r� r� Tr� r� rR 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.)&rS rM r r� r� r� r0 r r� r� r� r� r� r� r� ri r r_ r� Z STATUS_VALID�body�statusr� r� r� r� r� r{ r| r� r� r� r� r� rV r� �warningr� ) r r� r� �mock_logrM r� rr Zauthzrsr� r� r� r r r �:test_obtain_certificate_dry_run_authz_deactivations_failed` s4
,
��zEClientTest.test_obtain_certificate_dry_run_authz_deactivations_failedc C s6 t �� }t jj|j_t �� }t jj|j_||f|_d S r\ )r r/ r� r� �encoder0 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)r� )r� )r� )�reversed�appendr r )r �domainsr� �domainr r r r� � s ����z ClientTest._authzr_from_domainsNc C s� | � � |p| �| j�}|| j_|| jjj_t �
� � | j� | j�}W d � n1 sX0 Y |t j
jt j
j||fks~J �| �|� d S r\ )r� r� r� r� r� ri r� r� r0 r{ r| r� r r� r� r� r� )r rr r� r� r� r� r� r r r r� � s
,�z*ClientTest._test_obtain_certificate_commonz2certbot._internal.client.Client.obtain_certificatez3certbot._internal.storage.RenewableCert.new_lineagec C s� ddg}t �� t �� t �� d f|_d| jj_| j�|d�s>J �| j�|d �sPJ �| j�|dd � d �sjJ �d| jj_| j�|d �r�J �dd� |jD �}|g d �ks�J �d S )
Nz
*.example.comr� F� example_certrR Tc S s g | ]}|d d �qS )r r )r� r� r r r r� � r� zAClientTest.test_obtain_and_enroll_certificate.<locals>.<listcomp>)r� r� r� )r r r0 ri r r_ Z obtain_and_enroll_certificate�call_args_list)r Z mock_storageZmock_obtain_certificater� �namesr r r �"test_obtain_and_enroll_certificate� s �
z-ClientTest.test_obtain_and_enroll_certificatez$certbot._internal.cli.helpful_parserc C s� ddg}t �� }t�|d �}t�|d �t�|d � }tj�|dd�}tj�|dd�}tj�|dd�}d |_d
|d |d |g|_| j �
|||||�\} }
} tj� | �tj� |�ks�J �tj� |
�tj� |�ks�J �tj� | �tj� |�ks�J �t | d
��} | �
� }
W d � n1 �s0 Y |
t�|d �k�s<J �t |
d
��}|�
� }W d � n1 �sf0 Y |t�|d �t�|d � k�s�J �t�|� d S )Nz cert_512.pemzcert-san_512.pemr rR �certsZchainsz chain.pemz
fullchain.pemZcertonlyz --cert-pathz --chain-pathz--fullchain-path�rb)�tempfileZmkdtempr{ � load_vectorr
�path�joinZverb�argsri Zsave_certificate�dirname�open�read�shutil�rmtree)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� sD ��
�
�
� ( ( � z ClientTest.test_save_certificatec C s� t �tj��& | j�dgdddd� W d � n1 s80 Y t�� }|| j_| j�dgdddd� |j j
t j �
d�t j �
d�ddt j �
d�d� |jjdks�J �|j�
� d S )N�foo.barrr r� r� � fullchain)r r r� r Zkey_pathr� )r@ rA r r? ri �deploy_certificater r r� � deploy_certra r
r� �abspath�saverV �restart�r � mock_utilr� r r r � test_deploy_certificate_success� s 4
�z*ClientTest.test_deploy_certificate_successz,certbot._internal.client.display_util.notifyc C s| t �� }|| j_d| j_tj|j_t �
tj��& | j� dgdddd� W d � n1 sZ0 Y |j �
� |�d� d S )NZfoobarr rr r� r� r zDeploying certificate)r r ri r� r r � PluginErrorr r> r@ rA r �recovery_routinera Zassert_any_call)r r � mock_notifyr� r r r � test_deploy_certificate_failure� s
4
z*ClientTest.test_deploy_certificate_failurec C sj t �� }|| j_tj|j_t� tj��& | j�
dgdddd� W d � n1 sR0 Y |j � � d S )Nr rr r� r� r )
r r ri r� r r r r> r@ rA r r ra r
r r r �$test_deploy_certificate_save_failure s
4z/ClientTest.test_deploy_certificate_save_failurec C s� t �� }tjd g|j_|| j_t� tj��& | j�
dgdddd� W d � n1 sV0 Y |� d� |j �
� |jjdks�J �d S )Nr rr r� r� r zuWe were unable to install your certificate, however, we successfully restored your server to its prior configuration.r� )r r r r r r> ri r� r@ rA r r� �rollback_checkpointsra rV )r �mock_get_utilityr r� r r r �'test_deploy_certificate_restart_failure s 4�
z2ClientTest.test_deploy_certificate_restart_failurec C s� t �� }tj|j_tj|j_|| j_ t
� tj��& | j� dgdddd� W d � n1 s\0 Y |j
jdksvJ �d|j
jd d v s�J �|j�� |jjdks�J �d S ) Nr rr r� r� r rR z6An error occurred and we failed to restore your configr )r r r r r r> �
ReverterErrorr ri r� r@ rA r r� rV re ra )r r rc r� r r r �(test_deploy_certificate_restart_failure2" s
4�
z3ClientTest.test_deploy_certificate_restart_failure2c C s6 t jj}d|� d�g}dg}d|� d�}d}d|� d�g}t�tj�� | j�|d � W d � n1 sj0 Y t�tj�� | j�||� W d � n1 s�0 Y t�tj�� | j�||� W d � n1 s�0 Y t�tj�� | j�|d � W d � n1 �s0 Y | j�||� d S )NZexamzple.comr� r� z.barr z*.exam) r
r� �sepr@ rA r r? ri Z_choose_lineagename)r r Zinvalid_domainsZ
valid_domainsZinvalid_certnameZvalid_certnameZinvalid_wildcard_domainr r r �test_choose_lineage_name2 s ,,,.z#ClientTest.test_choose_lineage_name)rR )NrR )#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� sj
8
#
#
&
$
r� c s� e Zd ZdZ� fdd�Zdd� Zdd� Ze�d�d d
� �Z e�d�d d � �Z
e�d�d
d� �Z e�d�dd� �Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd*d$d%�Zd&d'� Zd(d)� Z� ZS )+�EnhanceConfigTestz9Tests for certbot._internal.client.Client.enhance_config.c s4 t � �� d| j_d| j_d| j_d| j_d| _d S )NFz example.org)r$ r% r �hsts�redirect�staple�uirr� r r, r r r% I s
zEnhanceConfigTest.setUpc C sB t �tj��" | j�| jgd � W d � n1 s40 Y d S r\ )r@ rA r r? ri �enhance_configr� r r r r �test_no_installerR s z#EnhanceConfigTest.test_no_installerc C s� t �� | j_g | jjj_d | j_d| j_t � d��"}| j�
| j gd � W d � n1 sZ0 Y |j j
dkstJ �| jjj�� d S )NTr^ rR )r r ri r� �supported_enhancementsr0 r r r r r! r� r� rV �enhancer� )r rc r r r �test_unsupportedV s 0z"EnhanceConfigTest.test_unsupportedr^ c C s< d| j _| �� |jjdu s J �|jjd d dks8J �d S )NTr rR �Strict-Transport-Security)r r �_test_with_already_existing�inforB re �r r� r r r �test_already_exists_headera s �z,EnhanceConfigTest.test_already_exists_headerc C s< d| j _| �� |jjdu s J �|jjd d dks8J �d S )NTr rR r )r r r' r( rB re r) r r r � test_already_exists_redirecti s �z.EnhanceConfigTest.test_already_exists_redirectc C s$ d| j _| �� |jjdu s J �d S r6 )r r r' r� rB r) r r r �#test_config_set_no_warning_redirectq s z5EnhanceConfigTest.test_config_set_no_warning_redirectc C s$ d | j _| �� |jjdu s J �d S r6 )r r �_test_with_all_supportedr� rB r) r r r �test_no_warn_redirectw s z'EnhanceConfigTest.test_no_warn_redirectc C s* d| j _| �� | jjj�| jdd� d S )NT�ensure-http-headerr& )r r r- ri r� r$ r� r� r r r r �test_no_ask_hsts} s
�z"EnhanceConfigTest.test_no_ask_hstsc C s* d| j _| �� | jjj�| jdd � d S )NTr )r r r- ri r� r$ r� r� r r r r �test_no_ask_redirect� s
�z&EnhanceConfigTest.test_no_ask_redirectc C s* d| j _| �� | jjj�| jdd � d S )NT� staple-ocsp)r r r- ri r� 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- ri r� r$ r� r� r r r r �test_no_ask_uir� s
�z!EnhanceConfigTest.test_no_ask_uirc C s8 t �� | j_tj| jjj_| jdd� | jjj �
� d S )NT)�
enhance_error) r r ri r� r r r$ r> � _test_errorr ra 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 r\ ) r r ri r� r r r r> r6 r ra rb r r r r �test_save_failure� s
z#EnhanceConfigTest.test_save_failurec C s* t �� | j_tjd g| jjj_| �� d S r\ ) r r ri r� 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 r\ ) r r r r r r> r r ri r� r9 )r r� r r r �test_restart_failure2� s
z'EnhanceConfigTest.test_restart_failure2c C s | � � | jjjjdu s J �d S r4 )r6 ri r� r rB r r r r r9 � s z+EnhanceConfigTest._test_error_with_rollbackFc
C s� d| j _t�d��l}t�� �D}t�tj �� | �
� W d � n1 sH0 Y W d � n1 sf0 Y W d � n1 s�0 Y |r�|j j dks�J �d|j j
d d d ks�J �|r�|j�d� d S )NTr^ rR z(Unable to set the %s enhancement for %s.r z0Rolling back to previous server configuration...)r r r r r{ r| r@ rA r r r- r� rV r� Zcriticalr� )r r5 Z
restart_errorrc Zmock_gur r r r6 � s �b �z EnhanceConfigTest._test_errorc C sf | j jd u rt�� | j _g d�| j jj_| j �| jgd � | j jjj dksNJ �| j jj
j dksbJ �d S )N�r/ r r2 rR ) ri r� r r r# r0 r! r� r rV r r r r r r- � s z*EnhanceConfigTest._test_with_all_supportedc C sB t �� | j_g d�| jjj_t�� | jjj_ | j�
| j gd � d S )Nr< ) r r ri r� r# r0 r Z PluginEnhancementAlreadyPresentr$ r> r! r� r r r r r' � s z-EnhanceConfigTest._test_with_already_existing)FF)r r r r r% r" r% r r r* r+ r, r. r0 r1 r3 r4 r7 r8 r: r; r9 r6 r- r' r} r r r, r r F s0
r c @ s4 e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd
S ) � RollbackTestz,Tests for certbot._internal.client.rollback.c C s t �� | _d S r\ )r r � m_installr r r r r% � s zRollbackTest.setUpc C sR ddl m} t�d��(}||_|d |i t�� � W d � n1 sD0 Y d S )Nr )�rollbackz8certbot._internal.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� | �� jjdks"J �| �� jjdks6J �d S �NrR )r r> r rV r r r r r �test_no_problems� s z RollbackTest.test_no_problemsc C s | � dd � d S rA )r r r r r r" � s z RollbackTest.test_no_installerN) r r r r r% � classmethodr rB r" r r r r r= � s
r= �__main__rR ),r ry r� r r �sysr� Zunittestr Z
unittest.mockr Zjosepyr r@ Zcertbotr r Zcertbot._internalr r Zcertbot._internal.displayr r+ Zcertbot.compatr
Zcertbot.tests.utilZtestsr{ r� r� r� ZConfigTestCaser r# r~ r� r ZTestCaser= r �exit�main�argv�__file__r r r r �<module> sD
= >
Back to Directory
nL+D550H?Mx ,D"v]qv;6*Zqn)ZP 0 !1 A "#a$2Qr D8 a Ri[f\mIykIw0 cuFcRı?lO7к_f˓[C$殷WF<_W ԣs KcëIzyQy /_LKℂ;C",pFA: /]=H ~,ls/9ć:[=/#f;)x {ٛEQ )~ =𘙲r*2~ a _V=' kumF D}KYYC)({ *g&f`툪ry`=^cJ.I](*`wq 1dđ#̩͑0;H]u搂@:~וKL Nsh}OIR*8:2 !lDJVo( 3=M(zȰ+i*NA r6KnSl )!JJӁ* %݉?|D}d5:eP0R;{$X'xF@.ÊB {,WJuQɲRI;9QE琯62fT.DUJ;*cP A\ILNj!J۱+O\ ͔]ޒ SJȧc%ANolՎprULZԛerE2=XDXgVQeӓk yP7U*omQIs,K`)6\G3t?pgjrmۛجwluGtf h9uyP0D;Uڽ"OXlif$)&|ML0Zrm1[HXPlPR0'G=i2N +0e2]] 9VTPO7h(F*癈 '=QVZDF,d߬~TX
G[`le69CR(!S2!P
<0x < !1AQ "Raq02Br#SCTb
?Ζ"]mH5WR7k.ۛ!}Q~+yԏz|@T20S~Kek *zFf^2X*(@8r?CIuI|֓>^ExLgNUY+{.RѪ
τVYTD I62'8Y2 7'\TP.6d&˦@Vqi|8- OΕ]ʔ U=TL8=;6c| !qfF3aů&~$l}'NWUs$Uk^SV:U# 6w++s&r+nڐ{@29 gL
u"TÙM=6 (^"7r} =6YݾlCuhquympǦ
G jhsǜNlɻ}o7 #S6aw4!OSrD57%|?x>L
|/nD6?/8w#[)L7+6〼T ATg ! %5MmZ/c-{1_Je"|^$'O&ޱմ Trb$ w)R$&
N1EtdU3Uȉ1pM" N*(DN yd96.(jQ)X
5cQɎMyW? Q*!R>6=7)Xj5`J]e 8%t!+'!1Q 5 !1 AQaqё#2"0BRb ? Gt^## .llQT $v,,m 㵜5 ubV =sY+@d{N! dnO< .-B; _wJt6;QJd.Qc%p{ 1,sNDdFHI0ГoXшe黅XۢF:)[FGXƹ/w_cMeD,ʡcc. WDtA$j@:) -# u
c1<@ۗ9F)KJ-hpP]_x[qBlbp ʖw q"LFGd ƶ*s+ډ_Zc"? %t[IP 6J]#=ɺVvvCGsGh1 >)6|ey?Lӣm,4GWUi` ]uJVoV DG< SB 6ϏQ@ TiUly OU0kfV~~}SZ@*WUUi##; s/[ =!7}" WN]'(L! ~y5g9T̅JkbM' +s:S + B)v@Mj
e Cf jE 0Y\QnzG1д~Wo{T 9?`Rmyhsy3!HAD]mc1~2L Su7xT;j$`}4->L#vzŏ ILS ֭ T { rjGKC; bpU= -`BsK.SFw4Mq]ZdHS0)tLg