179 lines
8.1 KiB
Python
179 lines
8.1 KiB
Python
# Copyright 2015 Intel Corporation
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import mock
|
|
from oslo_utils import timeutils
|
|
import pytz
|
|
import six
|
|
|
|
from cinder import exception
|
|
from cinder import objects
|
|
from cinder.tests.unit import fake_constants as fake
|
|
from cinder.tests.unit import objects as test_objects
|
|
from cinder.tests.unit.objects.test_consistencygroup import \
|
|
fake_consistencygroup
|
|
|
|
fake_cgsnapshot = {
|
|
'id': fake.CGSNAPSHOT_ID,
|
|
'user_id': fake.USER_ID,
|
|
'project_id': fake.PROJECT_ID,
|
|
'name': 'fake_name',
|
|
'description': 'fake_description',
|
|
'status': 'creating',
|
|
'consistencygroup_id': fake.CONSISTENCY_GROUP_ID,
|
|
}
|
|
|
|
|
|
class TestCGSnapshot(test_objects.BaseObjectsTestCase):
|
|
|
|
@mock.patch('cinder.db.sqlalchemy.api.cgsnapshot_get',
|
|
return_value=fake_cgsnapshot)
|
|
def test_get_by_id(self, cgsnapshot_get):
|
|
cgsnapshot = objects.CGSnapshot.get_by_id(self.context,
|
|
fake.CGSNAPSHOT_ID)
|
|
self._compare(self, fake_cgsnapshot, cgsnapshot)
|
|
|
|
@mock.patch('cinder.db.cgsnapshot_create',
|
|
return_value=fake_cgsnapshot)
|
|
def test_create(self, cgsnapshot_create):
|
|
fake_cgsnap = fake_cgsnapshot.copy()
|
|
del fake_cgsnap['id']
|
|
cgsnapshot = objects.CGSnapshot(context=self.context, **fake_cgsnap)
|
|
cgsnapshot.create()
|
|
self._compare(self, fake_cgsnapshot, cgsnapshot)
|
|
|
|
def test_create_with_id_except_exception(self):
|
|
cgsnapshot = objects.CGSnapshot(context=self.context,
|
|
**{'id': fake.CONSISTENCY_GROUP_ID})
|
|
self.assertRaises(exception.ObjectActionError, cgsnapshot.create)
|
|
|
|
@mock.patch('cinder.db.cgsnapshot_update')
|
|
def test_save(self, cgsnapshot_update):
|
|
cgsnapshot = objects.CGSnapshot._from_db_object(
|
|
self.context, objects.CGSnapshot(), fake_cgsnapshot)
|
|
cgsnapshot.status = 'active'
|
|
cgsnapshot.save()
|
|
cgsnapshot_update.assert_called_once_with(self.context, cgsnapshot.id,
|
|
{'status': 'active'})
|
|
|
|
@mock.patch('cinder.db.consistencygroup_update',
|
|
return_value=fake_consistencygroup)
|
|
@mock.patch('cinder.db.cgsnapshot_update')
|
|
def test_save_with_consistencygroup(self, cgsnapshot_update,
|
|
cgsnapshot_cg_update):
|
|
consistencygroup = objects.ConsistencyGroup._from_db_object(
|
|
self.context, objects.ConsistencyGroup(), fake_consistencygroup)
|
|
cgsnapshot = objects.CGSnapshot._from_db_object(
|
|
self.context, objects.CGSnapshot(), fake_cgsnapshot)
|
|
cgsnapshot.name = 'foobar'
|
|
cgsnapshot.consistencygroup = consistencygroup
|
|
self.assertEqual({'name': 'foobar',
|
|
'consistencygroup': consistencygroup},
|
|
cgsnapshot.obj_get_changes())
|
|
self.assertRaises(exception.ObjectActionError, cgsnapshot.save)
|
|
|
|
@mock.patch('oslo_utils.timeutils.utcnow', return_value=timeutils.utcnow())
|
|
@mock.patch('cinder.db.sqlalchemy.api.cgsnapshot_destroy')
|
|
def test_destroy(self, cgsnapshot_destroy, utcnow_mock):
|
|
cgsnapshot_destroy.return_value = {
|
|
'status': 'deleted',
|
|
'deleted': True,
|
|
'deleted_at': utcnow_mock.return_value}
|
|
cgsnapshot = objects.CGSnapshot(context=self.context,
|
|
id=fake.CGSNAPSHOT_ID)
|
|
cgsnapshot.destroy()
|
|
self.assertTrue(cgsnapshot_destroy.called)
|
|
admin_context = cgsnapshot_destroy.call_args[0][0]
|
|
self.assertTrue(admin_context.is_admin)
|
|
self.assertTrue(cgsnapshot.deleted)
|
|
self.assertEqual('deleted', cgsnapshot.status)
|
|
self.assertEqual(utcnow_mock.return_value.replace(tzinfo=pytz.UTC),
|
|
cgsnapshot.deleted_at)
|
|
|
|
@mock.patch('cinder.objects.consistencygroup.ConsistencyGroup.get_by_id')
|
|
@mock.patch('cinder.objects.snapshot.SnapshotList.get_all_for_cgsnapshot')
|
|
def test_obj_load_attr(self, snapshotlist_get_for_cgs,
|
|
consistencygroup_get_by_id):
|
|
cgsnapshot = objects.CGSnapshot._from_db_object(
|
|
self.context, objects.CGSnapshot(), fake_cgsnapshot)
|
|
# Test consistencygroup lazy-loaded field
|
|
consistencygroup = objects.ConsistencyGroup(
|
|
context=self.context, id=fake.CONSISTENCY_GROUP_ID)
|
|
consistencygroup_get_by_id.return_value = consistencygroup
|
|
self.assertEqual(consistencygroup, cgsnapshot.consistencygroup)
|
|
consistencygroup_get_by_id.assert_called_once_with(
|
|
self.context, cgsnapshot.consistencygroup_id)
|
|
# Test snapshots lazy-loaded field
|
|
snapshots_objs = [objects.Snapshot(context=self.context, id=i)
|
|
for i in [fake.SNAPSHOT_ID, fake.SNAPSHOT2_ID,
|
|
fake.SNAPSHOT3_ID]]
|
|
snapshots = objects.SnapshotList(context=self.context,
|
|
objects=snapshots_objs)
|
|
snapshotlist_get_for_cgs.return_value = snapshots
|
|
self.assertEqual(snapshots, cgsnapshot.snapshots)
|
|
snapshotlist_get_for_cgs.assert_called_once_with(
|
|
self.context, cgsnapshot.id)
|
|
|
|
@mock.patch('cinder.db.sqlalchemy.api.cgsnapshot_get')
|
|
def test_refresh(self, cgsnapshot_get):
|
|
db_cgsnapshot1 = fake_cgsnapshot.copy()
|
|
db_cgsnapshot2 = db_cgsnapshot1.copy()
|
|
db_cgsnapshot2['description'] = 'foobar'
|
|
|
|
# On the second cgsnapshot_get, return the CGSnapshot with an updated
|
|
# description
|
|
cgsnapshot_get.side_effect = [db_cgsnapshot1, db_cgsnapshot2]
|
|
cgsnapshot = objects.CGSnapshot.get_by_id(self.context,
|
|
fake.CGSNAPSHOT_ID)
|
|
self._compare(self, db_cgsnapshot1, cgsnapshot)
|
|
|
|
# description was updated, so a CGSnapshot refresh should have a new
|
|
# value for that field
|
|
cgsnapshot.refresh()
|
|
self._compare(self, db_cgsnapshot2, cgsnapshot)
|
|
if six.PY3:
|
|
call_bool = mock.call.__bool__()
|
|
else:
|
|
call_bool = mock.call.__nonzero__()
|
|
cgsnapshot_get.assert_has_calls([mock.call(self.context,
|
|
fake.CGSNAPSHOT_ID),
|
|
call_bool,
|
|
mock.call(self.context,
|
|
fake.CGSNAPSHOT_ID)])
|
|
|
|
|
|
class TestCGSnapshotList(test_objects.BaseObjectsTestCase):
|
|
@mock.patch('cinder.db.cgsnapshot_get_all',
|
|
return_value=[fake_cgsnapshot])
|
|
def test_get_all(self, cgsnapshot_get_all):
|
|
cgsnapshots = objects.CGSnapshotList.get_all(self.context)
|
|
self.assertEqual(1, len(cgsnapshots))
|
|
TestCGSnapshot._compare(self, fake_cgsnapshot, cgsnapshots[0])
|
|
|
|
@mock.patch('cinder.db.cgsnapshot_get_all_by_project',
|
|
return_value=[fake_cgsnapshot])
|
|
def test_get_all_by_project(self, cgsnapshot_get_all_by_project):
|
|
cgsnapshots = objects.CGSnapshotList.get_all_by_project(
|
|
self.context, self.project_id)
|
|
self.assertEqual(1, len(cgsnapshots))
|
|
TestCGSnapshot._compare(self, fake_cgsnapshot, cgsnapshots[0])
|
|
|
|
@mock.patch('cinder.db.cgsnapshot_get_all_by_group',
|
|
return_value=[fake_cgsnapshot])
|
|
def test_get_all_by_group(self, cgsnapshot_get_all_by_group):
|
|
cgsnapshots = objects.CGSnapshotList.get_all_by_group(
|
|
self.context, self.project_id)
|
|
self.assertEqual(1, len(cgsnapshots))
|
|
TestCGSnapshot._compare(self, fake_cgsnapshot, cgsnapshots[0])
|