File: //lib/python3/dist-packages/fail2ban/tests/__pycache__/failmanagertestcase.cpython-38.pyc
U
]^
# � @ 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 JaquierZGPL� 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 �� 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 _ddddg}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
Zcleanupr r �r Z 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 �getIP�
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
r/ � assertRaisesr r0 r r r r �
testbanNOK� s zAddFailure.testbanNOKc C sP | � � | j�� }| �|�� d� | j�� }| �|�� d� | �t| jj� d S )Nr )r r
r0 ZassertNotEqualr1 r8 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 r2 �id�servicer6 �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, r- r7 r9 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
rC r r Z _FailmanagerComplex__failManagerr �logLevel� _FailmanagerComplex__saved_llr r r r r � s zFailmanagerComplex.setUpc s t t| ��� | jt_d S �N)r
rC r rE r rD r r r r r � s zFailmanagerComplex.tearDownc c s~ G dd� dt �}d}ddddg}|| k rztddd�D ],}|| dk rX|| d7 < qbd||< q4|||�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�mapr4 r r r r �__str__� s z1FailmanagerComplex._ip_range.<locals>._ip.__str__c S s t | �S rF �r4 r r r r �__repr__� s z2FailmanagerComplex._ip_range.<locals>._ip.__repr__c S s t | �S rF rK 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)r? r@ rA rJ rL rM rR r r r r �_ip� s rS r � r ���� r )�listr= )ZmaxipsrS 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.255rV z 127.0.1.0z 127.0.3.233i�� z
127.0.255.255i�� z 127.1.0.0)rZ �unittestZF2BZfastr r4 )r r Zipr r r �testCheckIPGenerator� s z'FailmanagerComplex.testCheckIPGenerator) r? r@ rA r r � staticmethodrZ r] rB r r r r rC � s
rC )�
__author__Z
__copyright__Z __license__r\ Zserverr Zserver.failmanagerr r Z server.ipdnsr Z
server.ticketr ZTestCaser rC r r r r �<module> s 2