class GeneratorBaseTestCase(TestCase):
"""Classe de base pour les tests de génération"""
def setUp(self):
setup_db()
conf.hostsConf = {}
# Prepare temporary directory
self.tmpdir = setup_tmpdir()
self.basedir = os.path.join(self.tmpdir, "deploy")
self.old_conf_path = settings["vigiconf"]["confdir"]
settings["vigiconf"]["confdir"] = os.path.join(self.tmpdir, "conf.d")
os.mkdir(settings["vigiconf"]["confdir"])
self.testfactory = TestFactory(confdir=settings["vigiconf"]["confdir"])
self.host = Host(conf.hostsConf, "dummy.xml", "testserver1",
"192.168.1.1", "Servers")
self.host.set_attribute("collectorTimeout", "3")
# attention, le fichier dummy.xml doit exister ou l'hôte sera supprimé
# juste après avoir été inséré
open(os.path.join(self.tmpdir, "conf.d", "dummy.xml"), "w").close()
conffile = ConfFile.get_or_create("dummy.xml")
add_host("testserver1", conffile)
self.apps = self._get_apps()
self.genmanager = GeneratorManager(self.apps.values())
self.substitutions = {
'nagios_bin': 'nagios',
'nagios_init': '/etc/init.d/nagios',
'targetconfdir': '/etc/vigilo/vigiconf/',
'nagios_cfg': '/etc/nagios/nagios.cfg',
}
def tearDown(self):
DBSession.expunge_all()
teardown_db()
shutil.rmtree(self.tmpdir)
settings["vigiconf"]["confdir"] = self.old_conf_path
conf.hostsConf = {}
def _get_apps(self):
"""
Dictionnaire des instances d'applications à tester.
Exemple: {"nagios": Nagios(), "vigimap": VigiMap()}
"""
return {}
def _generate(self):
self.genmanager.generate(DummyRevMan())
def _validate(self):
deploydir = os.path.join(self.basedir, "localhost")
for appname, app in self.apps.iteritems():
__import__(app.__module__)
module = sys.modules[app.__module__]
validation_script = os.path.join(
os.path.dirname(module.__file__),
"validate.sh")
fh = open(validation_script, 'r')
contents = fh.read()
fh.close()
contents = contents % {
}
validation_script = os.path.join(self.tmpdir, "validate.sh")
fh = open(validation_script, 'w')
fh.write(contents % self.substitutions)
fh.close()
print appname
proc = Popen(["sh", validation_script, deploydir],
stdout=PIPE, stderr=STDOUT)
print proc.communicate()[0]
self.assertEqual(proc.returncode, 0)
class HostTemplates(unittest.TestCase):
def setUp(self):
setup_db()
testfactory = TestFactory(confdir=settings["vigiconf"]["confdir"])
self.hosttemplatefactory = HostTemplateFactory(testfactory)
self.hosttemplatefactory.register(HostTemplate("default"))
self.tpl = HostTemplate("testtpl1")
self.hosttemplatefactory.register(self.tpl)
conf.hostsConf = {}
self.host = Host(conf.hostsConf, "dummy", "testserver1",
"192.168.1.1", "Servers")
def tearDown(self):
conf.hostsConf = {}
teardown_db()
def test_add_test_simple(self):
"""Test for the add_test method, without test arguments"""
self.tpl.add_test("UpTime")
self.hosttemplatefactory.apply(self.host, "testtpl1")
self.assertTrue(conf.hostsConf["testserver1"]["services"].has_key(
"UpTime"), "add_test does not work without test args")
def test_add_test_args(self):
"""Test for the add_test method, with test arguments"""
self.tpl.add_test("Interface", {"label":"Loopback", "ifname":"lo"})
self.hosttemplatefactory.apply(self.host, "testtpl1")
self.assertTrue(conf.hostsConf["testserver1"]["SNMPJobs"]
[('Interface Loopback', 'service')]["params"]
== ["lo", "Loopback", "i", "c"],
"add_test does not work with test args")
def test_add_group_simple(self):
"""Test for the add_group method, with one argument only"""
self.tpl.add_group("/Test Group")
self.hosttemplatefactory.apply(self.host, "testtpl1")
self.assertTrue("/Test Group" in
conf.hostsConf["testserver1"]["otherGroups"],
"add_group does not work with one arg")
def test_add_group_multiple(self):
"""Test for the add_group method, with multiple arguments"""
self.tpl.add_group("/Test Group 1", "/Test Group 2")
self.hosttemplatefactory.apply(self.host, "testtpl1")
self.assertTrue("/Test Group 1" in
conf.hostsConf["testserver1"]["otherGroups"],
"add_group does not work with multiple args")
self.assertTrue("/Test Group 2" in
conf.hostsConf["testserver1"]["otherGroups"],
"add_group does not work with multiple args")
def test_add_attribute(self):
"""Test for the add_attribute method"""
self.tpl.add_attribute("TestAttr", "TestVal")
self.hosttemplatefactory.apply(self.host, "testtpl1")
self.assertEqual(conf.hostsConf["testserver1"]["TestAttr"],
"TestVal", "add_attribute does not work")
def test_add_default_service_weight(self):
"""Test for the add_weight method"""
self.tpl.add_weight("default_service_weight", 12)
self.hosttemplatefactory.apply(self.host, "testtpl1")
self.assertEqual(conf.hostsConf["testserver1"][
"default_service_weight"],
12, "add_weight does not work")
def test_add_default_service_warning_weight(self):
"""Test for the add_weight method"""
self.tpl.add_weight("default_service_warning_weight", 12)
self.hosttemplatefactory.apply(self.host, "testtpl1")
self.assertEqual(conf.hostsConf["testserver1"][
"default_service_warning_weight"],
12, "add_weight does not work")
def test_inherit_redefine_test(self):
self.tpl.add_test("Interface", {"ifname":"eth0", "label":"Label1"})
tpl2 = HostTemplate("testtpl2")
tpl2.add_parent("testtpl1")
tpl2.add_test("Interface", {"ifname":"eth0", "label":"Label2"})
self.hosttemplatefactory.register(tpl2)
# Reload the templates
self.hosttemplatefactory.load_templates()
intftest = None
for test in self.hosttemplatefactory.templates["testtpl2"]["tests"]:
if test["name"] == "Interface":
intftest = test
self.assertTrue(intftest is not None,
"inheritance does not work with tests")
self.assertEqual(intftest["args"]["label"], "Label2",
"child templates cannot redefine tests from parent templates")
def test_deepcopy(self):
"""
Test de la copie en profondeur
If the template data from the parent is not copied with
copy.deepcopy(), then the child's template data will propagate back
into the parent
"""
self.tpl.add_attribute("TestAttr1", "TestVal")
tpl2 = HostTemplate("testtpl2")
tpl2.add_parent("testtpl1")
tpl2.add_attribute("TestAttr2", "TestVal")
self.hosttemplatefactory.register(tpl2)
# Reload the templates
self.hosttemplatefactory.load_templates()
tpldata = self.hosttemplatefactory.templates["testtpl1"]
self.failIf(tpldata["attributes"].has_key("TestAttr2"),
"inheritence taints parent templates")
def test_defined_templates(self):
self.hosttemplatefactory.load_templates()
for tpl in self.hosttemplatefactory.templates.keys():
self.hosttemplatefactory.apply(self.host, tpl)
def test_parent_default(self):
tpl1 = HostTemplate("testtpl2")
tpl1.add_parent("testtpl1")
self.hosttemplatefactory.register(tpl1)
self.assertTrue("default" in
self.hosttemplatefactory.templates["testtpl1"]["parent"],
"The \"default\" template is not automatically added as "
"parent to other templates")
def test_add_nagios_hdirective(self):
"""Test for the add_nagios_directive method"""
self.tpl.add_nagios_directive("max_check_attempts", "5")
tpldata = self.hosttemplatefactory.templates["testtpl1"]
self.assertEquals(
tpldata["nagiosDirectives"]["host"]["max_check_attempts"],
"5")
def test_nagios_hdirs_apply_on_host(self):
self.tpl.add_nagios_directive("retry_interval", "8")
self.hosttemplatefactory.apply(self.host, "testtpl1")
testserver1 = conf.hostsConf['testserver1']
nagios_hdirs = testserver1.get('nagiosDirectives')["host"]
self.assertEquals(nagios_hdirs['retry_interval'], "8",
"retry_interval=8")
def test_nagios_sdirs_apply_on_all_service(self):
"""Nagios service directives for tests"""
self.tpl.add_nagios_directive("retry_interval", "8", target="services")
tpldata = self.hosttemplatefactory.templates["testtpl1"]
self.assertEquals(
tpldata["nagiosDirectives"]["services"]["retry_interval"],
"8")
def test_nagios_srvdirs_apply_on_service(self):
"""Nagios directives for tests"""
self.tpl.add_test("UpTime", directives={"testdir": "testdirvalue"})
self.hosttemplatefactory.apply(self.host, "testtpl1")
ndirs = conf.hostsConf["testserver1"]["nagiosSrvDirs"]
self.assertTrue("UpTime" in ndirs)
self.assertTrue("testdir" in ndirs["UpTime"])
self.assertEqual(ndirs["UpTime"]["testdir"], "testdirvalue")
def test_nonexistant_test(self):
"""
Une exception doit être levée si on cherche à ajouter un test inexistant.
"""
self.tpl.add_test("NonExistant")
self.assertRaises(ParsingError, self.hosttemplatefactory.apply,
self.host, "testtpl1")
def test_attributes_hierarchy_order(self):
"""
Les attributs doivent se faire surcharger dans l'ordre d'héritage
"""
self.tpl.add_attribute("snmpCommunity", "comm1")
self.hosttemplatefactory.apply(self.host, "testtpl1")
self.assertEquals(conf.hostsConf['testserver1']['snmpCommunity'],
"comm1", "La communauté SNMP doit être celle du dernier "
"template et pas celle d'un de ses parents")
def test_add_test_simple_active(self):
"""Ajout d'un service actif sur un hôte avec force-passive"""
self.host.set_attribute("force-passive", True)
self.tpl.add_test("HTTP")
self.hosttemplatefactory.apply(self.host, "testtpl1")
self.assertEqual("passive",
conf.hostsConf["testserver1"]["services"]["HTTP"]["type"])
