File: //lib/python3.9/site-packages/fail2ban/tests/__pycache__/banmanagertestcase.cpython-39.opt-2.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 JaquierZGPL� 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) r/ 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 r5 r% r )r r8 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+ r0 r9 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
rB 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
rB 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) rB � availablerL ZgetBanListExtendedCymruInforE rF Zfast�getZSkipTest)r Ztc�
cymru_infor r r � _getBanListExtendedCymruInfo� s
�
z4StatusExtendedCymruInfo._getBanListExtendedCymruInfoc C s, | � � }| �|| jg| jg| jgd�� d S )N�ZasnZcountryZrir)rS �assertDictEqualrI rJ rK )r rR r r r �
testCymruInfo� s ��z%StatusExtendedCymruInfo.testCymruInfoc C s | � | j�| �� �| jg� d S �N)r rL ZgeBanListExtendedASNrS rI r r r r �testCymruInfoASN� s �z(StatusExtendedCymruInfo.testCymruInfoASNc C s | � | j�| �� �| jg� d S rW )r rL ZgeBanListExtendedCountryrS rJ r r r r �testCymruInfoCountry� s �z,StatusExtendedCymruInfo.testCymruInfoCountryc C s | � | j�| �� �| jg� d S rW )r rL ZgeBanListExtendedRIRrS rK 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 ZnxdomainrT z8.0.0.0g <f�AZ3356rC rD r )�level�key) r rL r r r rS rU r; r- )r r) rR r r r �testCymruInfoNxdomain� s(
��
��z-StatusExtendedCymruInfo.testCymruInfoNxdomain)
r> r? r@ r r rP rS rV rX rY rZ r] rA r r r r rB � s
rB )�
__author__Z
__copyright__Z __license__rE Zutilsr r Zserver.banmanagerr Z server.ipdnsr Z
server.ticketr ZTestCaser rB r r r r �<module> s