def test_repr(self, mock_rpc_init):
# Test if a Service object is correctly represented, for example in
# log files.
serv = service.Service(self.host, self.binary, self.topic,
'masakari.tests.unit.test_service.FakeManager')
exp = "<Service: host=foo, binary=masakari-engine, " \
"manager_class_name=masakari.tests.unit." \
"test_service.FakeManager>"
self.assertEqual(exp, repr(serv))
def test_parent_graceful_shutdown(self, mock_rpc, mock_rpc_init,
mock_stop):
serv = service.Service(self.host, self.binary, self.topic,
'masakari.tests.unit.test_service.FakeManager')
serv.manager = mock.Mock()
serv.manager.service_name = self.topic
serv.start()
serv.stop()
serv.rpcserver.start.assert_called_once_with()
serv.rpcserver.stop.assert_called_once_with()
mock_stop.assert_called_once_with()
def test_parent_graceful_shutdown(self, mock_rpc, mock_rpc_init):
self.manager_mock = self.mox.CreateMock(FakeManager)
self.mox.StubOutWithMock(sys.modules[__name__],
'FakeManager',
use_mock_anything=True)
self.mox.StubOutWithMock(_service.Service, 'stop')
FakeManager(host=self.host).AndReturn(self.manager_mock)
self.manager_mock.service_name = self.topic
_service.Service.stop()
self.mox.ReplayAll()
serv = service.Service(self.host, self.binary, self.topic,
'masakari.tests.unit.test_service.FakeManager')
serv.start()
serv.stop()
serv.rpcserver.start.assert_called_once_with()
serv.rpcserver.stop.assert_called_once_with()
def test_reset(self, mock_rpc_init):
serv = service.Service(self.host, self.binary, self.topic,
'masakari.tests.unit.test_service.FakeManager')
with mock.patch.object(serv.manager, 'reset') as mock_reset:
serv.reset()
mock_reset.assert_called_once_with()
def test_message_gets_to_manager(self, mock_rpc_init):
serv = service.Service('test', 'test', 'test',
'masakari.tests.unit.test_service.FakeManager')
self.assertEqual('manager', serv.test_method())
