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)
