def test_updates_services_in_database(self):
# Pretend we're in a production environment.
self.patch(service_monitor_service,
"is_dev_environment").return_value = False
self.patch(proxyconfig, "is_config_present").return_value = True
service = self.pick_service()
state = ServiceState(SERVICE_STATE.ON, "running")
mock_ensureServices = self.patch(service_monitor, "ensureServices")
mock_ensureServices.return_value = succeed({service.name: state})
region = yield deferToDatabase(
transactional(factory.make_RegionController))
self.patch(RegionController.objects,
"get_running_controller").return_value = region
monitor_service = ServiceMonitorService(Clock())
yield monitor_service.startService()
yield monitor_service.stopService()
service = yield deferToDatabase(transactional(Service.objects.get),
node=region,
name=service.name)
self.assertThat(
service,
MatchesStructure.byEquality(
name=service.name,
status=SERVICE_STATUS.RUNNING,
status_info="",
),
)
def test_monitorServices_calls_ensureServices(self):
# Pretend we're in a production environment.
self.patch(service_monitor_service,
"is_dev_environment").return_value = False
monitor_service = ServiceMonitorService(Clock())
mock_ensureServices = self.patch(service_monitor, "ensureServices")
monitor_service.monitorServices()
self.assertThat(mock_ensureServices, MockCalledOnceWith())
def test_monitorServices_does_not_do_anything_in_dev_environment(self):
# Belt-n-braces make sure we're in a development environment.
self.assertTrue(service_monitor_service.is_dev_environment())
monitor_service = ServiceMonitorService(Clock())
mock_ensureServices = self.patch(service_monitor, "ensureServices")
with TwistedLoggerFixture() as logger:
monitor_service.monitorServices()
self.assertThat(mock_ensureServices, MockNotCalled())
self.assertDocTestMatches(
"Skipping check of services; they're not running under the "
"supervision of systemd.", logger.output)
def test_buildServices_builds_services_list(self):
self.patch(proxyconfig, "is_config_present").return_value = True
monitor_service = ServiceMonitorService(Clock())
service = self.pick_service()
state = ServiceState(SERVICE_STATE.ON, "running")
observed_services = yield monitor_service._buildServices(
{service.name: state})
expected_services = [{
"name": service.name,
"status": "running",
"status_info": ""
}]
self.assertEqual(expected_services, observed_services)
def test_monitorServices_handles_failure(self):
# Pretend we're in a production environment.
self.patch(
service_monitor_service, "is_dev_environment").return_value = False
monitor_service = ServiceMonitorService(Clock())
mock_ensureServices = self.patch(service_monitor, "ensureServices")
mock_ensureServices.return_value = fail(factory.make_exception())
with TwistedLoggerFixture() as logger:
monitor_service.monitorServices()
self.assertDocTestMatches("""\
Failed to monitor services and update database.
Traceback (most recent call last):
...""", logger.output)
def test_init_sets_up_timer_correctly(self):
monitor_service = ServiceMonitorService(
sentinel.advertisingService, sentinel.clock)
self.assertThat(monitor_service, MatchesStructure.byEquality(
call=(monitor_service.monitorServices, (), {}),
step=(60), advertisingService=sentinel.advertisingService,
clock=sentinel.clock))