def test_get_message_invalid_host_message_list(self): domain = "www.somedomain.com" k = mock.patch.dict(os.environ, {'NAGIOS_HOSTNAME': domain}) k.start() nagios_notify = NagiosNotify() nagios_servicestate = None nagios_hoststate = "DOWN" service_message = "HOST: {0} Yum Update needs updating" service_message_list = ["NAGIOS_HOSTNAME"] host_message = "HOST: {0} Host is down" invalid_host_message_list = ["NAG_HOSTNAME"] with self.assertRaises(SystemExit) as cm: nagios_notify._get_message(nagios_servicestate, nagios_hoststate, service_message, service_message_list, host_message, invalid_host_message_list) k.stop() self.assertEqual(1, cm.exception.code)
def test_get_message_host_message(self): domain = "www.somedomain.com" k = mock.patch.dict(os.environ, {'NAGIOS_HOSTNAME': domain}) k.start() nagios_notify = NagiosNotify() nagios_servicestate = None nagios_hoststate = "DOWN" service_message = "HOST: {0} Yum Update needs updating" service_message_list = ["NAGIOS_HOSTNAME"] host_message = "HOST: {0} Host is down" host_message_list = ["NAGIOS_HOSTNAME"] message = nagios_notify._get_message(nagios_servicestate, nagios_hoststate, service_message, service_message_list, host_message, host_message_list) k.stop() self.assertEqual("HOST: www.somedomain.com Host is down", message)