File: //usr/lib/python3/dist-packages/fail2ban/tests/__pycache__/banmanagertestcase.cpython-39.pyc
a
� �_:& � @ sd d Z dZdZddlZddlmZmZ ddlmZ dd l m
Z
G d
d � d ej �Z G d d
� d
ej �Z
dS )z
Cyril Jaquierz Copyright (c) 2004 Cyril JaquierZGPL� N� )� setUpMyTime�tearDownMyTime� )�
BanManager)� 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__� �C/usr/lib/python3/dist-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 �testAdd/ s
zAddFailure.testAddc C s> | � | j�| j�� | �| j�| j�� | �| j�� d� d S )Nr )r r r r � assertFalser r r r r r �testAddDuplicate6 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 �
testInListNOK` 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 ZgetIPr$ r r r! r r �strr" )r r( �c�ir r r �testBanTimeIncre 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( r. r r r � testUnbanx 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 r4 r$ r )r r7 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* r/ r8 r9 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 s` t t| ��� tj�� t� d| _d| _d| _ d| _
t | jd�}t � | _
| �| j
�|�� dS )r z
93.184.216.34Z15133ZEUZripenccr N)r rA r
�unittest�F2BZSkipIfNoNetworkr Z _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 rA 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 � )Ztimeoutr �errorFz.Skip test because service is not available: %s) rA � availablerG ZgetBanListExtendedCymruInforB rC Zfast�getZSkipTest)r Ztc�
cymru_infor r r � _getBanListExtendedCymruInfo� s
�
z4StatusExtendedCymruInfo._getBanListExtendedCymruInfoc C s, | � � }| �|| jg| jg| jgd�� d S )N�ZasnZcountryZrir)rM �assertDictEqualrD rE rF )r rL r r r �
testCymruInfo� s ��z%StatusExtendedCymruInfo.testCymruInfoc C s | � | j�| �� �| jg� d S �N)r rG ZgeBanListExtendedASNrM rD r r r r �testCymruInfoASN� s �z(StatusExtendedCymruInfo.testCymruInfoASNc C s | � | j�| �� �| jg� d S rQ )r rG ZgeBanListExtendedCountryrM rE r r r r �testCymruInfoCountry� s �z,StatusExtendedCymruInfo.testCymruInfoCountryc C s | � | j�| �� �| jg� d S rQ )r rG ZgeBanListExtendedRIRrM rF 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 ZnxdomainrN z8.0.0.0g <f�AZ3356ZUSZarinr )�level�key) r rG r r r rM rO r: r, )r r( rL r r r �testCymruInfoNxdomain� s(
��
��z-StatusExtendedCymruInfo.testCymruInfoNxdomain)
r= r> r? r
r rJ rM rP rR rS rT rW r@ r r r r rA � s
rA )�
__author__Z
__copyright__Z __license__rB Zutilsr r Zserver.banmanagerr Z
server.ticketr ZTestCaser rA r r r r �<module> s