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 sl d|_ t| jjd}d | j_d | j_d | j_t | j}t
||}|jd }ddl m
} ||| d S )NFzsample-renewal-ancient.conf
renewalparamsr renewal) 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 sN ddl m} d|_ddiddgd}|| j| i dd}|| j| d S )Nr r
Fztest.example.comz
/var/www/testz/var/www/other)Z webroot_mapr )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 d S )NZ
INVALID_VALUETsample-renewal.confNNNNr r
*certbot._internal.renewal.hooks.renew_hook)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 d S )
NZ
INVALID_CURVETZecdsazsample-renewal-ec.confZecr 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 |jt jd d S )NFTr r r r
r! )r r r# Znew_keyr$ r r r
r r r r r r% r& r r r' r( Zassert_called_withANY)r r r r r r* r r r r test_new_keyg s 0zRenewalTest.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 )NFTZrsai 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# r- 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 sP d|_ t| j}d|_t| jjd}ddlm } t
| 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)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 sB d|_ tj| jjddd}ddlm} t| j}| || d S )NFr r, r r
)
r r
r r r r r r9 r: r; )r r r r r< 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+ r. r0 r8 r
Zpatch_display_utilr= r> 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.renewalrD )clsargskwargsrD r r r _call s z'RestoreRequiredConfigElementsTest._callr c C s d|_ | | jddi d S )NFallow_subset_of_namesTruer rI r 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 )NFrJ mayber r5 r6 r r7 rI r r r r r r r "test_allow_subset_of_names_failure s zDRestoreRequiredConfigElementsTest.test_allow_subset_of_names_failurec C s6 d|_ dddi}| | j| tjjtjjg}d S )NF pref_challsz http-01, dns,)r splitrI r r ZHTTP01typDNS01r r r Zexpectedr r r test_pref_challs_list s
z7RestoreRequiredConfigElementsTest.test_pref_challs_listc C s* d|_ ddi}| | j| tjjg}d S )NFrS Zdns)r rI r r rW rV rX 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 )NFrS zfinding-a-shrubberyrP rQ r r r test_pref_challs_failure s z:RestoreRequiredConfigElementsTest.test_pref_challs_failurec C s d|_ | | jddi d S )NF must_staplerK rL rM 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\ rO rP rQ r r r test_must_staple_failure s z:RestoreRequiredConfigElementsTest.test_must_staple_failurec C s2 ddl m} d | j_d|_| | jd|ji d S )Nr ) constantsFserver)r r_ r r` r rI ZV1_URI)r r r_ 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)r` r r )rI r Z assertEqualr )r r r r r test_related_values s
z5RestoreRequiredConfigElementsTest.test_related_valuesN)r? r@ rA __doc__ classmethodrI r r' rB r r rN rR rY rZ r[ r] r^ ra rb r r r r rC s(
rC c @ sT e Zd ZdZdd Zdd Zedd Zdd Zd
d Z d d
Z
dd Z dd Z dS )DescribeResultsTestz r4 z>DescribeResultsTest._assert_success_output..)rj 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.rI r r% rx rl r r r test_no_renewal_attempts" s z,DescribeResultsTest.test_no_renewal_attemptsc C s. | tjdddgd d d | g d d S )NFry good.pem)P
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -z)Congratulations, all renewals succeeded: z good.pem (success)O- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -rz rl r r r test_successful_renewal& s z+DescribeResultsTest.test_successful_renewalc C sL | tjddg dgg g | ddg | jtddtdg d S ) NFry bad.pemr} r~ z@All %ss failed. The following certificates could not be renewed:r z bad.pem (failure)rI r r% rx rk rv rr rl r r r test_failed_renewal/ s
z'DescribeResultsTest.test_failed_renewalc C sV | tjddddgddgdgdg | g d | jtd
d td g d S )
NTry 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 rl r r r test_all_renewal: s
z$DescribeResultsTest.test_all_renewalN)
r? r@ rA rc rm rp rd rI rx r{ r r r r r r r re
s
re __main__ )rc r9 sysZunittestr r5 Zacmer Zcertbotr r r r Zcertbot.tests.utilZtestsutilr
ZConfigTestCaser rC ZTestCasere r? exitmainargv__file__r r r r s 1MA