File: //proc/self/root/lib/python3/dist-packages/twisted/test/__pycache__/test_logfile.cpython-38.pyc
U
W[3G � @ s� d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z m Z G dd� de j
�ZG dd� de j�ZG d d
� d
e j
�ZdS ) � )�division�absolute_importN)�unittest)�logfile�runtimec @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zej�� r�d e_d!d"� Zd#d$� Zd%d&� Zd'd(� Zd)S )*� LogFileTestsz%
Test the rotating log file.
c C s4 | � � | _t�| j� d| _tj�| j| j�| _d S )Nztest.log�Zmktemp�dir�os�makedirs�name�path�join��self� r �;/usr/lib/python3/dist-packages/twisted/test/test_logfile.py�setUp s
zLogFileTests.setUpc C s. t �| jd� t j�| j�r*t �| jd� dS )z�
Restore back write rights on created paths: if tests modified the
rights, that will allow the paths to be removed easily afterwards.
� N)r
�chmodr r
�existsr r r r �tearDown s zLogFileTests.tearDownc C s. t �| j| j�}| �|j� | �t|j� dS )zf
L{BaseLogFile.shouldRotate} is abstract and must be implemented by
subclass.
N) r Z BaseLogFiler r �
addCleanup�close� assertRaises�NotImplementedErrorZ shouldRotate�r �logr r r �test_abstractShouldRotate% s z&LogFileTests.test_abstractShouldRotatec C sr t �t�| j| j���,}|�d� |�d� |�� |�d� W 5 Q R X t| j ��}| �
|� � d� W 5 Q R X dS )zn
Log files can be written to, flushed and closed. Closing a log file
also flushes it.
�123�456�7890�
1234567890N) �
contextlib�closingr �LogFiler r �write�flush�openr
� assertEqual�read�r r �fr r r � test_writing/ s
zLogFileTests.test_writingc C sL t �tj| j| jdd����}|�d� |�d� |�d� | �tj �
d� | j ��� | � tj �
d� | j ��� |�d� | �tj �
d� | j ��� | �tj �
d� | j ��� | � tj �
d � | j ��� |�d
� | � tj �
d � | j ��� |�
� | �tj �
d � | j ��� | � tj �
d � | j ��� W 5 Q R X | �|�� d d
dg� dS )zq
Rotating log files autorotate after a period of time, and can also be
manually rotated.
�
)� rotateLengthr �4567890� 11111111111�{0}.1�{0}.2� �{0}.3�3�{0}.4� � � N)r# r$ r r% r r r& �
assertTruer
r
r �format� assertFalse�rotater) �listLogsr r r r �
test_rotation> s& �
$zLogFileTests.test_rotationc C s� t �t�| j| j���}|�d� W 5 Q R X t�| j| j�}| �|j� | � |j
d� | � |j � � |j
� |�d� | � |j
d� | � |j � � |j
� |j }|�
dd� | � |�� d� dS )a
Log files can be written to, closed. Their size is the number of
bytes written to them. Everything that was written to them can
be read, even if the writing happened on separate occasions,
and even if the log file was closed in between.
�
0123456789r. �abc�
r s
0123456789abcN)r# r$ r r% r r r& r r r) �size�_file�tell�seekr* r+ r r r � test_append\ s
zLogFileTests.test_appendc C s� t �| j| j�}| �|j� |�d� |�d� |�� |�d� |�� | � |�
� dg� t � |�
� ��4}|j�d� | � |�� dg� | � |�� g � W 5 Q R X t � |�d���*}| � |�� ddg� | � |�� g � W 5 Q R X | �t|jd� | �t|jd� |�� | � |�
� ddg� t � |�d���4}|j�d� | � |�� dg� | � |�� g � W 5 Q R X t � |�d���*}| � |�� ddg� | � |�� g � W 5 Q R X dS ) a[
Various tests for log readers.
First of all, log readers can get logs by number and read what
was written to those log files. Getting nonexistent log files
raises C{ValueError}. Using anything other than an integer
index raises C{TypeError}. As logs get older, their log
numbers increase.
zabc
zdef
zghi
r8 r r9 �1N)r r% r r r r r& r> r' r) r? r# r$ Z
getCurrentLogrE rG � readLines�getLogr �
ValueError� TypeError)r r �readerr r r �test_logReaderr s4
zLogFileTests.test_logReaderc C sF t | jd�� W 5 Q R X t�| j�}| �|j� | �g |�d�� dS )zB
L{LogReader.readLines} supports reading no line.
�wr N)r( r
r Z LogReaderr r r) rJ )r rN r r r �test_LogReaderReadsZeroLine� s
z(LogFileTests.test_LogReaderReadsZeroLinec C s| t | jd��� t�| jd� t�| j�tj }t�| j | j
�}| � |j� |� d� |�
� | �|t�| j�tj � dS )zH
Check rotated files have same permissions as original.
rP �� rB N)r( r
r r
r �stat�ST_MODEr r% r r r r& r> r) )r �moder r r r �test_modePreservation� s
z"LogFileTests.test_modePreservationc C s� t �| j| j�}| �|j� |�d� t�| jd� zt tj
� | jd�d�}W n t t
fk
rf Y nX |�� dS |�� |�d� |�� |j}| �|�� d� |�dd� | �|�� d � dS )
zH
Check it keeps working when permission on dir changes.
rB �m ZxxxrP N�def� r s abcdef)r r% r r r r r& r
r r( r
r �OSError�IOErrorr> r' rE r) rF rG r* r+ r r r �test_noPermission� s"
z LogFileTests.test_noPermissionc C s t j| j| jddd�}| �|j� |�d� |�d� | �tj �
d� | j ��� |�d� | �tj �
d� | j ��� |�d � | �tj �
d
� | j ��� t d
� | j ���}| �
|�� d� W 5 Q R X |�d � t d
� | j ���}| �
|�� d� W 5 Q R X | �tj �
d � | j ��� d
S )zl
Test it respect the limit on the number of files when maxRotatedFiles
is not None.
r. r: )r/ ZmaxRotatedFilesr1 Z 22222222222r2 Z 33333333333r3 Z 44444444444r5 Z 55555555555r7 N)r r% r r r r r& r; r
r
r r< r( r) r* r= �r r �fpr r r �test_maxNumberOfLog� s" �
z LogFileTests.test_maxNumberOfLogc C s� t j| j| jddd�}| �|j� t jj| jddd�}| �|j� | �|j|j� | �t j�
|j�|j� | �|j |j � | �|j |j � dS )z/
Test the fromFullPath method.
r. r �� defaultModeN)
r r% r r r r Z fromFullPathr
r) r
�abspathr/ ra )r �log1Zlog2r r r �test_fromFullPath� s z LogFileTests.test_fromFullPathc C s~ t | jd��, t�| jd� t�t�| j�tj �}W 5 Q R X t�| j | j
�}| � t�t�| j�tj �|� | � |j
� dS )zx
Test the default permission of the log file: if the file exist, it
should keep the permission.
�wbrR N)r( r
r
r rS �S_IMODErT r r% r r r) r r )r Z currentModerc r r r �test_defaultPermissions s "�z$LogFileTests.test_defaultPermissionsc C s` t j| j| jdd�}| �|j� t�t�| j �tj
�}t j �
� rP| �|d� n | �|d� dS )zG
Test specifying the permissions used on the log file.
�6 r` �$ N)r r% r r r r rS rf r
r
rT r �platform� isWindowsr) )r rc rU r r r �test_specifiedPermissions s
z&LogFileTests.test_specifiedPermissionsc C s� t �t�| j| j���@}|�d� tj� | jd�}t�
| j|� |� � |�d� W 5 Q R X t | j��}| �
|�� d� W 5 Q R X t |��}| �
|�� d� W 5 Q R X dS )z�
L{logfile.LogFile.reopen} allows to rename the currently used file and
make L{logfile.LogFile} create a new file.
Zhello1zsave.logZhello2N)r# r$ r r% r r r& r
r
r �renameZreopenr( r) r* )r rc ZsavePathr, r r r � test_reopen s
zLogFileTests.test_reopenz Can't test reopen on Windowsc C s( | � ttj| jd�}| �|jtj� dS )zR
Specifying an invalid directory to L{LogFile} raises C{IOError}.
Zthis_dir_does_not_existN)r r[ r r% r r) �errnoZENOENT)r �er r r �test_nonExistentDir2 s �z LogFileTests.test_nonExistentDirc C s` t j�� rd\}}d}n d\}}d}tj||dd�}| �|j� | �|j|� | �|j d� dS )z�
Opening a L{LogFile} which can be read and write but whose mode can't
be changed doesn't trigger an error.
)�NULr4 rr )Znullz/devz /dev/nullrW r` N)
r rj rk r r% r r r) r
ra )r r � directoryZ expectedPathr r r r �test_cantChangeFileMode; s
z$LogFileTests.test_cantChangeFileModec C s� t �| j| j�}| �|j� td�|j�d��}|� d� W 5 Q R X td�|j�d��}|� d� W 5 Q R X | �
dg|� � � dS )zl
L{LogFile.listLogs} doesn't choke if it encounters a file with an
unexpected name.
r2 rP r z {0}.bad-filer8 N) r r% r r r r r( r< r
r& r) r? r] r r r � test_listLogsWithBadlyNamedFilesN s z-LogFileTests.test_listLogsWithBadlyNamedFilesc
C sn t �| j| j�}| �|j� tdd�D ].}td�|j |�d��}|�
d� W 5 Q R X q&| � ddg|� � � dS ) zR
L{LogFile.listLogs} ignores log files which rotated suffix is 0.
r r: �{0}.{1}rP r r8 r9 N)
r r% r r r r �ranger( r< r
r& r) r? )r r �ir^ r r r �%test_listLogsIgnoresZeroSuffixedFiles^ s z2LogFileTests.test_listLogsIgnoresZeroSuffixedFilesN)�__name__�
__module__� __qualname__�__doc__r r r r- r@ rH rO rQ rV r\ r_ rd rg rl rn r rj rk �skiprq rt ru ry r r r r r s,
,
r c @ s e Zd ZdZdd� Zdd� ZdS )�RiggedDailyLogFile� c C s t j�| � | �� | _d S )N)r � DailyLogFile� _openFile�toDate�lastDater r r r r� p s z RiggedDailyLogFile._openFilec G s* |rt j|� d d� S t �| j�d d� S )Nr: )�time�gmtime�_clock)r �argsr r r r� v s zRiggedDailyLogFile.toDateN)rz r{ r| r� r� r� r r r r r m s r c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
j � � rLde _
dd� Zdd� Zdd� Zdd� ZdS )�DailyLogFileTestsz!
Test rotating log file.
c C s4 | � � | _t�| j� d| _tj�| j| j�| _d S )Nz
testdaily.logr r r r r r � s
zDailyLogFileTests.setUpc C sp t �t| j| j���,}|�d� |�d� |�� |�d� W 5 Q R X t| j��}| � |�
� d� W 5 Q R X dS )zO
A daily log file can be written to like an ordinary log file.
r r r! r" N) r# r$ r r r r& r' r( r
r) r* r+ r r r r- � s
z DailyLogFileTests.test_writingc s
t �j�j�� ��� j� � �fdd�td�D �}d� _� �d� d� _� �d� d� _� �d � ��t j
� |d
�� �� t j
� |d �� d � _� �d
� ��t j
� |d
�� ��t j
� |d �� �� t j
� |d �� d� _� �d� �� t j
� |d �� dS )z/
Daily log files rotate daily.
c s$ g | ] }�j d � �|d � �qS )�.�Q )r
�suffix)�.0Zday�r r r r �
<listcomp>� s z3DailyLogFileTests.test_rotation.<locals>.<listcomp>r: r� r i�� r0 r� r1 r r8 i � r4 r9 i� r6 N)
r r r r r rw r� r& r; r
r
r r= )r Zdaysr r� r r@ � s&
z DailyLogFileTests.test_rotationc C s� dddg}t | j| j�}| �|j� |D ]}|�|� q(|�� |�d�}| �|j� | �||� � � | �
t |jd� d|_ |�� |�
� |�d�}| �|j� | �||� � � dS ) zH
Test retrieving log files with L{DailyLogFile.getLog}.
z1
z2
z3
r� r� i�Q r N)r r r r r r& r' rK r) rJ r rL r� r> )r �datar �d�rr r r � test_getLog� s
z DailyLogFileTests.test_getLogc C sp t | j| j�}| �|j� d�|j|�|j��}t |d��}|�
d� W 5 Q R X |j }|� � | �
||j � dS )zu
L{DailyLogFile.rotate} doesn't do anything if they new log file already
exists on the disk.
rv rP r N)r r r r r r< r
r� r� r( r& rE r> r) )r r Z newFilePathr^ � previousFiler r r �test_rotateAlreadyExists� s z*DailyLogFileTests.test_rotateAlreadyExistsc C s\ t �| j| j�}| �|j� t�|jd� | �tj|jd� |j }|�
� | � ||j � dS )z�
L{DailyLogFile.rotate} doesn't do anything if the directory containing
the log files can't be written to.
ri i� N) r r� r r r r r
r rs rE r> r) �r r r� r r r �#test_rotatePermissionDirectoryNotOk� s z5DailyLogFileTests.test_rotatePermissionDirectoryNotOkzVMaking read-only directories on Windows is too complex for this test to reasonably do.c C sJ t �| j| j�}| �|j� t�|jd� |j }|�
� | � ||j � dS )zi
L{DailyLogFile.rotate} doesn't do anything if the log file can't be
written to.
ri N) r r� r r r r r
r r
rE r> r) r� r r r � test_rotatePermissionFileNotOk� s z0DailyLogFileTests.test_rotatePermissionFileNotOkc C s< t �| j| j�}| �|j� t�d�}| �d|� |�� dS )z~
Test that L{DailyLogFile.toDate} converts its timestamp argument to a
time tuple (year, month, day).
) �� r8 r8 r r r r r r )r� r8 r8 N)
r r� r r r r r� �mktimer) r� )r r Z timestampr r r � test_toDate s
z DailyLogFileTests.test_toDatec sT � fdd�}t �� j� j�}� �|j� � �td|� |�� }� � dddg|� dS )a�
Test that L{DailyLogFile.toDate} returns today's date by default.
By mocking L{time.localtime}, we ensure that L{DailyLogFile.toDate}
returns the first 3 values of L{time.localtime} which is the current
date.
Note that we don't compare the *real* result of L{DailyLogFile.toDate}
to the *real* current date, as there's a slight possibility that the
date changes between the 2 function calls.
c s � � d| � ttdd��S )Nr r � )r) �listrw )r� r r r �mock_localtime s zADailyLogFileTests.test_toDateDefaultToday.<locals>.mock_localtime� localtimer r8 r9 N)
r r� r r r r Zpatchr� r� r) )r r� r �logDater r r �test_toDateDefaultToday s z)DailyLogFileTests.test_toDateDefaultTodayc C sH t �| j| j�}| �|j� d}t�|d �}|�|�}| � ||� dS )zc
Test that L{DailyLogFile.toDate} uses its arguments to create a new
date.
)i� r. � )r r r r r r N)
r r� r r r r r� r� r� r) )r r �dateZsecondsr� r r r �#test_toDateUsesArgumentsToMakeADate& s
z5DailyLogFileTests.test_toDateUsesArgumentsToMakeADateN)rz r{ r| r} r r- r@ r� r� r� r rj rk r~ r� r� r� r� r r r r r� } s
� r� )Z
__future__r r r# ro r
rS r� Z
twisted.trialr Ztwisted.pythonr r ZTestCaser r� r r� r r r r �<module> s _