File: //usr/lib/python3.9/site-packages/fail2ban/tests/__pycache__/sockettestcase.cpython-39.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 Hiscocks�GPL� 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 r0 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>r4 )r+ r r: r< r= r> r r rC r r/ V s
�zSocket._stopServerThreadc C sZ | � � }| j�� | �� | �|�� � | j�� | �| j�� � | �tj �
| j �� d S r ) r@ r$ r( r/ � assertFalserA 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�testr1 �ERROR: test unpickle errorz!PROTO-error: load message failed:r T��allZSHUTDOWN)r@ r r: rK � assertEqual�sendr � assertLoggedr( r$ Zstop_communicationr. r/ rE rA r; r' rF rG 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 rM rN r1 c S s | � � S r )r( r r r r rD � � z0Socket.testSocketConnectBroken.<locals>.<lambda>zreset by peer|Broken pipec s � j �tj�d�d�S )Nr4 )�timeout)rS r< r= r> r �rU rV r r rD � rX ) r@ r r: rK r ZCSPROTOZENDr �found_terminatorZassertRaisesRegexr )r r+ � org_handlerr rZ 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 rL c S s t � S r )r r r r r rD � rX z.Socket.testStopByCommunicate.<locals>.<lambda>rO z Unexpected communication errorr TrP )r@ r r: rK rR rS r r[ rT r$ r. r/ rE rA )r r+ rU rV 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 )rc r r rD � rX z'Socket.testLoopErrors.<locals>.<lambda>r4 z1Server connection was closed: test errors in poll�0Too many errors - stop logging connection errorsTrP )
r �pollr@ r9 r r: r< r= r> rT )r rd r+ r rb 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$ r7 r@ r. r/ rE rA r; r' rF rG rH r r r �testSocketForce� s �
zSocket.testSocketForce)F)r r r r# r0 � staticmethodr2 r@ r/ rI rK rW r] r^ rg rj �
__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 ro r r �_poll� s z*ClientMisc.testErrorsInLoop.<locals>._poll)Zuse_pollztest *1*z test *10*z test *20*TrP re z test *21*z test *22*z test *23*)r rT ZassertNotLogged)r rp rq r ro r �testErrorsInLoop� s
zClientMisc.testErrorsInLoopc C s< t j}ttjd�t _zt�� t�� W |t _n|t _0 d S )Nrh )�sys�stdoutri r' �devnullr ZprintFormattedZ printWiki)r Z saved_stdoutr r r �testPrintFormattedAndWiki� s
z$ClientMisc.testPrintFormattedAndWikiN)r r r rr rv r r r r rm � s
rm ) �
__author__�
__copyright__� __license__r' rs r% r5 �timer< �utilsr � r Zserver.asyncserverr r r r
r Z server.utilsr Zclient.csocketr
r �objectr r! rm r r r r �<module> s& +