File: //usr/lib/python3.9/site-packages/fail2ban/tests/__pycache__/failmanagertestcase.cpython-39.pyc
a
���f��� @ sp d Z dZdZddlZddlmZ ddlmZmZ ddl m
Z
dd l m Z G d
d � d ej
�ZG d d
� d
ej
�ZdS )z
Cyril Jaquierz Copyright (c) 2004 Cyril Jaquier�GPL� N� )� failmanager)� FailManager�FailManagerEmpty)�IPAddr)�
FailTicketc s� e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Z� ZS )�
AddFailurec s t t| ��� d| _t� | _dS )� Call before every test case.N)�superr �setUp�_AddFailure__itemsr �_AddFailure__failManager��self�� __class__� �F/usr/lib/python3.9/site-packages/fail2ban/tests/failmanagertestcase.pyr % s zAddFailure.setUpc s t t| ��� dS )zCall after every test case.N)r r �tearDownr r r r r + s zAddFailure.tearDownc C s~ ddgddgddgddgddgddgddgddgddgddgddgddgdd gg
| _ | j D ] }| j�t|d
|d �� qZd S ) N�
193.168.0.128g �;f�Az
87.142.124.10�
100.100.10.10g e��Ag �e��Ag �f��Ag �g��A� �h��Ar � )r
r �
addFailurer )r �ir r r � _addDefItems/ s �
zAddFailure._addDefItemsc C sZ | � � | �| j�� d� | �| j�� d� | j�d� | �| j�� d� | j�d� d S )N� �
r )r � assertEqualr �sizeZ getFailTotalZ setFailTotalr r r r �testFailManagerAdd@ s z AddFailure.testFailManagerAddc C s d}|| j _g d�}d}|D ]4}|d8 }tdd| |g�}|�d� | j �|� q | j j}| �t|�d� |d }| �|�� t|�� | �t|� � �|� | �|� � |t|�| d � � tdd|�}|�t|�� | j �|� | j j}| �t|�d� |d }| �|�� dt|� � | �t|� � �|� | �|� � |t|�| d � � | j �|� | j j}| �t|�d� |d }| �|�� dt|� d � | �t|� � �|� | �|� � |t|�| d � � d | j _| j �|� | j j}|d }| �t|� � �d � |�
d � d S )
Nr )zabc
z123
zABC
z1234
�P �
z 127.0.0.1i�њ;r i�њ;r ) r �
maxMatchesr Z
setAttemptr Z_FailManager__failListr �lenZ
getAttemptZ
getMatchesZ
setMatches)r r$ Zfailuresr �f�ticketZ manFailListr r r � testFailManagerAdd_MaxMatchesH sJ
z(AddFailure.testFailManagerAdd_MaxMatchesc C sH | � � | �| j�� d� | j�d� | �| j�� d� | j�d� d S )NiX r )r r r Z
getMaxTimeZ
setMaxTimer r r r �testFailManagerMaxTimez s
z!AddFailure.testFailManagerMaxTimec C s6 | � � | j�d� | j�d� | �| j�� d� d S )Nr z
111.111.1.111r )r r Z
delFailurer r r r r r �testDel� s zAddFailure.testDelc C s. | � � d}| j�|� | �| j�� d� d S )Ng �5f�Ar �r r �cleanupr r �r � timestampr r r �
testCleanupOK� s zAddFailure.testCleanupOKc C s. | � � d}| j�|� | �| j�� d� d S )Ng �9f�Ar r+ r- r r r �testCleanupNOK� s zAddFailure.testCleanupNOKc C s� | � � | j�d� | j�� }| �|�� d� | �t|�� tt f�� t|�}t
|�}| �|d� | �|d� | � | � |� d� | �|�
� d� | �t|�d� d S )N� r z]FailTicket: ip=193.168.0.128 time=1167605999.0 bantime=None bancount=0 #attempts=5 matches=[]r z]FailTicket: ip=193.168.0.128 time=1000002000.0 bantime=None bancount=0 #attempts=5 matches=[])r r � setMaxRetry�toBanr �getID�
assertTrue�
isinstance�strr �repr� assertFalseZsetTimeZgetTime)r r' Z
ticket_strZ ticket_reprr r r � testbanOK� s,
��
�zAddFailure.testbanOKc C s( | � � | j�d� | �t| jj� d S )Nr# )r r r2 � assertRaisesr r3 r r r r �
testbanNOK� s zAddFailure.testbanNOKc C sP | � � | j�� }| �|�� d� | j�� }| �|�� d� | �t| jj� d S )Nr )r r r3 ZassertNotEqualr4 r; r )r r' r r r �
testWindow� s
zAddFailure.testWindowc C s� | j j}t� }|j}| �t|�t|�k� d }| �|�dd�� | �|�� � td|j�D ]}| �|�� � qZd|_ | �|�� � d|_ td|j�D ]}| �|�� � q�| �|�dd�� | �|�dd�� d S )NTr i �F)
r Z_FailManager__bgSvcr r5 �idZservicer9 �rangeZ_BgService__thresholdZ_BgService__serviceTime)r ZbgSvcZfailManager2ndZbgSvc2r r r r �
testBgService� s zAddFailure.testBgService)�__name__�
__module__� __qualname__r r r r! r( r) r* r/ r0 r: r< r= r@ �
__classcell__r r r r r # s 2r c s< e Zd Z� fdd�Z� fdd�Zedd� �Zdd� Z� ZS ) �FailmanagerComplexc s( t t| ��� t� | _tj| _dt_dS )r
r N)r rE r r Z _FailmanagerComplex__failManagerr �logLevel� _FailmanagerComplex__saved_llr r r r r � s zFailmanagerComplex.setUpc s t t| ��� | jt_d S �N)r rE r rG r rF r r r r r � s zFailmanagerComplex.tearDownc c sz G dd� dt �}d}g d�}|| k rvtddd�D ],}|| dk rT|| d7 < q^d||< q0|||�fV |d7 }q d S ) Nc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
z)FailmanagerComplex._ip_range.<locals>._ipc S s d� tt| ��S )N�.)�join�mapr7 r r r r �__str__� s z1FailmanagerComplex._ip_range.<locals>._ip.__str__c S s t | �S rH �r7 r r r r �__repr__� s z2FailmanagerComplex._ip_range.<locals>._ip.__repr__c S s t | �S rH rM r r r r �__key__� s z1FailmanagerComplex._ip_range.<locals>._ip.__key__c S s0 t | d d> | d d> B | d d> B | d B �S )Nr � r � r � r )�intr r r r �__hash__� s z2FailmanagerComplex._ip_range.<locals>._ip.__hash__N)rA rB rC rL rN rO rT r r r r �_ip� s rU r )� r r r r ���� r )�listr? )ZmaxipsrU r �c�nr r r � _ip_range� s
z FailmanagerComplex._ip_rangec C s� | � tjjsdnd�D ]�\}}|dkr8| �t|�d� q|dkrR| �t|�d� q|dkrl| �t|�d� q|dkr�| �t|�d � q|d
kr| �t|�d � qd S ) Ni i� � z 127.0.0.255rX z 127.0.1.0z 127.0.3.233i�� z
127.0.255.255i�� z 127.1.0.0)r\ �unittestZF2B�fastr r7 )r r �ipr r r �testCheckIPGenerator� s z'FailmanagerComplex.testCheckIPGenerator) rA rB rC r r � staticmethodr\ ra rD r r r r rE � s
rE )�
__author__�
__copyright__� __license__r^ �serverr Zserver.failmanagerr r Z server.ipdnsr Z
server.ticketr ZTestCaser rE r r r r �<module> s 2