Your IP : 216.73.216.172
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_handler�call�handlerr& r r r �test_it s4
� �z PreArgParseSetupTest.test_itN� �__name__�
__module__� __qualname__�__doc__� classmethodr r* �patchr2 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$ )�superr: �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 rB D s
�zPostArgParseSetupTest.setUpc s: | j �� | j�� | j�� | j�� tt| ��� d S �N)r/ �closerJ rK rG rA r: �tearDown�r, rP r r rT 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* r9 rO Z return_valuer r r rC Z
removeHandlerr r/ r' r" �calledr rL �exists�join�logs_dir� assertFalserM r) r rJ �levelrE 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)rC r r^ rU r r r �
test_debugu s z PostArgParseSetupTest.test_debugc C s d| j _| �� d S r_ )rC rE r^ rU r r r �
test_quiety s z PostArgParseSetupTest.test_quiet) r4 r5 r6 r7 r8 r rB rT r^ r` ra �
__classcell__r r rP r r: < 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 )�SetupLogFileHandlerTestz-Tests for certbot.log.setup_log_file_handler.c O s ddl m} |||�S )Nr )�setup_log_file_handler)r rd )r r
r rd r r r r � s z SetupLogFileHandlerTest._callc s t t| ��� d| j_d S )N�* )rA rc rB rC rD rU rP r r rB � 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 rC 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_commonrU r r r �test_success_with_rollover� s z2SetupLogFileHandlerTest.test_success_with_rolloverc C s d| j _| jdd� d S )Nr Frp )rC rD rr rU 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 )Nrf rg z.1)r rC rS r! r] r r Z formatterZ converter�time� localtimer rL rZ r[ rY )r, rq �log_filer1 r Z
expected_pathZ backup_pathr r r rr � s z,SetupLogFileHandlerTest._test_success_commonz0certbot.log.logging.handlers.RotatingFileHandlerc C s2 | � | jdd� |jd d }| �| jj|� d S )Nrf rg r Z backupCount)r rC � call_argsr! rD )r, rm Z backup_countr r r �test_max_log_backups_used� s z1SetupLogFileHandlerTest.test_max_log_backups_used)r4 r5 r6 r7 r8 r rB r* r9 ro rs rt rr ry rb r r rP r rc ~ s
rc 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> )
rH rI �stream�isattyr � getLogger�loggerr r r r? r1 r" )r, r? r r r rB � s
z ColoredStreamHandlerTest.setUpc C s | j �� d S rR �r1 rS rU r r r rT � 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 r1 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)r4 r5 r6 r7 rB rT r� r� r r r r rz � s
rz 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 r4 r� r r r� rH rI r} r( rJ r r% r1 r" )r, r% r r r rB � s
zMemoryHandlerTest.setUpc C s | j �� | j�� d S rR )r1 rS rJ rU r r r rT � s
zMemoryHandlerTest.tearDownc C s2 | � � | jjdd� | �| j�� | jd � d S )NT�Zforce�
)�_test_log_debugr1 �flushr! r} r� r� rU r r r �
test_flush� s z MemoryHandlerTest.test_flushc C s. | j �| j� | j�� | �| j�� d� d S )N� )r� Zcriticalr� r1 r� r! r} r� rU 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� rH rI r r( r1 Z setTargetr� r! r} r� r� rS )r, Z
new_streamZnew_stream_handlerr r r �test_target_reset� s
z#MemoryHandlerTest.test_target_resetc C s | j �| j� d S rR )r� r r� rU r r r r� � s z!MemoryHandlerTest._test_log_debugN)
r4 r5 r6 r7 rB rT 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@ r1 )r, r@ r r r rB � s zTempHandlerTest.setUpc C s | j �� d S rR r� rU r r r rT s zTempHandlerTest.tearDownc C s | � t�| jjd�� d S )Ni� )r' r Zcheck_permissionsr1 rL rU r r r �test_permissions s z TempHandlerTest.test_permissionsc C s$ | j �� | �tj�| j j�� d S rR )r1 rS r\ r rL rY rU r r r � test_delete s
zTempHandlerTest.test_deletec C sB | j �t�� � | j �� | �tj�| j j�� t� | j j� d S rR )
r1 �emitr* rN rS r' r rL rY �removerU r r r �test_no_delete
s
z TempHandlerTest.test_no_deleteN) r4 r5 r6 r7 rB rT 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._callrV c C sD t �� }d}ddi}| j|f|�|� |j||� |jjdd� d S )N)�somer
r� r Tr� )r* rN r r r� )r, Z mock_post_arg_parse_except_hookr/ r
r r r r r2 s �z!PreArgParseExceptHookTest.test_itNr3 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 rU r r r rB / 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 rj Z with_code)r� �Z acme_coder r �get_acme_errorI s zBPostArgParseExceptHookTest.test_acme_error.<locals>.get_acme_errorFr� ) �nextrH Ziterkeysr Z ERROR_CODESr� r� r rj 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.)rH rI r� �
BaseExceptionr �exc_infor* r9 r� ri r r �
SystemExitr� rk rl 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' rX rx 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 rR )r\ Z exceptionrX r' r r� )r, r� r� r r r r� s z/PostArgParseExceptHookTest._assert_quiet_outputN)r4 r5 r6 r7 r8 r rB 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 )Nrf r= �logfiles)r rL rZ �tempdirrF rS r� r' )r, rw �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� rk rl )r, r
r rn r r r r� � s
z ExitWithLogPathTest._test_commonN) r4 r5 r6 r7 r8 r r� r� r� r r r r r� � s
r� �__main__)#r7 r Zlogging.handlersr ru Zunittestr* rH Zacmer Zacme.magic_typingr Zcertbotr r r Zcertbot.compatr r Z
certbot.testsZ test_utilZTestCaser ZConfigTestCaser: rc rz r� r� r� r� ZTempDirTestCaser� r4 �mainr r r r �<module> s4 &B3 +]