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])