def test_run_checks_with_no_checks_defined(self, wait_on_event): check = Mock() check.passing = True discovery = Mock() service = Mock() service.name = "service" service.is_up = False service.discovery = "disco" service.checks = {} reporter = Reporter("etc/configs") reporter.configurables[Discovery] = { "disco": discovery } def set_shutdown(*args, **kwargs): reporter.shutdown.set() wait_on_event.side_effect = set_shutdown reporter.configurables[Service] = { "service": service } reporter.run_checks(service) wait_on_event.assert_called_once_with( reporter.shutdown, timeout=service.check_interval )
def test_run_checks_up_service_fails(self, wait_on_event): check = Mock() check.passing = False discovery = Mock() service = Mock() service.is_up = True service.discovery = "disco" service.checks = { "check1": check, } reporter = Reporter("etc/configs") reporter.configurables[Discovery] = { "disco": discovery } reporter.configurables[Service] = { service.name: service } def set_shutdown(*args, **kwargs): reporter.shutdown.set() wait_on_event.side_effect = set_shutdown reporter.run_checks(service) discovery.report_down.assert_called_once_with(service)
def test_update_discovery_clears_up_flags(self): service = Mock() discovery = Mock() service.discovery = "disco" service.checks = {} reporter = Reporter("etc/configs") reporter.configurables[Discovery] = { "disco": discovery } reporter.configurables[Service] = { "a_service": service } service.is_up = True reporter.update_configurable(Discovery, "disco", {}) self.assertEqual(service.is_up, None)