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 sv t jd }t j }d| |d< d|d< d|d< tjd|d$ | d |ksTJ W d n1 sh0 Y d S ) NPATH"RANDOM_NONSENSE_GARBAGE/blah/blah:RANDOM_NONSENSE_GARBAGESNAPTrueCERTBOT_SNAPPED certbot.compat.os.environ.copy return_value)r environcopyr patchr
)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 ksbJ d|d< | d |d ksJ |d= d|d< | d |d ksJ W d n1 s0 Y d S ) Nr r r r r r r r )r r r r r popr
)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 sD d| _ d| _d| _| dg\}}|dks4J |dks@J 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 s@ t jddd | ds J W d n1 s20 Y d S )N%certbot.util.filesystem.is_executableTr /path/to/exer r r
r r r r test_exe_exists] s z ExeExistsTest.test_exe_existsc C s@ t jddd | dr J W d n1 s20 Y d S )Nr: Fr r; 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 rA )r
argskwargsrA 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 rE m s
zLockDirUntilExit.setUpzcertbot.util.loggerz certbot.util.atexit_registerc C s t j| jd}t| | | j | | | | |jdksHJ |jd d }ddl m
} t |j dkstJ | |j
jdksJ d S )Nsubdirr6 r )rG )r pathjointempdirr mkdirr
call_count call_argsrH rG lenZ_LOCKSdebug)r Z
mock_registerZ mock_loggerrK Zregistered_funcrG r r r test_its s
zLockDirUntilExit.test_it) r r r! r" r# r
rE r r rU
__classcell__r r rI 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 rX )r rB rC rX r r r r
s zSetUpCoreDirTest._callz certbot.util.lock_dir_until_exitc C s@ t j| jd}| |dd t j|s.J |jdks t tj | | jd W d n1 s00 Y d S )N )r1 r2 r r] r
rM 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 )Nmakedirsrd i ) r r objectr r/ r0 r1 r2 r
)r rk r r r test_reraises_os_error s z*MakeOrVerifyDirTest.test_reraises_os_error) r r r! r" rE r
rg rh rj rm rV r r rI r r^ s r^ 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)rD rE r rM rN rO default_namer= rI r r rE s
zUniqueFileTest.setUpr` c C s ddl m} || j|S )Nr ) unique_file)r rq rp )r rc rq r r r r
s zUniqueFileTest._callc C sZ | \}}|d | t| }| dks8J W d n1 sL0 Y d S )Nrd )r
writecloseopenread)r fdnamefr r r test_returns_fd_for_writing s
z*UniqueFileTest.test_returns_fd_for_writingc C sP | d\}}| d\}}t|ds,J t|dsJ t|tsLJ d|v sXJ |D ]\}}| q\d S )N
r z
wow-0009.conf)rangeappendr
r r r strrs )r items_rx rw 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 r s
r 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 )Nr_ )rD rE r rM rN rO r= rI r r rE s
zSafelyRemoveTest.setUpc C s ddl m} || jS )Nr )
safely_remove)r r rM )r r r r r r
s zSafelyRemoveTest._callc C sJ t | jd W d n1 s"0 Y | tj| jrFJ d S )Nw)rt rM r
r r[ r= r r r test_exists$ s z SafelyRemoveTest.test_existsc C s | tj| jrJ d S N)r
r rM 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" rE r
r r r rV r r rI 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 ]}| |s J d| q d S )N)zcertbot@certbot.orgztbd.ade@gmail.comzabc_def.jdk@hotmail.museum
%s failed.r
r addrsr r r r test_valid_emails= s z SafeEmailTest.test_valid_emailsc C s, g d}|D ]}| |r J d| q d S )N)zcertbot@certbot..orgz.tbd.ade@gmail.comz ~/abc_def.jdk@hotmail.museumr r 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 r )argparseArgumentParserparserr= r r r rE 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 s | dd td }| jdg W d n1 s:0 Y |jdksRJ d|jd d v shJ d|jd d v s~J d S )N --old-optionr certbot.util.logger.warningr6
is deprecatedr
r r r
parse_argsrQ rR r Z mock_warnr r r test_warning_no_argY s ,z-AddDeprecatedArgumentTest.test_warning_no_argc C s | dd td }| jddg W d n1 s<0 Y |jdksTJ d|jd d v sjJ d|jd d v sJ d S )Nr r6 r 42r r r r r r r test_warning_with_arga s .z/AddDeprecatedArgumentTest.test_warning_with_argc C s| | dd t }tjd|d6 z| jdg W n tyH Y n0 W d n1 s^0 Y d| vsxJ d S )Nr rL z
sys.stdout)rY z-h) r
r StringIOr r r r
SystemExitgetvalue)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 s` t d0}| |_| dd | dd W d n1 s@0 Y t|jdks\J d S )Nzcertbot.util.configargparser r6 z
--old-option2rL )r r Z#ACTION_TYPES_THAT_DONT_NEED_A_VALUEr
rS )r typZmock_configargparser r r r s
*z/AddDeprecatedArgumentTest._test_constant_commonN) r r r! r" rE 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 | ddksJ d S )Nz example.comr 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 | ddksJ d S )Nzhttp.example.comr 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)wildcard no_wildcardr= r r r rE 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( | | jrJ | | j r$J d S r )r
r r r= r r r test_no_wildcard s z%IsWildcardDomainTest.test_no_wildcardc C s( | | jsJ | | j s$J d S r )r
r r r= r r r
test_wildcard s z"IsWildcardDomainTest.test_wildcardN)r r r! r" rE 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_| }t|dks,J d|v s8J d S )Nr zfirst debian third Zdebian)r rG Zliker Zget_systemd_os_likerS )r m_distrocbutilZid_likesr r r test_systemd_os_release_like s
z'OsInfoTest.test_systemd_os_release_likec C s dd l m} tjdddB d|j_g d|j_| d |j
dd ksNJ W d n1 sb0 Y d
g|j_| d
ksJ d S ) Nr platform.system_alias)r r r r 1.0)r somethingr T)ZprettyZwhatever) r rG r r versionr rw r0 Zget_os_info_uarN get_python_os_infor r r r r r test_get_os_info_ua
s &
z OsInfoTest.test_get_os_info_uac C s dd l m} tdV}d|j_d|j_d|_| dks>J d|j_d|j_| d ks^J W d n1 sr0 Y d S )
Nr zplatform.systemrw r r )rw r r else)r r )r rG r r idr r Z get_os_info)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$ | d dks@J W d n1 sT0 Y tjdddj td @}d
| _| d d ksJ | d d
ksJ W d n1 s0 Y W d n1 s0 Y tjdd
d" | dksJ W d n1 s 0 Y tjdddR tjddd" | dksZJ W d n1 sp0 Y W d n1 s0 Y W d n1 s0 Y d S )Nr zcertbot.util._USE_DISTROFr )Z
NonSystemDr r r Z
nonsystemd)darwinr r zsubprocess.runz42.42.42r r6 )freebsdz
9.3-RC3-p1r )r 9)windowsr r zplatform.win32_ver)Z4242952r )r r )r rG r r r r' )r r Zrun_mockr r r test_non_systemd_os_info* s4 2 P2z#OsInfoTest.test_non_systemd_os_infoc C s4 dd l m} d|j_d|j_| d dks0J d S )Nr r r r rG r r r r r r r r test_python_os_info_notfoundC s z'OsInfoTest.test_python_os_info_notfoundc C s0 dd l m} d|j_d|j_| dks,J d S )Nr testdistr )r r 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 MagicMockfuncrB rC r= r r r rE V s
zAtexitRegisterTest.setUpc O s ddl m} ||i |S )Nr )atexit_register)r r )r
rB rC r r r r r
[ s zAtexitRegisterTest._callc C s( | t | jj| ji | j d S r ) _test_commonr getpidr
Zassert_called_withrB rC r= r r r test_called` s z AtexitRegisterTest.test_calledc C s | j dd | jjdu s J d S )Nr ) initial_pidF)r r
calledr= 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 |jjsbJ |jj\}}|d }||dd i | W d n1 s0 Y d S )Nzcertbot.util._INITIAL_PIDzcertbot.util.atexitr r6 ) r r r
r
rB rC registerr rR )r r Z mock_atexitrB rC Z atexit_funcr r r r h s < z AtexitRegisterTest._test_commonN)
r r r! r" rE 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
rB rC r r r r r
} s zLooseVersionTest._callc C s4 d}|D ]&\}}| || |dksJ 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
try_risky_comparisonr Z comparisonsZv1Zv2r r r test_less_than s z LooseVersionTest.test_less_thanc C s4 d}|D ]&\}}| || |dksJ qd S )N))8.02r$ )1ar% )r 2.0.0)r r& r r r" r r r
test_equal s zLooseVersionTest.test_equalc C s4 d}|D ]&\}}| || |dksJ qd S )N)Z161z3.10a)z3.2.pl0z3.1.1.6r6 r r" r r r test_greater_than s z"LooseVersionTest.test_greater_thanc C sZ d}|D ]L\}}t t* | || |s6J W d q1 sJ0 Y qd S )N))z bookworm/sidr )r% r )r1 r2
ValueErrorr
r! r" 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
rB rC r- r r r r
s zParseLooseVersionTest._callc C s. d}|D ] \}}| || |k sJ qd S )N)r r r r r r r" r r r r# s z$ParseLooseVersionTest.test_less_thanc C s | d| dksJ d S )Nr$ r r= r r r r' s z ParseLooseVersionTest.test_equalc C s. d}|D ] \}}| || |ksJ qd S )Nr( 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 rF r r r r r1 rH r Zcertbot.compatr r Zcertbot.tests.utilZtestsrG Z test_utilZTestCaser r$ r8 ZTempDirTestCaser@ rW r^ rn r r r r r r r r r
r r, r exitmainargv__file__r r r r s@ "!$'4 ;';S /!