a
}|äg< ã @ s( d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z ddl m
Z
ddl Z ddl
Z
ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd
lmZ e j e d ¡¡Z G d
d„ de j ƒZ!G dd„ de j ƒZ"e#dkr$e $e
%ej&dd… e'g ¡¡ dS )z,Tests for certbot._internal.plugins.webroot.é )Úprint_functionN)Úmock)Ú
challenges)Ú achallenges)Úerrors)Ú
filesystem)Úos)Úutil)Ú acme_utilzrsa512_key.pemc @ sN e Zd ZdZejejdedZ dd„ Z
dd„ Z dd „ Z d
d „ Z
d d
„ Ze ¡ dd„ ƒZejejdde ¡ dd„ ƒƒZejejdde ¡ dd„ ƒƒZejejdddd„ ƒZe ¡ dd„ ƒZe ¡ dd„ ƒZe ¡ d d „ ƒZd d „ Z d d!„ Z e d"¡d#d$„ ƒZ e ¡ d%d&„ ƒZ!d'd(„ Z"d)d*„ Z#d+d,„ Z$d-d.„ Z%d/d0„ Z&e d1¡d2d3„ ƒZ'd4S )5ÚAuthenticatorTestz:Tests for certbot._internal.plugins.webroot.Authenticator.ú thing.com©ZchallbÚdomainZ account_keyc C s– ddl m} t ¡ | _tj | jd¡| _t | j¡ tj | jd¡| _
tj | jdd¡| _ tj | j d¡| _ t
j| jd| jid| _|| jdƒ| _d S ) Nr ©Ú
AuthenticatorÚwebrootú .well-knownúacme-challengeZ:ZXZhR3hmQURzNnBTUmIyTEF2OUlaZjE3RHQzanV4R0orUEN0OTJ3citvQQr )Ú webroot_pathÚ webroot_map)Ú!certbot._internal.plugins.webrootr ÚtempfileÚmkdtempZ workspacer ÚpathÚjoinr ÚmkdirÚpartial_root_challenge_pathÚroot_challenge_pathÚvalidation_pathr Ú MagicMockÚconfigÚauth©Úselfr © r$ úP/usr/lib/python3.9/site-packages/certbot/_internal/tests/plugins/webroot_test.pyÚsetUp# s$
ÿÿþÿzAuthenticatorTest.setUpc C s t | j¡ d S ©N)ÚshutilÚrmtreer ©r# r$ r$ r% ÚtearDown8 s zAuthenticatorTest.tearDownc C s | j ¡ }d S r' )r! Ú more_info)r# r, r$ r$ r% Útest_more_info; s
z AuthenticatorTest.test_more_infoc C s t ¡ }| j |¡ d S r' )r r r! Zadd_parser_arguments)r# Úaddr$ r$ r% Útest_add_parser_arguments@ s z+AuthenticatorTest.test_add_parser_argumentsc C s | j ¡ d S r' )r! Úpreparer* r$ r$ r% Ú test_prepareE s z AuthenticatorTest.test_preparec sL g | j _d| ji| j _|ƒ }tjdf|j_| j | j
g¡ |jj D ]‰ qBd S )Núotherthing.comé ) r r r r Ú display_utilÚOKÚmenuÚ return_valuer! ÚperformÚachallÚcall_args_list©r# Úmock_get_utilityÚ mock_displayr$ ©Úcallr% Útest_webroot_from_listH s z(AuthenticatorTest.test_webroot_from_listzTest specific to Windows)Úreasonc C s8 |ƒ }t jdf|j_| j | jg¡ | j | jg¡ d S )Nr3 )r4 r5 r6 r7 r! r8 r9 Úcleanupr; r$ r$ r% Ú(test_webconfig_file_generate_and_cleanupY s
z:AuthenticatorTest.test_webconfig_file_generate_and_cleanupc C s¨ |ƒ }t jdf|j_tj | jdd¡}t |¡ tj |d¡}t |dƒ}|
d¡ W d ƒ n1 sh0 Y | j | j
g¡ ddlm} | |¡}dd lm} d S )
Nr3 r r ú
web.configÚwÚ somethingr )Ú crypto_util)Ú_WEB_CONFIG_SHA256SUMS)r4 r5 r6 r7 r r r r ÚmakedirsÚopenÚwriter! r8 r9 ÚcertbotrG Z sha256sumr rH ) r# r< r= Úchallenge_pathÚwebconfig_pathÚfilerG Zwebconfig_hashrH r$ r$ r% Ú$test_foreign_webconfig_file_handlingd s
(
z6AuthenticatorTest.test_foreign_webconfig_file_handlingc C s¢ t jt tjddd¡dtd}| j| jj d< t
j | jdd¡}t
|¡ t
j |d¡}t|d ƒ}| d
¡ W d ƒ n1 s‚0 Y | j | j|g¡ d S ) Nó bingo©ÚtokenÚpendingzsecond-thing.comr
r r rD rE rF )r Ú"KeyAuthorizationAnnotatedChallenger
Úchall_to_challbr ÚHTTP01ÚKEYr r r r r r rI rJ rK r! r8 r9 )r# Zachall_2rM rN rO r$ r$ r% Ú'test_foreign_webconfig_multiple_domainsv s þ
(z9AuthenticatorTest.test_foreign_webconfig_multiple_domainsc sz g | j _d| ji| j _|ƒ }tjdff|j_t t
j ¡ | j
| jg¡ W d ƒ n1 s^0 Y |jjD ]‰ qpd S )Nr2 éÿÿÿÿ)r r r r r4 ÚCANCELr6 Ú side_effectÚpytestÚraisesr Ú PluginErrorr! r8 r9 r: r; r$ r> r% Ú&test_webroot_from_list_help_and_cancel† s . z8AuthenticatorTest.test_webroot_from_list_help_and_cancelc C s€ g | j _d| ji| j _|ƒ }tjdf|j_t d¡8}tj
dftj| jff|_ | j
| jg¡ W d ƒ n1 sr0 Y d S )Nú
something.comr ú'certbot.display.ops.validated_directoryrZ )r r r r r4 r5 r6 r7 r Úpatchr[ r\ r! r8 r9 ©r# r< r= Úmr$ r$ r% Útest_new_webroot– s
ÿ.z"AuthenticatorTest.test_new_webrootc C sš g | j _i | j _|ƒ }tjdf|j_t d¡X}tj df|_t
t j
¡ | j | jg¡ W d ƒ n1 sn0 Y W d ƒ n1 sŒ0 Y d S )Nr rb rZ )r r r r4 r5 r6 r7 r rc r[ r] r^ r r_ r! r8 r9 rd r$ r$ r% Ú!test_new_webroot_empty_map_cancel¥ s z3AuthenticatorTest.test_new_webroot_empty_map_cancelc C sL d | j _i | j _t tj¡ | j g ¡ W d ƒ n1 s>0 Y d S r' ) r r r r] r^ r r_ r! r8 r* r$ r$ r% Útest_perform_missing_root± s z+AuthenticatorTest.test_perform_missing_rootc C sú t j | jd¡| j_t j | jd¡}t|dƒ}| d¡ W d ƒ n1 sN0 Y t | jd¡ z6t|dƒ W d ƒ n1 sˆ0 Y t dƒ W nJ t
yæ t t
j¡ | j g ¡ W d ƒ n1 sØ0 Y Y n0 t | jd¡ d S ) NZnullZrndrE Zthingimyr Úrz9Warning, running tests as root skips permissions tests...iÀ )r r r r! Ú full_pathrJ rK r ÚchmodÚprintÚOSErrorr] r^ r r_ r8 )r# Zpermission_canaryÚfr$ r$ r% Ú"test_perform_reraises_other_errors· s ( 0z4AuthenticatorTest.test_perform_reraises_other_errorszJcertbot._internal.plugins.webroot.filesystem.copy_ownership_and_apply_modec C s" t tjdƒ|_| j | jg¡ d S )NÚmsg)rm ÚerrnoÚEACCESr\ r! r8 r9 )r# Zmock_ownershipr$ r$ r% Útest_failed_chownÇ s z#AuthenticatorTest.test_failed_chownc C s– t ¡ }g | j_d| ji| j_|ƒ }tjdftj|ff|j_ t
j t j
dtd}t d¡*}tj|f|_| j |g¡ W d ƒ n1 sˆ0 Y d S )Nz whatever.comr ra r
rb )r r r r r r r4 r5 r6 r\ r rU r
ÚHTTP01_PrX r rc r7 r! r8 )r# r< Z new_webrootr= r9 re r$ r$ r% Ú#test_perform_new_webroot_not_in_mapÌ s ÿÿ ,z5AuthenticatorTest.test_perform_new_webroot_not_in_mapc C sN | j ¡ | j | jg¡ t | j¡D ]"\}}}|D ]}tj ||¡}q4q&d S r' )r! r0 r8 r9 r Úwalkr r )r# ÚdirpathÚdirnamesÚ_Z directoryrj r$ r$ r% Útest_perform_permissionsÛ s
z*AuthenticatorTest.test_perform_permissionsc C s` | j ¡ | j | jg¡}t| jƒ}| ¡ }W d ƒ n1 sB0 Y | j | jg¡ d S r' )r! r0 r8 r9 rJ r ÚreadrB )r# Z responsesZ validation_fZ
validationr$ r$ r% Útest_perform_cleanupë s
&z&AuthenticatorTest.test_perform_cleanupc C s: t | j¡ | j ¡ | j | jg¡ | j | jg¡ d S r' )r r r r! r0 r8 r9 rB r* r$ r$ r% Ú"test_perform_cleanup_existing_dirsû s
z4AuthenticatorTest.test_perform_cleanup_existing_dirsc C sn t jt tjddd¡dtd}d}t | j ¡ | j
¡ | j
|| j
g¡ | j
| j
g¡ | j
|g¡ d S )NrQ rR rT r r
ZYmluZ28)r rU r
rV r rW rX r r r r! r0 r8 r9 rB )r# Z bingo_achallZbingo_validation_pathr$ r$ r% Ú(test_perform_cleanup_multiple_challenges s ÿý
z:AuthenticatorTest.test_perform_cleanup_multiple_challengesc C sR | j ¡ | j | jg¡ tj | jd¡}t |¡ | j
| jg¡ t |¡ d S )NZleftover) r! r0 r8 r9 r r r r r r rB Úrmdir)r# Z
leftover_pathr$ r$ r% Útest_cleanup_leftovers s
z(AuthenticatorTest.test_cleanup_leftoverszcertbot.compat.os.rmdirc C sB | j ¡ | j | jg¡ tƒ }tj|_||_| j | jg¡ d S r' ) r! r0 r8 r9 rm rq rr r\ rB )r# Z
mock_rmdirZos_errorr$ r$ r% Útest_cleanup_failure$ s
z&AuthenticatorTest.test_cleanup_failureN)(Ú__name__Ú
__module__Ú __qualname__Ú__doc__r rU r
rt rX r9 r& r+ r- r/ r1 Ú test_utilZpatch_display_utilr@ ÚunittestZskipIfr Z
POSIX_MODErC rP rY r` rf rg rh ro r rc rs ru rz r| r} r~ r€ r r$ r$ r$ r% r sL ÿ
r c @ sZ e Zd ZdZejejdedZ dd„ Z
dd„ Z dd „ Z d
d „ Z
d d
„ Zdd„ Zdd„ ZdS )ÚWebrootActionTestz#Tests for webroot argparse actions.r r
c C sF ddl m} t ¡ | _t ¡ | _| jjdddg d | | jd¡ d S )Nr r z-dz --domainsÚappend)ÚactionÚdefaultr )
r r r r r ÚargparseÚArgumentParserÚparserÚ add_argumentZinject_parser_optionsr" r$ r$ r% r& 8 s
ÿzWebrootActionTest.setUpc C s | j dt d| ji¡g¡}d S )Nz
--webroot-mapr )rŽ Ú
parse_argsÚjsonÚdumpsr )r# Úargsr$ r$ r% Útest_webroot_map_action@ s ÿz)WebrootActionTest.test_webroot_map_actionc C s, | j d | jj| j¡ ¡ ¡}| |¡}d S )Nz
-d {0} -w {1})rŽ r Úformatr9 r r ÚsplitÚ_get_config_after_perform©r# r“ r r$ r$ r% Útest_domain_before_webrootE s
ÿ
z,WebrootActionTest.test_domain_before_webrootc C s| t tj¡ | j d ¡ ¡ W d ƒ n1 s20 Y t tj¡ | j d ¡ ¡ W d ƒ n1 sn0 Y d S )Nz-d foo -w bar -w bazz-d foo -w bar -d baz -w qux)r] r^ r r_ rŽ r r– r* r$ r$ r% Ú test_domain_before_webroot_errorK s .z2WebrootActionTest.test_domain_before_webroot_errorc C s2 | j d | j| jjt ¡ ¡ ¡ ¡}| |¡}d S ©Nz-w {0} -d {1} -w {2} -d bar)
rŽ r r• r r9 r r r r– r— r˜ r$ r$ r% Útest_multiwebrootQ s
ÿ
z#WebrootActionTest.test_multiwebrootc C s, t ¡ }| j d | j| jj|¡ ¡ ¡}d S r› ) r r rŽ r r• r r9 r r– )r# Zother_webroot_pathr“ r$ r$ r% Ú(test_webroot_map_partial_without_performX s
ÿ
z:WebrootActionTest.test_webroot_map_partial_without_performc C s* ddl m} ||dƒ}| | jg¡ |jS )Nr r r )r r r8 r9 r )r# r r r! r$ r$ r% r— e s
z+WebrootActionTest._get_config_after_performN)r‚ rƒ r„ r… r rU r
rt rX r9 r& r” r™ rš rœ r r— r$ r$ r$ r% rˆ 2 s ÿ
rˆ Ú__main__r3 )(r… Z
__future__r rŒ rq r‘ r( Úsysr r‡ r ZjosepyZjoser] Zacmer rL r r Zcertbot.compatr r Zcertbot.displayr r4 Z
certbot.testsr
r† ZJWKRSAÚloadZ load_vectorrX ZTestCaser rˆ r‚ ÚexitÚmainÚargvÚ__file__r$ r$ r$ r% Ú s4 :