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)
