a
}|gX: @ s d Z ddlZddlZddlZddlmZ ddlZddlmZ ddlm Z ddlm
Z
ddl m Z ddl
m mZ G dd d ejZG d
d d ejZG d d
d
ejZedkreeejdd eg dS )z#Tests for certbot._internal.renewal N)mock)
challenges)
configuration)errors)storagec @ s e Zd Zejejddd Zejejddd Z eddd Z
edd d
Z ejejdd d Z ed
ejejddd Z
e ejejddd Zejejddd ZdS ) RenewalTest set_by_userc C s| d|_ t| jjd}d | j_d | j_d | j_t | j}t
||}|jd }ddl m
} ||| |jdgksxJ d S )NFzsample-renewal-ancient.conf
renewalparamsr renewalz /var/www/) return_value test_util make_lineageconfig
config_diraccountZemail webroot_pathr NamespaceConfigr
RenewableCertcertbot._internalr _restore_webroot_config)selfmock_set_by_userrc_pathr lineager r r H/usr/lib/python3.9/site-packages/certbot/_internal/tests/renewal_test.py!test_ancient_webroot_renewal_conf s
z-RenewalTest.test_ancient_webroot_renewal_confc C s ddl m} d|_ddiddgd}|| j| | jjddiksFJ | jjddgksZJ i dd}|| j| | jji ksJ | jjdgksJ d S )Nr r
Fztest.example.comz
/var/www/testz/var/www/other) webroot_mapr )r r r r r r r )r r r r r r r test_webroot_params_conservation! s z,RenewalTest.test_webroot_params_conservationz4certbot._internal.renewal._avoid_reuse_key_conflictsc C s d| j _d| j _d| j _t| j }t| j jd}t
||}t }d|j
_ddlm} t d" || j d || W d n1 s0 Y | j jdksJ d S ) NZ
INVALID_VALUETsample-renewal.confNNNNr r
*certbot._internal.renewal.hooks.renew_hook secp256r1)r elliptic_curve reuse_keydry_runr r r
r r r r r MagicMockobtain_certificater r r patch
renew_certr Z!unused_mock_avoid_reuse_conflictsr r r le_clientr r r r test_reuse_key_renewal_params9 s 0z)RenewalTest.test_reuse_key_renewal_paramsc C s d| j _d| j _d| j _d| j _t| j }tj| j j ddd}t
||}t
}d|j_ddlm} t d " || j d || W d n1 s0 Y | j jd
ksJ d S ) NZ
INVALID_CURVETecdsazsample-renewal-ec.confZecr! r r
r" r# )r r$ r% r& key_typer r r
r r r r r r' r( r r r r) r* r+ r r r test_reuse_ec_key_renewal_paramsN s" 0z,RenewalTest.test_reuse_ec_key_renewal_paramsc C s d|_ d| j_d| j_d| j_t| j}t| jj d}t
||}t
}d|j_ ddlm} t d" || jd || W d n1 s0 Y | jjdksJ | jjd ksJ | jjsJ |jt jd d S )
NFTr r! r r
r" r# r. )r r r% Znew_keyr& r r r
r r r r r r' r( r r r) r* r$ r0 Zassert_called_withANY)r r r r r r, r r r r test_new_keyg s$ 0 zRenewalTest.test_new_keyr" c C s d|_ d| j_d| j_d| j_d| j_t| j}t | jj
d}t ||}d|jd d< t
}d|j_ d d
lm} tjtjd d " || jd || W d n1 s0 Y d
d |_d| j_|| jd || d S )NFTrsai r r r% r! r r
z Unable to change the --key-type)matchc S s | dkS )Nr% r )varr r r z6RenewalTest.test_reuse_key_conflicts..)r r r% r0 Z rsa_key_sizer& r r r
r r r r r r' r( r r pytestraisesr Errorr* Z side_effect)r r Zunused_mock_renew_hookr r r r, r r r r test_reuse_key_conflicts s& 0
z$RenewalTest.test_reuse_key_conflictsc C sb d|_ t| j}d|_t| jjd}ddlm } t
| j}| ||}t
|jtjs^J d S )NFz sample-renewal-deprecated-optionz%sample-renewal-deprecated-option.confr r
)r r r r Zcertnamer
r r r r copydeepcopy reconstitute
isinstanceZmanual_public_ip_logging_okr r' )r r Zunused_mock_get_utilityr r r lineage_configZrenewal_candidater r r &test_remove_deprecated_config_elements s z2RenewalTest.test_remove_deprecated_config_elementsc C sP d|_ tj| jjddd}ddlm} t| j}| || |j
dksLJ d S )NFr r/ r r
r4 ) r r
r r r r r r= r> r? r0 )r r r r rA r r r test_absent_key_type_restored s z)RenewalTest.test_absent_key_type_restoredN)__name__
__module__ __qualname__r r) objectr r r r r- r1 r3 r< r
Zpatch_display_utilrB rC r r r r r s$
r c @ s e Zd ZdZedd Zeje j
ddd Z eje j
ddd Z eje j
dd d
Z
eje j
dd d Zeje j
dd
d Zeje j
ddd Zeje j
ddd Zeje j
ddd Zdd ZdS )!RestoreRequiredConfigElementsTestzETests for certbot._internal.renewal.restore_required_config_elements.c O s ddl m} ||i |S )Nr ) restore_required_config_elements)certbot._internal.renewalrI )clsargskwargsrI r r r _call s z'RestoreRequiredConfigElementsTest._callr c C s, d|_ | | jddi | jjdu s(J d S )NFallow_subset_of_namesTrueT)r rN r rO r r r r r "test_allow_subset_of_names_success s zDRestoreRequiredConfigElementsTest.test_allow_subset_of_names_successc C sL d|_ ddi}ttj | | j| W d n1 s>0 Y d S )NFrO mayber r9 r: r r; rN r r r r r r r "test_allow_subset_of_names_failure s zDRestoreRequiredConfigElementsTest.test_allow_subset_of_names_failurec C sF d|_ dddi}| | j| tjjtjjg}| jj|ksBJ d S )NF pref_challsz http-01, dns,) r splitrN r r ZHTTP01typDNS01rW r r r Zexpectedr r r test_pref_challs_list s
z7RestoreRequiredConfigElementsTest.test_pref_challs_listc C s: d|_ ddi}| | j| tjjg}| jj|ks6J d S )NFrW Zdns)r rN r r r[ rZ rW r\ r r r test_pref_challs_str s
z6RestoreRequiredConfigElementsTest.test_pref_challs_strc C sL d|_ ddi}ttj | | j| W d n1 s>0 Y d S )NFrW zfinding-a-shrubberyrT rU r r r test_pref_challs_failure s z:RestoreRequiredConfigElementsTest.test_pref_challs_failurec C s, d|_ | | jddi | jjdu s(J d S )NF must_staplerP T)r rN r r` rQ r r r test_must_staple_success s z:RestoreRequiredConfigElementsTest.test_must_staple_successc C sL d|_ ddi}ttj | | j| W d n1 s>0 Y d S )NFr` rS rT rU r r r test_must_staple_failure s z:RestoreRequiredConfigElementsTest.test_must_staple_failurec C sH ddl m} d | j_d|_| | jd|ji | jj|jd ksDJ d S )Nr ) constantsFserver)r rc r rd r rN ZV1_URIZ CLI_DEFAULTS)r r rc r r r test_ancient_server_renewal_conf s
zBRestoreRequiredConfigElementsTest.test_ancient_server_renewal_confc C s0 ddd}| | j| | | jj|d d S )Nzhttps://example.orgZsomehash)rd r r )rN r Z assertEqualr )r r r r r test_related_values s
z5RestoreRequiredConfigElementsTest.test_related_valuesN)rD rE rF __doc__ classmethodrN r r) rG r r rR rV r] r^ r_ ra rb re rf r r r r rH s(
rH c @ sT e Zd ZdZdd Zdd Zedd Zdd Zd
d Z d d
Z
dd Z dd Z dS )DescribeResultsTestz r8 z>DescribeResultsTest._assert_success_output..)rn assert_has_calls)r linesr r r _assert_success_output s z*DescribeResultsTest._assert_success_outputc C s* | tjddg g g g | dg d S )NTr& z%No simulated renewals were attempted.rN r r' r| rp r r r test_no_renewal_attempts" s z,DescribeResultsTest.test_no_renewal_attemptsc C s. | tjdddgd d d | g d d S )NFr} good.pem)P
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -z)Congratulations, all renewals succeeded: z good.pem (success)O- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -r~ rp r r r test_successful_renewal& s z+DescribeResultsTest.test_successful_renewalc C sL | tjddg dgg g | ddg | jtddtdg d S ) NFr} bad.pemr r z@All %ss failed. The following certificates could not be renewed:r z bad.pem (failure)rN r r' r| ro rz rv rp r r r test_failed_renewal/ s
z'DescribeResultsTest.test_failed_renewalc C sV | tjddddgddgdgdg | g d | jtd
d td g d S )
NTr} r z good2.pemr zbad2.pemzfoo.pem expires on 123z errored.conf)r z7The following certificates are not due for renewal yet:z" foo.pem expires on 123 (skipped)z+The following simulated renewals succeeded:z+ good.pem (success)
good2.pem (success)
zB
Additionally, the following renewal configurations were invalid: z errored.conf (parsefail)r zThe following %ss failed:zsimulated renewalz( bad.pem (failure)
bad2.pem (failure)r rp r r r test_all_renewal: s
z$DescribeResultsTest.test_all_renewalN)
rD rE rF rg rq rt rh rN r| r r r r r r r r ri
s
ri __main__ )rg r= sysZunittestr r9 Zacmer Zcertbotr r r r Zcertbot.tests.utilZtestsutilr
ZConfigTestCaser rH ZTestCaseri rD exitmainargv__file__r r r r s 1MA