a
}|ägn8 ã @ sz d Z ddlZddlZddlZddlZddlmZ ddlZddlZddl m
Z
ddl m Z ddl m
Z
ddl mZ ddl mZ dd lmZ dd
lmZ dd lmZ dd lmZ ej e d
¡¡Ze
je e
j dde j ¡dedZ G dd„ dej ƒZ!G dd„ dej ƒZ"G dd„ dej#ƒZ$G dd„ dej ƒZ%G dd„ dej ƒZ&G d d „ d ej'ƒZ(e)d krve *e +ej,d d… e-g ¡¡ dS ) z!Tests for certbot.plugins.common.é N)Úmock)Ú
challenges)Úmessages)Ú achallenges)Ú crypto_util)Úerrors)Ú
filesystem)Úos)Ú acme_util)Úutilzrsa512_key.pems token1)Útokenzencryption-example.demo)ZchallbÚdomainZ account_keyc @ s( e Zd ZdZdd„ Zdd„ Zdd„ ZdS ) ÚNamespaceFunctionsTestz7Tests for certbot.plugins.common.*_namespace functions.c C s ddl m} d|dƒks J ‚d S )Nr )Úoption_namespacezfoo-Úfoo)Úcertbot.plugins.commonr )Úselfr © r úO/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/common_test.pyÚtest_option_namespace s z,NamespaceFunctionsTest.test_option_namespacec C s ddl m} d|dƒks J ‚d S )Nr ©Údest_namespaceZfoo_r ©r r ©r r r r r Útest_dest_namespace# s z*NamespaceFunctionsTest.test_dest_namespacec C s ddl m} d|dƒks J ‚d S )Nr r Zfoo_bar_úfoo-barr r r r r Ú test_dest_namespace_with_dashes' s z6NamespaceFunctionsTest.test_dest_namespace_with_dashesN)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r r r r r r r s r c @ sX e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dd„ Z dS )Ú
PluginTestz'Test for certbot.plugins.common.Plugin.c C s@ ddl m} G dd„ d|ƒ}|| _t ¡ | _|| jdd| _d S )Nr )ÚPluginc @ s4 e Zd Zddœdd„Zedœdd„Zedd„ ƒZdS ) z$PluginTest.setUp..MockPluginN)Úreturnc S s d S ©Nr ©r r r r Úprepare3 s z,PluginTest.setUp..MockPlugin.preparec S s dS )NÚinfor r% r r r Ú more_info6 s z.PluginTest.setUp..MockPlugin.more_infoc S s |dddd d d S )Nr Údifferent_to_foo_baré ©ÚdestÚxÚyr )ÚclsÚaddr r r Úadd_parser_arguments9 s z9PluginTest.setUp..MockPlugin.add_parser_arguments)r r r r& Ústrr( Ú classmethodr1 r r r r Ú
MockPlugin2 s r4 r ©ÚconfigÚname)r r" Ú
plugin_clsr Ú MagicMockr6 Úplugin)r r" r4 r r r ÚsetUp/ s
zPluginTest.setUpc C s& d| j jksJ ‚| j| j jks"J ‚d S )Nr )r: r7 r6 r% r r r Ú test_initA s zPluginTest.test_initc C s d| j jksJ ‚d S )Nzmock-)r: r r% r r r r E s z PluginTest.test_option_namespacec C s d| j d¡ksJ ‚d S )Nz mock-foo_barÚfoo_bar)r: Z option_namer% r r r Útest_option_nameH s zPluginTest.test_option_namec C s d| j jksJ ‚d S )NZmock_)r: r r% r r r r K s z PluginTest.test_dest_namespacec C s, d| j d¡ksJ ‚d| j d¡ks(J ‚d S )NÚ mock_foo_barr r= )r: r, r% r r r Ú test_destN s zPluginTest.test_destc C s | j j| j d¡ksJ ‚d S )Nr )r6 r? r: Zconfr% r r r Ú test_confR s zPluginTest.test_confc C s. t ¡ }| j |d¡ |jjdddd d d S )Nr z--mock-foo-barr) r* r+ )r r9 r8 Zinject_parser_optionsÚ add_argumentÚassert_called_once_with)r Úparserr r r Útest_inject_parser_optionsU s
ÿz%PluginTest.test_inject_parser_optionsc C s@ d| j tjtjg¡v s J ‚d| j tjtjtjg¡v s r r@ rA rE rF r r r r r! , s r! c sˆ e Zd ZdZ‡ fdd„Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Zdd„ Zd d „ Z‡ ZS ) Ú
InstallerTestz+Tests for certbot.plugins.common.Installer.c sB t ƒ ¡ t | jj¡ ddlm} || jdd| _| jj | _ d S )Nr )ÚDummyInstallerZ Installerr5 )
Úsuperr; r Úmkdirr6 Z
config_dirZcertbot.tests.utilrH Ú installerÚreverter)r rH ©Ú __class__r r r; h s
ÿzInstallerTest.setUpc C s ddh}d}| d||¡ d S )Núfoo.barúbaz.quxúfoo bar baz quxÚadd_to_checkpoint©Ú_test_wrapped_method)r ÚfilesÚ
save_notesr r r Útest_add_to_real_checkpointq s z)InstallerTest.test_add_to_real_checkpointc C s | d¡ d S )NF©Ú _test_add_to_checkpoint_commonr% r r r Ú test_add_to_real_checkpoint2v s z*InstallerTest.test_add_to_real_checkpoint2c C s | d¡ d S )NTrX r% r r r Ú test_add_to_temporary_checkpointy s z.InstallerTest.test_add_to_temporary_checkpointc C s@ ddh}d}t j| jj|d}|r(d}nd}| ||||¡ d S )NrO rP rQ )Ú temporaryZadd_to_temp_checkpointrR )Ú functoolsÚpartialrK rR Ú_test_adapted_method)r r\ rU rV Úinstaller_funcÚreverter_func_namer r r rY | s
ÿz,InstallerTest._test_add_to_checkpoint_commonc C s | dd¡ d S )NZfinalize_checkpointr rS r% r r r Útest_finalize_checkpointŠ s z&InstallerTest.test_finalize_checkpointc C s | d¡ d S )NZrecovery_routinerS r% r r r Útest_recovery_routine s z#InstallerTest.test_recovery_routinec C s | d¡ d S )NZrevert_temporary_configrS r% r r r Ú test_revert_temporary_config s z*InstallerTest.test_revert_temporary_configc C s | dd¡ d S )NZrollback_checkpointsé* rS r% r r r Útest_rollback_checkpoints“ s z'InstallerTest.test_rollback_checkpointsc O s* t | j|ƒ}| j||g|¢R i |¤Ž dS )zÎTest a wrapped reverter method.
:param str name: name of the method to test
:param tuple args: position arguments to method
:param dict kwargs: keyword arguments to method
N)ÚgetattrrK r_ )r r7 ÚargsÚkwargsr` r r r rT – s z"InstallerTest._test_wrapped_methodc O s” t j | j|¡p}||i |¤Ž |j|i |¤Ž tj|_t tj
¡ ||i |¤Ž W d ƒ n1 sh0 Y W d ƒ n1 s†0 Y dS )a Test an adapted reverter method
:param callable installer_func: installer method to test
:param str reverter_func_name: name of the method on the
reverter that should be called
:param tuple passed_args: positional arguments passed from
installer method to the reverter method
:param dict passed_kargs: keyword arguments passed from
installer method to the reverter method
N) r ÚpatchÚobjectrL rC r Z
ReverterErrorZ side_effectÚpytestÚraisesZ PluginError)r r` ra Z passed_argsZ
passed_kwargsZ
reverter_funcr r r r_ ¡ s
z"InstallerTest._test_adapted_methodc C s" | j ¡ tj | j j¡s J ‚d S r$ )rK Zinstall_ssl_dhparamsr ÚpathÚisfileZ ssl_dhparamsr% r r r Útest_install_ssl_dhparamsµ s
z'InstallerTest.test_install_ssl_dhparamsc C s ddl m} t |¡S )Nr )ÚSSL_DHPARAMS_SRC)Úcertbot._internal.constantsrq r Ú sha256sum)r rq r r r Ú_current_ssl_dhparams_hash¹ s z(InstallerTest._current_ssl_dhparams_hashc C s$ ddl m} | ¡ |v s J dƒ‚d S )Nr )ÚALL_SSL_DHPARAMS_HASHESzwConstants.ALL_SSL_DHPARAMS_HASHES must be appended with the sha256 hash of self.config.ssl_dhparams when it is updated.)rr ru rt )r ru r r r Ú$test_current_file_hash_in_all_hashes½ s ÿz2InstallerTest.test_current_file_hash_in_all_hashes)r r r r r; rW rZ r[ rY rb rc rd rf rT r_ rp rt rv Ú
__classcell__r r rM r rG e s rG c @ s@ e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dS )ÚAddrTestz&Tests for certbot.plugins.common.Addr.c C sp ddl m} | d¡| _| d¡| _| d¡| _| d¡| _| d¡| _| d¡| _| d ¡| _ | d
¡| _
d S ) Nr ©ÚAddrú 192.168.1.1ú
192.168.1.1:*ú192.168.1.1:80ú [fe00::1]ú [fe00::1]:*ú [fe00::1]:80z [fe00::1]:5z [fe00:1:2:3:4:5:6:7:8:9]:8080) r rz Ú
fromstringÚaddr1Úaddr2Úaddr3Úaddr4Úaddr5Úaddr6Úaddr7Úaddr8©r rz r r r r; Ç s zAddrTest.setUpc C s( | j ¡ dksJ ‚| j ¡ dks$J ‚| j ¡ dks6J ‚| j ¡ dksHJ ‚| j ¡ dksZJ ‚| j ¡ dkslJ ‚| j ¡ dks~J ‚| j ¡ dksJ ‚| j ¡ dks¢J ‚| j ¡ dks´J ‚| j ¡ dksÆJ ‚| j ¡ dksØJ ‚| j ¡ dksêJ ‚| j ¡ dksüJ ‚| j ¡ dksJ ‚| j
¡ dks$J ‚d S ) Nr{ Ú Ú*Z80r~ zfe00:0:0:0:0:0:0:1Ú5zfe00:1:2:3:4:5:6:7) r‚ Zget_addrZget_portrƒ r„ r… r† r‡ Zget_ipv6_explodedrˆ r‰ r% r r r Útest_fromstringÒ s, ÿÿÿzAddrTest.test_fromstringc C sp t | jƒdksJ ‚t | jƒdks$J ‚t | jƒdks6J ‚t | jƒdksHJ ‚t | jƒdksZJ ‚t | jƒdkslJ ‚d S )Nr{ r| r} r~ r r€ )r2 r‚ rƒ r„ r… r† r‡ r% r r r Útest_strç s zAddrTest.test_strc C s” t | j d¡ƒdksJ ‚t | j d¡ƒdks0J ‚t | j d¡ƒdksHJ ‚t | j d¡ƒdks`J ‚t | j d¡ƒdksxJ ‚t | j d¡ƒd ksJ ‚d S )
NZ443z192.168.1.1:443r‹ r{ rŒ r| z
[fe00::1]:443r~ r )r2 r‚ Ú get_addr_objrƒ r… r† r% r r r Útest_get_addr_objï s zAddrTest.test_get_addr_objc C s | j | j d¡ksJ ‚| j | jks&J ‚| j dks4J ‚| j| j d¡ksJJ ‚| j| jksZJ ‚| jdkshJ ‚ddlm} | j| d¡ksˆJ ‚| j| d¡ksœJ ‚d S )Nr‹ i
r ry z
[fe00:0:0::1]z[fe00:0::0:0:1])r‚ rƒ r r… r† r rz r rŠ r r r Útest_eq÷ s zAddrTest.test_eqc
C sx ddl m} | j| jh}| d¡}| d¡}||h}||ks@J ‚| j| jh}| d¡}| d¡}||h} || kstJ ‚d S )Nr ry r{ r| r~ r )r rz r‚ rƒ r r… r† )
r rz Zset_aZaddr1bZaddr2bZset_bZset_cZaddr4bZaddr5bZset_dr r r Útest_set_inclusion s
zAddrTest.test_set_inclusionN)
r r r r r; rŽ r r‘ r’ r“ r r r r rx Ä s
rx c @ s( e Zd ZdZdd„ Zdd„ Zdd„ ZdS ) ÚChallengePerformerTestz4Tests for certbot.plugins.common.ChallengePerformer.c C s" t ¡ }ddlm} ||ƒ| _d S )Nr )ÚChallengePerformer)r r9 r r• Ú performer)r Z configuratorr• r r r r; s z ChallengePerformerTest.setUpc C s8 | j td¡ dt| j jƒks"J ‚dg| j jks4J ‚d S )Nr r* )r– Z add_challÚACHALLÚlenZachallsÚindicesr% r r r Útest_add_chall s z%ChallengePerformerTest.test_add_challc C s8 t t¡ | j ¡ W d ƒ n1 s*0 Y d S r$ )rl rm ÚNotImplementedErrorr– Zperformr% r r r Ú test_perform# s z#ChallengePerformerTest.test_performN)r r r r r; rš rœ r r r r r” s r” c s` e Zd ZdZ‡ fdd„Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dd„ Z ‡ Z
S )Ú InstallVersionControlledFileTestzATests for certbot.plugins.common.install_version_controlled_file.c s¶ t ƒ ¡ dg| _tj | jd¡| _tj | jd¡| _tj | jd¡| _ tj | jd¡| _
| j
| j fD ]J}t |dƒ}| |¡ W d ƒ n1 s”0 Y | j
t |¡¡ qfd S )NZ
someotherhashzoptions-ssl-dest.confz.options-ssl-conf.txtzoptions-ssl-old.confzoptions-ssl-src.confÚw)rI r; Úhashesr rn ÚjoinZtempdirÚ dest_pathÚ hash_pathÚold_pathÚ source_pathÚopenÚwriteÚappendr rs )r rn ÚfrM r r r; + s
(z&InstallVersionControlledFileTest.setUpc C s& ddl m} || j| j| j| jƒ d S )Nr )Ú install_version_controlled_file)r r© r¡ r¢ r¤ rŸ )r r© r r r Ú_call7 s ýz&InstallVersionControlledFileTest._callc C s t | j¡S r$ )r rs r¤ r% r r r Ú_current_file_hash> s z3InstallVersionControlledFileTest._current_file_hashc C s. t j | j¡sJ ‚t | j¡| ¡ ks*J ‚d S r$ )r rn ro r¡ r rs r« r% r r r Ú_assert_current_fileA s
ÿz5InstallVersionControlledFileTest._assert_current_filec C s& t j | j¡rJ ‚| ¡ | ¡ d S r$ )r rn ro r¡ rª r¬ r% r r r Ú test_no_fileF s z-InstallVersionControlledFileTest.test_no_filec C s" t dƒD ]}| ¡ | ¡ qd S )Né )Úrangerª r¬ )r Ú_r r r Útest_current_fileK s z2InstallVersionControlledFileTest.test_current_filec C s$ t | j| j¡ | ¡ | ¡ d S r$ )ÚshutilÚcopyfiler£ r¡ rª r¬ r% r r r Ú!test_prev_file_updates_to_currentQ s zBInstallVersionControlledFileTest.test_prev_file_updates_to_currentc C sÆ | ¡ t| jdƒ}| d¡ W d ƒ n1 s40 Y t d¡(}| ¡ |jjdu sbJ ‚W d ƒ n1 sv0 Y tj
| j¡s’J ‚t | j
¡| ¡ ksªJ ‚t | j¡| ¡ ksÂJ ‚d S )NÚaú a new line for the wrong hash
ú certbot.plugins.common.loggerF)rª r¥ r¡ r¦ r rj ÚwarningÚcalledr rn ro r rs r¤ r« )r Ú mod_ssl_confÚ mock_loggerr r r Ú3test_manually_modified_current_file_does_not_updateV s ( .
ÿ
ÿzTInstallVersionControlledFileTest.test_manually_modified_current_file_does_not_updatec C s t | jdƒ}| d¡ W d ƒ n1 s,0 Y t | jdƒ}| d¡ W d ƒ n1 sb0 Y t d¡0}| ¡ |jjd d dks˜J ‚W d ƒ n1 s¬0 Y t
| j ¡| ¡ ksÎJ ‚t d¡(}| ¡ |jj
du sòJ ‚W d ƒ n1 s0 Y d S ) Nrµ r¶ rž Zhashofanoldversionr· r zh%s has been manually modified; updated file saved to %s. We recommend updating %s for security purposes.F)r¥ r¡ r¦ r¢ r rj rª r¸ Z call_argsr rs r¤ r« r¹ )r rº r¨ r» r r r Ú&test_manually_modified_past_file_warnsc s (( ÿ&
ÿ zGInstallVersionControlledFileTest.test_manually_modified_past_file_warns)r r r r r; rª r« r¬ r r± r´ r¼ r½ rw r r rM r r ( s
r Ú__main__r* ).r r] r² ÚsysZunittestr ZjosepyZjoserl Zacmer r Zcertbotr r r Zcertbot.compatr r Z
certbot.testsr
r Z test_utilZJWKRSAÚloadZ load_vectorZAUTH_KEYZ"KeyAuthorizationAnnotatedChallengeZchall_to_challbZHTTP01ZSTATUS_PENDINGr— ZTestCaser r! ZConfigTestCaserG rx r” ZTempDirTestCaser r ÚexitÚmainÚargvÚ__file__r r r r Ú s> ÿý9_QL