File: //usr/lib/python3/dist-packages/fail2ban/tests/__pycache__/misctestcase.cpython-39.pyc
a
���`VF � @ s d Z dZdZddlZddlZddlZddlZddlZddlZddl Z ddl
m
Z
ddl m Z ddl
mZmZ d d
lmZmZmZmZmZmZmZmZmZ d d lm Z 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�getVerbosityFormat�
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 n0 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 n0 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�g d �� | � td�g d �� d S )
N� z �1z 1 2 �2z 1, 2 , z 1
2z 1
2, 3)r r �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_callJ 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. O s c C s t d�S )Nz?fail2ban-python -c 'import sys; print(tuple(sys.version_info))')r. r r r r �_getSysPythonVersionT s r3 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)�superr4 �setUp�unittest�F2BZ
SkipIfFast�os�path�join�dirname�__file__�exists�setup�sys� version_infoZSkipTestr3 �str�tuple)r r? ZsysVer�� __class__r r r6 Z s
���zSetupTest.setUpc C st | j s
d S t�� }tjjtjkr$dnd}z:t� dt
j | j ||f � | � t�
|� � W t�|� n t�|� 0 d S )Nz >/dev/null 2>&1r z2%s %s --dry-run install --disable-2to3 --root=%s%s)r? �tempfile�mkdtempr7 r8 � log_level�logging�DEBUGr9 �systemr@ �
executable�
assertTrue�listdir�shutil�rmtree)r �tmp� supdbgoutr r r �testSetupInstallDryRunj s �z SetupTest.testSetupInstallDryRunc
s& | j s
d S t�� � tj�d� }tjjt j
kr2dnd}�z�| � t� dt
j| j � |f �d� � fdd�}|ttd� ���}g d �}t|��|�r�d
d � }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�t
j�� �qZW t�� � t� dt
j| j |�r�|d ndf � |�r"tj�d��r"t�d� nPt�� � t� dt
j| j |�r�|d ndf � |�r tj�d��r t�d� 0 d S )NZbuildz >/dev/nullr z(%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�rQ r r �
<listcomp>� � zESetupTest.testSetupInstallRoot.<locals>.strippath.<locals>.<listcomp>r )�lrW 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)r9 r: r; )rU �f��baser r � <genexpr>� rY zISetupTest.testSetupInstallRoot.<locals>.recursive_glob.<locals>.<genexpr>)r9 �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�
z%s %s clean --all%sz 2>&1)r? rF rG r9 r: r> r7 r8 rH rI rJ r rK r@ rL �sortedr �set�
differencer; rM r. �splitrT �realpathrO rP ) r Z remove_buildrR r[ ZgotZneedrj ri ZmissingZ missing_fullr_ Z
installedPathr rW r �testSetupInstallRootz sh
�� ��
���
�
� �
�z SetupTest.testSetupInstallRoot)r$ r% r& r6 rS rt �
__classcell__r r rD r r4 X s r4 c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� 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 sd | � tttjdkrdndd� ttjdkr.dnd� td� td� tjdk r`dtjvr`td� d S )N)� s test�testz f2b-test::non-existing-encodings test�u testÏZPyPy)� assertRaises� Exceptionr r@ rA r �versionr" 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 ||_
n||_
0 | �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~ r{ 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: %rZ 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!)r{ )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�logrI ZNOTICE� assertLogged�pruneLog�noticer@ �getdefaultencodingZHandler�handlers�list)r r r� ry r� Z _org_handlerr r r �testSafeLogging� s0
z TestsUtilsTest.testSafeLoggingc s� dD ]�}t |d�}dd� �� �fdd�� zt� d�� W n tyP |� }Y n0 d|vrh| �d |� nd
|vr|| �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_functionrx zfail2ban-testcases�>Zcoverage�:)r
�printr �assertIn� assertNotIn)r r� �tb�sr r� r �
testTraceBack� s
z TestsUtilsTest.testTraceBackc O s | j t||g|�R i |�� 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^ )�KeyErrorrl 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� / rY zMTestsUtilsTest.testExtendedAssertRaisesRE.<locals>.<lambda>.<locals>.<lambda>)r� r r" r r r� . s
�)rz r� r� r� r� r� )r r� r r" r �testExtendedAssertRaisesRE s$ ��
�
�
�z)TestsUtilsTest.testExtendedAssertRaisesREc s2 � � dg d�� � � dd� � � dd� � � ddd� dD �� � �dg d�� � �dd� � �dd� � �ddd� dD �� � �d � jdd� � �d
� jd td �� � �d
� jddd� dD �� � �d� j dd� � �d� j dtd �� � �d� j ddd� dD �� t}� �� |�d� � �d� � jdddd� � jdddd� � jd ddd� � jdddd� � �� |�d � � �d� � jdddd� � jddddd� t j
j d }t j
_ � � t
t j
�d�t�� � �t j
�d d � �� d � � �t j
�d!�d!� � �t j
�d"�d"� dt j
_ zN� �t j
�d#d � �� d � � �t j
�d!�d$� � �t j
�d"�d"� W |t j
_ n
|t j
_ 0 � � t j
�d�� � �� |�d� � �d%� jd� � �d&� jdd� � jd'� jdddd� � jd(� jdddd� � jd)� jdddd*d+� � jd,� jd ddd� � jd-� jd ddd*d+� � jd&� jdddd� � �dd d.gidd d.gi� � �t� jdd d.gidd.d gi� � �dd/gd/dg� � jdd/ggd/dggd.d0� � jdd/ggd/dggdd1� � �t� fd2d �� � jddd/gidd/dgidd1� � �t� fd3d �� � jd4d/d5gd6d7gd8�gd7d6gd5d/gd9�d4gdd1� � jd4d/d5gd6d7gd8�gd7d6gd5d/gd9�d4gd:d0� � �t� fd;d �� � �d<d=gd>d=gg gfd<d=gd=d>gg gf� � �td?�td@�� � �t� jd?gd@g� � �t� jd?ggd@gg� � �dA� jdgd5d/g� � �dB� jdd/gd5d/g� d S )CN�a)r� �b�c�dZcbac s s | ]}|d kr|V qdS �r� Nr �rU r� r r r rb 7 rY z;TestsUtilsTest.testExtendedAssertMethods.<locals>.<genexpr>)r� r� r� Zcbdc s s | ]}|d kr|� � V qdS r� ��upperr� r r r rb ; rY z 'a' unexpectedly found in 'cba'z#1 unexpectedly found in \[0, 1, 2\]r rx z&'A' unexpectedly found in \['C', 'A'\]�Ac s s | ]}|d kr|� � V qdS r� r� r� r r r rb B rY 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 rb H rY z
test "xyz"ry Tr� ZzyxFZtest_zyxz
xxxx "xxx"c S s dS �N�2 r r r r r r� Z rY z:TestsUtilsTest.testExtendedAssertMethods.<locals>.<lambda>r� � g 9@c S s dS r� r r r r r r� _ rY 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 s4 � j dddgddgd�gddgddgd�dgdd �S )
Nr� r� r� r� r� r� r� Tr� r� r r" r r r� � s (�r ZA1ZA2�ABCZCBAz\['A'\] != \['C', 'B'\]z \['A', 'B'\] != \['B', 'C'\])r� r� r� �ranger� r� �debugr� ZassertNotLoggedr7 r8 ZfastZ assertFalse�
isinstanceZ maxWaitTime�boolr ZassertDictEqualrz r� r� r� )r r Zorgfastr r" r �testExtendedAssertMethods2 s� � ��� ��
��
�
� �
� �
�
� ,�,����z(TestsUtilsTest.testExtendedAssertMethodsc C s> | � td�d� | � tddd�d� | � tdddd�d� d S )Nr zA%(asctime)s %(name)-24s[%(process)d]: %(levelname)-7s %(message)sF)�paddingz<%(asctime)s %(name)s[%(process)d]: %(levelname)s %(message)s)Zaddtimer� z0%(name)s[%(process)d]: %(levelname)s %(message)s)r r
r" r r r �testVerbosityFormat� s
���z"TestsUtilsTest.testVerbosityFormatc 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 rI Z
StreamHandlerZ setFormatterZ
addHandler�error�getvaluerM 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: %rry r r rx znot all arguments converted)r� r� r7 r8 Zlog_lazyr� r� )r r r r r �testLazyLogging� s z TestsUtilsTest.testLazyLoggingN)
r$ r% r& rw r} r� r� r� r� r� r� r� r� r r r r rv � s * frv 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__rI r9 r@ r7 rF rO rd r �ior Zutilsr r r� Zhelpersr r r
r r r
r r r Z
server.mytimer ZTestCaser rA r. r3 r4 rv r� r r r r �<module> s0 ,
d