Current File : //usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyc
�
��&dc @ s� d Z d Z d Z d d l Z d d l m Z m Z d d l m Z d d l m
Z
d
e j f d � � YZ d e j f d
� � YZ
d S( s
Cyril Jaquiers Copyright (c) 2004 Cyril Jaquiert GPLi����Ni ( t setUpMyTimet tearDownMyTimei ( t
BanManager( t BanTickett
AddFailurec B sk e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z d
� Z RS( c C s< t t | � j � t � t d d � | _ t � | _ d S( s Call before every test case.s
193.168.0.128g �;f�AN( t superR t setUpR R t _AddFailure__ticketR t _AddFailure__banManager( t self( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyR # s c C s t t | � j � t � d S( s Call after every test case.N( R R t tearDownR ( R
( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyR * s c C s{ | j | j j | j � � | j | j j � d � | j | j j � d � | j j d � | j | j j � d � d S( Ni i ( t
assertTrueR t addBanTicketR t assertEqualt sizet getBanTotalt setBanTotal( R
( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testAdd/ s
c C sU | j | j j | j � � | j | j j | j � � | j | j j � d � d S( Ni ( R R R
R t assertFalseR R ( R
( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testAddDuplicate6 s c C sj | j j � } d } xNd d d d
f d f D]4\ } } t d d � } t d | � } | d k rt | j | � n | j | j j | � � | j | j j | � � | j | j j � d � | j j
| j � � } | j | j | � | j | � � | j | j | � | k � | j | � } | j | j
� d � | d k r. | j | j | � d � q. q. Wd S( Ni g �;f�Aid i i< i����s
193.168.0.128i ( g �;f�ANg �Tf�A( g �Tf�ANi� i�Q ( g �;f�Ai����( R t
getBanTimet NoneR t
setBanTimeR R
R R R t
getTicketByIDt getIDt getEndOfBanTimet getTime( R
t
defBanTimet prevEndOfBanTimet tnewt btnewt ticket1t ticket2t banticket( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testAddDuplicateWithTime; s* " c C sH | j | j j | j � � t d d � } | j | j j | � � d S( Ns
193.168.0.128g �;f�A( R R R
R R t
_inBanList( R
t ticket( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testInListOK[ s c C sH | j | j j | j � � t d d � } | j | j j | � � d S( Ns
111.111.1.111g �;f�A( R R R
R R R R$ ( R
R% ( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt
testInListNOK` s c C sp t | j j � | j j � � } d } x� d D]� } | j j | j � | d 7} | j | � | j | j j | � � | j t | j j
| j � � � d | j � | j � | | f � q. W| j j | j � | d 7} | j d � | j | j j | � � | j d � | j | j j | � � | j t | j j
| j � � � d | j � | j � d | f � d S( Ni i� i� i����i sF BanTicket: ip=%s time=%s bantime=%s bancount=%s #attempts=0 matches=[]( i� i� i����( R R t getIPR R R
R R R t strR ( R
R% t ct i( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testBanTimeIncre s" !
!'
!c C s8 | j j � } | j j � } | j | j j | j � � | j | j j | j � � | j | j j | � g � | j | j j | | d � | j g � | j | j j � d � | j | j j | j � � t
| j j � | d � } | j | j j | � � | j t
| j j | | d � � d � | j t
| j j | | d d � � d � x^ t d � D]P } t
d | | � } | j | j | � | d � | j | j j | � � qkW| j t
| j j | | d d � � d � | j t
| j j | | d
d � � d � | j | j j � d � d S( Ni i iX i s 193.168.0.%si
i i i
i2 ( R R R R R R
R$ R t unBanListR R R R t lent rangeR ( R
t btimet stimeR% R+ ( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testUnbanx s& * *. ..c C s� | j j � } | j j d � z� | j | j j | j � � | j | j j | j � � | j | j j | j j � | d � g � | j | j j
� d � Wd | j j | � Xd S( Ni����i ( R R R R R
R R$ R R- R R ( R
R0 ( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testUnbanPermanent� s - c C s t d d � t d d � g } | d j d � x | D] } | j j | � q6 W| j | j j d t d t � d d g � d S(
Ns 192.0.2.1g �;f�As 192.0.2.2i i����t orderedt withTimes: 192.0.2.1 2006-12-31 23:59:59 + 600 = 2007-01-01 00:09:59s9 192.0.2.2 2006-12-31 23:59:59 + -1 = 9999-12-31 23:59:59( R R R R
t assertSortedEqualt
getBanListt True( R
t ticketst t( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testBanList� s
(
t __name__t
__module__R R R R R# R& R' R, R2 R3 R; ( ( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyR " s t StatusExtendedCymruInfoc B s\ e Z d � Z d � Z e d f Z d � Z d � Z d � Z d � Z
d � Z d � Z RS( c C s� t t | � j � t j j � t � d | _ d | _ d | _ d | _
t | j d � } t � | _
| j | j
j | � � d S( s Call before every test case.s
93.184.216.34t 15133t EUt ripenccg �;f�AN( R R> R t unittestt F2Bt SkipIfNoNetworkR t _StatusExtendedCymruInfo__ban_ipt _StatusExtendedCymruInfo__asnt! _StatusExtendedCymruInfo__countryt _StatusExtendedCymruInfo__rirR R t$ _StatusExtendedCymruInfo__banManagerR R
( R
R% ( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyR � s
c C s t t | � j � t � d S( s Call after every test case.N( R R> R R ( R
( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyR � s c C s� t } | j d r= | j j d t j j r1 d n d � } n
| j d } | j d � r� t | f | _ t j d | d � � n | S( Ni t timeouti i i t errors. Skip test because service is not available: %s(
R> t availableRI t getBanListExtendedCymruInfoRB RC t fastt gett Falset SkipTest( R
t tct
cymru_info( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt _getBanListExtendedCymruInfo� s
c C sG | j � } | j | i | j g d 6| j g d 6| j g d 6� d S( Nt asnt countryt rir( RT t assertDictEqualRF RG RH ( R
RS ( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt
testCymruInfo� s
c C s, | j | j j | j � � | j g � d S( N( R RI t geBanListExtendedASNRT RF ( R
( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testCymruInfoASN� s c C s, | j | j j | j � � | j g � d S( N( R RI t geBanListExtendedCountryRT RG ( R
( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testCymruInfoCountry� s c C s, | j | j j | j � � | j g � d S( N( R RI t geBanListExtendedRIRRT RH ( R
( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testCymruInfoRIR� s c C s� t � | _ t d d � } | j | j j | � � | j � } | j | i d g d 6d g d 6d g d 6� t d d � } | j | j j | � � | j � } | j | i d d g d 6d d
g d 6d d g d 6d d
d t �d S( Ns 0.0.0.0g �;f�At nxdomainRU RV RW s 8.0.0.0g <f�At 3356t USt arint leveli����t key( R RI R R R
RT RX R6 R) ( R
R% RS ( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt testCymruInfoNxdomain� s
N(
R< R= R R R8 R RL RT RY R[ R] R_ Rf ( ( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyR> � s
( t
__author__t
__copyright__t __license__RB t utilsR R t server.banmanagerR t
server.ticketR t TestCaseR R> ( ( ( sE /usr/lib/python2.7/site-packages/fail2ban/tests/banmanagertestcase.pyt <module> s �
Mini Shell