def test_get_env(self): domain = "www.somedomain.com" k = mock.patch.dict(os.environ, {'NAGIOS_HOSTNAME': domain}) k.start() nagios_notify = NagiosNotify() env_value = nagios_notify._get_env("NAGIOS_HOSTNAME") k.stop() self.assertEqual(domain, env_value)
def test_get_env_test_data(self): domain = "www.somedomain.com" k = mock.patch.dict(os.environ, {'NAGIOS_HOSTNAME': domain}) k.start() nagios_notify = NagiosNotify() nagios_notify._TEST = True env_value = nagios_notify._get_env("NAGIOS_HOSTSTATE") k.stop() self.assertEqual("DOWN", env_value)