|
Server : Apache/2.4.41 (Ubuntu) System : Linux vmi1525618.contaboserver.net 5.4.0-105-generic #119-Ubuntu SMP Mon Mar 7 18:49:24 UTC 2022 x86_64 User : www-data ( 33) PHP Version : 8.2.12 Disable Function : NONE Directory : /lib/python3/dist-packages/certbot/tests/__pycache__/ |
Upload File : |
U
�]�; � @ sT d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z
ddl mZ ddl mZ ddlmZ dd lmZ ddlmZ G d
d � d ej�ZG d d
� d
ej�ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� dej�Z G dd� dej�Z G dd� dej�Z G dd� dej �Z e!d k�rPe�"� dS ) zTests for certbot.log.� N)�messages)�Optional)� constants)�errors)�util)�
filesystem)�osc @ sL e Zd ZdZedd� �Ze�d�e�d�e�d�e�d�dd � ����Zd
S ) �PreArgParseSetupTestz*Tests for certbot.log.pre_arg_parse_setup.c O s ddl m} |� S )Nr )�pre_arg_parse_setup)� certbot.logr
)�cls�args�kwargsr
� r �8/usr/lib/python3/dist-packages/certbot/tests/log_test.py�_call s zPreArgParseSetupTest._call�certbot.log.sysz%certbot.log.pre_arg_parse_except_hook� certbot.log.logging.getLoggerz certbot.log.util.atexit_registerc
C s� dg|_ tj|_| �� |� }|j�tj� | �|j j
d� d }|j j D ]D}|d d }|d krzt |tj
j�rz|}|j} qH| �t |tj�� qH| �t | tj�� |�tj� |�ddd� |j|ddddtjd� d S )Nz--debug� r � � T��debug�log_path)�argv�sys� version_infor �setLevel�assert_called_once_with�logging�DEBUG� assertEqual�
addHandlerZ
call_countZcall_args_list�
isinstance�handlers�
MemoryHandler�target�
assertTrue�
StreamHandlerZshutdown�
excepthook�mock�ANY)
�selfZ
mock_registerZmock_get�mock_except_hook�mock_sysZmock_root_logger�memory_handlerZcall�handlerr& r r r �test_it s4
� �z PreArgParseSetupTest.test_itN� �__name__�
__module__� __qualname__�__doc__� classmethodr r* �patchr1 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 )�PostArgParseSetupTestz+Tests for certbot.log.post_arg_parse_setup.c O s ddl m} |||�S )Nr )�post_arg_parse_setup)r r: )r r
r r: r r r r ? s zPostArgParseSetupTest._callc s� t t| ��� d| j_d| j_d| j_tjd | j_ t
t j d�| _ ddl
m} |t�� �| _ddl
m}m} |� | _| jj| _|| j�| _tj| j| jgd�| _d S ) NFi� �
verbose_count�wr ��ColoredStreamHandler)r% � TempHandler)r$ )�superr9 �setUp�configr �max_log_backups�quietr Z CLI_DEFAULTSr; �openr �devnullr r> �six�StringIO�stream_handlerr% r? � temp_handler�path� temp_pathr/ r* � MagicMock� root_logger)r, r> r% r? �� __class__r r rA D s
�zPostArgParseSetupTest.setUpc s: | j �� | j�� | j�� | j�� tt| ��� d S �N)r/ �closerI rJ rF r@ r9 �tearDown�r, rO r r rS U s
z PostArgParseSetupTest.tearDownc C s t �d��R}| j|_d}t �|��0}t �d��}tj|_| �| j� W 5 Q R X W 5 Q R X W 5 Q R X | jj� | j
� | � | jj j
� | � tj�tj�| jjd��� | �tj�| j�� |�ddd� |j ddd| jj| jjd� | jj}| jj�r| �|tj � n| �|| jj d � d S )
Nr �&certbot.log.post_arg_parse_except_hookr zletsencrypt.logr r r r �
) r* r8 rN Z return_valuer r r rB Z
removeHandlerr r/ r' r" �calledr rK �exists�join�logs_dir� assertFalserL r) r rI �levelrD r! r ZQUIET_LOGGING_LEVELr; )r, Zmock_get_loggerZexcept_hook_pathr- r. r\ r r r � test_common\ s8 *� � �
z!PostArgParseSetupTest.test_commonc C s d| j _| �� d S �NT)rB r r] rT r r r �
test_debugu s z PostArgParseSetupTest.test_debugc C s d| j _| �� d S r^ )rB rD r] rT r r r �
test_quiety s z PostArgParseSetupTest.test_quiet) r3 r4 r5 r6 r7 r rA rS r] r_ r` �
__classcell__r r rO r r9 < s
r9 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 )�SetupLogFileHandlerTestz-Tests for certbot.log.setup_log_file_handler.c O s ddl m} |||�S )Nr )�setup_log_file_handler)r rc )r r
r rc r r r r � s z SetupLogFileHandlerTest._callc s t t| ��� d| j_d S )N�* )r@ rb rA rB rC rT rO r r rA � s z SetupLogFileHandlerTest.setUpz1certbot.main.logging.handlers.RotatingFileHandlerc
C s` t |_z| �| jdd� W n6 tjk
rP } z| �dt|�k� W 5 d }~X Y n X | �d� d S )N�test.log� %(message)sz
--logs-dirzError not raised.) �IOError� side_effectr rB r �Errorr' �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_commonrT r r r �test_success_with_rollover� s z2SetupLogFileHandlerTest.test_success_with_rolloverc C s d| j _| jdd� d S )Nr Fro )rB rC rq rT r r r � test_success_without_rollover� s z5SetupLogFileHandlerTest.test_success_without_rolloverc C s� d}| � | j|d�\}}|�� | �|jtj� | �|jjt j
� t j �
| jj|�}| �||� t j �
| jj|d �}| �t j �|�|� d S )Nre rf z.1)r rB rR r! r\ r r Z formatterZ converter�time� localtimer rK rY rZ rX )r, rp �log_filer0 r Z
expected_pathZ backup_pathr r r rq � s z,SetupLogFileHandlerTest._test_success_commonz0certbot.log.logging.handlers.RotatingFileHandlerc C s2 | � | jdd� |jd d }| �| jj|� d S )Nre rf r Z backupCount)r rB � call_argsr! rC )r, rl Z backup_countr r r �test_max_log_backups_used� s z1SetupLogFileHandlerTest.test_max_log_backups_used)r3 r4 r5 r6 r7 r rA r* r8 rn rr rs rq rx ra r r rO r rb ~ s
rb c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �ColoredStreamHandlerTestz*Tests for certbot.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 r= )
rG rH �stream�isattyr � getLogger�loggerr r r r> r0 r" )r, r> r r r rA � s
z ColoredStreamHandlerTest.setUpc C s | j �� d S rQ �r0 rR rT r r r rS � s z!ColoredStreamHandlerTest.tearDownc C s, d}| j �|� | �| j�� d�|�� d S )Nz
I did a thingz{0}
)r 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 �� d S )NzI did another thingz
{0}{1}{2}
)
r r r0 Z red_levelr r 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)r3 r4 r5 r6 rA rS 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.log.MemoryHandlerc C sb t �t�| _| j�t j� d| _t�� | _ t �
| j �| _ ddl m
} || j �| _| j�| j� d S )Nzhi therer )r% )r r~ r3 r r r r� rG rH r| r( rI r r% r0 r" )r, r% r r r rA � s
zMemoryHandlerTest.setUpc C s | j �� | j�� d S rQ )r0 rR rI rT r r r rS � s
zMemoryHandlerTest.tearDownc C s2 | � � | jjdd� | �| j�� | jd � d S )NT�Zforce�
)�_test_log_debugr0 �flushr! r| r� r� rT r r r �
test_flush� s z MemoryHandlerTest.test_flushc C s. | j �| j� | j�� | �| j�� d� d S )N� )r Zcriticalr� r0 r� r! r| r� rT r r r �test_not_flushed� s
z"MemoryHandlerTest.test_not_flushedc C sh | � � t�� }t�|�}| j�|� | jjdd� | �| j �
� d� | �|�
� | j d � |� � d S )NTr� r� r� )
r� rG rH r r( r0 Z setTargetr� r! r| r� r� rR )r, Z
new_streamZnew_stream_handlerr r r �test_target_reset� s
z#MemoryHandlerTest.test_target_resetc C s | j �| j� d S rQ )r r r� rT r r r r� � s z!MemoryHandlerTest._test_log_debugN)
r3 r4 r5 r6 rA rS 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.log.TempHandler.c C s d| _ ddlm} |� | _d S )NFr )r? )�closedr r? r0 )r, r? r r r rA � s zTempHandlerTest.setUpc C s | j �� d S rQ r� rT r r r rS s zTempHandlerTest.tearDownc C s | � t�| jjd�� d S )Ni� )r' r Zcheck_permissionsr0 rK rT r r r �test_permissions s z TempHandlerTest.test_permissionsc C s$ | j �� | �tj�| j j�� d S rQ )r0 rR r[ r rK rX rT r r r � test_delete s
zTempHandlerTest.test_deletec C sB | j �t�� � | j �� | �tj�| j j�� t� | j j� d S rQ )
r0 �emitr* rM rR r' r rK rX �removerT r r r �test_no_delete
s
z TempHandlerTest.test_no_deleteN) r3 r4 r5 r6 rA rS r� r� r� r r r r r� � s r� c @ s. e Zd ZdZedd� �Ze�d�dd� �ZdS )�PreArgParseExceptHookTestz0Tests for certbot.log.pre_arg_parse_except_hook.c O s ddl m} |||�S )Nr )�pre_arg_parse_except_hook)r r� )r r
r r� r r r r s z PreArgParseExceptHookTest._callrU c C sD t �� }d}ddi}| j|f|�|� |j||� |jjdd� d S )N)�somer
r� r Tr� )r* rM r r r� )r, Z mock_post_arg_parse_except_hookr/ r
r r r r r1 s �z!PreArgParseExceptHookTest.test_itNr2 r r r r r� s
r� c @ sl e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�PostArgParseExceptHookTestz1Tests for certbot.log.post_arg_parse_except_hook.c O s ddl m} |||�S )Nr )�post_arg_parse_except_hook)r r� )r r
r r� r r r r * s z PostArgParseExceptHookTest._callc C s d| _ d| _d S )Nztest error messagezfoo.log)� error_msgr rT r r r rA / s z PostArgParseExceptHookTest.setUpc C s2 t }| j|dd�\}}| �|j|� | �|� d S �NF�r )�KeyboardInterrupt� _test_common�_assert_exception_logged�error�_assert_logfile_output�r, �exc_type� mock_logger�outputr r r �test_base_exception3 s z.PostArgParseExceptHookTest.test_base_exceptionc C s2 t }| j|dd�\}}| �|j|� | �|� d S )NTr� )�
ValueErrorr� r� r� r� r� r r r r_ 9 s z%PostArgParseExceptHookTest.test_debugc C s6 t j}| j|dd�\}}| �|j|� | �||� d S r� )r Z PluginErrorr� r� r �_assert_quiet_outputr� r r r �test_custom_error? s z,PostArgParseExceptHookTest.test_custom_errorc s^ t t�tj��� � fdd�}| j|dd�\}}| �|jtj� | � ||� | �
tj |k� d S )Nc s t jj� | d�S )z6Wraps ACME errors so the constructor takes only a msg.)Zdetail)r ri Z with_code)r� �Z acme_coder r �get_acme_errorI s zBPostArgParseExceptHookTest.test_acme_error.<locals>.get_acme_errorFr� ) �nextrG Ziterkeysr Z ERROR_CODESr� r� r ri r� r[ Z ERROR_PREFIX)r, r� r� r� r r� r �test_acme_errorE 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_errorR s z+PostArgParseExceptHookTest.test_other_errorc s� t �� � � fdd�}z|| j��W n� tk
r� t�� }t�d��z}||j_ t�d� ��Z z| j
||| j d�� W n0 t k
r� } z� �
t|�� W 5 d}~X Y n X | �d� W 5 Q R X W 5 Q R X Y nX � �� }||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_err\ s z:PostArgParseExceptHookTest._test_common.<locals>.write_errzcertbot.log.loggerzcertbot.log.sys.stderrr NzSystemExit not raised.)rG rH r� �
BaseExceptionr �exc_infor* r8 r� rh r r �
SystemExitr� rj rk r� )r, Z
error_typer r� r� r� Zexit_errr� r r� r r� X s( � $z'PostArgParseExceptHookTest._test_commonc C sJ | � |j� |jd }| � d|k� |d }|tjtjf}| �||� d S )Nr r� )r' rW rw r* r+ r! )r, Zlog_funcr� Z call_kwargsZactual_exc_infoZexpected_exc_infor r r r� r s
z3PostArgParseExceptHookTest._assert_exception_loggedc C s" | � d|k� | � | j|k� d S )NzPlease see the logfile)r' r )r, r� r r r r� { s z1PostArgParseExceptHookTest._assert_logfile_outputc C s0 | � |jj� | �|jj� | �| j|k� d S rQ )r[ Z exceptionrW r' r r� )r, r� r� r r r r� s z/PostArgParseExceptHookTest._assert_quiet_outputN)r3 r4 r5 r6 r7 r rA r� r_ r� r� r� r� r� r� r� r r r r r� ( s
r� c @ s4 e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd
S ) �ExitWithLogPathTestz)Tests for certbot.log.exit_with_log_path.c O s ddl m} |||�S )Nr )�exit_with_log_path)r r� )r r
r r� r r r r � s zExitWithLogPathTest._callc C sH t j�| jd�}t|d��� | �|�}| �d|k� | �||k� d S )Nre r< �logfiles)r rK rY �tempdirrE rR r� r' )r, rv �err_strr r r �
test_log_file� s
z!ExitWithLogPathTest.test_log_filec C s. | � | j�}| �d|k� | �| j|k� d S )Nr� )r� r� r' )r, r� r r r � test_log_dir� s z ExitWithLogPathTest.test_log_dirc
O sN z| j ||� W n. tk
r> } zt|� W Y �S d }~X Y nX | �d� d S )NzSystemExit was not raised.)r r� rj rk )r, r
r rm r r r r� � s
z ExitWithLogPathTest._test_commonN) r3 r4 r5 r6 r7 r r� r� r� r r r r r� � s
r� �__main__)#r6 r Zlogging.handlersr rt Zunittestr* rG Zacmer Zacme.magic_typingr Zcertbotr r r Zcertbot.compatr r Z
certbot.testsZ test_utilZTestCaser ZConfigTestCaser9 rb ry r� r� r� r� ZTempDirTestCaser� r3 �mainr r r r �<module> s4 &B3 +]