def test_sendStatus(self):
"""Envoi de l'état (sendStatus)"""
client = ClientStub("testhost", None, None)
self.settings["connector"]["status_service"] = "testservice"
sp = statuspublisher_factory(self.settings, client, [ProviderStub()])
sp.isConnected = lambda: True
d = client.stub_connect()
d.addCallback(lambda _x: sp.sendStatus())
def check(r):
output = client.channel.sent
print output
self.assertEqual(len(output), 2)
msg_perf = json.loads(output[0]["content"].body)
self.assertEqual(msg_perf["type"], "perf")
self.assertEqual(msg_perf["host"], self.localhn)
self.assertEqual(msg_perf["datasource"], "testservice-dummykey")
self.assertEqual(msg_perf["value"], "dummyvalue")
msg_cmd = json.loads(output[1]["content"].body)
self.assertEqual(msg_cmd["type"], "nagios")
self.assertEqual(msg_cmd["cmdname"],
"PROCESS_SERVICE_CHECK_RESULT")
self.assertTrue(msg_cmd["value"].startswith(
"%s;testservice;0;OK:" % self.localhn))
d.addCallback(check)
return d
def test_send_stats(self):
"""Relai d'un ensemble de statistiques"""
client = ClientStub("testhost", None, None)
self.settings["connector"]["status_service"] = "testsvc"
sp = statuspublisher_factory(self.settings, client)
sp.isConnected = lambda: True
client.stub_connect()
stats = {"key1": "value1", "key2": "value2", "key3": "value3"}
msg = {"type": "perf"}
sp._sendStats(stats, msg)
def check(r_):
output = client.channel.sent
print output
self.assertEqual(len(output), 3)
msg_out = [ json.loads(m["content"].body)
for m in output ]
msg_in = []
for k, v in stats.iteritems():
m = msg.copy()
m.update({"datasource": "testsvc-%s" % k, "value": v})
msg_in.append(m)
self.assertEqual(msg_in, msg_out)
d = wait(0.2)
d.addCallback(check)
return d
def test_force_exchange(self):
"""On force le nom de l'exchange à utiliser pour un message de perf"""
client = ClientStub("testhost", None, None)
self.settings["connector"]["self_monitoring_perf_exchange"] = "foo"
self.settings["connector"]["self_monitoring_nagios_exchange"] = "foo"
self.settings["connector"]["status_service"] = "dummyservice"
sp = statuspublisher_factory(self.settings, client, [ProviderStub()])
sp.isConnected = lambda: True
d = client.stub_connect()
d.addCallback(lambda _x: sp.sendStatus())
def check(r):
output = client.channel.sent
for msg in output:
print msg
self.assertEqual(msg["exchange"], "foo")
d.addCallback(check)
return d
def test_servicename(self):
"""On force le nom du service à utiliser"""
client = ClientStub("testhost", None, None)
self.settings["connector"]["hostname"] = "changedhost"
self.settings["connector"]["status_service"] = "changedsvc"
sp = statuspublisher_factory(self.settings, client,
[ProviderStub()])
sp.isConnected = lambda: True
d = client.stub_connect()
d.addCallback(lambda _x: sp.sendStatus())
def check(r):
output = client.channel.sent
msg_perf = json.loads(output[0]["content"].body)
self.assertEqual(msg_perf["type"], "perf")
self.assertEqual(msg_perf["datasource"],
"changedsvc-dummykey")
msg_cmd = json.loads(output[1]["content"].body)
self.assertEqual(msg_cmd["type"], "nagios")
self.assertTrue(msg_cmd["value"].startswith(
"changedhost;changedsvc;"))
d.addCallback(check)
return d
