File: //lib/python3/dist-packages/fail2ban/tests/__pycache__/failmanagertestcase.cpython-310.pyc
o
;s*b
# � @ 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__� �D/usr/lib/python3/dist-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 �� q-d 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 �size� getFailTotal� setFailTotalr r r r �testFailManagerAdd@ s z AddFailure.testFailManagerAddc C s d}|| j _g d�}d}|D ]}|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 �
setAttemptr �_FailManager__failListr �len�
getAttempt�
getMatches�
setMatches)r r& �failuresr �f�ticket� 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 �
getMaxTime�
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 �
delFailurer r r r r r �testDel� s zAddFailure.testDelc C �. | � � d}| j�|� | �| j�� d� d S )Ng �5f�Ar �r r �cleanupr r �r � timestampr r r �
testCleanupOK� � zAddFailure.testCleanupOKc C r7 )Ng �9f�Ar r8 r: r r r �testCleanupNOK� r= 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 �getIP�
assertTrue�
isinstance�strr �repr� assertFalse�setTime�getTime)r r/ �
ticket_str� ticket_reprr r r � testbanOK� s,
��
�zAddFailure.testbanOKc C s( | � � | j�d� | �t| jj� d S )Nr% )r r r@ � assertRaisesr rA r r r r �
testbanNOK� s zAddFailure.testbanNOKc C sP | � � | j�� }| �|�� d� | j�� }| �|�� d� | �t| jj� d S )Nr )r r rA �assertNotEqualrB rM r )r r/ r r r �
testWindow� s
zAddFailure.testWindowc C s� | j j}t� }|j}| �t|�t|�k� d }| �|�dd�� | �|�� � td|j�D ] }| �|�� � q-d|_ | �|�� � d|_ td|j�D ] }| �|�� � qJ| �|�dd�� | �|�dd�� d S )NTr i �F)
r �_FailManager__bgSvcr rC �id�servicerG �range�_BgService__threshold�_BgService__serviceTime)r �bgSvc�failManager2nd�bgSvc2r r r r �
testBgService� s zAddFailure.testBgService)�__name__�
__module__� __qualname__r r r r# r1 r4 r6 r< r> rL rN rP rZ �
__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 r_ r r � _FailmanagerComplex__failManagerr �logLevel� _FailmanagerComplex__saved_llr r r r r � s
zFailmanagerComplex.setUpc s t t| ��� | jt_d S �N)r r_ r rb r ra r r r r r � s zFailmanagerComplex.tearDownc c s� � G dd� dt �}d}g d�}|| k rAtddd�D ]}|| dk r+|| d7 < nd||< q|||�fV |d7 }|| k sd S 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�maprE r r r r �__str__� s z1FailmanagerComplex._ip_range.<locals>._ip.__str__c S � t | �S rc �rE r r r r �__repr__� � z2FailmanagerComplex._ip_range.<locals>._ip.__repr__c S rh rc ri r r r r �__key__� rk 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 0z2FailmanagerComplex._ip_range.<locals>._ip.__hash__N)r[ r\ r] rg rj rl rq r r r r �_ip� s
rr r )� r r r r ���� r )�listrT )�maxipsrr r �c�nr r r � _ip_range� s �
�z FailmanagerComplex._ip_rangec C s� | � tjjsdnd�D ]D\}}|dkr | �t|�d� q |dkr)| �t|�d� q |dkr6| �t|�d� q |dkrC| �t|�d � q |d
krO| �t|�d � q d S ) Ni i� � z 127.0.0.255ru z 127.0.1.0z 127.0.3.233i�� z
127.0.255.255i�� z 127.1.0.0)rz �unittest�F2B�fastr rE )r r �ipr r r �testCheckIPGenerator� s ��z'FailmanagerComplex.testCheckIPGenerator) r[ r\ r] r r � staticmethodrz r� r^ r r r r r_ � s
r_ )�
__author__�
__copyright__� __license__r| �serverr �server.failmanagerr r � server.ipdnsr �
server.ticketr �TestCaser r_ r r r r �<module> s 2