test/test_module_host_perfdata.py

   1 #!/usr/bin/env python2.6
   2 #Copyright (C) 2009-2010 :
   3 #    Gabes Jean, naparuba@gmail.com
   4 #    Gerhard Lausser, Gerhard.Lausser@consol.de
   5 #
   6 #This file is part of Shinken.
   7 #
   8 #Shinken is free software: you can redistribute it and/or modify
   9 #it under the terms of the GNU Affero General Public License as published by
  10 #the Free Software Foundation, either version 3 of the License, or
  11 #(at your option) any later version.
  12 #
  13 #Shinken is distributed in the hope that it will be useful,
  14 #but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 #GNU Affero General Public License for more details.
  17 #
  18 #You should have received a copy of the GNU Affero General Public License
  19 #along with Shinken.  If not, see <http://www.gnu.org/licenses/>.
  20
  21
  22 #
  23 # This file is used to test reading and processing of config files
  24 #
  25
  26 #It's ugly I know....
  27 import os
  28 from shinken_test import *
  29 sys.path.append("../shinken/modules")
  30 from host_perfdata_broker import *
  31 from shinken.brok import Brok
  32
  33 class TestConfig(ShinkenTest):
  34     #setUp is in shinken_test
  35
  36     #Change ME :)
  37     def test_host_perfdata(self):
  38         print self.conf.modules
  39         #get our modules
  40         mod = None
  41         for m in self.conf.modules:
  42             if m.module_type == 'host_perfdata':
  43                 mod = m
  44         self.assert_(mod is not None)
  45         self.assert_(mod.path == 'tmp/host-perfdata')
  46         self.assert_(mod.module_name == 'Host-Perfdata')
  47         self.assert_(mod.mode == 'a')
  48         #Warning, the r (raw) is important here
  49         print "Tempalte%stoto" % mod.template
  50 #        self.assert_(mod.template == r'$LASTHOSTCHECK\t$HOSTNAME$\t$HOSTOUTPUT$\t$HOSTSTATE$\t$HOSTPERFDATA$\n')
  51
  52         try :
  53             os.unlink(mod.path)
  54         except :
  55             pass
  56
  57         sl = get_instance(mod)
  58         print sl
  59         #Hack here :(
  60         sl.properties = {}
  61         sl.properties['to_queue'] = None
  62         sl.init()
  63         svc = self.sched.hosts.find_by_name("test_host_0")
  64         t = int(time.time())
  65         print "T", t
  66         self.scheduler_loop(1, [[svc, 2, 'BAD | value1=0 value2=0']])
  67         #manage all host check result broks
  68         for b in self.sched.broks.values():
  69             if b.type == 'host_check_result':
  70                 sl.manage_brok(b)
  71         self.sched.broks = {}
  72         sl.file.close() # the sl also has an open file handle
  73
  74         fd = open(mod.path)
  75         buf = fd.readline()
  76         print "BUF:", buf
  77         comparison = '%d\t%s\t%s\t%s\t%s\n' % (t, "test_host_0", 'BAD ', 'DOWN', ' value1=0 value2=0')
  78         print "Comparison:", comparison
  79         self.assert_(buf == comparison)
  80         fd.close()
  81         os.unlink(mod.path)
  82
  83         #Now change with a new template
  84         #and direct in the instance (do not do this in prod :) )
  85         mod.template = '$LASTHOSTCHECK$\t$HOSTNAME$\t$HOSTOUTPUT$\t$HOSTPERFDATA$\t$HOSTSTATE$\n'
  86         sl2 = get_instance(mod)
  87         sl2.init()
  88         print sl2.__dict__
  89         t = int(time.time())
  90         print "T", t
  91         self.scheduler_loop(1, [[svc, 2, 'BAD | value1=0 value2=0']])
  92         #manage all host check result broks
  93         for b in self.sched.broks.values():
  94             if b.type == 'host_check_result':
  95                 sl2.manage_brok(b)
  96         sl2.file.close()
  97
  98         fd = open(mod.path)
  99         buf = fd.readline()
 100         print "BUF:", buf
 101         comparison = '%d\t%s\t%s\t%s\t%s\n' % (t, "test_host_0", 'BAD ', ' value1=0 value2=0', 'DOWN')
 102         print "Comparison:", comparison
 103         self.assert_(buf == comparison)
 104         fd.close()
 105         os.unlink(mod.path)
 106
 107
 108 if __name__ == '__main__':
 109     unittest.main()
 110