def test_fill_obj_extarq_fields_with_dep_id(self, mock_get_devprof,
mock_get_ah, mock_get_obj_dep,
mock_obj_devprof):
in_db_extarq = self.fake_db_extarqs[0]
# Since state is not 'Bound', attach_handle_get_by_id is not called.
in_db_extarq['state'] = 'Initial'
in_db_extarq['deployable_id'] = '1'
db_devprof = fake_device_profile.get_db_devprofs()[0]
obj_devprof = fake_device_profile.get_obj_devprofs()[0]
mock_get_devprof.return_value = db_devprof
mock_obj_devprof.return_value = obj_devprof
mock_get_obj_dep.return_value = \
fake_deployable.fake_deployable_obj(self.context,
uuid=self.deployable_uuids[0])
out_db_extarq = objects.ExtARQ._fill_obj_extarq_fields(
self.context, in_db_extarq)
self.assertEqual(out_db_extarq['device_profile_name'],
db_devprof['name'])
self.assertEqual(out_db_extarq['attach_handle_type'], '')
self.assertEqual(out_db_extarq['attach_handle_info'], '')
devprof_group_id = out_db_extarq['device_profile_group_id']
self.assertEqual(out_db_extarq['device_profile_group'],
obj_devprof['groups'][devprof_group_id])
def test_create(self):
api_devprofs = fake_device_profile.get_api_devprofs()
api_devprof = api_devprofs[0]
db_devprofs = fake_device_profile.get_db_devprofs()
db_devprof = db_devprofs[0]
with mock.patch.object(self.dbapi,
'device_profile_create',
autospec=True) as mock_db_devprof_create:
mock_db_devprof_create.return_value = self.fake_device_profile
obj_devprof = objects.DeviceProfile(**api_devprof)
obj_devprof.create(self.context)
mock_db_devprof_create.assert_called_once_with(
self.context, db_devprof)
def test_update(self):
fake_db_devprofs = fake_device_profile.get_db_devprofs()
fake_obj_devprofs = fake_device_profile.get_obj_devprofs()
db_devprof = fake_db_devprofs[0]
db_devprof['created_at'] = None
db_devprof['updated_at'] = None
with mock.patch.object(self.dbapi,
'device_profile_get_by_uuid',
autospec=True) as mock_dp_get:
mock_dp_get.return_value = db_devprof
uuid = fake_db_devprofs[0]['uuid']
# Start with db_devprofs[0], corr. to fake_obj_devprofs[0]
obj_devprof = objects.DeviceProfile.get_by_uuid(self.context, uuid)
# Change contents to fake_obj_devprofs[1] except uuid
obj_devprof = fake_obj_devprofs[1]
obj_devprof['uuid'] = uuid
with mock.patch.object(self.dbapi,
'device_profile_update',
autospec=True) as mock_dp_update:
mock_dp_update.return_value = db_devprof
obj_devprof.save(self.context)
mock_dp_get.assert_called_once_with(self.context, uuid)
mock_dp_update.assert_called_once()