CasperSecurity
o
� �_� � @ sJ d Z dZdZddlmZ ddlZddlmZmZm Z G dd � d ej
�Z dS )
zSerg G. Brester (sebres)z?Copyright (c) 2015 Serg G. Brester, 2015- Fail2Ban Contributors�GPL� )�MyTime� N)�Ticket�
FailTicket� BanTicketc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TicketTestsc C s: t �� }ddg}ddg}g d�}td||�}| �|�� d� | �|�� |� | �|�� |� |�d� | �|�� d� |� d� | �|�
� d� | �|� d�d� | � |�
|d d d�� | �|�
|d d d�� |�d� | �|� d�d� | �|� � d� | � |�
|�� | �|�
|d d �� |�d
� | � |�
|d d �� |�d� t �� }ddg}td||�}|�d� | �|�� d� | �|�� |� | �|�� |� |�d� |�d � | �|�� d� | �|�� d � |�d� | �|�� d� |�d � | �|�� d � |�� | �|�� d � | �|�� d � | �|�� |� |�d
gd d� | �|�� d � | �|�� d� | �|�� |� | �|�� |� |�|d d� | �|�� |� | �|�� d� |�|d d� | �|�� |d � | �|�� d� |�|d d� | �|�� |d � | �|�� d� |�|d d� | �|�� |d � | �|�� d� | �|�� d � |�dd� | �|�� g d�d dd�� t|d�}| �||� | �|�� |�� � | �|�� d � | �|�� d� | �|�� |� | �|�� |�� � | �|�� |�� � | �|� � |� � � d S )N�first�second)r r
�third�
193.168.0.128r �
i �< � ���� � r � i � �country�DE)�matches�failuresr ��ticket)r �timer � assertEqual�getIP�getTime�
getMatches�
setAttempt�
getAttempt� setBanCount� getBanCount�
getBanTime� assertFalse�
isTimedOut�
assertTrue�
setBanTimer �setRetry�getRetry�inc�
adjustTime�setData�getData)�self�tmr �matches2�matches3�t�ft�ft2� r6 �?/usr/lib/python3/dist-packages/fail2ban/tests/tickettestcase.py�
testTicket! s�
�
zTicketTests.testTicketc C s� d}t dd�}g }dD ],}|D ]'}t|||� |r |�|� n|�|� |D ]}| �tt||��||v � q&qq t|d�}|D ] }| �tt||��� q?d S )N)�restored�banned�testr )TFTr ) r �setattr�append�remover �bool�getattrr r' )r/ �flagsr � trueflags�v�f�f2r6 r6 r7 �testTicketFlagsw s"
��
�zTicketTests.testTicketFlagsc C s� t dd ddg�}|�dddddd� | �|�� ddgd dddd
�� |�d d d d
�� | �|�� d d d�� |�ddd d
�� | �|�� ddd�� | �|�d�d� | �|�d�d� |jddd� | �|�� dddd
�� |�dddd� | �|�d�d� | �|�d�d� | �|�� dddd�� | �|�d�ddd�� | �|�dd� �ddi� |�dd � | �|�� ddd�� | �|�dd�d� |�dd� |jg d�� | �|�� ddd d �� |�i � | �|�� i � | �|�d d �d � d S ) Nr r r
�region�Hamburgr r �cityr )r r rG r rI �FR�Paris)rG r rI )rI r )rG r �Bremen)rG rI � Brandenburg�Berlin)rI rG r c S s | � � dkS )N�COUNTRY)�upper)�kr6 r6 r7 �<lambda>� s z,TicketTests.testTicketData.<locals>.<lambda>�Unknown� continent�Europe)r �RUrG �MoscowrV rW )rT r rG �anything�default)r r- r r. )r/ r3 r6 r6 r7 �testTicketData� s� �����
���
�� � �
�
�
zTicketTests.testTicketDataN)�__name__�
__module__� __qualname__r8 rF rZ r6 r6 r6 r7 r s V r ) �
__author__�
__copyright__� __license__�
server.mytimer �unittest�
server.ticketr r r �TestCaser r6 r6 r6 r7 �<module> s