def test_report_state_service_not_ready(self):
with mock.patch.object(service, 'db') as mock_db:
mock_db.service_get.return_value = service_ref
serv = service.Service(host, binary, topic, CONF.fake_manager)
serv.manager.is_service_ready = mock.Mock(return_value=False)
serv.start()
serv.report_state()
serv.manager.is_service_ready.assert_called_once()
mock_db.service_update.assert_not_called()
def test_report_state_newly_disconnected(self):
serv = service.Service(host, binary, topic, CONF.fake_manager)
serv.start()
serv.report_state()
self.assertTrue(serv.model_disconnected)
service.db.service_get_by_args.assert_called_once_with(
mock.ANY, host, binary)
service.db.service_create.assert_called_once_with(
mock.ANY, service_create)
service.db.service_get.assert_called_once_with(mock.ANY, mock.ANY)
def test_periodic_tasks(self, raise_on_error):
serv = service.Service(host, binary, topic, CONF.fake_manager)
self.mock_object(context, 'get_admin_context',
mock.Mock(side_effect=context.get_admin_context))
self.mock_object(serv.manager, 'periodic_tasks')
serv.periodic_tasks(raise_on_error=raise_on_error)
context.get_admin_context.assert_called_once_with()
serv.manager.periodic_tasks.assert_called_once_with(
utils.IsAMatcher(context.RequestContext),
raise_on_error=raise_on_error)
def test_message_gets_to_manager(self):
serv = service.Service('test', 'test', 'test', CONF.fake_manager)
serv.start()
self.assertEqual(serv.test_method(), 'manager')
