a
}|gs @ s d Z ddlZddlZddlZddlZddlZddlmZ ddlZddlZddl m
Z
ddl m Z ddl m
Z
ddlmZ ddlmZ dd lmZ dd
lmZ dd
lmZ G d d d ejZG d
d dejZG dd deZ G dd deZ G dd deZ G dd deZ G dd deZ G dd dejZ!G dd d ejZ"G d d d ejZ#G d d d ej$Z%e&d!kre'e(ej)d"d e*g dS )#)Tests for certbot._internal.cert_manager. N)mock)
configuration)errors)ALL_FOUR) storage_test)
filesystem)os)utilc s( e Zd ZdZ fddZdd Z ZS )BaseCertManagerTestz2Base class for setting up Cert Manager tests.
c s t d j_t jj d tj jj
dd _ fdd j D _ t
tj jjdd}|d W d n1 s0 Y d S ) NFZspecialarchive) example.org other.comc s i | ]}| | j| qS )_set_up_configdomains).0domainselfr M/usr/lib/python3.9/site-packages/certbot/_internal/tests/cert_manager_test.py
% s z-BaseCertManagerTest.setUp..z IGNORE.THISwz This file should be ignored!)supersetUpconfigquietr makedirsrenewal_configs_dirr pathjoin
config_dirr Z config_filesopenwrite)r Zjunk __class__r r r s
zBaseCertManagerTest.setUpc C s t tj| jj| t }|d ur | j |_d| j_d| j_| || jd |d dS ) Test deleteTr z6Deleted all files relating to certificate example.org.N)test_rc return_valueyesnor certnamer5 assert_called_once_with)r mock_delete_filesmock_lineage_for_certnameZ mock_notify mock_utilr r r test_delete_from_config_yesL s
z&DeleteTest.test_delete_from_config_yesc C s4 | j |_d| j_d| j_| |jdks0J dS )r7 Fr r N)r8 r9 r: r r; r5
call_countr r= r> r? r r r test_delete_from_config_no\ s
z%DeleteTest.test_delete_from_config_noc C s> | j |_tjdgf| j_d| j_| || jd dS )r7 r TN) r8 r9 display_utilOK checklistr: r5 r< r rB r r r "test_delete_interactive_single_yesh s
z-DeleteTest.test_delete_interactive_single_yesc C s> | j |_tjdgf| j_d| j_| |jdks:J dS )r7 r Fr Nr8 r9 rD rE rF r: r5 rA rB r r r !test_delete_interactive_single_not s
z,DeleteTest.test_delete_interactive_single_noc C s\ | j |_tjddgf| j_d| j_| || jd || jd |j dksXJ dS )r7 r other.orgT N)
r8 r9 rD rE rF r: r5 Zassert_any_callr rA rB r r r $test_delete_interactive_multiple_yes s
z/DeleteTest.test_delete_interactive_multiple_yesc C s@ | j |_tjddgf| j_d| j_| |jdks z=CertificatesTest.test_report_human_readable..zINVALID: EXPIREDrK )Zhoursz1 hourz 2 hour(s)TZVALIDINVALID )Zdaysz1 dayZunderz3 dayszINVALID: TEST_CERT, REVOKEDZ
indescribablez thrice.namedzINVALID:Zhorror)r9 datetimepytzr3 r2 ZnowZUTCr rd Z
target_expirynamesZ is_test_certZ timedeltaappendlenrefindallr r; ) r Z mock_revokedZ mock_serialro rp ZexpirycertZ
get_reportoutr rj r test_report_human_readable sl
z+CertificatesTest.test_report_human_readableN)r* r+ r, r- rV r rP rN rO r] r^ rb rg rx r r r r rQ s(
rQ c @ s6 e Zd ZdZedededdd ZdS )SearchLineagesTestz:Tests for certbot._internal.cert_manager._search_lineages. certbot.util.make_or_verify_dir,certbot._internal.storage.renewal_conf_filesr_ c C sF dg|_ tj|_ddlm} || jdd ddks8J |jsBJ d S )NZbadfiler r1 c S s | S rW r xr r r rk * rl z.check) r9 r CertStorageError side_effectr3 r2 Z_search_lineagesr rY )r ra mock_renewal_conf_filesmock_make_or_verify_dirr2 r r r test_cert_storage_error s
z*SearchLineagesTest.test_cert_storage_errorN)r* r+ r, r- r rP r r r r r ry s
ry c @ sn e Zd ZdZedededdd Zededdd Zededd d
Zd S ) LineageForCertnameTestz=Tests for certbot._internal.cert_manager.lineage_for_certnamerz 3certbot._internal.storage.renewal_file_for_certnamer_ c C sH d|_ tjdd}||_ ddlm} || jd|ks:J |jsDJ d S )N
somefile.conf example.comrh r r1 )r9 r Mockr3 r2 lineage_for_certnamer rY )r ra mock_renewal_conf_filer
mock_matchr2 r r r test_found_match1 s z'LineageForCertnameTest.test_found_matchc C s6 d|_ ddlm} || jdd u s(J |js2J d S )Nzother.com.confr r1 r )r9 r3 r2 r r rY r r r r2 r r r
test_no_match= s z$LineageForCertnameTest.test_no_matchc C s: t |_ddlm} || jdd u s,J |js6J d S )Nr r1 r )r r r r3 r2 r r rY r r r r test_no_renewal_fileE s
z+LineageForCertnameTest.test_no_renewal_fileN) r* r+ r, r- r rP r r r r r r r r . s r c @ sR e Zd ZdZedededdd Zededdd Zd S )
DomainsForCertnameTestz=Tests for certbot._internal.cert_manager.domains_for_certnamerz r r_ c C sX d|_ tjdd}ddg}||j_ ||_ ddlm} || jd|ksJJ |jsTJ d S )Nr r rh r r r1 ) r9 r r rq r3 r2 domains_for_certnamer rY )r ra r r r r r2 r r r r Q s z'DomainsForCertnameTest.test_found_matchc C s6 d|_ ddlm} || jdd u s(J |js2J d S )Nr r r1 r
)r9 r3 r2 r r rY r r r r r ` s z$DomainsForCertnameTest.test_no_matchN)r* r+ r, r- r rP r r r r r r r N s r c s e Zd ZdZ fddZdd Zede dd Z
e d d
Z e ed d d
Z e eddd Z
e eddd Ze eddd Z ZS )RenameLineageTestz7Tests for certbot._internal.cert_manager.rename_lineagec s t d| j_d| j_d S )Nr Zafter)r r r r; new_certnamer r# r r r l s
zRenameLineageTest.setUpc O s ddl m} |j|i |S r0 )r3 r2 Zrename_lineage)r rT rU r2 r r r r5 q s zRenameLineageTest._callr{ c C s d | j _d| j _g |_ttj | | j W d n1 sD0 Y dg|_| }t j
df|j _ttj | | j W d n1 s0 Y t j df|j _ttj | | j W d n1 s0 Y d S )Ntwozone.confr )
r r; r r9 pytestraisesr Errorr5 rD CANCELmenurE )r mock_get_utilityr util_mockr r r test_no_certnameu s **z"RenameLineageTest.test_no_certnamec C s d| j _d | j _| }tjdf|j_tt j
| | j W d n1 sR0 Y tj d f|j_tt j
| | j W d n1 s0 Y d S )Nonename)
r r; r rD r inputr9 r r r r r5 rE )r r r r r r test_no_new_certname s *z&RenameLineageTest.test_no_new_certnamer6 c C sR d| j _d| j _d |_ttj | | j W d n1 sD0 Y d S )Nr r ) r r; r r9 r r r ConfigurationErrorr5 )r r> unused_get_utilityr r r test_no_existing_certname s
z+RenameLineageTest.test_no_existing_certnamez7certbot._internal.storage.RenewableCert._check_symlinksc C sR d|_ | | j ddlm} || j| jj}|d usCertPathToLineageTest.test_options_fullchain..r )r9 r r r r5 r mock_acceptable_matchesr r r test_options_fullchain
s
z,CertPathToLineageTest.test_options_fullchainc C sB dd g|_ tj| jjddd}|| j_d| | jks>J d S )Nc S s | j S rW )r r| r r r rk rl z>CertPathToLineageTest.test_options_cert_path..r r zcert.pem)r9 r r r r r r r5 )r r Ztest_cert_pathr r r test_options_cert_path s z,CertPathToLineageTest.test_options_cert_pathc sB t j jjddd j_ fddg|_d jks>J d S )Narchiver z
cert11.pemc s | dS )Nrv r r| r r r rk rl zACertPathToLineageTest.test_options_archive_cert..r r r r r r r9 r5 r r r r test_options_archive_cert s
z/CertPathToLineageTest.test_options_archive_certc sB t j jjddd j_ fddg|_d jks>J d S )Nr r zfullchain11.pemc s | dS )Nr r r| r r r rk $ s zFCertPathToLineageTest.test_options_archive_fullchain..r r r r r test_options_archive_fullchain s
z4CertPathToLineageTest.test_options_archive_fullchainc C s" | j | j_d| | jks J d S r )r r r r5 r r r r test_only_path( s
z$CertPathToLineageTest.test_only_path)r* r+ r, r- r r5 r r r r rP r r r r r r. r r r# r r s
r c sT e Zd ZdZ fddZdd Zdd Zedd d
Z edd d Z
Z S )
MatchAndCheckOverlapszfTests for certbot._internal.cert_manager.match_and_check_overlaps w/o overlapping
archive dirs.c sB t | j | tj| jj ddd| _
| j
| j_ d S r r r r# r r r 1 s
zMatchAndCheckOverlaps.setUpc C s ddl m} |||||S )Nr )match_and_check_overlaps)rS r )r r Zacceptable_matchesZ
match_funcZrv_funcr r r r r5 9 s zMatchAndCheckOverlaps._callc s: ddl m} dg j| fdddd ks6J d S )Nr )_acceptable_matchesr c s j jS rW )r r r| r r r rk @ rl z8MatchAndCheckOverlaps.test_basic_match..c S s | j S rW rh r| r r r rk @ rl )rS r r5 r )r r r r r r = s z&MatchAndCheckOverlaps.test_basic_matchz/certbot._internal.cert_manager._search_lineagesc C sH g |_ ttj" | | jd d d W d n1 s:0 Y d S rW )r9 r r r r r5 r r Zmock_search_lineagesr r r test_no_matchesB s z%MatchAndCheckOverlaps.test_no_matchesc C sL ddg|_ ttj" | | jd d d W d n1 s>0 Y d S )NZspiderZdance)r9 r r r ZOverlappingMatchFoundr5 r r r r r test_too_many_matchesH s
z+MatchAndCheckOverlaps.test_too_many_matches) r* r+ r, r- r r5 r r rP r r r. r r r# r r - s
r c @ s e Zd ZdZdd Zededdd Zededdd Zededd
d Z ededd d
Z
ededdd Z ededdd Z dS )GetCertnameTestr c C s4 t }| | _| |j t | _d | j_ d S rW )
rN rO startr Z
addCleanupstopr rd r r; )r Zget_utility_patchr r r r R s
zGetCertnameTest.setUpr{ z2certbot._internal.storage.lineagename_for_filenamec C sl dg|_ d|_ ddlm} d}tjdf| j_ |j| jddddgksLJ || jj d d v shJ d S ) Nexample.com.confr r r1 zWhich certificate would youverbFallow_multiple
r9 r3 r2 rD rE r r
get_certnamesr call_argsr mock_name
mock_filesr2 promptr r r test_get_certnamesY s z"GetCertnameTest.test_get_certnamesc C sn dg|_ d|_ ddlm} d}tjdf| j_ |j| jdd|ddgksNJ | jj d d |ksjJ d S ) Nr r r r1
custom promptr Fr Z
custom_promptr r r r r test_get_certnames_custom_prompte s
z0GetCertnameTest.test_get_certnames_custom_promptc C sn dg|_ d|_ ddlm} tjdf| j_ tt j
" |j | j ddd W d n1 s`0 Y d S )Nr r r r1 erroring_anywayFr )
r9 r3 r2 rD r r r r r r r r r r r r r2 r r r test_get_certnames_user_aborts s z-GetCertnameTest.test_get_certnames_user_abortc C sn dg|_ d|_ ddlm} d}tjdgf| j_ |j| jddddgksNJ || jj d d v sjJ d S ) Nr r r r1 z Which certificate(s) would your Tr
r9 r3 r2 rD rE r rF r r r r r r r !test_get_certnames_allow_multiple} s z1GetCertnameTest.test_get_certnames_allow_multiplec C sp dg|_ d|_ ddlm} d}tjdgf| j_ |j| jdd|ddgksPJ | jj d d |kslJ d S ) Nr r r r1 r r Tr r r r r r /test_get_certnames_allow_multiple_custom_prompt s
z?GetCertnameTest.test_get_certnames_allow_multiple_custom_promptc C sn dg|_ d|_ ddlm} tjg f| j_ tt j
" |j | j ddd W d n1 s`0 Y d S )Nr r r r1 r Tr )
r9 r3 r2 rD r r rF r r r r r r r r r r ,test_get_certnames_allow_multiple_user_abort s z s< ,W ^&@"U