¡@
OpenStack Study: test_extended_snapshot_attributes.py**** CubicPower OpenStack Study ****
# Copyright 2012 OpenStack Foundation # All Rights Reserved. # # 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. from lxml import etree import webob from cinder.api.contrib import extended_snapshot_attributes from cinder.openstack.common import jsonutils from cinder import test from cinder.tests.api import fakes from cinder import volume UUID1 = '00000000-0000-0000-0000-000000000001' UUID2 = '00000000-0000-0000-0000-000000000002'
**** CubicPower OpenStack Study **** def _get_default_snapshot_param(): return {'id': UUID1, 'volume_id': 12, 'status': 'available', 'volume_size': 100, 'created_at': None, 'display_name': 'Default name', 'display_description': 'Default description', 'project_id': 'fake', 'progress': '0%'}
**** CubicPower OpenStack Study **** def fake_snapshot_get(self, context, snapshot_id): param = _get_default_snapshot_param() return param
**** CubicPower OpenStack Study **** def fake_snapshot_get_all(self, context, search_opts=None): param = _get_default_snapshot_param() return [param]
**** CubicPower OpenStack Study **** class ExtendedSnapshotAttributesTest(test.TestCase): content_type = 'application/json' prefix = 'os-extended-snapshot-attributes:'
**** CubicPower OpenStack Study **** def setUp(self): super(ExtendedSnapshotAttributesTest, self).setUp() self.stubs.Set(volume.api.API, 'get_snapshot', fake_snapshot_get) self.stubs.Set(volume.api.API, 'get_all_snapshots', fake_snapshot_get_all)
**** CubicPower OpenStack Study **** def _make_request(self, url): req = webob.Request.blank(url) req.headers['Accept'] = self.content_type res = req.get_response(fakes.wsgi_app()) return res
**** CubicPower OpenStack Study **** def _get_snapshot(self, body): return jsonutils.loads(body).get('snapshot')
**** CubicPower OpenStack Study **** def _get_snapshots(self, body): return jsonutils.loads(body).get('snapshots')
**** CubicPower OpenStack Study **** def assertSnapshotAttributes(self, snapshot, project_id, progress): self.assertEqual(snapshot.get('%sproject_id' % self.prefix), project_id) self.assertEqual(snapshot.get('%sprogress' % self.prefix), progress)
**** CubicPower OpenStack Study **** def test_show(self): url = '/v2/fake/snapshots/%s' % UUID2 res = self._make_request(url) self.assertEqual(res.status_int, 200) self.assertSnapshotAttributes(self._get_snapshot(res.body), project_id='fake', progress='0%')
**** CubicPower OpenStack Study **** def test_detail(self): url = '/v2/fake/snapshots/detail' res = self._make_request(url) self.assertEqual(res.status_int, 200) for i, snapshot in enumerate(self._get_snapshots(res.body)): self.assertSnapshotAttributes(snapshot, project_id='fake', progress='0%')
**** CubicPower OpenStack Study **** class ExtendedSnapshotAttributesXmlTest(ExtendedSnapshotAttributesTest): content_type = 'application/xml' ext = extended_snapshot_attributes prefix = '{%s}' % ext.Extended_snapshot_attributes.namespace
**** CubicPower OpenStack Study **** def _get_snapshot(self, body): return etree.XML(body)
**** CubicPower OpenStack Study **** def _get_snapshots(self, body): return etree.XML(body).getchildren()
|