Fix : fix hot module under windows test. (at leat I hope...)
[shinken.git] / test / test_module_redis_retention.py
blob82c8fe0f48db994704dbc659192814b516428cfb
1 #!/usr/bin/env python2.6
2 #Copyright (C) 2009-2010 :
3 # Gabes Jean, naparuba@gmail.com
4 # Gerhard Lausser, Gerhard.Lausser@consol.de
6 #This file is part of Shinken.
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.
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.
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/>.
23 # This file is used to test reading and processing of config files
26 import os
28 from shinken_test import unittest, ShinkenTest
30 from shinken.log import logger
31 from shinken.objects.module import Module
32 from shinken.modules import redis_retention_scheduler
33 from shinken.modules.redis_retention_scheduler import get_instance
36 modconf = Module()
37 modconf.module_name = "RedisRetention"
38 modconf.module_type = redis_retention_scheduler.properties['type']
39 modconf.properties = redis_retention_scheduler.properties.copy()
42 class TestConfig(ShinkenTest):
43 #setUp is in shinken_test
45 #Change ME :)
46 def test_redis_retention(self):
47 print self.conf.modules
48 #get our modules
49 mod = redis_retention_scheduler.Redis_retention_scheduler(modconf, 'localhost')
51 sl = get_instance(mod)
52 print "Instance", sl
53 #Hack here :(
54 sl.properties = {}
55 sl.properties['to_queue'] = None
56 sl.init()
57 l = logger
59 #updte the hosts and service in the scheduler in the retentino-file
60 sl.hook_save_retention(self.sched)
62 #Now we change thing
63 svc = self.sched.hosts.find_by_name("test_host_0")
64 self.assert_(svc.state == 'PENDING')
65 print "State", svc.state
66 svc.state = 'UP' #was PENDING in the save time
68 r = sl.hook_load_retention(self.sched)
69 self.assert_(r == True)
71 #search if the host is not changed by the loading thing
72 svc2 = self.sched.hosts.find_by_name("test_host_0")
73 self.assert_(svc == svc2)
75 self.assert_(svc.state == 'PENDING')
77 # Now make real loops with notifications
78 self.scheduler_loop(10, [[svc, 2, 'CRITICAL | bibi=99%']])
79 #updte the hosts and service in the scheduler in the retentino-file
80 sl.hook_save_retention(self.sched)
82 r = sl.hook_load_retention(self.sched)
83 self.assert_(r == True)
87 if __name__ == '__main__':
88 unittest.main()