Current File : //lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyc
�
��&dc @ s
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 d d l m
Z
d d l m Z d d l
m Z d d l m Z m Z e j d k r� d d l m Z m Z n d d l m Z m Z d � Z e j d k r d e f d � � YZ n d S( i����N( t wraps( t timeout( t SSLErrori ( t
CallingMap( t DummyJail( t IPAddr( t LogCaptureTestCaset
CONFIG_DIRi ( t HTTPErrort URLErrorc s t � � � f d � � } | S( sY Helper to decorate tests to skip in case of timeout/http-errors like "502 bad gateway".
c s� y � | | � SWn� t t t t f k
r� } t | t � r� d t | � k r� t | d � s | j d k r� | j d k r� � q� n t j d | � � n Xd S( Ns timed outt codei� i� s Skip test because of %s(
R R R R t
isinstancet strt hasattrR
t unittestt SkipTest( t selft argst e( t f( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyt wrapper( s ". ( R ( R R ( ( R sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyt skip_if_not_available% s i t BadIPsActionTestc B s� e Z e d f Z d Z d Z e d � � Z d � Z e d � � Z
e d � � Z e d � � Z e d � � Z
e d � � Z e d � � Z RS( c C su t t | � j � t j j � t � | _ | j j j d � t
j j t
d d � } t j d r� t j s� t j s� | j j j | � t _ n t j j t _ | j j j | � n t j j d t j j r� d n d � t _ n t j d s
t j d t j d � � n | j j j d
| d i d d
6d d 6d d 6d d 6t j j rSd n d d 6�| j j d
| _ d S( s Call before every test case.t tests action.ds badips.pyi R i i s. Skip test because service is not available: %si t badipst initOptst ssht categoryt banactiont 2wt agei t scorei i< N( t superR t setUpR t F2Bt SkipIfNoNetworkR t jailt actionst addt ost patht joinR t availablet modActiont pythonModulet _load_python_modulet Actiont isAvailablet fastR t action( R t pythonModuleName( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyR! : s*
-
c C s6 | j j r | j j j � n t t | � j � d S( s Call after every test case.N( R1 t _timert cancelR R t tearDown( R ( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyR5 \ s c C s� | j j � } | j d | � | j t | � d k � | j t t | j d d � | j t t | j d d � d | j _ d S( NR i
R s invalid-categoryt mail( R1 t
getCategoriest assertInt
assertTruet lent assertRaisest
ValueErrort setattrt bancategory( R t
categories( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyt testCategoryc s c C s8 | j t t | j d d � d | j _ d | j _ d S( NR i����i t 3( R; R< R= R1 R ( R ( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyt testScores s c C s, | j t t | j d d � d | j _ d S( NR s invalid-actionR ( R; R< R= R1 R ( R ( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyt
testBanactiony s c C sT | j t t | j d d � | j t t | j d d � d | j _ d | j _ d S( Nt updateperiodi���i i� t 900( R; R< R= R1 RD ( R ( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyt testUpdateperiod� s c C s{ | j j � | j t | j j � d k d t | j j � | j j f � | j j � | j t | j j � d k � d S( Ni
s %s is fewer as 10: %ri ( R1 t startR9 R: t
_bannedipst stop( R ( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyt
testStartStop� s
#
c C s� t i t d � d 6� } | j j | � | j d d t �| j � d | j _ d | d <| j t j
j | j j | � | j d d d t d t �d S(
Ns 192.0.2.1t ips badips.com: bant waits0 f2b-this-category-dont-available-test-suite-onlyt s
IP is invalids invalid categoryt all(
R R R1 t bant assertLoggedt Truet pruneLogt _categoryR; R R, R t False( R t aInfo( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyt testBanIP� s
N( t __name__t
__module__RQ t NoneR* R, R+ R R! R5 R@ RB RC RF RJ RV ( ( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyR 4 s " ( i ( i i ( R' R t syst functoolsR t socketR t sslR t actiontestcaseR t dummyjailR t servertestcaseR t utilsR R t version_infot urllib.errorR R t urllib2R R ( ( ( sG /usr/lib/python2.7/site-packages/fail2ban/tests/action_d/test_badips.pyt <module> s
Mini Shell