Current File : //usr/lib/python2.7/site-packages/fail2ban/tests/utils.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 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 d d l m Z m Z d d
l m Z d d l m Z d d l m Z d d
l! m! Z! e e" � a# d Z$ e j% j& d d � Z( e( s�e j) j* e j) j+ d d � � r�d Z( q�d Z( n e j) j* e j) j+ e( d � � Z, e j- d e j) j. e j) j. e j) j. e j) j/ e0 � � � � � d e j1 f d � � YZ2 d d � Z3 d � Z4 d e2 f d � � YZ5 d � Z6 d � Z7 e8 e d � s�d e9 f d � � YZ: e: e _: e j; j< Z= d � Z< e< e j; _< n d � Z> d! � Z? e j% j& d" d � Z@ d# � ZA d$ � ZB d d d% � ZC d d lD ZD d d lE ZE e8 e jF d& � s0d d' � ZG eG e jF _G n d( eH eI d d) � ZJ eJ e jF _J e8 e jF d* � s{d+ � ZK eK e jF _K n eL r�d d, � ZM eM e jF _M d d- � ZN eN e jF _N n e jF jO ZP d. � ZQ e jF jR ZS d/ � ZT eQ e jF _O eT e jF _R d0 e jF f d1 � � YZU e jV ZV d S(2 s Yaroslav Halchenkos% Copyright (c) 2013 Yaroslav Halchenkot GPLi����N( t StringIO( t wrapsi ( t getLoggert str2LogLevelt getVerbosityFormatt
uni_decode( t IPAddrt DNSUtils( t MyTime( t Utils( t asyncserver( t versioni �Bt FAIL2BAN_CONFIG_DIRt configs
fail2ban.confs
/etc/fail2bant
PYTHONPATHt DefaultTestOptionsc B s e Z d � Z RS( c C sS i
d d 6d d 6t d 6d d 6d d 6t d 6t d 6t d 6t d 6t d
6| _ d S( Nt log_levelt verbosityt log_lazyt
log_tracebackt full_tracebackt fastt memory_dbt no_gamint
no_networkt negate_re( t Nonet Truet Falset __dict__( t self( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt __init__H s
( t __name__t
__module__R ( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR G s t c C s� t j } t j d d t j d | d d t � } | j | d d d d d
d d d �| d
d d d d d
d d d �| d d d d d d t d
d d d �| d d d d d d
t d d �| d d d d d d d d �| d d d d d d! d d" �| d# d$ d d d d% d d& �| d' d( d d d d) d d* �| d+ d, d d d d- d d. �| d/ d0 d d d d1 �| d2 d d d d3 �g � | S(4 Nt usages %s [OPTIONS] [regexps]
i R s %prog s -ls --log-levelt destR t defaultt helps4 Log level for the logger to use during running testss -vt actiont countR s Increase verbositys --verbosityt storet types' Set numerical level of verbosity (0..4)s --log-directt store_falseR s! Prevent lazy logging inside testss -ns --no-networkt
store_trueR s) Do not run tests that require the networks -gs
--no-gaminR s' Do not run tests that require the gamins -ms --memory-dbR s/ Run database tests using memory instead of files -fs --fastR sQ Try to increase speed of the tests, decreasing of wait intervals, memory databases -is --ignoreR sA negate [regexps] filter to ignore tests matched specified regexpss -ts --log-tracebacks. Enrich log-messages with compressed tracebackss --full-tracebacksB Either to make the tracebacks full, not compressed (as by default)(
t optparset Optiont OptionParsert syst argvR t add_optionsR t intR ( t docR/ t p( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt getOptParserS sN
c C s� t d � a d } | j d k r@ t | j � } t j | � n t j t j � t j | _ | j } | d k r| d k r� d no | t j
k r� d nZ | t j k r� d nE | t t j
t j � k r� d n$ | t t j t j � k r� d n d } | | _ n t j t j � } d } | j rNd d l m } | j rAd pDd
| } n t j } t | | � } | j | | � � t j | � | j d k r�d t t t j � j d d
� f GHn | S( Nt fail2bani i i i i s %(message)s( t FormatterWithTraceBacks %(tb)ss %(tbc)ss1 Fail2ban %s test suite. Python %s. Please wait...s
R# ( R t logSysR R R t setLevelt loggingt CRITICALt levelR t
HEAVYDEBUGt DEBUGt mint INFOt NOTICEt WARNINGt ERRORt
StreamHandlerR1 t stdoutR t helpersR9 R t FormatterR t setFormattert
addHandlerR t strt replace( t optst llevR RG t fmtRI ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt initProcess s: !!
$t F2Bc B sM e Z d Z d Z d � Z d � Z d � Z d � Z d � Z e d � Z
RS( i< i c C s; | j | _ | j r* t | _ t | _ n i | j d <d S( Nt share_config( R R R R R ( R RN ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR � s
c C s d S( N( ( R ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt
SkipIfFast� s c C s d S( N( ( R ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt SkipIfNoNetwork� s c K s� t s� | j d � r' t j d � � n x� d D]� } | j | � } | d
k rU q. n t j j | � d d k r~ | d 7} n t j j t j j t
| d | � � s. t j d | | f � � q. q. Wn d
S( s4 Helper to check action/filter config is available
t stocks/ Skip test because of missing stock-config filesR( t filteri R# s .confs .ds- Skip test because of missing %s-config for %rN( s actions filter( t STOCKt gett unittestt SkipTestR t ost patht splitextt existst joint
CONFIG_DIR( R t kwargst tt v( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt SkipIfCfgMissing� s
(c s � f d � } | S( s> Helper decorator to check action/filter config is available
c s" t � � � � f d � � } | S( Nc s t j j � � � | | | � S( N( RZ RR Re ( R t argsRb ( t decargst f( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt wrapper� s ( R ( Rh Ri ( Rg ( Rh s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt
_deco_wrapper� s ( ( R Rg Rj ( ( Rg s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt skip_if_cfg_missing� s c C sP t | t � r! | r! | j } n | j rL t | t � rL t | � d } n | S( Ni
( t
isinstancet boolt MAX_WAITTIMER R4 t float( R t wtime( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt maxWaitTime� s
( R! R" Rn t MID_WAITTIMER RT RU Re Rk R Rq ( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRR � s
c s t � � � f d � � } | S( s� Helper decorator to create a temporary directory
Directory gets removed after function returns, regardless
if exception was thrown of not
c s> t j d d � } z � | | | | � SWd t j | � Xd S( Nt prefixs f2b-temp( t tempfilet mkdtempt shutilt rmtree( R Rf Rb t tmp( Rh ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRi � s ( R ( Rh Ri ( ( Rh s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt with_tmpdir� s c s t � � � f d � � } | S( s@ Helper decorator to execute test in alternate (fixed) test time.c s* t � z � | | | � SWd t � Xd S( N( t setUpMyTimet tearDownMyTime( R Rf Rb ( Rh ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRi � s ( R ( Rh Ri ( ( Rh s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt
with_alt_time� s R[ c B s e Z RS( ( R! R" ( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR[ s c C sw | d t k rc | j r6 | j j t | d � � n) | j r_ | j j d � | j j � n d St | | | � d S( Ni i t s( R[ t showAllt streamt writelnRL t dotst writet flusht
_org_AddError( R t testt err( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt addError s c s� | s t t � � } n t | � t _ t j j rf d t _ d t _ d t _ d � } | t j _ n< d t _ d t _ d t _ t
j � � f d � } | t
_ t j j r� d � } | t j _
n d t j d
<t
j � t j t � t j } | j d d d
d' � xd t d � D]V } | j d | d � | j d | d � | j d | d � | j d | d � qW| j d d � | j d d � t j j r�| j d d � | j d d � | j d d � t j } x� d t � f d t � f d t � f d t � f d t � f d! t � f d" t t d# � t d$ � g � f d% t t d# � t d$ � g � f f D] } | j | � qfWt j j r�x0 t j � D] } | j | t j d& � � q�Wq�n d S(( Ng{�G�zd?g����Mb@?g-C ��6?c S s t j d � � d S( Ns Skip test because of "--fast"( RZ R[ ( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt F2B_SkipIfFast" s g�������?g{�G�zt?c s- | d k r t d | � � n � | � d S( Ng �?sT [BAD-CODE] To long sleep interval: %s, try to use conditional Utils.wait_for instead( t
ValueError( Rd ( t
_org_sleep( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt
_new_sleep, s c S s t j d � � d S( Ns# Skip test because of "--no-network"( RZ R[ ( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt F2B_SkipIfNoNetwork3 s s CET-01CEST-02,M3.5.0,M10.5.0t TZt maxCounti' t maxTimei i< i s
192.0.2.%ss
198.51.100.%ss 203.0.113.%ss 2001:db8::%ss test-hosts 2001:db8::ffffs
test-others
87.142.124.10s 192.0.2.888s 8.8.4.4s
dns.googles 999.999.999.999s
abcdef.abcdefs
192.168.0.s
failed.dns.chs! doh1.2.3.4.buga.xxxxx.yyy.invalids 1.2.3.4.buga.xxxxx.yyy.invalids example.coms" 2606:2800:220:1:248:1893:25c8:1946s
93.184.216.34s www.example.comt localhosti, ( RQ R RR RZ R R
t DEFAULT_SLEEP_TIMEt DEFAULT_SLEEP_INTERVALt DEFAULT_SHORT_INTERVALRT t timet sleepR RU R\ t environt tzsetR t setAlternateNowt TEST_NOWR t CACHE_ipToNamet
setOptionst xranget setR t CACHE_nameToIpR t getSelfNamest dnsToIp( RN R� R� R� t ct i( ( R� s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt initTests s`
!+ c C s d S( N( ( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt
mtimesleepe s R� c C s t j t � d S( N( R t setTimeR� ( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRz m s c C s
d t _ d S( N( R R t myTime( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR{ s s c s� t � � d d l m } d d l m } d d l m } d d l m } d d l m } d d l m } d d l m } d d l m } d d
l m
}
d d l m } d d l m } d d
l m
}
d d l m } d d l m } d d l m } d d l m } d d l m } � s/t j � } n. d t j f � � � f d � � Y� � � } | j t j | j � � | j t j | j � � | j t j | j � � | j t j | j � � | j t j | j � � | j t j |
j � � | j t j | j � � | j t j | j � � | j t j | j � � | j t j | j � � | j t j | j � � y) d d l! } | j t j | j" � � Wn t# k
r�n X| j t j | j$ � � | j t j | j% � � | j t j | j& � � | j t j | j' � � | j t j | j( � � | j t j | j) � � | j t j | j* � � | j t j | j+ � � | j t j |
j, � � | j t j |
j- � � | j t j |
j. � � | j t j |
j/ � � | j t j | j0 � � | j t j | j1 � � | j t j | j2 � � | j t j | j3 � � | j t j | j4 � � | j t j | j5 � � | j t j | j6 � � | j t j | j7 � � | j t j | j8 � � | j t j | j9 � � | j t j | j: � � | j t j | j; � � | j t j | j< � � | j t j | j � � | j t j | j= � � | j t j | j> � � | j t j | j? � � | j t j | j@ � � | j t j | jA � � | j t j | jB � � t jC � } d d l mD } x� tE jF tE jG jH tE jG jI | jJ � � � D]Z } | jK d � r| jL d � r| j | jM d | jN tE jG jO | � d f � � qqWd d lP mQ } | g } y^ t jR jS s�t jR jT r�t# d t jR jS t jR jT f � � n d d lU mV } | jW | � Wn$ t# k
r} tX jY d | � n Xy! d d lZ m[ } | jW | � Wn$ t# k
rW} tX jY d! | � n Xx- | D]% } | j t j | j\ | � � � q_Wy3 d d" l] m^ } | j t j | j_ | � � � Wn$ t# k
r�} tX jY d# | � n X| j t j | j` � � | S($ Ni ( t banmanagertestcase( t clientbeautifiertestcase( t clientreadertestcase( t tickettestcase( t failmanagertestcase( t filtertestcase( t servertestcase( t datedetectortestcase( t actiontestcase( t actionstestcase( t sockettestcase( t misctestcase( t databasetestcase( t observertestcase( t samplestestcase( t fail2banclienttestcase( t fail2banregextestcaset FilteredTestSuitec s< e Z g � D] Z e j e � ^ q
Z � � f d � Z RS( c s� g } x� | D]� } t | t j � r8 | j | � q
n t | � } xG | j D]< } | j | � } � j sr | n | rN | j | � PqN qN Wq
Wx$ | D] } t � | � j | � q� Wd S( N(
Rl RZ t TestSuitet addTestRL t _regexpst searchR t appendt super( R t suitet matchedR� R} t rt m( R� RN ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR� � s
( R! R" R� t ret compileR� R� ( ( R� RN t regexps( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR� � s "i����( t action_dt test_s .pys %s.%si i ( t
FilterPolls Skip, fast: %s, no_gamin: %s( t FilterGamins2 Skipping gamin backend testing. Got exception '%s'( t FilterPyinotifys9 I: Skipping pyinotify backend testing. Got exception '%s'( t
FilterSystemds7 I: Skipping systemd backend testing. Got exception '%s'(a R� R# R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� RZ R� R� t makeSuitet Transmittert JailTestst
RegexTestst LoggingTestst ServerConfigReaderTestst CommandActionTestt ExecuteActionst TicketTestst
AddFailuret FailmanagerComplext dnst StatusExtendedCymruInfot ImportErrort BeautifierTestt ConfigReaderTestt JailReaderTestt FilterReaderTestt JailsReaderTestt JailsReaderTestCachet Sockett
ClientMisct HelpersTestt SetupTestt TestsUtilsTestt
MyTimeTestt DatabaseTestt ObserverTestt BanTimeIncrt
BanTimeIncrDBt IgnoreIPt BasicFiltert LogFilet LogFileMonitort LogFileFilterPollt IgnoreIPDNSt GetFailurest
DNSUtilsTestst DNSUtilsNetworkTestst DateDetectorTestt CustomDateFormatsTestt FilterSamplesRegext Fail2banClientTestt Fail2banServerTestt Fail2banRegexTestt
TestLoaderR� R\ t listdirR] t abspatht dirnamet __file__t
startswitht endswitht loadTestsFromNameR! R^ t server.filterpollR� RR R R t server.filtergaminR� R� R: t warningt server.filterpyinotifyR� t get_monitor_failures_testcaset server.filtersystemdR� t% get_monitor_failures_journal_testcaset TransmitterLogging( R� RN R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� t testsR� t
testloaderR� t file_R� t filtersR� t eR� t Filter_R� ( ( R� RN R� s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt gatherTestsw s�
%
( . %
#t assertDictEqualc C s� | j t | t � d � | j t | t � d � | | k r� d | | f } d d j t j t j | � j � t j | � j � � � } | p� | | } | j | � n d S( Ns" First argument is not a dictionarys# Second argument is not a dictionarys %r != %rs
(
t assert_Rl t dictR` t difflibt ndifft pprintt pformatt
splitlinest fail( R t d1t d2t msgt standardMsgt diff( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR ' s i c
s� d � � | r t � n | � f d � � � � � f d � � y � | | | | | � Wn� t k
r� } t | t � r� | j d n
t | � d } d d j t j t j
| � j � t j
| � j � � � } | p� | | } | j | � n Xd S( s� Compare complex elements (like dict, list or tuple) in sorted order until
level 0 not reached (initial level = -1 meant all levels),
or if nestedOnly set to True and some of the objects still contains nested lists or dicts.
c S s= t | t � r) t d � | j � D� � St d � | D� � S( Nc s s' | ] } t | t t t f � Vq d S( N( Rl R t listt tuple( t .0Rd ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pys <genexpr>; s c s s' | ] } t | t t t f � Vq d S( N( Rl R R R ( R Rd ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pys <genexpr>< s ( Rl R t anyt
itervalues( Rd ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt
_is_nested9 s c sE t | t t t f � rA t t � � f d � | D� � d � �S| S( Nc 3 s | ] } � | � � Vq d S( N( ( R Rd ( t _nest_sortedt key( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pys <genexpr>B s R% ( Rl R� R R t sorted( Rd R% ( R$ ( R% s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR$ @ s )c s t | � t | � k r1 t d | | f � � n | r� | r� � | � r� � | � r� | | k rh d St d | | f � � n t | t � rMt | t � rMxs| j � D]� \ } } | | } t | t t t f � r!t | t t t f � r!� | | | d k r| d n d | | � q� | | k r� t d | | f � � q� q� Wn� � | d | �} � | d | �} x� t | | � D]� \ } } t | t t t f � r�t | t t t f � r�� | | | d k r�| d n d | | � q�| | k r�t d | | f � � q�q�Wd S( Ns %r != %ri i R% ( t lenR� Rl R t iteritemsR R t zip( t at bR> t
nestedOnlyR% t kt v1t v2( t _assertSortedEqualR# R$ ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR0 E s( '
0/ 0/ i s
within:s
N(
R& t ExceptionRl R� Rf RL R` R R R R R R (
R R* R+ R> R, R% R R
R R ( ( R0 R# R$ s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt assertSortedEqual3 s , t assertRaisesRegexpc O s} y | | | � WnK | k
r^ } t j | t | � � d k ry | j d | | f � qy n X| j d t | d � � d S( Ns "%s" does not match "%s"s
%s not raisedR! ( R� R� RL R R t getattr( R t excclst regexpt funRf Rb R
( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR3 j s c C s� | } t } | d k rU t | d � rU t | t � rU t j | � \ } } t } n | | k r� | rv t | � } n | p� d | | f } | j | � n d S( Nt __iter__s %r was not found in %r(
R R t hasattrRl t
basestringt itertoolst teeR R R ( R R* R+ R t bbt wrap( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt assertInv s + c C s� | } t } | d k rU t | d � rU t | t � rU t j | � \ } } t } n | | k r� | rv t | � } n | p� d | | f } | j | � n d S( NR8 s %r unexpectedly found in %r(
R R R9 Rl R: R; R< R R R ( R R* R+ R R= R> ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt assertNotIn� s + c C st t j j t j k rB t j j d � t j d d | j
� � n t | � t j j d k rp t
j
� | _ n d S(
Ns
t =i
s %s i i s
==========s ========== %s s ====================( RZ RR R R< R@ R1 t stderrR� R: t debugt idt
_org_setUpR R� t __startTime( R ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt _customSetUp� s
c C s= t j j d k r9 t j j d t j � | j f � n d S( Ni s
%.3fs -- ( RZ RR R R1 RB R� R� RF ( R ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt _customTearDown� s t LogCaptureTestCasec B sx e Z d e j f d � � YZ d � Z d � Z d � Z d � Z d � Z d
d � Z d � Z e
e j d � � Z RS( t _MemHandlerc B sJ e Z d Z e d � Z d d � Z d � Z d � Z d � Z d � Z
RS( s� Logging handler helper
Affords not to delegate logging to StreamHandler at all,
format lazily on demand in getvalue.
Increases performance inside the LogCaptureTestCase tests, because there
the log level set to DEBUG.
c C sk t j � | _ d | _ d | _ t � | _ d | _ t � | _ t
j j | � | rg | j
| _ n d S( NR# i ( t threadingt Lockt _lockt _valt _dirtyR t _recst
_nolckCntrR t _strmR< t HandlerR t _handle_lazyt handle( R t lazy( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyR � s c C s[ | r t d | � � n d | _ | j �* d | _ t � | _ | j j d � Wd QXd S( s) Truncate the internal buffer and records.s. invalid size argument: %r, should be None or 0R# i N( R1 RN RM RO R RP RR t truncate( R t size( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRW � s
c C s� yX | j � d } y | j j | � Wn- t k
rV | j j | j d d � � n XWn' t k
r� } | j j d | � n Xd S( Ns
s UTF-8RM s Error by logging handler: %r( t
getMessageRR R� t UnicodeEncodeErrort encodeR1 ( R t recordR R
( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt __write� s
$c C s | j s | j S| j j t � } | j d @r� | ss | j d 7_ | j d k rZ | j Sd | _ | j j � n z( | j d M_ | j } t � | _ Wd | j j � XxC | D] } | j | � q� Wn% | r� | j d M_ | j j � n | j
j � | _ | j S( s& Return current buffer as whole string.i i i i i Ni����i����( RO RN RM t acquireR RQ RP R t releaset _MemHandler__writeRR t getvalue( R t lckt recsR\ ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRa � s,
c C s0 | j | � | j � | j d O_ Wd QXd S( s- Handle the specified record direct (not lazy)i N( R` RM RO ( R R\ ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRU � s
c C s3 | j �$ | j j | � | j d O_ Wd QXd S( s* Lazy handle the specified record on demandi N( RM RP R� RO ( R R\ ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRT � s
N( R! R" t __doc__R R R RW R` Ra RU RT ( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRJ � s
" c C s� t j | _ t j | _ t j t j j � | _
| j
g t _ | j t j k rf t j | j 7_ n | j t j d k r� t j
t j d � n t t | � j � d S( Ni ( R: R> t
_old_levelt handlerst
_old_handlersRI RJ RZ RR R t _logR< R@ R; R� t setUp( R ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRi � s c C s9 | j � | j t _ | j t _ t t | � j � d S( s Call after every test case.N( t pruneLogRg R: Rf Re R> R� RI t tearDown( R ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRk s
c O sx | j j � } | j d t � sO x | D] } | | k r( t Sq( Wt rt t Sn% x | D] } | | k rV t SqV Wt Sd S( Nt all( Rh Ra RY R R ( R R} Rb t loggedt s_( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt
_is_logged s
c s � j d d � } | rN t j j | � } t j � � � f d � | � } n � j � � � } � j d t � s� | s� j j
� } � j d � | r� d | n d | f � qnb | s� j j
� } xJ � D]? } | | k r� � j d | | rd | n d | f � q� q� Wn d S( sk Assert that one of the strings was logged
Preferable to assertTrue(self._is_logged(..)))
since provides message with the actual log.
Parameters
----------
s : string or list/set/tuple of strings
Test should succeed if string (or any of the listed) is present in the log
all : boolean (default False) if True should fail if any of s not logged
t waitc s � j � � � S( N( Ro ( ( Rb R} R ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt <lambda>2 s Rl s/ None among %r was found in the log%s: ===
%s===s , waited %sR# s( %r was not found in the log%s: ===
%s===N( RY R RZ RR Rq R
t wait_forRo R Rh Ra R ( R R} Rb Rp t resRm Rn ( ( Rb R} R s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt assertLogged# s $ $
c O s� | j j � } t | � d k ro | j d t � ro x | D] } | | k r; d Sq; W| j d | | f � n7 x4 | D], } | | k rv | j d | | f � qv qv Wd S( s Assert that strings were not logged
Parameters
----------
s : string or list/set/tuple of strings
Test should succeed if the string (or at least one of the listed) is not
present in the log
all : boolean (default False) if True should fail if any of s logged
i Rl Ns6 All of the %r were found present in the log: ===
%s===s" %r was found in the log: ===
%s===( Rh Ra R' RY R R ( R R} Rb Rm Rn ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt assertNotLoggedD s
%
c C s1 | j j d � | r- t j d d | � n d S( Ni RA i s %s s =====s ===== %s s =====( Rh RW R: RC ( R t logphase( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRj Y s c C s
| j j � S( N( Rh Ra ( R ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt getLog^ s c C s~ | t j k s$ t j � t j k rz | d | d � x0 t j | � D] } | j d � } | | � qF W| d d � n d S( s@ Helper which outputs content of the file at HEAVYDEBUG loglevelss ---- s ----s
t -i N( R: RC t getEffectiveLevelR< R@ t fileinputt inputt rstrip( t fnRU t line( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt dumpFilea s $N( R! R" R< RS RJ Ri Rk Ro Rt Ru R Rj Rw t staticmethodR: RC R ( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyRI � s X ! (W t
__author__t
__copyright__t __license__Rz R; R< R. R\ R� Rt Rv R1 R� RK RZ t cStringIOR t functoolsR RH R R R R t server.ipdnsR R t
server.mytimeR t server.utilsR
t serverR R R! R: R� R� RY R Ra R] R_ R` RX t putenvR� R� R� t ValuesR R7 RQ RR Ry R| R9 R1 R[ t _TextTestResultR� R� R� R� t old_TZRz R{ R R R t TestCaseR R t reprR2 R3 R R? R@ Ri RE RG Rk t
_org_tearDownRH RI t
pid_exists( ( ( s8 /usr/lib/python2.7/site-packages/fail2ban/tests/utils.pyt <module> s� " ! !$ , 82
O �
4
�
Mini Shell