JFIF $
d d 7
Viewing File: /usr/lib/python3.9/site-packages/certbot/_internal/tests/__pycache__/log_test.cpython-39.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� |jj dks<J �d }|jj
D ]B}|d d }|d u rzt |tj j
�rz|}|j} qHt |tj�sHJ �qHt | tj�s�J �|�tj� |�ddd� |j|dddddtjd� d S ) Nz--debug� r � � TF��debug�quiet�log_path)�argv�sys� version_infor �setLevel�assert_called_once_withr �DEBUG�
addHandlerZ
call_countZcall_args_list�
isinstance�handlers�
MemoryHandler�target�
StreamHandlerr �
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 �patchr8 �
__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 rA )r
r r rA 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_DEFAULTSrB �openr �devnullr rE �io�StringIO�stream_handlerr. rH � temp_handler�path� temp_pathr5 r � MagicMock� root_logger)r rE r. rH r r r rI I s
�zPostArgParseSetupTest.setUpc s6 | j �� | j�� | j�� | j�� t� �� d S r )r5 �closerP rQ rM r r r r r r r [ s
z PostArgParseSetupTest.tearDownc
C sT 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� | jjjs�J �tj �|�s�J �tj �| j�r�J �|�ddd� |j
ddd| jj| jj|d� | jj}| jj�r@|tjk�sPJ �n|tjk�sPJ �d S ) Nr �0certbot._internal.log.post_arg_parse_except_hookr zletsencrypt.logr r r r! )r r> 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<J �W Y d }~nd }~0 0 | �d� d S )N�test.log� %(message)sz
--logs-dirzError not raised.)�IOError� side_effectr rJ r �Error�str�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 Frn )rJ rK rp r r r r � test_success_without_rollover� s z5SetupLogFileHandlerTest.test_success_without_rolloverc C s� d}| � | j|d�\}}|�� |jtjks0J �|jjtj ksBJ �t
j � | jj
|�}||ks`J �t
j � | jj
|d �}t
j �|�|ks�J �d S )Nrd re z.1)r rJ rV r\ r r* � formatterZ converter�time� localtimer rR rX rY r[ )r ro �log_filer7 r$ Z
expected_pathZ backup_pathr r r rp � s z,SetupLogFileHandlerTest._test_success_commonz:certbot._internal.log.logging.handlers.RotatingFileHandlerc C s2 | � | jdd� |jd d }| jj|ks.J �d S )Nrd re r Z backupCount)r rJ � call_argsrK )r rk Z backup_countr r r �test_max_log_backups_used� s z1SetupLogFileHandlerTest.test_max_log_backups_used)r9 r: r; r< r= r rI r r> 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 �<lambda>� � z0ColoredStreamHandlerTest.setUp.<locals>.<lambda>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� | d�S )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.<locals>.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.<locals>.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 �<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