File: //lib/python3/dist-packages/fail2ban/tests/__pycache__/banmanagertestcase.cpython-38.pyc
U
ܜ^�$ � @ 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 sh e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z � Z
S )�
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 k r@|�|� | �| 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� z`| �| j �| j�� | �| j �| j�� | �| j �| j� � | d �g � | �| j �
� d� W 5 | j �|� X d S )Nr r ) r r r! r r r r& r r4 r$ r )r r7 r r r �testUnbanPermanent� s
"z AddFailure.testUnbanPermanent)�__name__�
__module__� __qualname__r
r r r r% r) r* r/ r8 r9 �
__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 r> 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 r> 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) r> � availablerD ZgetBanListExtendedCymruInfor? r@ Zfast�getZSkipTest)r Ztc�
cymru_infor r r � _getBanListExtendedCymruInfo� s
�
z4StatusExtendedCymruInfo._getBanListExtendedCymruInfoc C s, | � � }| �|| jg| jg| jgd�� d S )N�ZasnZcountryZrir)rJ �assertDictEqualrA rB rC )r rI r r r �
testCymruInfo� s ��z%StatusExtendedCymruInfo.testCymruInfoc C s | � | j�| �� �| jg� d S �N)r rD ZgeBanListExtendedASNrJ rA r r r r �testCymruInfoASN� s �z(StatusExtendedCymruInfo.testCymruInfoASNc C s | � | j�| �� �| jg� d S rN )r rD ZgeBanListExtendedCountryrJ rB r r r r �testCymruInfoCountry� s �z,StatusExtendedCymruInfo.testCymruInfoCountryc C s | � | j�| �� �| jg� d S rN )r rD ZgeBanListExtendedRIRrJ rC 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 ZnxdomainrK z8.0.0.0g <f�AZ3356ZUSZarinr )�level�key) r rD r r r rJ rL ZassertSortedEqualr, )r r( rI r r r �testCymruInfoNxdomain� s*
��
� �z-StatusExtendedCymruInfo.testCymruInfoNxdomain)
r: r; r< r
r rG rJ rM rO rP rQ rT r= r r r r r> � s
r> )�
__author__Z
__copyright__Z __license__r? Zutilsr r Zserver.banmanagerr Z
server.ticketr ZTestCaser r> r r r r �<module> s |