def test_create(self):
with mock.patch.object(self.dbapi, 'create_magnum_service',
autospec=True) as mock_create_magnum_service:
mock_create_magnum_service.return_value = self.fake_magnum_service
ms_dict = {'host': 'fakehost', 'binary': 'fake-bin'}
ms = objects.MagnumService(self.context, **ms_dict)
ms.create(self.context)
mock_create_magnum_service.assert_called_once_with(ms_dict)
def get_test_magnum_service_object(context, **kw):
"""Return a test magnum_service object.
Get a magnum_service from DB layer and return an object with
appropriate attributes.
"""
db_magnum_service = db_utils.get_test_magnum_service(**kw)
magnum_service = objects.MagnumService(context)
for key in db_magnum_service:
setattr(magnum_service, key, db_magnum_service[key])
return magnum_service
def update_magnum_service(self, ctx):
LOG.debug('Update magnum_service')
if self.magnum_service_ref is None:
self.magnum_service_ref = \
objects.MagnumService.get_by_host_and_binary(
ctx, self.host, self.binary)
if self.magnum_service_ref is None:
magnum_service_dict = {
'host': self.host,
'binary': self.binary
}
self.magnum_service_ref = objects.MagnumService(
ctx, **magnum_service_dict)
self.magnum_service_ref.create()
self.magnum_service_ref.report_state_up()
