|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/fail2ban/tests/__pycache__/ |
Upload File : |
o
;s*bVF � @ 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 dkredd� Z ndd� Z dd� Z!G dd� dej �Z"G dd� de�Z#G dd� dej �Z$dS )zYaroslav Halchenkoz%Copyright (c) 2013 Yaroslav Halchenko�GPL� 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 s8 zt d�� t� \}}| �|d� | �|d� Y 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 s: zt dd �� t� \}}| �|d� | �|d� Y 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)�
subprocess� check_outputr �rstrip��cmdr+ �retr r r �_sh_callJ s r1 c C s, dd l }|j| d|jd�j�� }t|��� S )Nr T)r* �stdout)r+ �Popen�PIPEr2 �readr r- r. r r r r1 O s c C s t d�S )Nz?fail2ban-python -c 'import sys; print(tuple(sys.version_info))')r1 r r r r �_getSysPythonVersionT � r6 c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )� SetupTestc s� t t| ��� tj�� tj�tj� t
�ddd�}tj� |�r!|p"d | _ | j s1t
jdkr1t�d��t� }|ttt
j��krKt�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)�superr8 �setUp�unittest�F2B�
SkipIfFast�os�path�join�dirname�__file__�exists�setup�sys� version_info�SkipTestr6 �str�tuple)r rD �sysVer�� __class__r r r: Z s
����zSetupTest.setUpc C sr | j sd S t�� }tjjtjkrdnd}z t� dt
j | j ||f � | � t�
|� � W t�|� d S t�|� w )Nz >/dev/null 2>&1r z2%s %s --dry-run install --disable-2to3 --root=%s%s)rD �tempfile�mkdtempr; r< � log_level�logging�DEBUGr> �systemrE �
executable�
assertTrue�listdir�shutil�rmtree)r �tmp� supdbgoutr r r �testSetupInstallDryRunj s �z SetupTest.testSetupInstallDryRunc
s" | j sd S t�� � tj�d� }tjjt j
krdnd}z�| � t� dt
j| j � |f �d� � fdd�}|ttd� ���}g d �}t|��|�r~d
d � }i }t|��|�D ]}tj�� |�} tj�| �rk||| d ��pld ||< qU| j ||d
||� |f d� dD ]}
| jtj�tj�� |
��d|
d� q�td� d ��d�} | �t| �dk� | D ]} | � tj�| �tj�t
j�� q�W t�� � t� dt
j| j |r�|d ndf � |r�tj�d�r�t�d� d S d S d S t�� � t� dt
j| j |r�|d ndf � |�rtj�d��rt�d� w w w )N�buildz >/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�rX r r �
<listcomp>� s zESetupTest.testSetupInstallRoot.<locals>.strippath.<locals>.<listcomp>r )�lr_ r r � strippath� s z1SetupTest.testSetupInstallRoot.<locals>.strippathz%s/*)�etc�usr�varc sF g }t �| �D ]\� }}t�|| |�}|�� fdd�|D �� q|S )Nc 3 s � | ]
}t j�� |�V qd S �N)r> r? r@ )r] �f��baser r � <genexpr>� s � zISetupTest.testSetupInstallRoot.<locals>.recursive_glob.<locals>.<genexpr>)r> �walk�fnmatch�filter�extend)�treeroot�pattern�results�dirs�files� goodfilesr rh 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)rD rM rN r> r? rC r; r< rO rP rQ r rR rE rS �sortedr �set�
differencer@ rT r1 �splitr\ �realpathrV rW ) r � remove_buildrY rb �got�needru rs �missing� missing_fullrg �
installedPathr r_ r �testSetupInstallRootz sn
�� ��
�����
��
�� �z SetupTest.testSetupInstallRoot)r% r& r' r: rZ r� �
__classcell__r r rK r r8 X s r8 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.py�samplez/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 sl | � tttjdkr dndd� ttjdkrdnd� td� td� tjdk r2dtjvr4td� d S d S d S )N)� s test�testz f2b-test::non-existing-encodings test�u test�PyPy)� assertRaises� Exceptionr rE rF 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||_
w | �d� d S )Nc @ s e Zd Zddd�Zdd� ZdS )z,TestsUtilsTest.testSafeLogging.<locals>.Testr c S s
|| _ d S rf )�err)r r� r r r �__init__� s
z5TestsUtilsTest.testSafeLogging.<locals>.Test.__init__c S s | j rtd��dS )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: %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 @ � e Zd Zdd� ZdS )z5TestsUtilsTest.testSafeLogging.<locals>._ErrorHandlerc S s t d��)Nzerror in handler test!)r� )r �recordr r r �handle� r7 z<TestsUtilsTest.testSafeLogging.<locals>._ErrorHandler.handleN)r% r& r' r� r r r r �
_ErrorHandler� s r� ztest 3aztest 3b�OK) � DefLogSys�logrP �NOTICE� assertLogged�pruneLog�noticerE �getdefaultencoding�Handler�handlers�list)r r r� r� r� � _org_handlerr r r �testSafeLogging� s0
z TestsUtilsTest.testSafeLoggingc s� dD ]B}t |d�}dd� �� �fdd�� zt� d�� W n ty( |� }Y nw d|vr4| �d |� n
d
|vr>| �d |� | �d |� qd S ) N)TF)�compressc S s t � �rf )r r r r r �
func_raise� s z0TestsUtilsTest.testTraceBack.<locals>.func_raisec s | r
� | d � d S �� d S )Nr r )�i��
deep_functionr� r r r� s
z3TestsUtilsTest.testTraceBack.<locals>.deep_functionr� zfail2ban-testcases�>�coverage�:)r �printr �assertIn� assertNotIn)r r� �tb�sr r� r �
testTraceBack� s
� �z TestsUtilsTest.testTraceBackc O s | j t||g|�R i |�� d S rf )�assertRaisesRegex�AssertionError)r �regexp�funr �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 | ��rf )�KeyErrorrw r r r �_key_err r7 z;TestsUtilsTest.testExtendedAssertRaisesRE.<locals>._key_err�^failed$�failed�
__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� / s zMTestsUtilsTest.testExtendedAssertRaisesRE.<locals>.<lambda>.<locals>.<lambda>)r� r r# r r r� . s
�)r� 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
_ z'� �t j
�d#d � �� d � � �t j
�d!�d$� � �t j
�d"�d"� W |t j
_ n|t j
_ w � � 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�d�cbac s s � | ] }|d kr|V qdS �r� Nr �r] r� r r r rj 7 s � z;TestsUtilsTest.testExtendedAssertMethods.<locals>.<genexpr>)r� r� r� �cbdc s � � | ] }|d kr|� � V qdS r� ��upperr� r r r rj ; � � z 'a' unexpectedly found in 'cba'z#1 unexpectedly found in \[0, 1, 2\]r r� z&'A' unexpectedly found in \['C', 'A'\]�Ac s r� r� r� r� r r r rj B r� z'a' was not found in 'xyz'�xyzz 5 was not found in \[0, 1, 2\]� z!'A' was not found in \['C', 'B'\]c s r� )r� Nr� r� r r r rj H r� z
test "xyz"r� Tr� �zyxF�test_zyxz
xxxx "xxx"c S � dS �N�2 r r r r r r� Z � z:TestsUtilsTest.testExtendedAssertMethods.<locals>.<lambda>r� � g 9@c S r� 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 s4 � j dddgddgd�gddgddgd�dgdd �S )
Nr� r� r� r� r� r� r� Tr� r� r r# r r r� � s (�r �A1�A2�ABC�CBAz\['A'\] != \['C', 'B'\]z \['A', 'B'\] != \['B', 'C'\])r� r� r� �ranger� r� �debugr� �assertNotLoggedr; r< �fast� assertFalse�
isinstance� maxWaitTime�boolr �assertDictEqualr� r� r� r� )r r �orgfastr 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)�addtimer� 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)s�fail2ban_tests�XXXz: XXX�|�
r r )r r r
rP �
StreamHandler� setFormatter�
addHandler�error�getvaluerT r- �endswith�indexr )r �strout� Formatter�fmtr �outr� �pindexr r r �testFormatterWithTraceBack� s
,z)TestsUtilsTest.testFormatterWithTraceBackc C s2 t }|�dtjj� |�dddd� | �d� d S )Nzlazy logging: %rr� r r r� znot all arguments converted)r� r� r; r< �log_lazyr� r� )r r r r r �testLazyLogging� s z TestsUtilsTest.testLazyLoggingN)
r% r& r' r� r� r� r� r� r� r� r� r r r r r r r� � s * f r� c @ r� )�
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 )N�1y6mo30w15d12h35m25si]��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 � str2secondsr �float)r �str2secr r r �testStr2Seconds� s $zMyTimeTest.testStr2SecondsN)r% r& r' r r r r r r � s r )%�
__author__�
__copyright__� __license__rP r> rE r; rM rV rl r �ior �utilsr r r� �helpersr r
r r r
r r r r �
server.mytimer �TestCaser rF r1 r6 r8 r� r r r r r �<module> s2 ,
d