File: //usr/lib/python3.9/site-packages/fail2ban/tests/__pycache__/banmanagertestcase.cpython-39.pyc
a
�M^��;�� @ sp d Z dZdZddlZddlmZmZ ddlmZ dd l m
Z
dd
l m Z G d d � d ej
�ZG d
d� dej
�ZdS )z
Cyril Jaquierz Copyright (c) 2004 Cyril Jaquier�GPL� N� )� setUpMyTime�tearDownMyTime� )�
BanManager)�DNSUtils)� BanTicketc sp e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
� ZS )�
AddFailurec s, t t| ��� t� tdd�| _t� | _dS )� Call before every test case.�
193.168.0.128� �;f�AN)�superr
�setUpr r �_AddFailure__ticketr �_AddFailure__banManager��self�� __class__� �E/usr/lib/python3.9/site-packages/fail2ban/tests/banmanagertestcase.pyr $ s zAddFailure.setUpc s t t| ��� t� dS �zCall after every test case.N)r r
�tearDownr r r r r r + s zAddFailure.tearDownc C sZ | � | j�| j�� | �| j�� d� | �| j�� d� | j�d� | �| j�� d� d S )Nr r )�
assertTruer � addBanTicketr � assertEqual�sizeZ getBanTotalZ setBanTotalr r r r �testAdd0 s
zAddFailure.testAddc C s> | � | j�| j�� | �| j�| j�� | �| j�� d� d S )Nr )r r r r � assertFalser r r r r r �testAddDuplicate7 s zAddFailure.testAddDuplicatec C s� | j �� }d}dD ]�\}}tdd�}td|�}|d ur@|�|� | �| j �|�� | �| j �|�� | �| j �� d� | j � |�
� �}| �|� |�|� |�� | �|� |�|k� |� |�}| �|� � d� |dkr| �|�|�d� qd S )Nr ))r
N)g �Tf�AN)r
i�Q )r
���r r
r r! )
r �
getBanTimer �
setBanTimer r r r r �
getTicketByID�getIDZgetEndOfBanTime�getTime)r Z
defBanTimeZprevEndOfBanTimeZtnewZbtnewZticket1Zticket2Z banticketr r r �testAddDuplicateWithTime< s"
z#AddFailure.testAddDuplicateWithTimec C s4 | � | j�| j�� tdd�}| � | j�|�� d S )Nr r
)r r r r r �
_inBanList�r �ticketr r r � testInListOK\ s
zAddFailure.testInListOKc C s4 | � | j�| j�� tdd�}| �| j�|�� d S )Nz
111.111.1.111r
)r r r r r r r( r) r r r �
testInListNOKa s
zAddFailure.testInListNOKc C s t | j�� | j�� �}d}dD ]f}| j�| j� |d7 }|�|� | �| j�|�� | �t | j�
|�� ��d|�� |�� ||f � q | j�| j� |d7 }|�d� | �| j�|�� |�d� | �| j�|�� | �t | j�
|�� ��d|�� |�� d|f � d S )Nr )�� i� r! r zFBanTicket: ip=%s time=%s bantime=%s bancount=%s #attempts=0 matches=[]r! r- ) r r r% r& r r r# r r �strr$ )r r* �c�ir r r �testBanTimeIncrf s"
�
�zAddFailure.testBanTimeIncrc C s� | j �� }| j�� }| �| j �| j�� | �| j �| j�� | �| j �|�g � | �| j �|| d �| jg� | �| j � � d� | �| j �| j�� t
| j� � |d �}| � | j �|�� | �t
| j �|| d ��d� | �t
| j �|| d d ��d� td�D ]>}t
d| |�}|�|�|�|d � | �| j �|�� �q| �t
| j �|| d d ��d� | �t
| j �|| d d ��d � | �| j � � d� d S )
Nr r iX � z 193.168.0.%s�
r �2 � )r r" r r& r r r( r � unBanListr r r% r �len�ranger# )r �btimeZstimer* r0 r r r � testUnbany s&
$ $$zAddFailure.testUnbanc C s� | j �� }| j �d� zl| �| j �| j�� | �| j �| j�� | �| j �| j� � | d �g � | �| j �
� d� W | j �|� n| j �|� 0 d S )Nr! r ) r r" r# r r r r( r r6 r& r )r r9 r r r �testUnbanPermanent� s
"z AddFailure.testUnbanPermanentc C sX t dd�t dd�g}|d �d� |D ]}| j�|� q&| �| jjddd�dd g� d S )
Nz 192.0.2.1r
z 192.0.2.2r r! T)ZorderedZwithTimez:192.0.2.1 2006-12-31 23:59:59 + 600 = 2007-01-01 00:09:59z9192.0.2.2 2006-12-31 23:59:59 + -1 = 9999-12-31 23:59:59)r r# r r �assertSortedEqualZ
getBanList)r Ztickets�tr r r � testBanList� s ���zAddFailure.testBanList)�__name__�
__module__� __qualname__r r r r r' r+ r, r1 r: r; r> �
__classcell__r r r r r
# s r
c s\ e Zd Z� fdd�Z� fdd�ZdZdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z � Z S )�StatusExtendedCymruInfoc sn t t| ��� tj�� t� ttt �
d���| _ d| _ d| _
d| _t| j d�}t� | _| �| j�|�� dS )r zresolver1.opendns.comZ36692�US�arinr
N)r rC r �unittest�F2BZSkipIfNoNetworkr �next�iterr ZdnsToIpZ _StatusExtendedCymruInfo__ban_ip� _StatusExtendedCymruInfo__asn�!_StatusExtendedCymruInfo__country� _StatusExtendedCymruInfo__rirr r �$_StatusExtendedCymruInfo__banManagerr r r) r r r r � s
z StatusExtendedCymruInfo.setUpc s t t| ��� t� dS r )r rC r r r r r r r � s z StatusExtendedCymruInfo.tearDown)TNc C s^ t }|jd r*| jjtjjr dndd�}n
|jd }|�d�rZd|f|_t�d|d ��|S ) Nr r � )�timeoutr �errorFz.Skip test because service is not available: %s) rC � availablerM ZgetBanListExtendedCymruInforF rG �fast�getZSkipTest)r Ztc�
cymru_infor r r � _getBanListExtendedCymruInfo� s
�
z4StatusExtendedCymruInfo._getBanListExtendedCymruInfoc C s, | � � }| �|| jg| jg| jgd�� d S )N�ZasnZcountryZrir)rU �assertDictEqualrJ rK rL )r rT r r r �
testCymruInfo� s ��z%StatusExtendedCymruInfo.testCymruInfoc C s | � | j�| �� �| jg� d S �N)r rM ZgeBanListExtendedASNrU rJ r r r r �testCymruInfoASN� s �z(StatusExtendedCymruInfo.testCymruInfoASNc C s | � | j�| �� �| jg� d S rY )r rM ZgeBanListExtendedCountryrU rK r r r r �testCymruInfoCountry� s �z,StatusExtendedCymruInfo.testCymruInfoCountryc C s | � | j�| �� �| jg� d S rY )r rM ZgeBanListExtendedRIRrU rL r r r r �testCymruInfoRIR� s �z(StatusExtendedCymruInfo.testCymruInfoRIRc C s� t � | _tdd�}| �| j�|�� | �� }| �|dgdgdgd�� tdd�}| �| j�|�� | �� }| j|ddgddgdd gd�d
td � d S ) Nz0.0.0.0r
ZnxdomainrV z8.0.0.0g <f�AZ3356rD rE r! )�level�key) r rM r r r rU rW r< r. )r r* rT r r r �testCymruInfoNxdomain� s(
��
��z-StatusExtendedCymruInfo.testCymruInfoNxdomain)
r? r@ rA r r rQ rU rX rZ r[ r\ r_ rB r r r r rC � s
rC )�
__author__�
__copyright__� __license__rF �utilsr r Zserver.banmanagerr Z server.ipdnsr Z
server.ticketr ZTestCaser
rC r r r r �<module> s