def test_device_obj_with_snapshot(self):
snap = self.snap_obj
backup_device_info = objects.BackupDeviceInfo()
backup_device_info.snapshot = snap
backup_device_info.secure_enabled = (
self.backup_device_dict['secure_enabled'])
backup_device_obj = backup_device_info.device_obj
self.assertIsInstance(backup_device_obj, objects.Snapshot)
self.assertEqual(snap, backup_device_obj)
def test_device_obj_with_volume(self):
vol = self.vol_obj
backup_device_info = objects.BackupDeviceInfo()
backup_device_info.volume = vol
backup_device_info.secure_enabled = (
self.backup_device_dict['secure_enabled'])
backup_device_obj = backup_device_info.device_obj
self.assertIsInstance(backup_device_obj, objects.Volume)
self.assertEqual(vol, backup_device_obj)
def test_is_snapshot_both_volume_and_snapshot_raises_error(self):
snap = self.snap_obj
vol = self.vol_obj
backup_device_info = objects.BackupDeviceInfo()
backup_device_info.snapshot = snap
backup_device_info.volume = vol
backup_device_info.secure_enabled = (
self.backup_device_dict['secure_enabled'])
self.assertRaises(exception.ProgrammingError, getattr,
backup_device_info, 'is_snapshot')
def test_to_primitive_with_volume(self, mock_fake_vol):
vol_obj = self.vol_obj
self.backup_device_dict['backup_device'] = fake_vol
backup_device_info = objects.BackupDeviceInfo()
backup_device_info.volume = vol_obj
backup_device_info.secure_enabled = (
self.backup_device_dict['secure_enabled'])
backup_device_ret_dict = backup_device_info.to_primitive(self.context)
self.assertEqual(self.backup_device_dict['secure_enabled'],
backup_device_ret_dict['secure_enabled'])
self.assertFalse(backup_device_ret_dict['is_snapshot'])
self.assertEqual(self.backup_device_dict['backup_device'],
backup_device_ret_dict['backup_device'])
def test_to_primitive_with_snapshot(self, mock_fake_snap):
snap_obj = self.snap_obj
backup_device_info = objects.BackupDeviceInfo()
backup_device_info.snapshot = snap_obj
backup_device_info.secure_enabled = (
self.backup_device_dict['secure_enabled'])
backup_device_ret_dict = backup_device_info.to_primitive(self.context)
self.assertEqual(self.backup_device_dict['secure_enabled'],
backup_device_ret_dict['secure_enabled'])
self.assertTrue(backup_device_ret_dict['is_snapshot'])
# NOTE(sborkows): since volume in sqlalchemy snapshot is a sqlalchemy
# object too, to compare snapshots we need to convert their volumes to
# dicts.
snap_actual_dict = fake_snap
snap_ref_dict = backup_device_ret_dict['backup_device']
snap_actual_dict['volume'] = self.vol_obj.obj_to_primitive()
snap_ref_dict['volume'] = snap_ref_dict['volume']
self.assertEqual(snap_actual_dict, snap_ref_dict)
def test_is_snapshot_neither_volume_nor_snapshot_raises_error(self):
backup_device_info = objects.BackupDeviceInfo()
backup_device_info.secure_enabled = (
self.backup_device_dict['secure_enabled'])
self.assertRaises(exception.ProgrammingError, getattr,
backup_device_info, 'is_snapshot')
