a
}|ägM> ã @ sx d Z ddlZddlZddlZddlZddlZddlmZ ddlZddlm Z ddl
Z
ddl m Z ddl
mZ ddl
mZ ddlmZ dd lmZ dd
lmZ ddlmZ G d d „ d ejƒZG d
d„ dejƒZG dd„ dejƒZG dd„ dejƒZ G dd„ dejƒZ G dd„ dejƒZ G dd„ dejƒZ G dd„ dejƒZ G dd „ d ej!ƒZ"e#d krte $e
%ej&d d… e'g ¡¡ dS ) z Tests for certbot._internal.log.é N)ÚOptional)Úmock)Úmessages)Úerrors)Úutil)Ú constants)Ú
filesystem)Úosc s\ e Zd ZdZedd„ ƒZ‡ fdd„Ze d¡e d¡e d¡e d ¡d
d „ ƒƒƒƒZ ‡ Z
S ) ÚPreArgParseSetupTestz4Tests for certbot._internal.log.pre_arg_parse_setup.c O s ddl m} |ƒ S )Nr )Úpre_arg_parse_setup)Úcertbot._internal.logr )ÚclsÚargsÚkwargsr © r úD/usr/lib/python3.9/site-packages/certbot/_internal/tests/log_test.pyÚ_call s zPreArgParseSetupTest._callc s t ¡ tƒ ¡ d S ©N)ÚloggingÚshutdownÚsuperÚtearDown©Úself©Ú __class__r r r s z PreArgParseSetupTest.tearDownúcertbot._internal.log.sysz/certbot._internal.log.pre_arg_parse_except_hookú'certbot._internal.log.logging.getLoggerz*certbot._internal.log.util.atexit_registerc
C sÔ dg|_ tj|_| ¡ |ƒ }|j tj¡ |jj dks rU Z return_valuer& r' r rJ r rR ÚjoinÚlogs_dirZ
removeHandlerr) r5 r+ ÚcalledÚexistsrS r1 r" r# rP Úlevelr ZQUIET_LOGGING_LEVELZDEFAULT_LOGGING_LEVEL)r Zmock_get_loggerZexcept_hook_pathr3 r4 r$ r\ r r r Ú test_commonb s. fÿ þ
z!PostArgParseSetupTest.test_commonc C s d| j _| ¡ d S ©NT)rJ r" r] r r r r Ú
test_debug} s z PostArgParseSetupTest.test_debugc C s d| j _| ¡ d S r^ )rJ r# r] r r r r Ú
test_quiet s z PostArgParseSetupTest.test_quiet) r9 r: r; r< r= r rI r r] r_ r` r? r r r r r@ A s
r@ c sh e Zd ZdZedd„ ƒZ‡ fdd„Ze d¡dd„ ƒZ d d
„ Z
d d „ Z d
d„ Z e d¡dd„ ƒZ
‡ ZS )ÚSetupLogFileHandlerTestz7Tests for certbot._internal.log.setup_log_file_handler.c O s ddl m} ||i |¤ŽS )Nr )Úsetup_log_file_handler)r rb )r
r r rb r r r r ‰ s z SetupLogFileHandlerTest._callc s t ƒ ¡ d| j_d S )Né* )r rI rJ rK r r r r rI Ž s
z SetupLogFileHandlerTest.setUpz;certbot._internal.main.logging.handlers.RotatingFileHandlerc
C s` t |_z| | jdd¡ W n6 tjyP } z dt|ƒv s rm rq rr rp rx r? r r r r ra † s
ra c @ s0 e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
S ) ÚColoredStreamHandlerTestz4Tests for certbot._internal.log.ColoredStreamHandlerc C sX t ¡ | _dd„ | j_t ¡ | _| j tj¡ ddl m
} || jƒ| _ | j | j ¡ d S )Nc S s dS r^ r r r r r Ú¾ ó z0ColoredStreamHandlerTest.setUp..r rD )
rN rO ÚstreamÚisattyr Ú getLoggerÚloggerr( r* r rE r7 r+ )r rE r r r rI ¼ s
z ColoredStreamHandlerTest.setUpc C s | j ¡ d S r ©r7 rV r r r r r Æ s z!ColoredStreamHandlerTest.tearDownc C s, d}| j |¡ | j ¡ d |¡ks(J ‚d S )Nz
I did a thingz{0}
)r r" r| ÚgetvalueÚformat©r Úmsgr r r Ú test_formatÉ s z$ColoredStreamHandlerTest.test_formatc C s> d}t j| j_| j |¡ | j ¡ d t j
|t j ¡ks:J ‚d S )NzI did another thingz
{0}{1}{2}
) r r* r7 Z red_levelr r" r| r r‚ r Z ANSI_SGR_REDZANSI_SGR_RESETrƒ r r r Útest_format_and_red_levelÎ s
þÿz2ColoredStreamHandlerTest.test_format_and_red_levelN)r9 r: r; r< rI r r… r† r r r r ry ¹ s
ry c @ s@ e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dS )ÚMemoryHandlerTestz-Tests for certbot._internal.log.MemoryHandlerc C sb t t¡| _| j t j¡ d| _t ¡ | _ t
| j ¡| _ ddl m
} || j ƒ| _| j | j¡ d S )Nzhi therer rF )r r~ r9 r r( r* r„ rN rO r| r0 rP r r. r7 r+ )r r. r r r rI Û s
zMemoryHandlerTest.setUpc C s | j ¡ | j ¡ d S r )r7 rV rP r r r r r æ s
zMemoryHandlerTest.tearDownc C s2 | ¡ | jjdd | j ¡ | jd ks.J ‚d S )NT©ÚforceÚ
)Ú_test_log_debugr7 Úflushr| r r„ r r r r Ú
test_flushê s z MemoryHandlerTest.test_flushc C s. | j | j¡ | j ¡ | j ¡ dks*J ‚d S )NÚ )r Zcriticalr„ r7 rŒ r| r r r r r Útest_not_flushedï s
z"MemoryHandlerTest.test_not_flushedc C sh | ¡ t ¡ }t |¡}| j |¡ | jjdd | j ¡ dksFJ ‚| ¡ | j
d ks\J ‚| ¡ d S )NTrˆ rŽ rŠ ) r‹ rN rO r r0 r7 Z setTargetrŒ r| r r„ rV )r Z
new_streamZnew_stream_handlerr r r Útest_target_resetõ s
z#MemoryHandlerTest.test_target_resetc C s | j | j¡ d S r )r r" r„ r r r r r‹ s z!MemoryHandlerTest._test_log_debugN)
r9 r: r; r< rI r r r r r‹ r r r r r‡ Ù s r‡ c @ s8 e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd S )
ÚTempHandlerTestz,Tests for certbot._internal.log.TempHandler.c C s d| _ ddlm} |ƒ | _d S )NFr rG )Úclosedr rH r7 )r rH r r r rI s zTempHandlerTest.setUpc C s | j ¡ d S r r€ r r r r r s zTempHandlerTest.tearDownc C s t | jjd¡sJ ‚d S )Ni€ )r Zcheck_permissionsr7 rR r r r r Útest_permissions s z TempHandlerTest.test_permissionsc C s" | j ¡ tj | j j¡r J ‚d S r )r7 rV r rR r[ r r r r Ú test_delete s
zTempHandlerTest.test_deletec C s@ | j t ¡ ¡ | j ¡ tj | j j¡s.J ‚t | j j¡ d S r ) r7 Úemitr rT rV r rR r[ Úremover r r r Útest_no_delete s
z TempHandlerTest.test_no_deleteN) r9 r: r; r< rI r r“ r” r— r r r r r‘ s r‘ c @ s. e Zd ZdZedd„ ƒZe d¡dd„ ƒZdS )ÚPreArgParseExceptHookTestz:Tests for certbot._internal.log.pre_arg_parse_except_hook.c O s ddl m} ||i |¤ŽS )Nr )Úpre_arg_parse_except_hook)r r™ )r
r r r™ r r r r s z PreArgParseExceptHookTest._callrW c C sN t ¡ }d}ddi}| j|g|¢R i |¤Ž |j|i |¤Ž |jjdd d S )N)Úsomer rš r Trˆ )r rT r r) rŒ )r Z mock_post_arg_parse_except_hookr5 r r r r r r8 # s ÿÿz!PreArgParseExceptHookTest.test_itN) r9 r: r; r< r= r r r> r8 r r r r r˜ s
r˜ c @ s~ e Zd ZdZedd„ ƒZdd„ Zdd„ Zdd „ Zd
d „ Z d d
„ Z
dd„ Z dd„ Z dd„ Z
d dd„Zdd„ Zdd„ Zdd „ Zd S ) ÚPostArgParseExceptHookTestz;Tests for certbot._internal.log.post_arg_parse_except_hook.c O s ddl m} ||i |¤ŽS )Nr )Úpost_arg_parse_except_hook)r rœ )r
r r rœ r r r r 2 s z PostArgParseExceptHookTest._callc C s d| _ d| _d S )Nztest error messagezfoo.log)Ú error_msgr$ r r r r rI 7 s z PostArgParseExceptHookTest.setUpc C s2 t }| j|dd\}}| |j|¡ | |¡ d S ©NF©r" )Ú
BaseExceptionÚ _test_commonÚ_assert_exception_loggedÚerrorÚ_assert_logfile_output©r Úexc_typeÚ mock_loggerÚoutputr r r Útest_base_exception; s z.PostArgParseExceptHookTest.test_base_exceptionc C s2 t }| j|dd\}}| |j|¡ | |¡ d S )NTrŸ )Ú
ValueErrorr¡ r¢ r£ r¤ r¥ r r r r_ A s z%PostArgParseExceptHookTest.test_debugc C s6 t }| j|ddd\}}| |j|¡ d|vs2J ‚d S )NT)r" r# úSee the logfile)rª r¡ r¢ r£ r¥ r r r r` G s z%PostArgParseExceptHookTest.test_quietc C s6 t j}| j|dd\}}| |j|¡ | ||¡ d S rž )r Z PluginErrorr¡ r¢ r" Ú_assert_quiet_outputr¥ r r r Útest_custom_errorM s z,PostArgParseExceptHookTest.test_custom_errorc sZ t ttjƒƒ‰ ‡ fdd„}| j|dd\}}| |jtj¡ | ||¡ tj |vsVJ ‚d S )Nc s t jjˆ | dS )z6Wraps ACME errors so the constructor takes only a msg.)Zdetail)r rh Z with_code)r„ ©Z acme_coder r Úget_acme_errorW s zBPostArgParseExceptHookTest.test_acme_error..get_acme_errorFrŸ )
ÚnextÚiterr Z ERROR_CODESr¡ r¢ r" rh r¬ Z ERROR_PREFIX)r r¯ r§ r¨ r r® r Útest_acme_errorS s z*PostArgParseExceptHookTest.test_acme_errorc C s4 t }| j|dd\}}| |j|¡ | ||¡ d S rž )rª r¡ r¢ r" r¬ r¥ r r r Útest_other_error` s z+PostArgParseExceptHookTest.test_other_errorc C s& t }| j|dd\}}|j d¡ d S )NFrŸ z Exiting due to user request.)ÚKeyboardInterruptr¡ r£ r) r¥ r r r Útest_keyboardinterruptf s z1PostArgParseExceptHookTest.test_keyboardinterruptFc s t ¡ ‰ ‡ fdd„}z|| jƒ‚W nÐ tyò t ¡ }t d¡œ}||j_ t dˆ ¡h z| j
|||| j dœŽ W n2 t y¦ } zˆ
t|ƒ¡ W Y d}~nd}~0 0 | d¡ W d ƒ n1 sÆ0 Y W d ƒ n1 sä0 Y Y n0 ˆ ¡ }||fS )z,Returns the mocked logger and stderr output.c s ˆ | d ¡ dS )zWrite error to mock_err.r N)Úwrite)r Z
unused_kwargs©Zmock_errr r Ú write_erro s z:PostArgParseExceptHookTest._test_common..write_errz certbot._internal.log.loggerz certbot._internal.log.sys.stderrr! NzSystemExit not raised.)rN rO r r r& Úexc_infor r> r£ rg r r$ Ú
SystemExitr¶ ri rj r ) r Z
error_typer" r# r¸ r¹ r§ Zexit_errr¨ r r· r r¡ k s$
ÿ $Lz'PostArgParseExceptHookTest._test_commonc C sF |j s
J ‚|jd }d|v s J ‚|d }|tjtjf}||ksBJ ‚d S )Nr r¹ )rZ rw r r2 )r Zlog_funcr¦ Z call_kwargsZactual_exc_infoZexpected_exc_infor r r r¢ … s
z3PostArgParseExceptHookTest._assert_exception_loggedc C s d|v s J ‚| j |v sJ ‚d S )Nr« )r$ )r r¨ r r r r¤ Ž s z1PostArgParseExceptHookTest._assert_logfile_outputc C s. |j jdu sJ ‚|jjs J ‚| j|v s*J ‚d S )NF)Ú exceptionrZ r" r )r r§ r¨ r r r r¬ ’ s z/PostArgParseExceptHookTest._assert_quiet_outputN)F)r9 r: r; r< r= r rI r© r_ r` r r² r³ rµ r¡ r¢ r¤ r¬ r r r r r› 0 s
r› c @ s4 e Zd ZdZedd„ ƒZdd„ Zdd„ Zdd „ Zd
S ) ÚExitWithAdviceTestz1Tests for certbot._internal.log.exit_with_advice.c O s ddl m} ||i |¤ŽS )Nr )Úexit_with_advice)r r½ )r
r r r½ r r r r š s zExitWithAdviceTest._callc C sD t j | jd¡}t|dƒ ¡ | |¡}d|vs4J ‚||v s@J ‚d S )Nrd rC Úlogfiles)r rR rX ÚtempdirrL rV r¡ )r rv Úerr_strr r r Ú
test_log_fileŸ s
z ExitWithAdviceTest.test_log_filec C s* | | j¡}d|v sJ ‚| j|v s&J ‚d S )Nr¾ )r¡ r¿ )r rÀ r r r Ú test_log_dir§ s z ExitWithAdviceTest.test_log_dirc
O sP z| j |i |¤Ž W n, ty@ } zt|ƒW Y d }~S d }~0 0 | d¡ d S )NzSystemExit was not raised.)r rº ri rj )r r r rl r r r r¡ s
z ExitWithAdviceTest._test_commonN) r9 r: r; r< r= r rÁ r r¡ r r r r r¼ ˜ s
r¼ Ú__main__r )(r< rN r Zlogging.handlersr& rt Útypingr Zunittestr ZpytestZacmer Zcertbotr r Zcertbot._internalr Zcertbot.compatr r Z
certbot.testsZ test_utilZTestCaser
ZConfigTestCaser@ ra ry r‡ r‘ r˜ r› ZTempDirTestCaser¼ r9 ÚexitÚmainr% Ú__file__r r r r Ú s6 +E3 +h