← Back
Editing: 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 _
Save File
Cancel