def test_get_volume_snapshot(self): volume_snapshot = fake_volume.FakeVolumeSnapshot( display_name=CONF.volume_snapshot_name_template % self.snapshot['id']) self.stubs.Set(self._driver.volume_api, 'get_all_snapshots', mock.Mock(return_value=[volume_snapshot])) result = self._driver._get_volume_snapshot(self._context, self.snapshot['id']) self.assertEqual(result, volume_snapshot)
def test_get_volume_snapshot_error(self): volume_snapshot = fake_volume.FakeVolumeSnapshot( display_name=CONF.volume_snapshot_name_template % self.snapshot['id']) self.stubs.Set( self._driver.volume_api, 'get_all_snapshots', mock.Mock(return_value=[volume_snapshot, volume_snapshot])) self.assertRaises(exception.ManilaException, self._driver._get_volume_snapshot, self._context, self.share['id'])
def test_delete_snapshot(self): fake_vol_snap = fake_volume.FakeVolumeSnapshot() self.stubs.Set(self._driver, '_get_volume_snapshot', mock.Mock(return_value=fake_vol_snap)) self.stubs.Set(self._driver.volume_api, 'delete_snapshot', mock.Mock()) self.stubs.Set(self._driver.volume_api, 'get_snapshot', mock.Mock(side_effect=exception.VolumeSnapshotNotFound( snapshot_id=fake_vol_snap['id']))) self._driver.delete_snapshot(self._context, fake_vol_snap, share_server=self.server) self._driver._get_volume_snapshot.assert_called_once() self._driver.volume_api.delete_snapshot.assert_called_once() self._driver.volume_api.get_snapshot.assert_called_once()
def test_create_snapshot(self): fake_vol = fake_volume.FakeVolume() fake_vol_snap = fake_volume.FakeVolumeSnapshot() self.stubs.Set(self._driver, '_get_volume', mock.Mock(return_value=fake_vol)) self.stubs.Set(self._driver.volume_api, 'create_snapshot_force', mock.Mock(return_value=fake_vol_snap)) self._driver.create_snapshot(self._context, self.snapshot, share_server=self.server) self._driver._get_volume.assert_called_once() self._driver.volume_api.create_snapshot_force.assert_called_once_with( self._context, fake_vol['id'], CONF.volume_snapshot_name_template % self.snapshot['id'], '')
def test_allocate_container_with_snaphot(self): fake_vol = fake_volume.FakeVolume() fake_vol_snap = fake_volume.FakeVolumeSnapshot() self.stubs.Set(self._driver, '_get_volume_snapshot', mock.Mock(return_value=fake_vol_snap)) self.stubs.Set(self._driver.volume_api, 'create', mock.Mock(return_value=fake_vol)) result = self._driver._allocate_container(self._context, self.share, self.snapshot) self.assertEqual(result, fake_vol) self._driver.volume_api.create.assert_called_once_with( self._context, self.share['size'], CONF.volume_name_template % self.share['id'], '', snapshot=fake_vol_snap)