File manager - Edit - /usr/lib/python3/dist-packages/fail2ban/tests/action_d/__pycache__/test_badips.cpython-310.pyc
Back
o ;s*b � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ ddl m Z ddl m Z ddl mZ dd lmZmZ ejd krFd d lmZmZ nd d lmZmZ d d � ZejdkraG dd� de�ZdS dS )� N��wraps��timeout)�SSLError� )� CallingMap)� DummyJail)�IPAddr)�LogCaptureTestCase� CONFIG_DIR)� )� HTTPError�URLErrorc s t � �� fdd��}|S )zYHelper to decorate tests to skip in case of timeout/http-errors like "502 bad gateway". c sx z � | g|�R � W S t tttfy; } z"t|t�s0dt|�vr0t|d�r/|jdkr0|jdkr0� t� d| ��d }~ww )Nz timed out�code�� i� zSkip test because of %s) r r r r � isinstance�str�hasattrr �unittest�SkipTest)�self�args�e��f� �E/usr/lib/python3/dist-packages/fail2ban/tests/action_d/test_badips.py�wrapper( s ��z&skip_if_not_available.<locals>.wrapperr )r r r r r �skip_if_not_available% s r )r � c s� e Zd ZdZdZdZe� fdd��Z� fdd�Zedd� �Z ed d � �Z ed d � �Z ed d� �Z edd� �Z edd� �Z� ZS )�BadIPsActionTest)TNNc s� t t| ��� tj�� t� | _| jj� d� t j � t dd�}tjd rLtjs>tjs2| jj�|�t_tjjt_| jj�|� tjjtjjrGdndd�t_tjd s[t�dtjd ��| jjj d |d dd d tjjrkdndd�d� | jjd | _dS )z Call before every test case.�testzaction.dz badips.pyr r � r z.Skip test because service is not available: %s� �badips�ssh�2w� r �< )�category� banaction�age�scorer )�initOptsN)�superr! �setUpr �F2B�SkipIfNoNetworkr �jail�actions�add�os�path�joinr � available� modAction� pythonModule�_load_python_module�Action� isAvailable�fastr �action)r �pythonModuleName�� __class__r r r0 : s, � zBadIPsActionTest.setUpc s& | j jr | j j�� tt| ��� dS )zCall after every test case.N)r@ �_timer�cancelr/ r! �tearDown�r rB r r rF \ s zBadIPsActionTest.tearDownc C s\ | j �� }| �d|� | �t|�dk� | �tt| j dd� | �tt| j dd� d| j _d S )Nr& � r* zinvalid-category�mail) r@ � getCategories�assertIn� assertTrue�len� assertRaises� ValueError�setattr� bancategory)r � categoriesr r r � testCategoryc s � � z BadIPsActionTest.testCategoryc C s( | � tt| jdd� d| j_d| j_d S )Nr- ���r �3)rN rO rP r@ r- rG r r r � testScores s zBadIPsActionTest.testScorec C s | � tt| jdd� d| j_d S )Nr+ zinvalid-actionr" )rN rO rP r@ r+ rG r r r � testBanactiony s � z BadIPsActionTest.testBanactionc C s< | � tt| jdd� | � tt| jdd� d| j_d| j_d S )N� updateperiodi���r i� �900)rN rO rP r@ rX rG r r r �testUpdateperiod� s � � z!BadIPsActionTest.testUpdateperiodc C sZ | j �� | �t| j j�dkdt| j j�| j jf � | j �� | �t| j j�dk� d S )NrH z%s is fewer as 10: %rr )r@ �startrL rM � _bannedips�stoprG r r r � testStartStop� s � z BadIPsActionTest.testStartStopc C sn t dtd�i�}| j�|� | jddd� | �� d| j_d|d< | �tj j | jj|� | jdd dd d � d S ) N�ipz 192.0.2.1zbadips.com: banT)�waitz0f2b-this-category-dont-available-test-suite-only� z IP is invalidzinvalid categoryF)r` �all) r r r@ �ban� assertLogged�pruneLog� _categoryrN r! r; r )r �aInfor r r � testBanIP� s � zBadIPsActionTest.testBanIP)�__name__� __module__� __qualname__r9 r; r: r r0 rF rS rV rW rZ r^ rh � __classcell__r r rB r r! 4 s& ! r! )r6 r �sys� functoolsr �socketr �sslr �actiontestcaser � dummyjailr �servertestcaser �utilsr r � version_info� urllib.errorr r r r! r r r r �<module> s" �
| ver. 1.4 |
Github
|
.
| PHP 8.2.9 | Generation time: 0.02 |
proxy
|
phpinfo
|
Settings