JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__/log_test.cpython-39.opt-1.pyc
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 k�rte�$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� d }|jj D ]2}|d d }|d u r8t
|tj j �r8|}|j
} q8q8|�tj� |�ddd� |j|dddddtjd� d S ) Nz--debugr � � � TF��debug�quiet�log_path)�argv�sys� version_infor �setLevel�assert_called_once_withr �DEBUG�
addHandlerZcall_args_list�
isinstance�handlers�
MemoryHandler�targetr �
excepthookr �ANY)
r Z
mock_registerZmock_get�mock_except_hook�mock_sysZmock_root_logger�memory_handler�call�handlerr/ 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� fdd�Z� fdd�Zdd � 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_count�wr ��ColoredStreamHandler�r. �� TempHandler)r- )r �setUp�configr" �max_log_backupsr# r Z CLI_DEFAULTSrA �openr �devnullr rD �io�StringIO�stream_handlerr. rG � temp_handler�pathZ 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 s�0 Y tj �
| jj d�}| jj �
| j� |�ddd� |j
ddd| jj| jj|d� | jj}| jjr�n 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 �join�logs_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� 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 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 )N�test.log� %(message)szError not raised.)�IOError� side_effectr rI r �Error�fail)r � mock_handler�errr r r � test_failure� s z$SetupLogFileHandlerTest.test_failurec C s | j dd� d S )NT��should_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�� | _| j�tj� ddl m
} || j�| _ | j� | j � d S )Nc S s dS rZ r r r r r �<lambda>� � z0ColoredStreamHandlerTest.setUp.<locals>.<lambda>r rC )
rM rN �stream�isattyr � getLogger�loggerr( 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�| _| j�t 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 �NT��force)�_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� �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._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 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 rH 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 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� | d�S )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.<locals>.get_acme_errorFr� ) �next�iterr 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�\}}|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.<locals>.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& �time�typingr Zunittestr ZpytestZacmer Zcertbotr r Zcertbot._internalr Zcertbot.compatr r Z
certbot.testsZ test_utilZTestCaser
ZConfigTestCaser? r] rq r} r� r� r� ZTempDirTestCaser� r8 �exit�mainr% �__file__r r r r �<module> s6 +E3 +h
Back to Directory
nL+D550H?Mx ,D"v]qv;6*Zqn)ZP 0 !1 A "#a$2Qr D8 a Ri[f\mIykIw0 cuFcRı?lO7к_f˓[C$殷WF<_W ԣs KcëIzyQy /_LKℂ;C",pFA: /]=H ~,ls/9ć:[=/#f;)x {ٛEQ )~ =𘙲r*2~ a _V=' kumF D}KYYC)({ *g&f`툪ry`=^cJ.I](*`wq 1dđ#̩͑0;H]u搂@:~וKL Nsh}OIR*8:2 !lDJVo( 3=M(zȰ+i*NA r6KnSl )!JJӁ* %݉?|D}d5:eP0R;{$X'xF@.ÊB {,WJuQɲRI;9QE琯62fT.DUJ;*cP A\ILNj!J۱+O\ ͔]ޒ SJȧc%ANolՎprULZԛerE2=XDXgVQeӓk yP7U*omQIs,K`)6\G3t?pgjrmۛجwluGtf h9uyP0D;Uڽ"OXlif$)&|ML0Zrm1[HXPlPR0'G=i2N +0e2]] 9VTPO7h(F*癈 '=QVZDF,d߬~TX
G[`le69CR(!S2!P
<0x < !1AQ "Raq02Br#SCTb
?Ζ"]mH5WR7k.ۛ!}Q~+yԏz|@T20S~Kek *zFf^2X*(@8r?CIuI|֓>^ExLgNUY+{.RѪ
τVYTD I62'8Y2 7'\TP.6d&˦@Vqi|8- OΕ]ʔ U=TL8=;6c| !qfF3aů&~$l}'NWUs$Uk^SV:U# 6w++s&r+nڐ{@29 gL
u"TÙM=6 (^"7r} =6YݾlCuhquympǦ
G jhsǜNlɻ}o7 #S6aw4!OSrD57%|?x>L
|/nD6?/8w#[)L7+6〼T ATg ! %5MmZ/c-{1_Je"|^$'O&ޱմ Trb$ w)R$&
N1EtdU3Uȉ1pM" N*(DN yd96.(jQ)X
5cQɎMyW? Q*!R>6=7)Xj5`J]e 8%t!+'!1Q 5 !1 AQaqё#2"0BRb ? Gt^## .llQT $v,,m 㵜5 ubV =sY+@d{N! dnO< .-B; _wJt6;QJd.Qc%p{ 1,sNDdFHI0ГoXшe黅XۢF:)[FGXƹ/w_cMeD,ʡcc. WDtA$j@:) -# u
c1<@ۗ9F)KJ-hpP]_x[qBlbp ʖw q"LFGd ƶ*s+ډ_Zc"? %t[IP 6J]#=ɺVvvCGsGh1 >)6|ey?Lӣm,4GWUi` ]uJVoV DG< SB 6ϏQ@ TiUly OU0kfV~~}SZ@*WUUi##; s/[ =!7}" WN]'(L! ~y5g9T̅JkbM' +s:S + B)v@Mj
e Cf jE 0Y\QnzG1д~Wo{T 9?`Rmyhsy3!HAD]mc1~2L Su7xT;j$`}4->L#vzŏ ILS ֭ T { rjGKC; bpU= -`BsK.SFw4Mq]ZdHS0)tLg