def test_mongodb_uri_environ(self, zapi, mongo_mock):
from redisapi.hc import ZabbixHealthCheck
ZabbixHealthCheck()
mongo_mock.assert_called_with("mongodb://localhost:27017/")
os.environ["MONGODB_URI"] = "0.0.0.0"
self.addCleanup(self.remove_env, "MONGODB_URI")
ZabbixHealthCheck()
mongo_mock.assert_called_with("0.0.0.0")
def test_running_without_the_ZABBIX_INTERFACE_variable(self):
del os.environ["ZABBIX_INTERFACE"]
with self.assertRaises(Exception) as cm:
from redisapi.hc import ZabbixHealthCheck
ZabbixHealthCheck()
exc = cm.exception
self.assertEqual(
(u"You must define the ZABBIX_INTERFACE environment variable.",),
exc.args,
)
def setUp(self, zabbix_mock):
url = "http://zbx.com"
user = "******"
password = "******"
os.environ["ZABBIX_URL"] = url
os.environ["ZABBIX_USER"] = user
os.environ["ZABBIX_PASSWORD"] = password
os.environ["ZABBIX_HOST"] = "1"
os.environ["ZABBIX_INTERFACE"] = "1"
self.addCleanup(self.remove_env, "REDIS_SERVER_HOST")
zapi_mock = mock.Mock()
zabbix_mock.return_value = zapi_mock
from redisapi.hc import ZabbixHealthCheck
self.hc = ZabbixHealthCheck()
zabbix_mock.assert_called_with(url)
zapi_mock.login.assert_called_with(user, password)
