a
}|äg`f ã @ s€ d Z ddlZddlZddlmZ ddlZddlZddlZddl m
Z
m Z ddl Z ddl m
Z
ddlZddlmZ ddlmZ ddlmZmZ dd lmZ dd
lmZ dd lmZ dd lm Z dd
l m Z ddl m Z ddl m! m"Z# ddl m$Z$ e j% &¡ Z'G dd„ de$ƒZ(G dd„ de j)ƒZ*e+e
dœdd„Z,e+e
edœdd„Z-G dd„ de j)ƒZ.e/d kr|e 0e 1ej2d d… e3g ¡¡ dS ) z Tests for certbot._internal.cli.é N)Úreload)ÚAnyÚList)Úmock)Ú
challenges)Úerrors)ÚArgumentSourceÚNamespaceConfig)Úcli)Ú constants)Ú flag_default)Údisco)Ú
filesystem)Úos)ÚTempDirTestCasec @ s e Zd ZdZdd„ ZdS )Ú TestReadFilezTest cli.read_filec C sÒ t ¡ }z¸t | j¡ t t j | jd¡¡}t j |¡}t
t j ¡ t
|¡ W d ƒ n1 sd0 Y d}t|dƒ}| |¡ W d ƒ n1 sœ0 Y t
|¡\}}W t |¡ n t |¡ 0 d S )NÚfoos bar
Úwb)r ÚgetcwdÚchdirZtempdirr ÚrealpathÚpathÚjoinÚrelpathÚpytestÚraisesÚargparseÚArgumentTypeErrorr
Z read_fileÚopenÚwrite)ÚselfZcurr_dirZ real_pathÚ
relative_pathZ
test_contentsÚfr Úcontents© r$ úD/usr/lib/python3.9/site-packages/certbot/_internal/tests/cli_test.pyÚtest_read_file s ( (zTestReadFile.test_read_fileN)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r& r$ r$ r$ r% r s r c @ s e Zd ZdZdd„ ZdS )ÚFlagDefaultTestzTests cli.flag_defaultc C s t jdkr n d S )NÚnt)r Úname©r r$ r$ r% Útest_default_directories@ s
z(FlagDefaultTest.test_default_directoriesN)r' r( r) r* r/ r$ r$ r$ r% r+ = s r+ )ÚattrÚvaluec C s d S ©Nr$ )Ú namespacer0 r1 r$ r$ r% Ú assert_set_by_user_with_valueK s r4 )r0 r1 Úsourcec C s d S r2 r$ )r3 r0 r1 r5 r$ r$ r% Úassert_value_and_sourceP s r6 c @ sÚ e Zd ZdZeee edœdd„ƒZeee edœdd„ƒZ ee dœdd „Z
e d
¡d d „ ƒZ
d
d„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd „ Zd d „ Zd d „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Z d+d,„ Z d-d.„ Z e d/¡d0d1„ ƒZ e d2¡d3d4„ ƒZ d5d6„ Z!d7d8„ Z"d9d:„ Z#d;d<„ Z$d=d>„ Z%d?d@„ Z&dAdB„ Z'dCdD„ Z(dEdF„ Z)dGdH„ Z*dIdJ„ Z+dKdL„ Z,dMdN„ Z-dOdP„ Z.dQdR„ Z/dSdT„ Z0dUdV„ Z1dWdX„ Z2dYdZ„ Z3d[d\„ Z4d]d^„ Z5e d/¡d_d`„ ƒZ6dadb„ Z7dcdd„ Z8deS )fÚ ParseTestz Test the cli args entrypoint)ÚargsÚreturnc C s t t| ¡S )z)Get result of cli.prepare_and_parse_args.)r
Úprepare_and_parse_argsÚPLUGINS©r8 r$ r$ r% Ú_unmocked_parseX s zParseTest._unmocked_parsec C s6 t ¡ t | ¡W d ƒ S 1 s(0 Y dS )zJMocks certbot._internal.display.obj.get_display and calls _unmocked_parse.N)Ú test_utilÚpatch_display_utilr7 r= r< r$ r$ r% Úparse] s
zParseTest.parser< c sÚ t ¡ ‰ ‡ fdd„}tjdˆ d t ¡ x}||ƒ j_t d¡D t t
¡ | |¡ W d ƒ n1 sn0 Y W d ƒ n1 sŒ0 Y W d ƒ n1 sª0 Y W d ƒ n1 sÈ0 Y ˆ ¡ S )z8Run a command, and return the output string for scrutinyc s ˆ | ¡ d S r2 )r )Úmessager8 Úkwargs©Úoutputr$ r% Ú write_msgh s z)ParseTest._help_output..write_msgz!certbot._internal.main.sys.stdout)Únewz!certbot._internal.main.sys.stderrN)
ÚioÚStringIOr Úpatchr> r? Z notificationÚ side_effectr r Ú
SystemExitr= Úgetvalue)r r8 rE Zmock_get_utilityr$ rC r% Ú _help_outputc s
‚zParseTest._help_outputz*certbot._internal.cli.helpful.flag_defaultc sÒ t ¡ ¶‰ ˆ ¡ ‡ fdd„}||_| dg¡}t|dg tjƒ tˆ j dƒ}|
d¡ W d ƒ n1 sl0 Y | dg¡}t|ddgtj ƒ | dg¡}t|dg tj ƒ W d ƒ n1 sÄ0 Y d S ) Nc s | dkrt tj| ¡S ˆ jgS )NZ config_files)ÚcopyÚdeepcopyr Z CLI_DEFAULTSr- )Úv©Ú
tmp_configr$ r% Úz s ÿz0ParseTest.test_cli_ini_domains..ÚcertonlyÚdomainsÚwzdomains = example.comú example.comÚrenew)
ÚtempfileÚNamedTemporaryFileÚcloserJ r@ r6 r ÚDEFAULTr r- r Ú CONFIG_FILEÚRUNTIME)r Zmock_flag_defaultZshimr3 Úfile_hr$ rQ r% Útest_cli_ini_domainst s
ÿ ( z ParseTest.test_cli_ini_domainsc C s | g ¡}dD ]}qd S )N)Ú
config_dirZlogs_dirZwork_dir©r@ )r r3 Údr$ r$ r% Ú test_no_args‰ s
zParseTest.test_no_argsc C sZ d}d}d}d}t d¡, | dd|ddd dd
dg ¡}W d ƒ n1 sL0 Y d S ) NÚcertÚkeyÚchainÚ fullchainz certbot._internal.main.installÚinstallz --cert-pathz
--key-pathz --chain-pathz--fullchain-path)r rI r@ )r re rf rg rh r3 r$ r$ r% Útest_install_abspath s
þ$z ParseTest.test_install_abspathc C s¸ | dg¡ | ddg¡}| ddg¡}dtv r0| dg¡}dtv rFn | ddg¡}| ddg¡}| ddg¡}| ddg¡}| dd g¡}| ddg¡}| dd
g¡}| dg¡}d S ) Nú--helpÚallú-hZnginxÚpluginsrX Z
automationÚrevokeri Úregister)rM r; ©r Úoutr$ r$ r% Ú test_helpŸ s zParseTest.test_helpc C s: | dg¡ | ddg¡}| ddg¡}| ddg¡}d S )NÚhelprl ri ro )rM rq r$ r$ r% Útest_help_no_dashesÙ s
z ParseTest.test_help_no_dashesc C sÈ ddg}| |¡}t|ddgƒ ddg}| |¡}t|ddgƒ ddg}| |¡}t|dg d¢ƒ ddg}| |¡}t|ddgƒ ddg}| |¡}t|ddgƒ dd g}| |¡}t|ddd
gƒ d S ) Nz-drW rU ztrailing.period.com.ztrailing.period.comz-example.com,another.net,third.org,example.com)rW ú another.netz third.orgz --domainsz#example.com,another.net,example.comrv ©r@ r4 )r Ú
short_argsr3 Z long_argsr$ r$ r% Útest_parse_domainsì s(
ÿ
z ParseTest.test_parse_domainsc C s˜ ddg}| |¡}tjjtjjg}t|d|ƒ ddg}t d¡D t t
¡ | |¡ W d ƒ n1 sl0 Y W d ƒ n1 sŠ0 Y d S )Nz--preferred-challengesz http, dnsÚ pref_challszjumping-over-the-moonz
sys.stderr) r@ r ÚHTTP01ÚtypZDNS01r4 r rI r r rK )r rx r3 Zexpectedr$ r$ r% Útest_preferred_challenges s
z#ParseTest.test_preferred_challengesc C s | d ¡ ¡}t|ddƒ d S )Nú--server example.comÚserverrW )r@ Úsplitr4 ©r r3 r$ r$ r% Útest_server_flag s zParseTest.test_server_flagc C s, | dg¡}t|ddƒ t|ddtjƒ d S )Nú
--must-stapleÚ must_stapleTÚstaple)r@ r4 r6 r r^ r r$ r$ r% Útest_must_staple_flag s z ParseTest.test_must_staple_flagc C s* | ddg¡}t|ddƒ t|ddƒ d S )Nrƒ z
--staple-ocspr„ Tr… rw r r$ r$ r% Ú&test_must_staple_and_staple_ocsp_flags s z0ParseTest.test_must_staple_and_staple_ocsp_flagsc
C sZ z$| |¡ | d d |¡¡¡ W n0 tjyT } z|D ]}q:W Y d }~n
d }~0 0 d S )Nz6The following flags didn't conflict with --server: {0}z, )r@ ZfailÚformatr r ÚError)r Z parser_argsZconflicting_argsÚerrorÚargr$ r$ r% Ú _check_server_conflict_message" s
ÿÿz(ParseTest._check_server_conflict_messagec C sF dg}| |¡}t|ddƒ t|dtjƒ |d ¡ 7 }| |d¡ d S )Nú --stagingÚstagingTr r~ )r@ r4 r Ú STAGING_URIr€ rŒ )r rx r3 r$ r$ r% Útest_staging_flag- s
zParseTest.test_staging_flagc C sˆ t |ddƒ t|ddtjƒ t|ddtjƒ t|dtjtjƒ |rdt|ddtjƒ t|ddtjƒ n t|ddtjƒ t|ddtjƒ d S ) NÚdry_runTZbreak_my_certsrŽ r ZtosZ register_unsafely_without_emailF)r4 r6 r r^ r r r\ )r r3 Zexisting_accountr$ r$ r% Ú_assert_dry_run_flag_worked6 s z%ParseTest._assert_dry_run_flag_workedc C sœ t ¡ }d |¡ ¡ }t tj¡ | |¡ W d ƒ n1 sB0 Y | | |dg ¡d¡ | | |dg ¡d¡ | | |dg ¡d¡ t
j |t
j¡}t |¡ t t
j |d¡¡ | | |dg ¡d¡ | | |dg ¡d¡ | | |dg ¡d¡ |dg7 }| |dd g ¡}t|d
d ƒ | |dt
jg ¡}t|d
t
jƒ | |dt d
¡g ¡}t|d
t
jtjƒ d g}| |g d ¢ |¡ d S )
Nz--dry-run --config-dir {0}ZauthFrT rX Zfake_account_dirTú--serverrW r r )r“ rW r )rY Zmkdtemprˆ r€ r r r r‰ r@ r’ r r r r Z ACCOUNTS_DIRr Úmkdirr4 r r
r r6 r r^ rŒ )r ra rx Z account_dirÚconfigÚ conflictsr$ r$ r% Útest_dry_run_flagC s< (ÿÿÿ
ÿzParseTest.test_dry_run_flagc C s* d}t |¡}| d |¡ ¡ ¡}d}d S )NZ rsa_key_sizez--rsa-key-size {0}ra ©r
r r@ rˆ r€ )r Zkey_size_optionZkey_size_valuer• Zconfig_dir_optionr$ r$ r% Útest_user_set_rsa_key_sizek s
z$ParseTest.test_user_set_rsa_key_sizec C s | d¡}| d¡}d S )Nz--apachez--installer webrootrb )r r• r$ r$ r% Ú)test_user_set_installer_and_authenticatorw s
z3ParseTest.test_user_set_installer_and_authenticatorc C s& d}t |¡}| d |¡ ¡ ¡}d S )NZelliptic_curvez--elliptic-curve {0}r˜ )r Zelliptic_curve_optionZelliptic_curve_option_valuer• r$ r$ r% Ú test_user_set_ecdsa_key_option€ s
z(ParseTest.test_user_set_ecdsa_key_optionc C sZ d}t |¡}| d |¡ ¡ ¡}t t¡ | d¡ W d ƒ n1 sL0 Y d S )NZkey_typez--key-type {0}z--key-type foo)r
r r@ rˆ r€ r r rK )r Zkey_type_optionZkey_type_valuer• r$ r$ r% Ú test_user_set_invalid_key_type† s
z(ParseTest.test_user_set_invalid_key_typez&certbot._internal.hooks.validate_hooksc C s" d ¡ }tj ¡ }t ||¡}d S )Nzrenew --deploy-hook foo)r€ r
ÚPluginsRegistryÚfind_allr
r: )r Zunused_mock_validate_hooksr8 rn r• r$ r$ r% Útest_user_set_deploy_hook s
z#ParseTest.test_user_set_deploy_hookz3certbot._internal.plugins.webroot._validate_webrootc C s( d ¡ }d|_tj ¡ }t ||¡}d S )Nz%renew -w /var/www/html -d example.comz
/var/www/html)r€ Z return_valuer
r rž r
r: )r Zmock_validate_webrootr8 rn r• r$ r$ r% Útest_user_set_webroot_map– s
z#ParseTest.test_user_set_webroot_mapc C sL t j ¡ D ]\}}| d|g¡}q
t j ¡ D ]\}}| d| ¡ g¡}q,d S )Nz--reason)r ZREVOCATION_REASONSÚitemsr@ Úupper)r ÚreasonÚcoder3 r$ r$ r% Ú test_encode_revocation_reasonž s z'ParseTest.test_encode_revocation_reasonc C sx t tj¡ | d ¡ ¡ W d ƒ n1 s00 Y t tj¡ | d ¡ ¡ W d ƒ n1 sj0 Y d S )Nzrenew --force-interactivez-n --force-interactive©r r r r‰ r@ r€ r. r$ r$ r% Útest_force_interactive¦ s ,z ParseTest.test_force_interactivec C sf t d¡H t t¡ | d ¡ ¡ W d ƒ n1 s:0 Y W d ƒ n1 sX0 Y d S )Nú certbot._internal.cli.sys.stderrz"--renew-hook foo --deploy-hook bar©r rI r r rK r@ r€ r. r$ r$ r% Útest_deploy_hook_conflict¬ s z#ParseTest.test_deploy_hook_conflictc C s4 d}| d|d|dg¡}t|d|ƒ t|d|ƒ d S )Nr ú --renew-hookú
--deploy-hookú--disable-hook-validationÚ deploy_hookÚ
renew_hookrw ©r r1 r3 r$ r$ r% Ú#test_deploy_hook_matches_renew_hook± s þ z-ParseTest.test_deploy_hook_matches_renew_hookc C s0 d}| d|dg¡}t|d|ƒ t|d|ƒ d S )Nr r¬ r r® r¯ rw r° r$ r$ r% Ú test_deploy_hook_sets_renew_hook¹ s ÿ z*ParseTest.test_deploy_hook_sets_renew_hookc C sf t d¡H t t¡ | d ¡ ¡ W d ƒ n1 s:0 Y W d ƒ n1 sX0 Y d S )Nr¨ z"--deploy-hook foo --renew-hook barr© r. r$ r$ r% Útest_renew_hook_conflictÀ s z"ParseTest.test_renew_hook_conflictc C s4 d}| d|d|dg¡}t|d|ƒ t|d|ƒ d S )Nr r¬ r« r r® r¯ rw r° r$ r$ r% Ú#test_renew_hook_matches_deploy_hookÅ s þ z-ParseTest.test_renew_hook_matches_deploy_hookc C s$ d}| d|dg¡}t|d|ƒ d S )Nr r« r r¯ rw r° r$ r$ r% Ú'test_renew_hook_does_not_set_renew_hookÍ s
ÿz1ParseTest.test_renew_hook_does_not_set_renew_hookc C sž t d¡€ t t¡ | d ¡ ¡ W d ƒ n1 s:0 Y t t¡ | d ¡ ¡ W d ƒ n1 sr0 Y W d ƒ n1 s0 Y d S )Nr¨ z--max-log-backups fooz--max-log-backups -42r© r. r$ r$ r% Útest_max_log_backups_errorÔ s
, z$ParseTest.test_max_log_backups_errorc C s& d}| d|g¡}t|dt|ƒƒ d S )NZ42z--max-log-backupsZmax_log_backups)r@ r4 Úintr° r$ r$ r% Ú test_max_log_backups_successÛ s z&ParseTest.test_max_log_backups_successc C sl | g ¡}t|dg tjƒ t|dg tjƒ tjjg|_dg|_| g ¡}t|dg tjƒ t|dg tjƒ d S )NrU rz rW ) r@ r6 r r\ r r{ r| rz rU r r$ r$ r% Útest_unchanging_defaultsà s
z"ParseTest.test_unchanging_defaultsc C s | dg¡}t|ddƒ d S )Nz--no-directory-hooksÚdirectory_hooksFrw r r$ r$ r% Útest_no_directory_hooks_setì s z%ParseTest.test_no_directory_hooks_setc C s | g ¡}t|ddtjƒ d S )Nrº T)r@ r6 r r\ r r$ r$ r% Ú test_no_directory_hooks_unsetð s
z'ParseTest.test_no_directory_hooks_unsetc C s | dg¡}t|ddƒ d S )Nz--delete-after-revokeÚdelete_after_revokeTrw r r$ r$ r% Útest_delete_after_revokeô s z"ParseTest.test_delete_after_revokec C s | g ¡}d S r2 rb r r$ r$ r% Ú test_delete_after_revoke_defaultø s
z*ParseTest.test_delete_after_revoke_defaultc C s | dg¡}t|ddƒ d S )Nz--no-delete-after-revoker½ Frw r r$ r$ r% Útest_no_delete_after_revokeý s z%ParseTest.test_no_delete_after_revokec C s> t tj¡ | d ¡ ¡ W d ƒ n1 s00 Y d S )Nz(--allow-subset-of-names -d *.example.orgr¦ r. r$ r$ r% Ú test_allow_subset_with_wildcard s z)ParseTest.test_allow_subset_with_wildcardc C s dD ]}dD ]}q qd S )N)rm rk )rl rn z dns-route53r$ )r Z help_flagZtopicr$ r$ r% Útest_route53_no_revert s z ParseTest.test_route53_no_revertc C s> t tj¡ | ddg¡ W d ƒ n1 s00 Y d S )Nz--hstsz --auto-hsts)r r r r‰ r@ r. r$ r$ r% Ú$test_parse_args_hosts_and_auto_hosts
s z.ParseTest.test_parse_args_hosts_and_auto_hostsc C sä t dƒ}d}d}| dg¡}t|d|tjƒ t ¡ œ}| ¡ t|j dƒ }|
d|› ¡ W d ƒ n1 sp0 Y | dd|j g¡}t|d|tj ƒ | dd|j d|g¡}t|d|tj ƒ W d ƒ n1 sÖ0 Y d S ) Nr zconfigfile.bizzcommandline.edurT rV z server = z-cr“ )
r r@ r6 r r\ rY rZ r[ r r- r r] Z COMMAND_LINE)r Z
DEFAULT_VALUEZCONFIG_FILE_VALUEZCOMMAND_LINE_VALUEr3 rR r_ r$ r$ r% Ú)test_parse_with_multiple_argument_sources s(
.þýz3ParseTest.test_parse_with_multiple_argument_sourcesc C s® | d¡}t|ddgƒ t|ddƒ t ¡ n}| ¡ t|jdƒ}| d¡ W d ƒ n1 sb0 Y | dd |jg¡}t|ddgƒ W d ƒ n1 s 0 Y d S )
Nz --preferred-chal dns --no-dirrz zdns-01rº FrV zpreferred-chal = dnsrT z--config)r@ r4 rY rZ r[ r r- r )r r3 rR r_ r$ r$ r% Útest_abbreviated_arguments+ s
(þz$ParseTest.test_abbreviated_argumentsc C sB | d¡}t|ddgƒ tj ¡ }t |ddg¡}t|ddƒ d S )Nz-d=example.comrU rW Úrunz--pre-hook="foo=bar"Zpre_hookz "foo=bar")r@ r4 r
r rž r
r: )r Zunsused_mock_validate_hooksr3 rn r$ r$ r% Útest_argument_with_equals> s
z#ParseTest.test_argument_with_equalsc C sv | d¡}t|ddƒ t|ddƒ | d¡}t|ddƒ t|ddƒ | d¡}t|ddƒ t|ddƒ t|d d
ƒ d S ) Nz-tvZ text_modeTZ
verbose_counté z-tvvvé z-tvm foo@example.comZemailzfoo@example.comrw r r$ r$ r% Útest_adjacent_short_argsH s
z"ParseTest.test_adjacent_short_argsc C s | ddg¡}t|ddgƒ d S )NrT z
-d foo.comrU zfoo.comrw r r$ r$ r% Ú test_arg_with_contained_spacesV s z(ParseTest.test_arg_with_contained_spacesN)9r' r( r) r* Ú staticmethodr Ústrr r= r@ rM r rI r` rd rj rs ru ry r} r‚ r† r‡ rŒ r r’ r— r™ rš r› rœ rŸ r r¥ r§ rª r± r² r³ r´ rµ r¶ r¸ r¹ r» r¼ r¾ r¿ rÀ rÁ r rà rÄ rÅ rÇ rÊ rË r$ r$ r$ r% r7 U sl
:
(
r7 Ú__main__rÈ )4r* r rN Ú importlibr Z
reload_modulerG ÚsysrY Útypingr r Zunittestr r Zacmer Zcertbotr Zcertbot.configurationr r Zcertbot._internalr
r Z certbot._internal.cli.cli_utilsr Zcertbot._internal.pluginsr
Zcertbot.compatr r Zcertbot.tests.utilZtestsÚutilr> r r rž r; r ZTestCaser+ rÍ r4 r6 r7 r' ÚexitÚmainÚargvÚ__file__r$ r$ r$ r% Ú sB