Current File : //lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyc
�
��&dc @ s! d Z d Z d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m
Z
d d l m Z d d l
m Z m Z m Z m Z m Z d d
l m Z d d l m Z d d l m
Z
d � Z d
e f d � � YZ d e
f d � � YZ d e
f d � � YZ d S( s Steven Hiscockss" Copyright (c) 2013 Steven Hiscockst GPLi����Ni ( t LogCaptureTestCasei ( t protocol( t asyncoret RequestHandlert loopt AsyncServert AsyncServerException( t Utils( t CSocketc G s t d � � d S( Ns test unpickle error( t Exception( t args( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt TestMsgError, s t TestMsgc B s# e Z e d f d � Z d � Z RS( c C s
| | _ d S( N( t unpickle( t selfR ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt __init__/ s c C s | j S( N( R ( R ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt
__reduce__1 s ( ( t __name__t
__module__R R R ( ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR
. s t Socketc B s} e Z d � Z d � Z e d � � Z e d � Z d � Z d � Z d � Z
d � Z d � Z d � Z
d
� Z d � Z RS( c C sw t j | � t t | � j � t | � | _ t j d d � \ } } t j | � t j
| � | | _ d | _
d S( s Call before every test case.s
fail2ban.socks
f2b-socketN( R t setUpt superR R t servert tempfilet mkstempt ost closet removet sock_namet Nonet serverThread( R t sock_fdR ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR 7 s
c C s4 | j r# | j j � | j � n t j | � d S( s Call after every test case.N( R R t stopt _stopServerThreadR t tearDown( R ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR# B s
c C s | S( s< Test transmitter proceed method which just returns first arg( ( t message( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt proceedI s c C sp t j d | j j d | j | f � | _ } t | _ | j � | j t j
| j j t j
j d � � � | S( Nt targetR i
( t threadingt ThreadR t startR R t Truet daemont
assertTrueR t wait_fort isActivet unittestt F2Bt maxWaitTime( R t forceR ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt _createServerThreadN s %
+c s; | j � t j � f d � t j j d � � d | _ d S( Nc s � j � p � j t j � S( N( t is_alivet joinR t DEFAULT_SLEEP_TIME( ( R ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt <lambda>Z s
i
( R R R- R/ R0 R1 R ( R ( ( R sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR" W s c C sy | j � } | j j � | j � | j | j � � | j j � | j | j j � � | j t j j
| j � � d S( N( R3 R R R" t assertFalseR4 R! R. R t patht existsR ( R R ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt testStopPerCloseUnexpected^ s
c C s, y t | j � SWn t k
r' } d SXd S( N( R R R
R ( R t e( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt
_serverSocketk s c C sL | j � } t j | j d � } d d d g } | j | j | � | � | j | j t � g g � d � | j d d d t �| j | j | � | � | j � | j � | j
j � t j | j d � } | j | j | � d g � | j
j � | j
� | j | j � � | j | j
j � � | j t j j | j � � d S(
Ni t At testR$ s ERROR: test unpickle errors! PROTO-error: load message failed:s test unpickle errort allt SHUTDOWN( R3 R R- R= t assertEqualt sendR
t assertLoggedR* R R t stop_communicationR! R" R8 R4 R. R R9 R: R ( R R t clientt testMessage( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt
testSocketq s" "
c s� | j � } t j | j d � � d d d t j j g g � t j } z/ d � t _ | j t
d � � f d � � Wd | t _ Xd S( Ni R> R? R$ c S s
| j � S( N( R ( R ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR7 � s s reset by peer|Broken pipec s � j � d t j j d � �S( Nt timeouti
( RC R/ R0 R1 ( ( RF RG ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR7 � s ( R3 R R- R= R t CSPROTOt ENDR t found_terminatort assertRaisesRegexpR
( R R t org_handler( ( RF RG sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt testSocketConnectBroken� s c C s� | j � } t j | j d � } d d d g } | j | j | � | � t j } z) d � t _ | j | j | � d � Wd | t _ X| j d d d t �| j
j � | j � | j
| j � � d S(
Ni R> R? R$ c S s t � S( N( R ( R ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR7 � s s ERROR: test unpickle errors Unexpected communication errors test unpickle errorR@ ( R3 R R- R= RB RC R RL RD R* R R! R" R8 R4 ( R R RF RG RN ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt testStopByCommunicate� s
c s� t j � i d d 6� � � f d � } zG | t _ | j � } | j t j � f d � t j j d � � � Wd � t _ X| j d d d t
�d S( Ni t cntrc s9 � d c d 7<� d d k r/ t d � � n � | � S( NRQ i i2 s test errors in poll( t RuntimeError( R ( t errt org_poll( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt _produce_error� s c s � d d k S( NRQ i2 ( ( ( RS ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR7 � s i
s1 Server connection was closed: test errors in polls0 Too many errors - stop logging connection errorsR@ ( R t pollR3 R, R R- R/ R0 R1 RD R* ( R RU R ( ( RS RT sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt testLoopErrors� s
2
c C s� t | j d � j � | j t | j j | j t � | j t � } | j j
� | j � | j | j
� � | j | j j � � | j t j j | j � � d S( Nt w( t openR R t assertRaisesR R R) t FalseR3 R* R! R" R8 R4 R. R R9 R: ( R R ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt testSocketForce� s
( R R R R# t staticmethodR% R[ R3 R" R; R= RH RO RP RW R\ ( ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR 5 s
! t
ClientMiscc B s e Z d � Z d � Z RS( c s~ i d d 6� � f d � } � f d � } t | d | �| j d d d d t �| j d
� | j d d d
d t �d S( Ni RQ c s � d d k S( NRQ i( ( ( ( t phase( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt _active� s c s( � d c d 7<t d � d � � d S( NRQ i s test *%d*( R
( R ( R_ ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt _poll� s t use_polls test *1*s test *10*s test *20*R@ s0 Too many errors - stop logging connection errorss test *21*s test *22*s test *23*( R RD R* t assertNotLogged( R R` Ra ( ( R_ sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt testErrorsInLoop� s
c C sG t j } t t j d � t _ z t j � t j � Wd | t _ Xd S( NRX ( t syst stdoutRY R t devnullR t printFormattedt printWiki( R t saved_stdout( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt testPrintFormattedAndWiki� s
( R R Rd Rk ( ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyR^ � s
( t
__author__t
__copyright__t __license__R Re R R' t timeR/ t utilsR t R t server.asyncserverR R R R R t server.utilsR t client.csocketR R t objectR
R R^ ( ( ( sA /usr/lib/python2.7/site-packages/fail2ban/tests/sockettestcase.pyt <module> s$ ( �
Mini Shell