🔐 Sid Gifari File Manager Pro
v8.0.5 | 2026-06-14 15:43:58 | PHP 8.1.34
📂
/ (Root)
/
usr
/
lib
/
python3.9
/
site-packages
/
certbot
/
_internal
/
tests
/
__pycache__
📍 /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__
🔄 Refresh
✏️
Editing: client_test.cpython-39.pyc
Read Only
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 = >
💾 Save Changes
❌ Cancel