Current File : //lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyc
�
��&dc @ s� d Z d Z d Z d d l Z d d l m 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 failmanager( t FailManagert FailManagerEmpty( t IPAddr( t
FailTickett
AddFailurec B s} e Z d � Z d � 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 � d | _ t � | _ d S( s Call before every test case.N( t superR t setUpt Nonet _AddFailure__itemsR t _AddFailure__failManager( t self( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyR % s c C s t t | � j � d S( s Call after every test case.N( R R t tearDown( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyR
+ s c C s� d d g d d g d d g d d g d d g d d g d d g d d g d d g d d g d d g d d g d d g g
| _ x2 | j D]' } | j j t | d
| d � � q� Wd S( Nu
193.168.0.128g �;f�As
87.142.124.10s
100.100.10.10g e��Ag �e��Ag �f��Ag �g��Ag �h��Ai i ( R
R t
addFailureR ( R t i( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt _addDefItems/ s c C sy | j � | j | j j � d � | j | j j � d � | j j d � | j | j j � d � | j j d � d S( Ni i
i ( R t assertEqualR t sizet getFailTotalt setFailTotal( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt testFailManagerAdd@ s
c C s� d } | | j _ d d d d g } d } xN | D]F } | d 8} t d d | | g � } | j d
� | j j | � q1 W| j j } | j t | � d
� | d } | j | j � t | � � | j t | j � � | � | j | j � | t | � | � t d d | � } | j t | � � | j j | � | j j } | j t | � d
� | d } | j | j � d t | � � | j t | j � � | � | j | j � | t | � | � | j j | � | j j } | j t | � d
� | d } | j | j � d t | � d
� | j t | j � � | � | j | j � | t | � | � d | j _ | j j | � | j j } | d } | j t | j � � d � | j
d � d S(
Ni s abc
s 123
s ABC
s 1234
iP i
s 127.0.0.1i�њ;i i i�њ;( R t
maxMatchesR t
setAttemptR t _FailManager__failListR t lent
getAttemptt
getMatchest
setMatchesR ( R R t failuresR t ft tickett manFailList( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt testFailManagerAdd_MaxMatchesH sJ
$
$
$ $
c C s` | j � | j | j j � d � | j j d � | j | j j � d � | j j d � d S( NiX i
( R R R t
getMaxTimet
setMaxTime( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt testFailManagerMaxTimez s
c C sG | j � | j j d � | j j d � | j | j j � d � d S( Ns
193.168.0.128s
111.111.1.111i ( R R t
delFailureR R ( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt testDel� s
c C s= | j � d } | j j | � | j | j j � d � d S( Ng �5f�Ai ( R R t cleanupR R ( R t timestamp( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt
testCleanupOK� s
c C s= | j � d } | j j | � | j | j j � d � d S( Ng �9f�Ai ( R R R' R R ( R R( ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt testCleanupNOK� s
c C s� | j � | j j d � | j j � } | j | j � d � | j t | j � t t f � � t | � } t
| � } | j | d � | j | d � | j | � | j d � | j | j
� d � | j t | � d � d S( Ni s
193.168.0.128s] FailTicket: ip=193.168.0.128 time=1167605999.0 bantime=None bancount=0 #attempts=5 matches=[]g �h��As] FailTicket: ip=193.168.0.128 time=1000002000.0 bantime=None bancount=0 #attempts=5 matches=[]( R R t setMaxRetryt toBanR t getIPt
assertTruet
isinstancet strR t reprt assertFalset setTimet getTime( R R t
ticket_strt ticket_repr( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt testbanOK� s&
"
c C s4 | j � | j j d � | j t | j j � d S( Ni
( R R R+ t assertRaisesR R, ( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt
testbanNOK� s
c C sn | j � | j j � } | j | j � d � | j j � } | j | j � d � | j t | j j � d S( Ns
100.100.10.10( R R R, t assertNotEqualR- R8 R ( R R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt
testWindow� s
c C s* | j j } t � } | j } | j t | � t | � k � d } | j | j t t � � | j | j � � x- t d | j
� D] } | j | j � � q� Wd | _ | j | j � � d | _ x- t d | j
� D] } | j | j � � q� W| j | j t t � � | j | j t t � � d S( Ni i �(
R t _FailManager__bgSvcR R. t idR t servicet TrueR2 t ranget _BgService__thresholdt _BgService__serviceTimet False( R t bgSvct failManager2ndt bgSvc2R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt
testBgService� s ( t __name__t
__module__R R
R R R! R$ R& R) R* R7 R9 R; RG ( ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyR # s 2 t FailmanagerComplexc B s2 e Z d � Z d � Z e d � � Z d � Z RS( c C s8 t t | � j � t � | _ t j | _ d t _ d S( s Call before every test case.i N( R RJ R R t _FailmanagerComplex__failManagerR t logLevelt _FailmanagerComplex__saved_ll( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyR � s c C s# t t | � j � | j t _ d S( N( R RJ R
RM R RL ( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyR
� s c c s� d t f d � � Y} d } d d d d g } xv | | k r� xH t d d d � D]4 } | | d k rz | | c d 7<Pn d | | <qP W| | | � f V| d 7} q1 Wd S( Nt _ipc B s, e Z d � Z d � Z d � Z d � Z RS( c S s d j t t | � � S( Nt .( t joint mapR0 ( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt __str__� s c S s
t | � S( N( R0 ( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt __repr__� s c S s
t | � S( N( R0 ( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt __key__� s c S s2 t | d d >| d d >B| d d >B| d B� S( Ni i i i i i i ( t int( R ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt __hash__� s ( RH RI RR RS RT RV ( ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyRN � s i i i i����i� i ( t listR@ ( t maxipsRN R t ct n( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt _ip_range� s
c C s� x� | j t j j s d n d � D]� \ } } | d k rS | j t | � d � q" | d k rx | j t | � d � q" | d k r� | j t | � d � q" | d k r� | j t | � d � q" | d
k r" | j t | � d � q" q" Wd S( Ni i� i� s 127.0.0.255i� s 127.0.1.0s 127.0.3.233i�� s
127.0.255.255i�� s 127.1.0.0( R[ t unittestt F2Bt fastR R0 ( R R t ip( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt testCheckIPGenerator� s . ( RH RI R R
t staticmethodR[ R` ( ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyRJ � s ( t
__author__t
__copyright__t __license__R\ t serverR t server.failmanagerR R t server.ipdnsR t
server.ticketR t TestCaseR RJ ( ( ( sF /usr/lib/python2.7/site-packages/fail2ban/tests/failmanagertestcase.pyt <module> s �
Mini Shell