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
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
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
46 def test_redis_retention(self
):
47 print self
.conf
.modules
49 mod
= redis_retention_scheduler
.Redis_retention_scheduler(modconf
, 'localhost')
51 sl
= get_instance(mod
)
55 sl
.properties
['to_queue'] = None
59 #updte the hosts and service in the scheduler in the retentino-file
60 sl
.hook_save_retention(self
.sched
)
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__':