/usr/lib/python3/dist-packages/fail2ban/tests/__pycache__
Edit: /usr/lib/python3/dist-packages/fail2ban/tests/__pycache__/sockettestcase.cpython-310.pyc (7782B)
o
;s*bˆ ã @ 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ƒ‚)Nútest unpickle error©Ú Exception©Úargs© r ú?/usr/lib/python3/dist-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 sz e Zd Z‡ fdd„Zdd„ Zedd„ ƒZddd „Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Z‡ ZS )ÚSocketc sV t | ¡ tt| ƒ ¡ t| ƒ| _t dd¡\}}t |¡ t
|¡ || _ d| _ dS )z Call before every test case.z
fail2ban.sockz
f2b-socketN)
r ÚsetUpÚsuperr! r
ÚserverÚtempfileÚmkstempÚosÚcloseÚremoveÚ sock_nameÚ serverThread)r Úsock_fdr* ©Ú __class__r r r" 7 s
z Socket.setUpc C s& | j r | j ¡ | ¡ t | ¡ dS )zCall after every test case.N)r+ r$ ÚstopÚ_stopServerThreadr ÚtearDownr r r r r1 B s
zSocket.tearDownc C s | S )z
Z s
z*Socket._stopServerThread..r5 )r+ r r; r= r> r? r r rE r r0 W s
ÿ
zSocket._stopServerThreadc C sZ | ¡ }| j ¡ | ¡ | | ¡ ¡ | j ¡ | | j ¡ ¡ | tj
| j ¡¡ d S r ) rA r$ r( r0 Ú assertFalserB r/ r< r' ÚpathÚexistsr* ©r r+ r r r ÚtestStopPerCloseUnexpected^ s
z!Socket.testStopPerCloseUnexpectedc
C s4 zt | jƒW S ty } zW Y d }~d S d }~ww r )r
r* r )r Úer r r Ú
_serverSocketk 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Útestr2 úERROR: test unpickle errorz!PROTO-error: load message failed:r T©ÚallÚSHUTDOWN)rA r r; rM Ú assertEqualÚsendr Ú assertLoggedr( r$ Ústop_communicationr/ r0 rG rB r< r' rH rI r* )r r+ ÚclientÚ testMessager r r Ú
testSocketq s"
zSocket.testSocketc sf | ¡ }t | jd¡‰ dddtjjgg‰tj}zdd„ t_| t
d‡ ‡fdd„¡ W |t_d S |t_w ) Nr rO rP r2 c S s | ¡ S r )r( r r r r rF › s z0Socket.testSocketConnectBroken..zreset by peer|Broken pipec s ˆ j ˆtj d¡dS )Nr5 )Útimeout)rV r= r> r? r ©rY rZ r r rF s ) rA r r; rM r ÚCSPROTOÚENDr Úfound_terminatorÚassertRaisesRegexr )r r+ Ú org_handlerr r] r ÚtestSocketConnectBroken’ s
ÿz Socket.testSocketConnectBrokenc C sš | ¡ }t | jd¡}g d¢}| | |¡|¡ tj}zdd„ t_| | |¡d¡ W |t_n|t_w | jdddd | j
¡ | ¡ | |
¡ ¡ d S )
Nr rN c S s t ƒ S r )r r r r r rF « s z.Socket.testStopByCommunicate..rQ z Unexpected communication errorr TrR )rA r r; rM rU rV r r` rW r$ r/ r0 rG rB )r r+ rY rZ rb r r r ÚtestStopByCommunicate¡ s
z Socket.testStopByCommunicatec st t j‰ddi‰ ‡ ‡fdd„}z |t _| ¡ }| t ‡ fdd„tj d¡¡¡ W ˆt _nˆt _w | j dd d
d d S ) NÚcntrr c s, ˆ d d7 < ˆ d dk rt dƒ‚ˆ| Ž S )Nre r é2 ztest errors in poll)Ú RuntimeErrorr ©ÚerrÚorg_pollr r Ú_produce_error½ s z-Socket.testLoopErrors.._produce_errorc s ˆ d dkS )Nre rf r r )ri r r rF Ç s z'Socket.testLoopErrors..r5 z1Server connection was closed: test errors in pollú0Too many errors - stop logging connection errorsTrR )
r ÚpollrA r: r r; r= r> r? rW )r rk r+ r rh r ÚtestLoopErrors¹ s $
ÿzSocket.testLoopErrorsc C sx t | jdƒ ¡ | t| jj| jd¡ | d¡}| j ¡ | ¡ |
| ¡ ¡ |
| j ¡ ¡ |
t
j | j¡¡ d S )NÚwFT)Úopenr* r( Ú assertRaisesr r$ r8 rA r/ r0 rG rB r< r' rH rI rJ r r r ÚtestSocketForceÏ s ÿ
zSocket.testSocketForce)F)r r r r" r1 Ú staticmethodr3 rA r0 rK rM r[ rc rd rn rr Ú
__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 )Nre r c s ˆ d dk S )Nre é( r r ©Úphaser r Ú_activeä s z,ClientMisc.testErrorsInLoop.._activec s ˆ d d7 < t dˆ d ƒ‚)Nre r z test *%d*r r rw r r Ú_pollæ s z*ClientMisc.testErrorsInLoop.._poll)Úuse_pollztest *1*z test *10*z test *20*TrR rl z test *21*z test *22*z test *23*)r rW ÚassertNotLogged)r ry rz r rw r ÚtestErrorsInLoopâ s
zClientMisc.testErrorsInLoopc C s: t j}ttjdƒt _zt ¡ t ¡ W |t _d S |t _w )Nro )ÚsysÚstdoutrp r' Údevnullr ÚprintFormattedÚ printWiki)r Ú saved_stdoutr r r ÚtestPrintFormattedAndWikiï s
z$ClientMisc.testPrintFormattedAndWikiN)r r r r} r„ r r r r ru à s
ru ) Ú
__author__Ú
__copyright__Ú __license__r' r~ r% r6 Útimer= Úutilsr Ú r Úserver.asyncserverr r r r
r Ú server.utilsr Úclient.csocketr
r Úobjectr r! ru r r r r Ú s( ,