def setUp(self, zabbix_mock, mongo_mock): os.environ["ZABBIX_URL"] = self.url = "http://zbx.com" os.environ["ZABBIX_USER"] = self.user = "******" os.environ["ZABBIX_PASSWORD"] = self.password = "******" os.environ["ZABBIX_HOST"] = "1" os.environ["ZABBIX_HOST_GROUP"] = "2" zapi_mock = mock.Mock() zapi_mock.trigger.get.return_value = { "result": [{ "comments": "teste" }] } zabbix_mock.return_value = zapi_mock instance_mock = mock.Mock() mongo_mock.return_value = instance_mock from healthcheck.backends import Zabbix self.backend = Zabbix() zabbix_mock.assert_called_with(self.url) zapi_mock.login.assert_called_with(self.user, self.password) mongo_mock.assert_called_with() instance_mock.conn.assert_called_with() self.backend.storage = mock.Mock()
def get_manager(): from healthcheck.backends import Zabbix managers = { "zabbix": Zabbix(), } manager = os.environ.get("API_MANAGER", "zabbix") manager_class = managers.get(manager) if manager_class: return manager_class raise ValueError("{0} is not a valid manager".format(manager))
def test_add_url(self): backend = Zabbix() url = "http://mysite.com" backend.add_url(url)