14e779bbac
Now that we no longer support py27, we can use the standard library unittest.mock module instead of the third party mock lib. Change-Id: I2de669d8e89b8daeb7ee5405ffab35af6307c40b
200 lines
8.5 KiB
Python
200 lines
8.5 KiB
Python
# 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 unittest import mock
|
|
|
|
from django.urls import reverse
|
|
|
|
from openstack_dashboard.api import cinder
|
|
from openstack_dashboard.test import helpers as test
|
|
|
|
|
|
INDEX_URL = reverse('horizon:project:vg_snapshots:index')
|
|
|
|
|
|
class GroupSnapshotTests(test.TestCase):
|
|
@mock.patch.object(cinder, 'group_snapshot_get')
|
|
@mock.patch.object(cinder, 'group_create_from_source')
|
|
def test_create_group_from_snapshot(self,
|
|
mock_group_create_from_source,
|
|
mock_group_snapshot_get):
|
|
group = self.cinder_groups.first()
|
|
vg_snapshot = self.cinder_group_snapshots.first()
|
|
formData = {'vg_snapshot_id': vg_snapshot.id,
|
|
'name': 'test VG SS Create',
|
|
'description': 'test desc'}
|
|
|
|
mock_group_snapshot_get.return_value = vg_snapshot
|
|
mock_group_create_from_source.return_value = group
|
|
|
|
url = reverse('horizon:project:vg_snapshots:create_group',
|
|
args=[vg_snapshot.id])
|
|
res = self.client.post(url, formData)
|
|
self.assertNoFormErrors(res)
|
|
self.assertRedirectsNoFollow(
|
|
res, reverse('horizon:project:volume_groups:index'))
|
|
|
|
mock_group_snapshot_get.assert_called_once_with(
|
|
test.IsHttpRequest(), vg_snapshot.id)
|
|
mock_group_create_from_source.assert_called_once_with(
|
|
test.IsHttpRequest(),
|
|
formData['name'],
|
|
group_snapshot_id=formData['vg_snapshot_id'],
|
|
description=formData['description'])
|
|
|
|
@mock.patch.object(cinder, 'group_snapshot_get')
|
|
@mock.patch.object(cinder, 'group_create_from_source')
|
|
def test_create_group_from_snapshot_exception(
|
|
self,
|
|
mock_group_create_from_source,
|
|
mock_group_snapshot_get):
|
|
vg_snapshot = self.cinder_group_snapshots.first()
|
|
new_cg_name = 'test VG SS Create'
|
|
formData = {'vg_snapshot_id': vg_snapshot.id,
|
|
'name': new_cg_name,
|
|
'description': 'test desc'}
|
|
|
|
mock_group_snapshot_get.return_value = vg_snapshot
|
|
mock_group_create_from_source.side_effect = \
|
|
self.exceptions.cinder
|
|
|
|
url = reverse('horizon:project:vg_snapshots:create_group',
|
|
args=[vg_snapshot.id])
|
|
res = self.client.post(url, formData)
|
|
self.assertNoFormErrors(res)
|
|
# There are a bunch of backslashes for formatting in the message from
|
|
# the response, so remove them when validating the error message.
|
|
self.assertIn('Unable to create group "%s" from snapshot.'
|
|
% new_cg_name,
|
|
res.cookies.output().replace('\\', ''))
|
|
self.assertRedirectsNoFollow(res, INDEX_URL)
|
|
|
|
mock_group_snapshot_get.assert_called_once_with(
|
|
test.IsHttpRequest(), vg_snapshot.id)
|
|
mock_group_create_from_source.assert_called_once_with(
|
|
test.IsHttpRequest(),
|
|
formData['name'],
|
|
group_snapshot_id=formData['vg_snapshot_id'],
|
|
description=formData['description'])
|
|
|
|
@mock.patch.object(cinder, 'group_snapshot_list')
|
|
@mock.patch.object(cinder, 'group_snapshot_delete')
|
|
@mock.patch.object(cinder, 'group_list')
|
|
def test_delete_group_snapshot(self,
|
|
mock_group_list,
|
|
mock_group_snapshot_delete,
|
|
mock_group_snapshot_list):
|
|
vg_snapshots = self.cinder_group_snapshots.list()
|
|
vg_snapshot = self.cinder_group_snapshots.first()
|
|
|
|
mock_group_snapshot_list.return_value = vg_snapshots
|
|
mock_group_snapshot_delete.return_value = None
|
|
mock_group_list.return_value = self.cinder_groups.list()
|
|
|
|
form_data = {'action': 'volume_vg_snapshots__delete_vg_snapshot__%s'
|
|
% vg_snapshot.id}
|
|
res = self.client.post(INDEX_URL, form_data, follow=True)
|
|
self.assertEqual(res.status_code, 200)
|
|
self.assertIn("Scheduled deletion of Snapshot: %s" % vg_snapshot.name,
|
|
[m.message for m in res.context['messages']])
|
|
|
|
self.assert_mock_multiple_calls_with_same_arguments(
|
|
mock_group_snapshot_list, 2,
|
|
mock.call(test.IsHttpRequest()))
|
|
mock_group_snapshot_delete.assert_called_once_with(
|
|
test.IsHttpRequest(), vg_snapshot.id)
|
|
self.assert_mock_multiple_calls_with_same_arguments(
|
|
mock_group_list, 2,
|
|
mock.call(test.IsHttpRequest()))
|
|
|
|
@mock.patch.object(cinder, 'group_snapshot_list')
|
|
@mock.patch.object(cinder, 'group_snapshot_delete')
|
|
@mock.patch.object(cinder, 'group_list')
|
|
def test_delete_group_snapshot_exception(self,
|
|
mock_group_list,
|
|
mock_group_snapshot_delete,
|
|
mock_group_snapshot_list):
|
|
vg_snapshots = self.cinder_group_snapshots.list()
|
|
vg_snapshot = self.cinder_group_snapshots.first()
|
|
|
|
mock_group_snapshot_list.return_value = vg_snapshots
|
|
mock_group_snapshot_delete.side_effect = self.exceptions.cinder
|
|
mock_group_list.return_value = self.cinder_groups.list()
|
|
|
|
form_data = {'action': 'volume_vg_snapshots__delete_vg_snapshot__%s'
|
|
% vg_snapshot.id}
|
|
res = self.client.post(INDEX_URL, form_data, follow=True)
|
|
self.assertEqual(res.status_code, 200)
|
|
self.assertIn("Unable to delete snapshot: %s" % vg_snapshot.name,
|
|
[m.message for m in res.context['messages']])
|
|
|
|
self.assert_mock_multiple_calls_with_same_arguments(
|
|
mock_group_snapshot_list, 2,
|
|
mock.call(test.IsHttpRequest()))
|
|
mock_group_snapshot_delete.assert_called_once_with(
|
|
test.IsHttpRequest(), vg_snapshot.id)
|
|
self.assert_mock_multiple_calls_with_same_arguments(
|
|
mock_group_list, 2,
|
|
mock.call(test.IsHttpRequest()))
|
|
|
|
@mock.patch.object(cinder, 'group_snapshot_get')
|
|
@mock.patch.object(cinder, 'group_get')
|
|
@mock.patch.object(cinder, 'volume_type_get')
|
|
@mock.patch.object(cinder, 'volume_list')
|
|
def test_detail_view(self,
|
|
mock_volume_list,
|
|
mock_volume_type_get,
|
|
mock_group_get,
|
|
mock_group_snapshot_get):
|
|
vg_snapshot = self.cinder_group_snapshots.first()
|
|
group = self.cinder_groups.first()
|
|
volume_type = self.cinder_volume_types.first()
|
|
volumes = self.cinder_volumes.list()
|
|
|
|
mock_group_snapshot_get.return_value = vg_snapshot
|
|
mock_group_get.return_value = group
|
|
mock_volume_type_get.return_value = volume_type
|
|
mock_volume_list.return_value = volumes
|
|
|
|
url = reverse(
|
|
'horizon:project:vg_snapshots:detail',
|
|
args=[vg_snapshot.id])
|
|
res = self.client.get(url)
|
|
self.assertNoFormErrors(res)
|
|
self.assertEqual(res.status_code, 200)
|
|
|
|
mock_group_snapshot_get.assert_called_once_with(
|
|
test.IsHttpRequest(), vg_snapshot.id)
|
|
mock_group_get.assert_called_once_with(
|
|
test.IsHttpRequest(), group.id)
|
|
mock_volume_type_get.assert_called_once_with(
|
|
test.IsHttpRequest(), volume_type.id)
|
|
search_opts = {'group_id': group.id}
|
|
mock_volume_list.assert_called_once_with(
|
|
test.IsHttpRequest(), search_opts=search_opts)
|
|
|
|
@mock.patch.object(cinder, 'group_snapshot_get')
|
|
def test_detail_view_with_exception(self, mock_group_snapshot_get):
|
|
vg_snapshot = self.cinder_group_snapshots.first()
|
|
|
|
mock_group_snapshot_get.side_effect = self.exceptions.cinder
|
|
|
|
url = reverse(
|
|
'horizon:project:vg_snapshots:detail',
|
|
args=[vg_snapshot.id])
|
|
res = self.client.get(url)
|
|
self.assertNoFormErrors(res)
|
|
self.assertRedirectsNoFollow(res, INDEX_URL)
|
|
|
|
mock_group_snapshot_get.assert_called_once_with(
|
|
test.IsHttpRequest(), vg_snapshot.id)
|