def test_get_env_list_invalid_env(self):
domain = "www.somedomain.com"
k = mock.patch.dict(os.environ, {'NAGIOS_HOSTNAME': domain})
k.start()
nagios_notify = NagiosNotify()
invalid_list = ["NAG_HOSTNAME"]
with self.assertRaises(SystemExit) as cm:
nagios_notify._get_env_list(invalid_list)
k.stop()
self.assertEqual(1, cm.exception.code)
def test_get_env_list(self):
domain = "www.somedomain.com"
k = mock.patch.dict(os.environ, {
'NAGIOS_HOSTNAME': domain,
"NAGIOS_HOSTSTATE": "DOWN"
})
k.start()
nagios_notify = NagiosNotify()
valid_list = ["NAGIOS_HOSTNAME", "NAGIOS_HOSTSTATE"]
valid_return = nagios_notify._get_env_list(valid_list)
k.stop()
self.assertEqual(2, len(valid_return))
self.assertEqual(domain, valid_return[0])
self.assertEqual("DOWN", valid_return[1])
