127 lines
5.0 KiB
Python
127 lines
5.0 KiB
Python
# Copyright 2011 Denali Systems, Inc.
|
|
# 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.
|
|
|
|
import mock
|
|
import webob
|
|
|
|
from nova.api.openstack.compute.legacy_v2.contrib import volumes as volumes_v2
|
|
from nova.api.openstack.compute import volumes as volumes_v21
|
|
from nova import exception
|
|
from nova import test
|
|
from nova.tests.unit.api.openstack import fakes
|
|
from nova.volume import cinder
|
|
|
|
FAKE_UUID = 'aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa'
|
|
|
|
|
|
class SnapshotApiTestV21(test.NoDBTestCase):
|
|
controller = volumes_v21.SnapshotController()
|
|
validation_error = exception.ValidationError
|
|
|
|
def setUp(self):
|
|
super(SnapshotApiTestV21, self).setUp()
|
|
fakes.stub_out_networking(self)
|
|
fakes.stub_out_rate_limiting(self.stubs)
|
|
self.stubs.Set(cinder.API, "create_snapshot",
|
|
fakes.stub_snapshot_create)
|
|
self.stubs.Set(cinder.API, "create_snapshot_force",
|
|
fakes.stub_snapshot_create)
|
|
self.stubs.Set(cinder.API, "delete_snapshot",
|
|
fakes.stub_snapshot_delete)
|
|
self.stubs.Set(cinder.API, "get_snapshot", fakes.stub_snapshot_get)
|
|
self.stubs.Set(cinder.API, "get_all_snapshots",
|
|
fakes.stub_snapshot_get_all)
|
|
self.stubs.Set(cinder.API, "get", fakes.stub_volume_get)
|
|
self.req = fakes.HTTPRequest.blank('')
|
|
|
|
def _test_snapshot_create(self, force):
|
|
snapshot = {"volume_id": '12',
|
|
"force": force,
|
|
"display_name": "Snapshot Test Name",
|
|
"display_description": "Snapshot Test Desc"}
|
|
body = dict(snapshot=snapshot)
|
|
resp_dict = self.controller.create(self.req, body=body)
|
|
self.assertIn('snapshot', resp_dict)
|
|
self.assertEqual(snapshot['display_name'],
|
|
resp_dict['snapshot']['displayName'])
|
|
self.assertEqual(snapshot['display_description'],
|
|
resp_dict['snapshot']['displayDescription'])
|
|
self.assertEqual(snapshot['volume_id'],
|
|
resp_dict['snapshot']['volumeId'])
|
|
|
|
def test_snapshot_create(self):
|
|
self._test_snapshot_create(False)
|
|
|
|
def test_snapshot_create_force(self):
|
|
self._test_snapshot_create(True)
|
|
|
|
def test_snapshot_create_invalid_force_param(self):
|
|
body = {'snapshot': {'volume_id': '1',
|
|
'force': '**&&^^%%$$##@@'}}
|
|
self.assertRaises(self.validation_error,
|
|
self.controller.create, self.req, body=body)
|
|
|
|
def test_snapshot_delete(self):
|
|
snapshot_id = '123'
|
|
result = self.controller.delete(self.req, snapshot_id)
|
|
|
|
# NOTE: on v2.1, http status code is set as wsgi_code of API
|
|
# method instead of status_int in a response object.
|
|
if isinstance(self.controller, volumes_v21.SnapshotController):
|
|
status_int = self.controller.delete.wsgi_code
|
|
else:
|
|
status_int = result.status_int
|
|
self.assertEqual(202, status_int)
|
|
|
|
@mock.patch.object(cinder.API, 'delete_snapshot',
|
|
side_effect=exception.SnapshotNotFound(snapshot_id=FAKE_UUID))
|
|
def test_delete_snapshot_not_exists(self, mock_mr):
|
|
self.assertRaises(webob.exc.HTTPNotFound, self.controller.delete,
|
|
self.req, FAKE_UUID)
|
|
|
|
def test_snapshot_delete_invalid_id(self):
|
|
self.assertRaises(webob.exc.HTTPNotFound, self.controller.delete,
|
|
self.req, '-1')
|
|
|
|
def test_snapshot_show(self):
|
|
snapshot_id = '123'
|
|
resp_dict = self.controller.show(self.req, snapshot_id)
|
|
self.assertIn('snapshot', resp_dict)
|
|
self.assertEqual(str(snapshot_id), resp_dict['snapshot']['id'])
|
|
|
|
def test_snapshot_show_invalid_id(self):
|
|
self.assertRaises(webob.exc.HTTPNotFound, self.controller.show,
|
|
self.req, '-1')
|
|
|
|
def test_snapshot_detail(self):
|
|
resp_dict = self.controller.detail(self.req)
|
|
self.assertIn('snapshots', resp_dict)
|
|
resp_snapshots = resp_dict['snapshots']
|
|
self.assertEqual(3, len(resp_snapshots))
|
|
|
|
resp_snapshot = resp_snapshots.pop()
|
|
self.assertEqual(102, resp_snapshot['id'])
|
|
|
|
def test_snapshot_index(self):
|
|
resp_dict = self.controller.index(self.req)
|
|
self.assertIn('snapshots', resp_dict)
|
|
resp_snapshots = resp_dict['snapshots']
|
|
self.assertEqual(3, len(resp_snapshots))
|
|
|
|
|
|
class SnapshotApiTestV2(SnapshotApiTestV21):
|
|
controller = volumes_v2.SnapshotController()
|
|
validation_error = webob.exc.HTTPBadRequest
|