a
}|g` @ s d Z ddlZddlZddlmZ ddlZddlZddlZddlm Z ddl
Z
ddl m Z ddl
mZ ddl
mZ ddlm mZ G dd d ejZG d
d d ejZG d d
d
ejZG dd dejZG dd dejZG dd dejZG dd dejZ G dd dejZ G dd dejZ G dd dejZ G d d d ejZ G d d d ejZ!G d d! d!ejZ"G d"d# d#ejZ#G d$d% d%ejZ$G d&d' d'ejZ%G d(d) d)ejZ&G d*d+ d+ejZ'e(d,kre)e
*ej+d-d e,g dS ).zTests for certbot.util. N)reload)mock)errors)
filesystem)osc @ s, e Zd ZdZedd Zdd Zdd ZdS ) EnvNoSnapForExternalCallsTestz6Tests for certbot.util.env_no_snap_for_external_calls.c C s ddl m} | S )Nr ) env_no_snap_for_external_calls) certbot.utilr )clsr r E/usr/lib/python3.9/site-packages/certbot/_internal/tests/util_test.py_call s z#EnvNoSnapForExternalCallsTest._callc C sb t jd }t j }d| |d< d|d< d|d< tjd|d W d n1 sT0 Y d S ) NPATH"RANDOM_NONSENSE_GARBAGE/blah/blah:RANDOM_NONSENSE_GARBAGESNAPTrueCERTBOT_SNAPPED certbot.compat.os.environ.copy return_value)r environcopyr patch)selfZ
original_path
env_copy_dictr r r test_removed s
z*EnvNoSnapForExternalCallsTest.test_removedc C s t j }d|d |d< | }tjd|d> |dd |dd d|d< |d= d|d< W d n1 st0 Y d S ) Nr r r r r r r r )r r r r r pop)r Zenv_copy_dict_unmodifiedr r r r test_noop" s
z'EnvNoSnapForExternalCallsTest.test_noopN)__name__
__module__ __qualname____doc__ classmethodr
r r r r r r r s
r c @ sR e Zd ZdZedd Zeddd Zeddd Z edd d
Z
d S )
RunScriptTestz"Tests for certbot.util.run_script.c C s ddl m} ||S )Nr )
run_script)r r% )r
paramsr% r r r r
6 s zRunScriptTest._callzcertbot.util.subprocess.runc C s, d| _ d| _d| _| dg\}}dS )z.These will be changed soon enough with reload.r stdoutstderrtestN)
returncoder' r( r
)r mock_runouterrr r r test_default; s
zRunScriptTest.test_defaultc C sB t |_ttj | dg W d n1 s40 Y d S )Nr) )OSError side_effectpytestraisesr SubprocessErrorr
r r+ r r r test_bad_processF s z RunScriptTest.test_bad_processc C sD d| _ ttj | dg W d n1 s60 Y d S )N r) )r* r1 r2 r r3 r
r4 r r r test_failureM s zRunScriptTest.test_failureN) r r r! r" r# r
r r r. r5 r7 r r r r r$ 4 s
r$ c @ s, e Zd ZdZedd Zdd Zdd ZdS )
ExeExistsTestz"Tests for certbot.util.exe_exists.c C s ddl m} ||S )Nr )
exe_exists)r r9 )r
Zexer9 r r r r
X s zExeExistsTest._callc C s2 t jddd W d n1 s$0 Y d S )N%certbot.util.filesystem.is_executableTr r r r r r r test_exe_exists] s z ExeExistsTest.test_exe_existsc C s2 t jddd W d n1 s$0 Y d S )Nr: Fr r; r< r r r test_exe_not_existsa s z!ExeExistsTest.test_exe_not_existsN)r r r! r" r# r
r= r> r r r r r8 U s
r8 c sH e Zd ZdZedd Z fddZededdd Z Z
S )
LockDirUntilExitz+Tests for certbot.util.lock_dir_until_exit.c O s ddl m} ||i |S )Nr )lock_dir_until_exit)r r@ )r
argskwargsr@ r r r r
h s zLockDirUntilExit._callc s t dd l}t|j d S )Nr )supersetUpr
reload_moduleutil)r certbot __class__r r rD m s
zLockDirUntilExit.setUpzcertbot.util.loggerz certbot.util.atexit_registerc C s^ t j| jd}t| | | j | | | | |jd d }ddlm } | d S )Nsubdirr )rF )
r pathjointempdirr mkdirr
call_argsrG rF )r Z
mock_registerZ mock_loggerrJ Zregistered_funcrF r r r test_its s
zLockDirUntilExit.test_it) r r r! r" r# r
rD r r rP
__classcell__r r rH r r? f s
r? c @ s< e Zd ZdZdd Zeddd Zeddd Zd
S ) SetUpCoreDirTestz/Tests for certbot.util.make_or_verify_core_dir.c O s ddl m} ||i |S )Nr )set_up_core_dir)r rS )r rA rB rS r r r r
s zSetUpCoreDirTest._callz certbot.util.lock_dir_until_exitc C s" t j| jd}| |dd d S )Nnew Fr rK rL rM r
)r Z mock_lockZnew_dirr r r test_success s z SetUpCoreDirTest.test_successz certbot.util.make_or_verify_dirc C sF t |_ttj | | jdd W d n1 s80 Y d S )NrU F)r/ r0 r1 r2 r Errorr
rM )r Zmock_make_or_verifyr r r r7 s z SetUpCoreDirTest.test_failureN) r r r! r" r
r r rW r7 r r r r rR s
rR c sH e Zd ZdZ fddZdd Zdd Zdd Zd
d Zd d
Z Z
S )MakeOrVerifyDirTestzTests for certbot.util.make_or_verify_dir.
Note that it is not possible to test for a wrong directory owner,
as this testing script would have to be run as root.
c s. t tj| jd| _t| jd d S )Nfoo )rC rD r rK rL rM r rN r< rH r r rD s
zMakeOrVerifyDirTest.setUpc C s ddl m} |||ddS )Nr )make_or_verify_dirT)strict)r r\ )r Z directorymoder\ r r r r
s zMakeOrVerifyDirTest._callc C s t j| jd}| |d d S )Nbari rV )r rK r r r test_creates_dir_when_missing s z1MakeOrVerifyDirTest.test_creates_dir_when_missingc C s | | jd d S )Nr[ )r
rK r< r r r (test_existing_correct_mode_does_not_fail s z t tj | | jd W d n1 s00 Y d S )N )r1 r2 r rX r
rK r< r r r test_existing_wrong_mode_fails s z2MakeOrVerifyDirTest.test_existing_wrong_mode_failsc C sp t jtdN}t |_tt | dd W d n1 sD0 Y W d n1 sb0 Y d S )Nmakedirsr_ i ) r r objectr r/ r0 r1 r2 r
)r rd r r r test_reraises_os_error s z*MakeOrVerifyDirTest.test_reraises_os_error) r r r! r" rD r
r` ra rc rf rQ r r rH r rY s rY c sB e Zd ZdZ fddZd
ddZdd Zd d
Zd d Z Z S )UniqueFileTestz#Tests for certbot.util.unique_file.c s t tj| jd| _d S )Nzfoo.txt)rC rD r rK rL rM default_namer< rH r r rD s
zUniqueFileTest.setUpr[ c C s ddl m} || j|S )Nr ) unique_file)r ri rh )r r^ ri r r r r
s zUniqueFileTest._callc C sJ | \}}|d | t|}W d n1 s<0 Y d S )Nr_ )r
writecloseopen)r fdnamefr r r test_returns_fd_for_writing s
z*UniqueFileTest.test_returns_fd_for_writingc C s0 | d\}}| d\}}| | d S )NrU r[ r
rk )r fd1name1fd2name2r r r test_right_mode s z UniqueFileTest.test_right_modec
C sd | \}}| \}}| \}}tj|}tj|}tj|} | | | d S N)r
r rK basenamerk )
r rr rs rt ru Zfd3Zname3Z basename1Z basename2Z basename3r r r test_default_exists s
z"UniqueFileTest.test_default_exists)r[ )
r r r! r" rD r
rp rv ry rQ r r rH r rg s
rg c @ s2 e Zd ZdZd ddZdd Zdd Zd d
Zd S )
UniqueLineageNameTestz+Tests for certbot.util.unique_lineage_name. c C s ddl m} || j||S )Nr )unique_lineage_name)r r| rM )r filenamer^ r| r r r r
s zUniqueLineageNameTest._callc C s | d\}}| d S )Nwowrq )r ro rK r r r
test_basic s z UniqueLineageNameTest.test_basicc C sH g }t dD ]}|| d q |d \}}|D ]\}}| q2d S )N
r~ )rangeappendr
rk )r items_ro rn r r r
test_multiple s z#UniqueLineageNameTest.test_multiplec C sl t jdttjdD tt | d W d n1 s@0 Y W d n1 s^0 Y d S )Nz certbot.compat.filesystem.open)r0 r~ )r r r/ errnoZEIOr1 r2 r
r< r r r r7 s z"UniqueLineageNameTest.test_failureN)r{ )r r r! r" r
r r r7 r r r r rz s
rz c s@ e Zd ZdZ fddZdd Zdd Zdd Zd
d Z Z S ) SafelyRemoveTestz%Tests for certbot.util.safely_remove.c s t tj| jd| _d S )NrZ )rC rD r rK rL rM r< rH r r rD s
zSafelyRemoveTest.setUpc C s ddl m} || jS )Nr )
safely_remove)r r rK )r r r r r r
s zSafelyRemoveTest._callc C s8 t | jd W d n1 s"0 Y | d S )Nw)rl rK r
r< r r r test_exists$ s z SafelyRemoveTest.test_existsc C s | d S rw r
r< r r r test_missing* s z SafelyRemoveTest.test_missingc C sf t dH}t|_tt | W d n1 s:0 Y W d n1 sX0 Y d S )Nzcertbot.util.os.remove)r r r/ r0 r1 r2 r
)r Z mock_remover r r test_other_error_passthrough/ s z-SafelyRemoveTest.test_other_error_passthrough)
r r r! r" rD r
r r r rQ r r rH r r s r c @ s, e Zd ZdZedd Zdd Zdd ZdS )
SafeEmailTestzTest safe_email.c C s ddl m} ||S )Nr )
safe_email)r r )r
addrr r r r r
8 s zSafeEmailTest._callc C s g d}|D ]}q d S )N)zcertbot@certbot.orgztbd.ade@gmail.comzabc_def.jdk@hotmail.museumr r addrsr r r r test_valid_emails= s z SafeEmailTest.test_valid_emailsc C s g d}|D ]}q d S )N)zcertbot@certbot..orgz.tbd.ade@gmail.comz ~/abc_def.jdk@hotmail.museumr r r r r test_invalid_emailsF s z!SafeEmailTest.test_invalid_emailsN)r r r! r" r# r
r r r r r r r 6 s
r c @ sP e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dS )AddDeprecatedArgumentTestz Test add_deprecated_argument.c C s t | _d S rw )argparseArgumentParserparserr< r r r rD R s z AddDeprecatedArgumentTest.setUpc C s ddl m} || jj|| d S )Nr )add_deprecated_argument)r r r add_argument)r
argument_namenargsr r r r r
U s z AddDeprecatedArgumentTest._callc C sH | dd td }| jdg W d n1 s:0 Y d S )N --old-optionr certbot.util.logger.warningr
r r r
parse_argsr Z mock_warnr r r test_warning_no_argY s ,z-AddDeprecatedArgumentTest.test_warning_no_argc C sJ | dd td }| jddg W d n1 s<0 Y d S )Nr r6 r 42r r r r r test_warning_with_arga s .z/AddDeprecatedArgumentTest.test_warning_with_argc C sl | dd t }tjd|d6 z| jdg W n tyH Y n0 W d n1 s^0 Y d S )Nr z
sys.stdout)rT z-h)r
ioStringIOr r r r
SystemExit)r r' r r r test_helpi s $z#AddDeprecatedArgumentTest.test_helpc C s | t dS )zTest when ACTION_TYPES_THAT_DONT_NEED_A_VALUE is a set.
This variable is a set in configargparse versions < 0.12.0.
N)_test_constant_commonsetr< r r r test_set_constants s z+AddDeprecatedArgumentTest.test_set_constantc C s | t dS )zTest when ACTION_TYPES_THAT_DONT_NEED_A_VALUE is a tuple.
This variable is a tuple in configargparse versions >= 0.12.0.
N)r tupler< r r r test_tuple_constant{ s z-AddDeprecatedArgumentTest.test_tuple_constantc C sN t d0}| |_| dd | dd W d n1 s@0 Y d S )Nzcertbot.util.configargparser r6 z
--old-option2r )r r Z#ACTION_TYPES_THAT_DONT_NEED_A_VALUEr
)r typZmock_configargparser r r r s
*z/AddDeprecatedArgumentTest._test_constant_commonN) r r r! r" rD r
r r r r r r r r r r r P s
r c @ sX e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dS )EnforceLeValidityzTest enforce_le_validity.c C s ddl m} ||S )Nr )enforce_le_validity)r r )r domainr r r r r
s zEnforceLeValidity._callc C s: t tj | d W d n1 s,0 Y d S )Nz..r1 r2 r ConfigurationErrorr
r< r r r test_sanity s z EnforceLeValidity.test_sanityc C s: t tj | d W d n1 s,0 Y d S )Nzhello_world.example.comr r< r r r test_invalid_chars s z$EnforceLeValidity.test_invalid_charsc C s: t tj | d W d n1 s,0 Y d S )Nz-a.example.comr r< r r r test_leading_hyphen s z%EnforceLeValidity.test_leading_hyphenc C s: t tj | d W d n1 s,0 Y d S )Nza-.example.comr r< r r r test_trailing_hyphen s z&EnforceLeValidity.test_trailing_hyphenc C s: t tj | d W d n1 s,0 Y d S )NZcomr r< r r r test_one_label s z EnforceLeValidity.test_one_labelc C s d S rw r r< r r r test_valid_domain s z#EnforceLeValidity.test_valid_domainc C sp t tj | d W d n1 s,0 Y t tj | d W d n1 sb0 Y d S )Nzhttp://example.comzhttps://example.comr r< r r r test_input_with_scheme s (z(EnforceLeValidity.test_input_with_schemec C s d S rw r r< r r r !test_valid_input_with_scheme_name s z3EnforceLeValidity.test_valid_input_with_scheme_nameN)
r r r! r" r
r r r r r r r r r r r r r s r c @ sp e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dd Zdd ZdS )EnforceDomainSanityTestzTest enforce_domain_sanity.c C s ddl m} ||S )Nr )enforce_domain_sanity)r r )r r r r r r r
s z EnforceDomainSanityTest._callc C s@ t tj | dd W d n1 s20 Y d S )N eichhörnchen.example.comzutf-8)r1 r2 r r r
encoder< r r r test_nonascii_str s z)EnforceDomainSanityTest.test_nonascii_strc C s: t tj | d W d n1 s,0 Y d S )Nr r r< r r r test_nonascii_unicode s z-EnforceDomainSanityTest.test_nonascii_unicodec C s> d}t tj | | W d n1 s00 Y d S )NA aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar )r Z long_domainr r r
test_too_long s z%EnforceDomainSanityTest.test_too_longc C s d dddd}| | d S )Nz{0}.{1}.{2}.{3}?aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa?bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbZ?cccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccccZ?dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddformatr
)r Znot_too_long_domainr r r test_not_too_long s z)EnforceDomainSanityTest.test_not_too_longc C s> d}t tj | | W d n1 s00 Y d S )Nzfizz..example.comr )r Zempty_label_domainr r r test_empty_label s z(EnforceDomainSanityTest.test_empty_labelc C s> d}t tj | | W d n1 s00 Y d S )Nz
example.com..r )r Zempty_trailing_label_domainr r r test_empty_trailing_label s z1EnforceDomainSanityTest.test_empty_trailing_labelc C s> d}t tj | | W d n1 s00 Y d S )N@aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar r Zlong_label_domainr r r test_long_label_1 s z)EnforceDomainSanityTest.test_long_label_1c C sF d dd}ttj | | W d n1 s80 Y d S )N {0}.{1}.comr r )r r1 r2 r r r
r r r r test_long_label_2 s z)EnforceDomainSanityTest.test_long_label_2c C s d dd}| | d S )Nr r r r )r Znot_too_long_label_domainr r r test_not_long_label s z+EnforceDomainSanityTest.test_not_long_labelc C s> d}t tj | | W d n1 s00 Y d S )N r )r Z empty_domainr r r test_empty_domain s z)EnforceDomainSanityTest.test_empty_domainc C s | d d S )Nzthis.is.xn--ls8h.tldr r< r r r test_punycode_ok s z(EnforceDomainSanityTest.test_punycode_okN)r r r! r" r
r r r r r r r r r r r r r r r r s r c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) IsWildcardDomainTestz Tests for is_wildcard_domain.c C s d| _ d| _d S )Nz
*.example.orgz example.org)ZwildcardZ no_wildcardr< r r r rD s zIsWildcardDomainTest.setUpc C s ddl m} ||S )Nr )is_wildcard_domain)r r )r r r r r r r
s zIsWildcardDomainTest._callc C s d S rw r r< r r r test_no_wildcard s z%IsWildcardDomainTest.test_no_wildcardc C s d S rw r r< r r r
test_wildcard s z"IsWildcardDomainTest.test_wildcardN)r r r! r" rD r
r r r r r r r s
r c @ s e Zd ZdZedeej
dddd Z edeej
dddd Z edeej
ddd d
Z
d d Zedeej
ddd
d Zedeej
dddd ZdS )
OsInfoTestz Test OS / distribution detectionzcertbot.util.distrolinuxzrequires Linuxc C s dd l m} d|j_| }d S )Nr zfirst debian third)r rF Zliker Zget_systemd_os_like)r m_distrocbutilZid_likesr r r test_systemd_os_release_like s z'OsInfoTest.test_systemd_os_release_likec C s\ dd l m} tjddd$ d|j_g d|j_W d n1 sD0 Y dg|j_d S )Nr platform.system_alias)r r r r 1.0)r somethingr Zwhatever)r rF r r versionr rn r0 r r r r r r test_get_os_info_ua
s
z OsInfoTest.test_get_os_info_uac C s` dd l m} td6}d|j_d|j_d|_d|j_d|j_W d n1 sR0 Y d S )Nr zplatform.systemrn r r r else)r rF r r idr r )r r r Z
mock_platformr r r test_get_os_info s zOsInfoTest.test_get_os_infoc
C s\ dd l m} tdd, tjddd W d n1 s@0 Y tjdddB td}d | _W d n1 s0 Y W d n1 s0 Y tjdd
d W d n1 s0 Y tjdd d@ tjd d
d W d n1 s0 Y W d n1 s.0 Y W d n1 sN0 Y d S )Nr zcertbot.util._USE_DISTROFr )Z
NonSystemDr r r )darwinr r zsubprocess.runz42.42.42)Zfreebsdz
9.3-RC3-p1r )Zwindowsr r zplatform.win32_ver)Z4242Z952r )r rF r r r' )r r Zrun_mockr r r test_non_systemd_os_info* s. < z#OsInfoTest.test_non_systemd_os_infoc C s dd l m} d|j_d|j_d S )Nr r r rF r r r r r r r test_python_os_info_notfoundC s z'OsInfoTest.test_python_os_info_notfoundc C s dd l m} d|j_d|j_d S )Nr Ztestdistr r r r r r test_python_os_info_customK s z%OsInfoTest.test_python_os_info_customN)r r r! r" r r unittestZ
skipUnlesssysplatform
startswithr r r r r r r r r r r s"
r c @ s< e Zd ZdZdd Zedd Zdd Zdd Zd
d Z d S )
AtexitRegisterTestz'Tests for certbot.util.atexit_register.c C s t | _d| _ddi| _d S )N)hiZanswer* )r Z MagicMockfuncrA rB r< r r r rD V s
zAtexitRegisterTest.setUpc O s ddl m} ||i |S )Nr )atexit_register)r r )r
rA rB r r r r r
[ s zAtexitRegisterTest._callc C s( | t | jj| ji | j d S rw ) _test_commonr getpidr Zassert_called_withrA rB r< r r r test_called` s z AtexitRegisterTest.test_calledc C s | j dd d S )Nr ) initial_pid)r r< r r r test_not_calledd s z"AtexitRegisterTest.test_not_calledc C s t d| t d.}| j| jg| jR i | j W d n1 sL0 Y |jj\}}|d }||dd i | W d n1 s0 Y d S )Nzcertbot.util._INITIAL_PIDzcertbot.util.atexitr r6 )r r r
r rA rB registerrO )r r Z mock_atexitrA rB Z atexit_funcr r r r h s < z AtexitRegisterTest._test_commonN)
r r r! r" rD r# r
r r r r r r r r T s
r c @ s< e Zd ZdZedd Zdd Zdd Zdd Zd
d Z d S )
LooseVersionTestzTest for certbot.util.LooseVersion.
These tests are based on the original tests for
distutils.version.LooseVersion at
https://github.com/python/cpython/blob/v3.10.0/Lib/distutils/tests/test_version.py#L58-L81.
c O s ddl m} ||i |S )Nr ) LooseVersion)r r )r
rA rB r r r r r
} s zLooseVersionTest._callc C s d}|D ]\}}qd S )N)z1.5.1z1.5.2b2z3.4jz
1996.07.12Z2g6Z11gz0.960923z 2.2beta29z1.13++z5.5.kw)2.0z2.0.1)abr r Z comparisonsZv1Zv2r r r test_less_than s z LooseVersionTest.test_less_thanc C s d}|D ]\}}qd S )N))8.02r )1ar )r 2.0.0)r r
r r r r r
test_equal s zLooseVersionTest.test_equalc C s d}|D ]\}}qd S N))Z161z3.10a)z3.2.pl0z3.1.1.6r r r r r test_greater_than s z"LooseVersionTest.test_greater_thanc C s@ d}|D ]2\}}t t W d q1 s00 Y qd S )N))z bookworm/sid9)r r )r1 r2
ValueErrorr r r r test_incomparible s z"LooseVersionTest.test_incomparibleN)
r r r! r" r# r
r r r
r r r r r r t s
r c @ s4 e Zd ZdZedd Zdd Zdd Zdd Zd
S ) ParseLooseVersionTestzTest for certbot.util.parse_loose_version.
These tests are based on the original tests for
distutils.version.LooseVersion at
https://github.com/python/cpython/blob/v3.10.0/Lib/distutils/tests/test_version.py#L58-L81.
c O s ddl m} ||i |S )Nr )parse_loose_version)r r )r
rA rB r r r r r
s zParseLooseVersionTest._callc C s d}|D ]\}}qd S )N)r r r r r r r r r r r s z$ParseLooseVersionTest.test_less_thanc C s d S rw r r< r r r r s z ParseLooseVersionTest.test_equalc C s d}|D ]\}}qd S r r r r r r r
s z'ParseLooseVersionTest.test_greater_thanN) r r r! r" r# r
r r r
r r r r r s
r __main__r6 )-r" r r importlibr rE r r r r r1 rG r Zcertbot.compatr r Zcertbot.tests.utilZtestsrF Z test_utilZTestCaser r$ r8 ZTempDirTestCaser? rR rY rg rz r r r r r r r r r r r exitmainargv__file__r r r r s@ "!$'4 ;';S /!