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 ejed
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)ZchallbdomainZ account_keyc @ s( e Zd ZdZdd Zdd Zdd ZdS ) NamespaceFunctionsTestz7Tests for certbot.plugins.common.*_namespace functions.c C s ddl m} d S )Nr )option_namespace)certbot.plugins.commonr )selfr r O/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/common_test.pytest_option_namespace s z,NamespaceFunctionsTest.test_option_namespacec C s ddl m} d S Nr )dest_namespacer r r r r r r test_dest_namespace# s z*NamespaceFunctionsTest.test_dest_namespacec C s ddl m} d S r r 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 ZddddZedddZedd 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 )Ninfor r# r r r more_info6 s z.PluginTest.setUp..MockPlugin.more_infoc S s |dddd d d S )Nzfoo-bardifferent_to_foo_bar destxyr )clsaddr r r add_parser_arguments9 s z9PluginTest.setUp..MockPlugin.add_parser_arguments)r r r r$ strr& classmethodr/ r r r r
MockPlugin2 s r2 r configname)r r
plugin_clsr MagicMockr4 Zplugin)r r r2 r r r setUp/ s
zPluginTest.setUpc C s d S r" r r# r r r test_initA s zPluginTest.test_initc C s d S r" r r# r r r r E s z PluginTest.test_option_namespacec C s d S r" r r# r r r test_option_nameH s zPluginTest.test_option_namec C s d S r" r r# r r r r K s z PluginTest.test_dest_namespacec C s d S r" r r# r r r test_destN s zPluginTest.test_destc C s d S r" r r# 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 r7 r6 Zinject_parser_options add_argumentassert_called_once_with)r parserr r r test_inject_parser_optionsU s
z%PluginTest.test_inject_parser_optionsc C s d S r" r r# r r r test_fallback_auth_hint] s z"PluginTest.test_fallback_auth_hintN)
r r r r r8 r9 r r: r r; r< r@ rA r r r r r , s r c s e Zd ZdZ fddZdd 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 Installerr3 )
superr8 r mkdirr4 Z
config_dirZcertbot.tests.utilrC installerreverter)r rC __class__r r r8 h s
zInstallerTest.setUpc C s ddh}d}| d|| d S )Nfoo.barbaz.quxfoo bar baz quxadd_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 )NTrS 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 )NrJ rK rL ) temporaryZadd_to_temp_checkpointrM ) functoolspartialrF rM _test_adapted_method)r rW rP rQ installer_funcreverter_func_namer r r rT | s
z,InstallerTest._test_add_to_checkpoint_commonc C s | dd d S )NZfinalize_checkpointZfoorN r# r r r test_finalize_checkpoint s z&InstallerTest.test_finalize_checkpointc C s | d d S )NZrecovery_routinerN r# r r r test_recovery_routine s z#InstallerTest.test_recovery_routinec C s | d d S )NZrevert_temporary_configrN r# r r r test_revert_temporary_config s z*InstallerTest.test_revert_temporary_configc C s | dd d S )NZrollback_checkpoints* rN r# r r r test_rollback_checkpoints s z'InstallerTest.test_rollback_checkpointsc O s* t | j|}| j||g|R i | dS )zTest 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)getattrrF rZ )r r5 argskwargsr[ r r r rO 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 s0 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 patchobjectrG r> r Z
ReverterErrorZ side_effectpytestraisesZ PluginError)r r[ r\ Z passed_argsZ
passed_kwargsZ
reverter_funcr r r rZ s
z"InstallerTest._test_adapted_methodc C s | j d S r" )rF Zinstall_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.constantsrj r sha256sum)r rj r r r _current_ssl_dhparams_hash s z(InstallerTest._current_ssl_dhparams_hashc C s ddl m} d S )Nr )ALL_SSL_DHPARAMS_HASHES)rk rn )r rn r r r $test_current_file_hash_in_all_hashes s z2InstallerTest.test_current_file_hash_in_all_hashes)r r r r r8 rR rU rV rT r] r^ r_ ra rO rZ ri rm ro
__classcell__r r rH r rB e s rB 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:*z192.168.1.1:80 [fe00::1] [fe00::1]:*z [fe00::1]:80z [fe00::1]:5z [fe00:1:2:3:4:5:6:7:8:9]:8080) r rs
fromstringaddr1addr2Zaddr3addr4addr5Zaddr6Zaddr7Zaddr8r rs r r r r8 s zAddrTest.setUpc C s d S r" r r# r r r test_fromstring s zAddrTest.test_fromstringc C s d S r" r r# r r r test_str s zAddrTest.test_strc C s d S r" r r# r r r test_get_addr_obj s zAddrTest.test_get_addr_objc C s ddl m} d S )Nr rr )r rs r} r r r test_eq s zAddrTest.test_eqc
C s` ddl m} | j| jh}|d}|d}||h}| j| jh}|d}|d}||h} d S )Nr rr rt ru rv rw )r rs ry rz rx r{ r| )
r rs Zset_aZaddr1bZaddr2bZset_bZset_cZaddr4bZaddr5bZset_dr r r test_set_inclusion s
zAddrTest.test_set_inclusionN)
r r r r r8 r~ r r r r r r r r rq s
rq c @ s( e Zd ZdZdd Zdd Zdd ZdS ) ChallengePerformerTestz4Tests for certbot.plugins.common.ChallengePerformer.c C s" t }ddlm} ||| _d S )Nr )ChallengePerformer)r r7 r r performer)r Z configuratorr r r r r8 s z ChallengePerformerTest.setUpc C s | j td d S )Nr )r Z add_challACHALLr# 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" )rg rh NotImplementedErrorr Zperformr# r r r test_perform# s z#ChallengePerformerTest.test_performN)r r r r r8 r r r r r r r s r c s` e Zd ZdZ fddZdd 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 s0 Y | j
t| qfd S )NZ
someotherhashzoptions-ssl-dest.confz.options-ssl-conf.txtzoptions-ssl-old.confzoptions-ssl-src.confw)rD r8 hashesr pathjoinZtempdir dest_path hash_pathold_path source_pathopenwriteappendr rl )r r frH r r r8 + 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 | jS r" )r rl r r# r r r _current_file_hash> s z3InstallVersionControlledFileTest._current_file_hashc C s d S r" r r# r r r _assert_current_fileA s z5InstallVersionControlledFileTest._assert_current_filec C s | | d S r" )r r r# r r r test_no_fileF s z-InstallVersionControlledFileTest.test_no_filec C s" t dD ]}| | 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" )shutilcopyfiler r r r r# r r r !test_prev_file_updates_to_currentQ s zBInstallVersionControlledFileTest.test_prev_file_updates_to_currentc C st | t| jd}|d W d n1 s40 Y td}| W d n1 sf0 Y d S )Na a new line for the wrong hash
certbot.plugins.common.logger)r r r r r re )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 td}| W d n1 s0 Y td}| W d n1 s0 Y d S )Nr r r Zhashofanoldversionr )r r r r r re 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 r8 r r r r r r r r rp r r rH r r ( s
r __main__r( ).r rX r sysZunittestr ZjosepyZjoserg Zacmer r Zcertbotr r r Zcertbot.compatr r Z
certbot.testsr
r Z test_utilZJWKRSAloadZ load_vectorZAUTH_KEYZ"KeyAuthorizationAnnotatedChallengeZchall_to_challbZHTTP01ZSTATUS_PENDINGr ZTestCaser r ZConfigTestCaserB rq r ZTempDirTestCaser r exitmainargv__file__r r r r s> 9_QL