File: //lib/python3/dist-packages/fail2ban/tests/__pycache__/misctestcase.cpython-38.pyc
U
]^�B � @ s d Z dZdZddlZddlZddlZddlZddlZddlZddl Z ddl
m
Z
ddl m Z ddl
mZmZ d d
lmZmZmZmZmZmZmZmZ d d lmZ G d d
� d
ej �Z ej dkr�dd� Z ndd� Z dd� Z G dd� dej �Z!G dd� de�Z"G dd� dej �Z#dS )zYaroslav Halchenkoz%Copyright (c) 2013 Yaroslav HalchenkoZGPL� N)�glob)�StringIO� )�LogCaptureTestCase�logSys� )�formatExceptionInfo� mbasename� TraceBack�FormatterWithTraceBack� getLogger�
splitwords�
uni_decode�
uni_string)�MyTimec @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� HelpersTestc C s@ z t d��W n. t� \}}| �|d� | �|d� Y nX d S )NzVery bad exception�
ValueError�r r � assertEqual��self�name�args� r �=/usr/lib/python3/dist-packages/fail2ban/tests/misctestcase.py� testFormatExceptionInfoBasic+ s
z(HelpersTest.testFormatExceptionInfoBasicc C sB zt dd ��W n. t� \}}| �|d� | �|d� Y nX d S )NzVery badr z('Very bad', None)r r r r r � testFormatExceptionConvertArgs3 s
z*HelpersTest.testFormatExceptionConvertArgsc C s� | � td �g � | � td�g � | � td�g � | � td�dg� | � td�ddg� | � td�ddg� | � td�ddg� | � td�ddd g� d S )
N� z �1z 1 2 �2z 1, 2 , z 1
2z 1
2, 3�3)r r
�r r r r �testsplitwords< s zHelpersTest.testsplitwordsN)�__name__�
__module__� __qualname__r r r" r r r r r ) s r �r � c C s" dd l }|j| dd�}t|��� S )Nr T)�shell)�
subprocessZ check_outputr �rstrip��cmdr) Zretr r r �_sh_callH s r- c C s, dd l }|j| d|jd�j�� }t|��� S )Nr T)r( �stdout)r) �Popen�PIPEr. �readr r* r+ r r r r- M s c C s t d�S )Nz?fail2ban-python -c 'import sys; print(tuple(sys.version_info))')r- r r r r �_getSysPythonVersionR s r2 c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )� SetupTestc s� t t| ��� tj�� tj�tj� t
�ddd�}tj� |�rB|pDd | _ | j sbt
jdkrbt�d��t� }|ttt
j��kr�t�dttt
j��|f ��d S )Nz..zsetup.pyr& zLSeems to be running not out of source distribution -- cannot locate setup.pyziSeems to be running with python distribution %s -- install can be tested only with system distribution %s)�superr3 �setUp�unittest�F2BZ
SkipIfFast�os�path�join�dirname�__file__�exists�setup�sys� version_infoZSkipTestr2 �str�tuple)r r> ZsysVer�� __class__r r r5 X s
���zSetupTest.setUpc C sj | j s
d S t�� }tjjtjkr$dnd}z0t
� dt j
| j ||f � | �t
�|� � W 5 t� |� X d S )Nz >/dev/null 2>&1r z2%s %s --dry-run install --disable-2to3 --root=%s%s)r> �tempfile�mkdtempr6 r7 � log_level�logging�DEBUG�shutil�rmtreer8 �systemr? �
executable�
assertTrue�listdir)r �tmp� supdbgoutr r r �testSetupInstallDryRunh s �z SetupTest.testSetupInstallDryRunc
s� | j s
d S t�� � tj�d� }tjjt j
kr2dnd}�zN| �t�
dtj| j � |f �d� � fdd �}|ttd
� ���}d d d
g}t|��|��rdd� }i }t|��|�D ]4}tj�� |�} tj�| �r�||| d��p�d ||< q�| j||d||� |f d� dD ]*}
| jtj�tj�� |
��d|
d� �qtd� d ��d�} | �t| �dk� | D ]$} | �tj�| �tj�tj�� �q^W 5 t � � � t�
dtj| j |�r�|d ndf � |�r�tj�d��r�t � d� X d S )NZbuildz >/dev/nullr z%s %s clean --all%sz 2>&1z(%s %s install --disable-2to3 --root=%s%sr c s � fdd�| D �S )Nc s g | ]}|t � �d d� �qS )r N)�len)�.0�x�rP r r �
<listcomp>� s zESetupTest.testSetupInstallRoot.<locals>.strippath.<locals>.<listcomp>r )�lrV r r � strippath� s z1SetupTest.testSetupInstallRoot.<locals>.strippathz%s/*�etcZusr�varc sF g }t �| �D ]2\� }}t�|| |�}|�� fdd�|D �� q|S )Nc 3 s | ]}t j�� |�V qd S �N)r8 r9 r: )rT �f��baser r � <genexpr>� s zISetupTest.testSetupInstallRoot.<locals>.recursive_glob.<locals>.<genexpr>)r8 �walk�fnmatch�filter�extend)Ztreeroot�patternZresults�dirs�filesZ goodfilesr r^ r �recursive_glob� s
z6SetupTest.testSetupInstallRoot.<locals>.recursive_glob�*z6Got: %s Needed: %s under %s. Files under new paths: %s��msg)zetc/fail2ban/fail2ban.confzetc/fail2ban/jail.confz
Can't find %szfind z -name fail2ban-python�
)r> rE rF r8 r9 r= r6 r7 rG rH rI rJ rK rL r? rM r �sortedr �set�
differencer: rN r- �splitrS �realpath) r Z remove_buildrQ rY ZgotZneedrh rg ZmissingZ missing_fullr] Z
installedPathr rV r �testSetupInstallRootx s`
��
��
���
�
�z SetupTest.testSetupInstallRoot)r# r$ r% r5 rR rr �
__classcell__r r rC r r3 V s r3 c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestsUtilsTestc C sT | � td�d� | � td�d� | � td�d� | � td�d� | � td�d� d S ) Nz sample.pyZsamplez/long/path/sample.pyz/long/path/__init__.pyz
path.__init__z/long/path/base.pyz path.basez/long/path/base)r r r! r r r �
testmbasename� s
z TestsUtilsTest.testmbasenamec C sP | � tttjdkrdndd� ttjdkr.dnd� td� td� td� d S )N)� s test�testz f2b-test::non-existing-encodings test�u testÏ)� assertRaises� Exceptionr r? r@ r r! r r r �testUniConverters� s �z TestsUtilsTest.testUniConvertersc C s" t }G dd� d�}|� }|�tjd|� | �dd� | �� |�d|� | �dd� | �dt�� � |d�}|�tjd ||� | jd
d d d
� |�d||� | jdd d d
� | �d� G dd� dtj �}|j
}z:t |j
�|_
| j
|� g7 _
|�tjd� |�d� W 5 ||_
X | �d� d S )Nc @ s e Zd Zddd�Zdd� ZdS )z,TestsUtilsTest.testSafeLogging.<locals>.Testr c S s
|| _ d S r\ )�err)r r{ r r r �__init__� s z5TestsUtilsTest.testSafeLogging.<locals>.Test.__init__c S s | j rtd��ndS d S )N�no represenation for test!u. conv-error (òðåòèé), unterminated utf Ï)r{ ry r! r r r �__repr__� s
z5TestsUtilsTest.testSafeLogging.<locals>.Test.__repr__N)r )r# r$ r% r| r~ r r r r �Test� s
r z test 1a: %r� Tracebackr} z test 1b: %rz.[phase 2] test error conversion by encoding %sr ztest 2a: %r, %sztest 2azError by logging handlerF��allztest 2b: %r, %sztest 2bz*[phase 3] test unexpected error in handlerc @ s e Zd Zdd� ZdS )z5TestsUtilsTest.testSafeLogging.<locals>._ErrorHandlerc S s t d��d S )Nzerror in handler test!)ry )r �recordr r r �handle� s z<TestsUtilsTest.testSafeLogging.<locals>._ErrorHandler.handleN)r# r$ r% r� r r r r �
_ErrorHandler� s r� ztest 3aztest 3bZOK) � DefLogSys�logrH ZNOTICE� assertLogged�pruneLog�noticer? �getdefaultencodingZHandlerZhandlers�list)r r r rw r� Z _org_handlerr r r �testSafeLogging� s0
z TestsUtilsTest.testSafeLoggingc s� dD ]�}t |d�}dd� �� �fdd�� zt� d�� W n tk
rR |� }Y nX d|krj| �d |� nd
|kr~| �d |� | �d |� qd S ) N)TF)�compressc S s
t � �d S r\ )r r r r r �
func_raise� s z0TestsUtilsTest.testTraceBack.<locals>.func_raisec s | r� | d � n�� d S )Nr r )�i��
deep_functionr� r r r� � s z3TestsUtilsTest.testTraceBack.<locals>.deep_functionrv zfail2ban-testcases�>Zcoverage�:)r
�printr �assertIn� assertNotIn)r r� �tb�sr r� r �
testTraceBack� s
z TestsUtilsTest.testTraceBackc O s | j t||f|�|� d S r\ )ZassertRaisesRegex�AssertionError)r ZregexpZfunr �kwargsr r r �_testAssertionErrorRE s z$TestsUtilsTest._testAssertionErrorREc sr dd� }� � t� jd|d� � � t� jd� jd� � �d� fdd�� � � t� jd td
� � �d � fd d�� d S )
Nc S s t | ��d S r\ )�KeyErrorrj r r r �_key_err s z;TestsUtilsTest.testExtendedAssertRaisesRE.<locals>._key_err�^failed$Zfailed�
__failed__z$failed.* does not match .*__failed__c s � � d� jd�S )Nr� r� )r� �failr r! r r �<lambda>$ s �z;TestsUtilsTest.testExtendedAssertRaisesRE.<locals>.<lambda>r r zJ0 AssertionError not raised X.* does not match .*AssertionError not raisedc s � � d� fdd��S )Nz ^0 AssertionError not raised X$c s � � dtd�S )Nr r )r� �intr r! r r r� , � zMTestsUtilsTest.testExtendedAssertRaisesRE.<locals>.<lambda>.<locals>.<lambda>)r� r r! r r r� + s
�)rx r� r� r� r� r� )r r� r r! r �testExtendedAssertRaisesRE s2 � �
� �
�z)TestsUtilsTest.testExtendedAssertRaisesREc s� � � dddddg� � � dd� � � dd� � � ddd� dD �� � �ddddg� � �dd � � �dd
� � �dd d� dD �� � �d � jdd� � �d
� jdttd��� � �d� jddd� dD �� � �d� j dd� � �d� j dttd��� � �d� j ddd� dD �� t}� �� |�d� � �d� � jdddd � � j dd d d � � j d d dd � � jdd d d � � �� |�d � � � d� � j ddd d � � j ddd dd � t
j j d }t
j _ � �
tt
j �d�t�� � �t
j �d!d"� �� d#� � �t
j �d$�d$� � �t
j �d%�d%� dt
j _ zF� �t
j �d&d"� �� d#� � �t
j �d$�d'� � �t
j �d%�d%� W 5 |t
j _ X � �
t
j �d �� � �� |�d� � �d(� j d� � �d)� j dd� � jd*� j dddd � � jd+� jdd dd � � jd,� jdd dd-d.� � jd/� jd d d d � � jd0� jd d d d-d.� � jd)� j ddd d � � �ddd1giddd1gi� � �t� jddd1gidd1dgi� � �dd2gd2dg� � jdd2ggd2dggd1d3� � jdd2ggd2dggd d4� � �t� fd5d"�� � jddd2gidd2dgid d4� � �t� fd6d"�� � jd7d2d8gd9d:gd;�gd:d9gd8d2gd<�d7gd d4� � jd7d2d8gd9d:gd;�gd:d9gd8d2gd<�d7gd=d3� � �t� fd>d"�� � �d?� jdgd8d2g� � �d@� jdd2gd8d2g� d S )AN�a�b�c�d)r� r� r� r� Zcbac s s | ]}|d kr|V qdS �r� Nr �rT r� r r r r` 4 s z;TestsUtilsTest.testExtendedAssertMethods.<locals>.<genexpr>)r� r� r� Zcbdc s s | ]}|d kr|� � V qdS r� ��upperr� r r r r` 8 s z 'a' unexpectedly found in 'cba'z#1 unexpectedly found in \[0, 1, 2\]r rv z&'A' unexpectedly found in \['C', 'A'\]�Ac s s | ]}|d kr|� � V qdS r� r� r� r r r r` ? s z'a' was not found in 'xyz'Zxyzz 5 was not found in \[0, 1, 2\]� z!'A' was not found in \['C', 'B'\]c s s | ]}|d kr|� � V qdS )r� Nr� r� r r r r` E s z
test "xyz"rw Tr� ZzyxFZtest_zyxz
xxxx "xxx"c S s dS �N�2 r r r r r r� W r� z:TestsUtilsTest.testExtendedAssertMethods.<locals>.<lambda>r� � g 9@c S s dS r� r r r r r r� \ r� g @z.* was found in the logz+All of the .* were found present in the logzwas found in the logzwas not found in the logz&was not found in the log, waited 1e-06g���ư>)r� �waitz"None among .* was found in the logz0None among .* was found in the log, waited 1e-06r �B)�level)�
nestedOnlyc s � j ddggddggddd�S �Nr� r� r T)r� r� ��assertSortedEqualr r! r r r� } s
�c s" � j dddgidddgiddd�S r� r� r r! r r r� � s
��Z�C�E�F�r� r� �r� r� ���c s0 � � dddgddgd�gddgddgd�dg�S )Nr� r� r� r� r� r� r� r� r r! r r r� � s �z\['A'\] != \['C', 'B'\]z \['A', 'B'\] != \['B', 'C'\])r� r� r� r� �ranger� r� �debugr� ZassertNotLoggedr6 r7 ZfastZ assertFalse�
isinstanceZ maxWaitTime�boolr ZassertDictEqualrx r� r� )r r Zorgfastr r! r �testExtendedAssertMethods/ s �
� � �
� �
� � � � � � � �
� ,�,� � �z(TestsUtilsTest.testExtendedAssertMethodsc C s� t � }t}d}td�}t�|�}|�||�� |�|� |�d� |�� }| � |�
� � d�� |� d�}| � |dk� | �
|d |� ||d |d d � � d S ) Nz %(tb)s | %(tbc)s : %(message)sZfail2ban_testsZXXXz: XXX�|�
r r )r r r rH Z
StreamHandlerZ setFormatterZ
addHandler�error�getvaluerN r* �endswith�indexr )r ZstroutZ FormatterZfmtr �outr� Zpindexr r r �testFormatterWithTraceBack� s
z)TestsUtilsTest.testFormatterWithTraceBackc C s2 t }|�dtjj� |�dddd� | �d� d S )Nzlazy logging: %rrw r r rv znot all arguments converted)r� r� r6 r7 Zlog_lazyr� r� )r r r r r �testLazyLogging� s z TestsUtilsTest.testLazyLoggingN) r# r$ r% ru rz r� r� r� r� r� r� r� r r r r rt � s * ^rt c @ s e Zd Zdd� ZdS )�
MyTimeTestc C s� t j}| �|d�d� | �|d�d� | �|d�d� | �|d�d� | �|d�d � | �|d
�d � | �|d �d � | �t|d
��d d d d� | �t|d��d d d d� d S )NZ1y6mo30w15d12h35m25si]��z 2yy 3mo 4ww 10dd 5hh 30mm 20ssi$�mz>2 years 3 months 4 weeks 10 days 5 hours 30 minutes 20 secondsz!1 year + 1 month - 1 week + 1 dayi��zI2 * 0.5 yea + 1*1 mon - 3*1/3 wee + 2/2 day - (2*12 hou 3*20 min 80 sec) g @��Az(2*.5y+1*1mo-3*1/3w+2/2d-(2*12h3*20m80s) z!1ye -2mo -3we -4da -5ho -6mi -7seiQ pz1 month�< � g p>@z1 yearg �v@)r Z str2secondsr �float)r Zstr2secr r r �testStr2Seconds� s zMyTimeTest.testStr2SecondsN)r# r$ r% r� r r r r r� � s r� )$�
__author__Z
__copyright__Z __license__rH r8 r? r6 rE rJ rb r �ior Zutilsr r r� Zhelpersr r r
r r r
r r Z
server.mytimer ZTestCaser r@ r- r2 r3 rt r� r r r r �<module> s. (
d p