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 ejZG d
d dejZG dd dejZG dd dejZ G dd dejZ G dd dejZ G dd dejZ G dd dejZ 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 fddZedededed 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 )clsargskwargsr r D/usr/lib/python3.9/site-packages/certbot/_internal/tests/log_test.py_call s zPreArgParseSetupTest._callc s t t d S N)loggingshutdownsupertearDownself __class__r r r s z PreArgParseSetupTest.tearDowncertbot._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|_| | }|jtj d }|jj D ]2}|d d }|d u r8t
|tj j r8|}|j
} q8q8|tj |ddd |j|dddddtjd d S ) Nz--debugr TFdebugquietlog_path)argvsys version_infor setLevelassert_called_once_withr DEBUG
addHandlerZcall_args_list
isinstancehandlers
MemoryHandlertargetr
excepthookr ANY)
r Z
mock_registerZmock_getmock_except_hookmock_sysZmock_root_loggermemory_handlercallhandlerr/ r r r test_it$ s" z PreArgParseSetupTest.test_it) __name__
__module__ __qualname____doc__ classmethodr r r patchr7
__classcell__r r r r r
s
r
c sP e Zd ZdZedd Z fddZ fddZdd Zd
d Z d d
Z
Z S )PostArgParseSetupTestz5Tests for certbot._internal.log.post_arg_parse_setup.c O s ddl m} ||i |S )Nr )post_arg_parse_setup)r r@ )r
r r r@ r r r r D s zPostArgParseSetupTest._callc s t d| j_d| j_d| j_tjd | j_t t
j d| _ ddl m
} |t | _ddl m} ddl m} | | _| jj| _|| j| _tj| j| jgd | _d S )
NFi
verbose_countwr ColoredStreamHandlerr. TempHandler)r- )r setUpconfigr" max_log_backupsr# r Z CLI_DEFAULTSrA openr devnullr rD ioStringIOstream_handlerr. rG temp_handlerpathZ temp_pathr4 r MagicMock root_logger)r rD r. rG r r r rH I s
zPostArgParseSetupTest.setUpc s6 | j | j | j | j t d S r )r4 closerO rP rL r r r r r r r [ s
z PostArgParseSetupTest.tearDownc
C s t d}| j|_d}t |N}t d$}tj|_| | j W d n1 sX0 Y W d n1 sv0 Y W d n1 s0 Y tj
| jj d}| jj
| j |ddd |j
ddd| jj| jj|d | jj}| jjrn d S ) Nr 0certbot._internal.log.post_arg_parse_except_hookr zletsencrypt.logr r r r! )r r= rS Z return_valuer& r' r rI r rQ joinlogs_dirZ
removeHandlerr) r4 r0 r" r# rO level)r Zmock_get_loggerZexcept_hook_pathr2 r3 r$ rX r r r test_commonb s( f z!PostArgParseSetupTest.test_commonc C s d| j _| d S NT)rI r" rY r r r r
test_debug} s z PostArgParseSetupTest.test_debugc C s d| j _| d S rZ )rI r# rY r r r r
test_quiet s z PostArgParseSetupTest.test_quiet) r8 r9 r: r; r< r rH r rY r[ r\ r> r r r r r? A s
r? c sh e Zd ZdZedd Z fddZeddd Z d d
Z
d d Z d
d Z eddd Z
ZS )SetupLogFileHandlerTestz7Tests for certbot._internal.log.setup_log_file_handler.c O s ddl m} ||i |S )Nr )setup_log_file_handler)r r^ )r
r r r^ r r r r s z SetupLogFileHandlerTest._callc s t d| j_d S )N* )r rH rI rJ r r r r rH s
z SetupLogFileHandlerTest.setUpz;certbot._internal.main.logging.handlers.RotatingFileHandlerc
C sP t |_z| | jdd W n& tjy@ } z W Y d }~nd }~0 0 | d d S )Ntest.log %(message)szError not raised.)IOError side_effectr rI r Errorfail)r mock_handlererrr r r test_failure s z$SetupLogFileHandlerTest.test_failurec C s | j dd d S )NTshould_rollover)_test_success_commonr r r r test_success_with_rollover s z2SetupLogFileHandlerTest.test_success_with_rolloverc C s d| j _| jdd d S )Nr Fri )rI rJ rk r r r r test_success_without_rollover s z5SetupLogFileHandlerTest.test_success_without_rolloverc C sL d}| | j|d\}}| tj| jj|}tj| jj|d }d S )Nr` ra z.1)r rI rT r rQ rV rW )r rj log_filer6 r$ Z
expected_pathZ backup_pathr r r rk s z,SetupLogFileHandlerTest._test_success_commonz:certbot._internal.log.logging.handlers.RotatingFileHandlerc C s" | | jdd |jd d }d S )Nr` ra r Z backupCount)r rI call_args)r rf Z backup_countr r r test_max_log_backups_used s z1SetupLogFileHandlerTest.test_max_log_backups_used)r8 r9 r: r; r< r rH r r= rh rl rm rk rp r> r r r r r] s
r] c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) ColoredStreamHandlerTestz4Tests for certbot._internal.log.ColoredStreamHandlerc C sX t | _dd | j_t | _| jtj ddl m
} || j| _ | j | j d S )Nc S s dS rZ r r r r r z0ColoredStreamHandlerTest.setUp..r rC )
rM rN streamisattyr getLoggerloggerr( r* r rD r6 r+ )r rD r r r rH s
z ColoredStreamHandlerTest.setUpc C s | j d S r r6 rT r r r r r s z!ColoredStreamHandlerTest.tearDownc C s d}| j | d S )Nz
I did a thing)rw r" r msgr r r test_format s z$ColoredStreamHandlerTest.test_formatc C s d}t j| j_| j| d S )NzI did another thing)r r* r6 Z red_levelrw r" ry r r r test_format_and_red_level s
z2ColoredStreamHandlerTest.test_format_and_red_levelN)r8 r9 r: r; rH r r{ r| r r r r rq s
rq 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| _| jt j d| _t | _ t
| j | _ ddl m
} || j | _| j| j d S )Nzhi therer rE )r rv r8 rw r( r* rz rM rN rt
StreamHandlerrO r r. r6 r+ )r r. r r r rH s
zMemoryHandlerTest.setUpc C s | j | j d S r )r6 rT rO r r r r r s
zMemoryHandlerTest.tearDownc C s | | jjdd d S NTforce)_test_log_debugr6 flushr r r r
test_flush s z MemoryHandlerTest.test_flushc C s | j | j | j d S r )rw Zcriticalrz r6 r r r r r test_not_flushed s
z"MemoryHandlerTest.test_not_flushedc C s@ | t }t|}| j| | jjdd | d S r ) r rM rN r r~ r6 Z setTargetr rT )r Z
new_streamZnew_stream_handlerr r r test_target_reset s
z#MemoryHandlerTest.test_target_resetc C s | j | j d S r )rw r" rz r r r r r s z!MemoryHandlerTest._test_log_debugN)
r8 r9 r: r; rH 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 rF )closedr rG r6 )r rG r r r rH s zTempHandlerTest.setUpc C s | j d S r rx r r r r r s zTempHandlerTest.tearDownc C s d S r r r r r r test_permissions s z TempHandlerTest.test_permissionsc C s | j d S r rx r r r r test_delete s
zTempHandlerTest.test_deletec C s, | j t | j t| j j d S r )r6 emitr rR rT r removerQ r r r r test_no_delete s
z TempHandlerTest.test_no_deleteN) r8 r9 r: r; rH r r r r r r r r r s r c @ s. e Zd ZdZedd Zeddd 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._callrU c C sN t }d}ddi}| j|g|R i | |j|i | |jjdd d S )N)somer r r Tr )r rR r r) r )r Z mock_post_arg_parse_except_hookr4 r r r r r r7 # s z!PreArgParseExceptHookTest.test_itN) r8 r9 r: r; r< r r r= r7 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 ddZdd 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 rH 7 s z PostArgParseExceptHookTest.setUpc C s2 t }| j|dd\}}| |j| | | d S NFr" )
BaseException _test_common_assert_exception_loggederror_assert_logfile_outputr exc_type mock_loggeroutputr 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 s* t }| j|ddd\}}| |j| d S )NT)r" r# )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 sL t ttj fdd}| j|dd\}}| |jtj | || d S )Nc s t jj | dS )z6Wraps ACME errors so the constructor takes only a msg.)Zdetail)r rd Z with_code)rz Z acme_coder r get_acme_errorW s zBPostArgParseExceptHookTest.test_acme_error..get_acme_errorFr ) nextiterr Z ERROR_CODESr r r" rd r )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\}}|jd 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|| jW n ty t }td}||j_ td h z| j
|||| j d W n2 t y } z
t| W Y d}~nd}~0 0 | d W d n1 s0 Y W d n1 s0 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_kwargsZmock_errr r write_erro s z:PostArgParseExceptHookTest._test_common..write_errz certbot._internal.log.loggerz certbot._internal.log.sys.stderrr! NzSystemExit not raised.)rM rN r r r& exc_infor r= r rc r r$
SystemExitr strre getvalue) r Z
error_typer" r# r r r Zexit_errr r r r r k s$
$Lz'PostArgParseExceptHookTest._test_commonc C s$ |j d }|d }|tjtjf}d S )Nr r )ro r r1 )r Zlog_funcr Z call_kwargsZactual_exc_infoZexpected_exc_infor r r r s
z3PostArgParseExceptHookTest._assert_exception_loggedc C s d S r r )r r r r r r s z1PostArgParseExceptHookTest._assert_logfile_outputc C s d S r r )r r r r r r r s z/PostArgParseExceptHookTest._assert_quiet_outputN)F)r8 r9 r: r; r< r rH 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 s, t j| jd}t|d | |}d S )Nr` rB )r rQ rV tempdirrK rT r )r rn err_strr r r
test_log_file s
z ExitWithAdviceTest.test_log_filec C s | | j}d S r )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 r re )r r r rg r r r r s
z ExitWithAdviceTest._test_commonN) r8 r9 r: r; r< r r r r r r r r r s
r __main__r )(r; rM r Zlogging.handlersr& timetypingr Zunittestr ZpytestZacmer Zcertbotr r Zcertbot._internalr Zcertbot.compatr r Z
certbot.testsZ test_utilZTestCaser
ZConfigTestCaser? r] rq r} r r r ZTempDirTestCaser r8 exitmainr% __file__r r r r s6 +E3 +h