File: //lib/python3.9/site-packages/fail2ban/tests/__pycache__/sockettestcase.cpython-39.opt-2.pyc
a
<J��i� @ s� d Z dZdZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z dd l
mZmZmZmZmZ dd
lmZ dd lmZ ddl m
Z
d d
� ZG dd� de�ZG dd� de
�ZG dd� de
�ZdS )zSteven Hiscocksz"Copyright (c) 2013 Steven HiscocksZGPL� N� )�LogCaptureTestCase� )�protocol)�asyncore�RequestHandler�loop� AsyncServer�AsyncServerException)�Utils)�CSocketc G s t d��d S )N�test unpickle error�� Exception��args� r �A/usr/lib/python3.9/site-packages/fail2ban/tests/sockettestcase.py� TestMsgError, s r c @ s$ e Zd Zedffdd�Zdd� ZdS )�TestMsgr c C s
|| _ d S �N��unpickle)�selfr r r r �__init__/ s zTestMsg.__init__c C s | j S r r �r r r r �
__reduce__1 s zTestMsg.__reduce__N)�__name__�
__module__� __qualname__r r r r r r r r . s r c s~ e Zd Z� fdd�Z� fdd�Zedd� �Zddd �Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Z� ZS )�Socketc sL t t| ��� t| �| _t�dd�\}}t�|� t� |� || _
d| _ dS )z Call before every test case.z
fail2ban.sockz
f2b-socketN) �superr �setUpr �server�tempfile�mkstemp�os�close�remove� sock_name� serverThread)r Zsock_fdr) �� __class__r r r" 7 s
z Socket.setUpc s* | j r| j�� | �� tt| ��� dS )zCall after every test case.N)r* r# �stop�_stopServerThreadr! r �tearDownr r+ r r r/ A s
zSocket.tearDownc C s | S )z<Test transmitter proceed method which just returns first argr )�messager r r �proceedH s zSocket.proceedFc C sN t j| jj| j|fd� | _}d|_|�� | �t� | jj
t j �
d��� |S )N)�targetr T�
)� threading�Threadr# �startr) r* �daemon�
assertTruer �wait_for�isActive�unittest�F2B� maxWaitTime)r �forcer* r r r �_createServerThreadM s � zSocket._createServerThreadc s, | j � t�� fdd�tj�d�� d | _ d S )Nc s � � � p� �tj�S r )�is_alive�joinr ZDEFAULT_SLEEP_TIMEr �r* r r �<lambda>Y s
z*Socket._stopServerThread.<locals>.<lambda>r3 )r* r r9 r; r< r= r r rB r r. V s
�zSocket._stopServerThreadc C sZ | � � }| j�� | �� | �|�� � | j�� | �| j�� � | �tj �
| j �� d S r ) r? r# r' r. � assertFalser@ r- r: r& �path�existsr) �r r* r r r �testStopPerCloseUnexpected] s
z!Socket.testStopPerCloseUnexpectedc
C s8 z t | j�W S ty2 } zW Y d }~d S d }~0 0 d S r )r r) r )r �er r r �
_serverSocketj s zSocket._serverSocketc C s� | � � }t�| jd�}g d�}| �|�|�|� | �|�t� gg�d� | jdddd� | �|�|�|� |�� |�� | j �
� t�| jd�}| �|�|�dg� | j � � | � � | �
|�� � | �
| j �� � | �
tj�| j�� d S ) Nr ��A�testr0 �ERROR: test unpickle errorz!PROTO-error: load message failed:r
T��allZSHUTDOWN)r? r r9 rJ � assertEqual�sendr � assertLoggedr' r# Zstop_communicationr- r. rD r@ r: r& rE rF r) )r r* �client� testMessager r r �
testSocketp s"
zSocket.testSocketc sh | � � }t�| jd�� dddtjjgg�tj}z,dd� t_| � t
d� �fdd�� W |t_n|t_0 d S ) Nr rL rM r0 c S s | � � S r )r' r r r r rC � � z0Socket.testSocketConnectBroken.<locals>.<lambda>zreset by peer|Broken pipec s � j �tj�d�d�S )Nr3 )�timeout)rR r; r< r= r �rT rU r r rC � rW ) r? r r9 rJ r ZCSPROTOZENDr �found_terminatorZassertRaisesRegexr )r r* � org_handlerr rY r �testSocketConnectBroken� s
�z Socket.testSocketConnectBrokenc C s� | � � }t�| jd�}g d�}| �|�|�|� tj}z&dd� t_| �|�|�d� W |t_n|t_0 | jdddd � | j �
� | � � | � |�
� � d S )
Nr rK c S s t � S r )r r r r r rC � rW z.Socket.testStopByCommunicate.<locals>.<lambda>rN z Unexpected communication errorr
TrO )r? r r9 rJ rQ rR r rZ rS r# r- r. rD r@ )r r* rT rU r[ r r r �testStopByCommunicate� s
z Socket.testStopByCommunicatec st t j�ddi� � �fdd�}z:|t _| �� }| �t�� fdd�tj�d��� W �t _n�t _0 | j dd d
d � d S ) N�cntrr c s, � d d7 < � d dk r$t d���| � S )Nr^ r �2 ztest errors in poll)� RuntimeErrorr ��errZorg_pollr r �_produce_error� s z-Socket.testLoopErrors.<locals>._produce_errorc s � d dkS )Nr^ r_ r r )rb r r rC � rW z'Socket.testLoopErrors.<locals>.<lambda>r3 z1Server connection was closed: test errors in poll�0Too many errors - stop logging connection errorsTrO )
r �pollr? r8 r r9 r; r< r= rS )r rc r* r ra r �testLoopErrors� s $�zSocket.testLoopErrorsc C sx t | jd��� | �t| jj| jd� | �d�}| j�� | � � | �
|� � � | �
| j� � � | �
t
j�| j�� d S )N�wFT)�openr) r' Z assertRaisesr
r# r6 r? r- r. rD r@ r: r& rE rF rG r r r �testSocketForce� s �
zSocket.testSocketForce)F)r r r r" r/ � staticmethodr1 r? r. rH rJ rV r\ r] rf ri �
__classcell__r r r+ r r 5 s
!r c @ s e Zd Zdd� Zdd� ZdS )�
ClientMiscc s^ ddi� � fdd�}� fdd�}t ||d� | jdd d
d d � | �d
� | jdddd d � d S )Nr^ r c s � d dk S )Nr^ �( r r �Zphaser r �_active� s z,ClientMisc.testErrorsInLoop.<locals>._activec s$ � d d7 < t d� d ��d S )Nr^ r z test *%d*r r rn r r �_poll� s z*ClientMisc.testErrorsInLoop.<locals>._poll)Zuse_pollztest *1*z test *10*z test *20*TrO rd z test *21*z test *22*z test *23*)r rS ZassertNotLogged)r ro rp r rn r �testErrorsInLoop� s
zClientMisc.testErrorsInLoopc C s< t j}ttjd�t _zt�� t�� W |t _n|t _0 d S )Nrg )�sys�stdoutrh r& �devnullr ZprintFormattedZ printWiki)r Z saved_stdoutr r r �testPrintFormattedAndWiki� s
z$ClientMisc.testPrintFormattedAndWikiN)r r r rq ru r r r r rl � s
rl ) �
__author__Z
__copyright__Z __license__r& rr r$ r4 �timer; Zutilsr � r Zserver.asyncserverr r r r r
Z server.utilsr Zclient.csocketr r �objectr r rl r r r r �<module> s& +