# 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 django import http from django.urls import reverse from mox3.mox import IsA from openstack_dashboard import api from openstack_dashboard.dashboards.admin.volume_types.qos_specs \ import views from openstack_dashboard.test import helpers as test class QosSpecsTests(test.BaseAdminViewTests): @test.create_stubs({api.cinder: ('qos_spec_get',), }) def test_manage_qos_spec(self): qos_spec = self.cinder_qos_specs.first() index_url = reverse( 'horizon:admin:volume_types:qos_specs:index', args=[qos_spec.id]) api.cinder.qos_spec_get(IsA(http.HttpRequest), qos_spec.id)\ .AndReturn(qos_spec) self.mox.ReplayAll() res = self.client.get(index_url) self.assertTemplateUsed( res, 'admin/volume_types/qos_specs/index.html') rows = res.context['table'].get_rows() specs = self.cinder_qos_specs.first().specs for row in rows: key = row.cells['key'].data self.assertIn(key, specs) self.assertEqual(row.cells['value'].data, specs.get(key)) @test.create_stubs({api.cinder: ('qos_spec_create',)}) def test_create_qos_spec(self): formData = {'name': 'qos-spec-1', 'consumer': 'back-end'} api.cinder.qos_spec_create(IsA(http.HttpRequest), formData['name'], {'consumer': formData['consumer']}).\ AndReturn(self.cinder_qos_specs.first()) self.mox.ReplayAll() res = self.client.post( reverse('horizon:admin:volume_types:create_qos_spec'), formData) redirect = reverse('horizon:admin:volume_types:index') self.assertNoFormErrors(res) self.assertRedirectsNoFollow(res, redirect) self.assertMessageCount(success=1) @test.create_stubs({api.cinder: ('volume_type_list_with_qos_associations', 'volume_encryption_type_list', 'qos_spec_list', 'qos_spec_delete',)}) def test_delete_qos_spec(self): qos_spec = self.cinder_qos_specs.first() formData = {'action': 'qos_specs__delete__%s' % qos_spec.id} api.cinder.volume_type_list_with_qos_associations( IsA(http.HttpRequest)).\ AndReturn(self.cinder_volume_types.list()) api.cinder.volume_encryption_type_list(IsA(http.HttpRequest))\ .AndReturn(self.cinder_volume_encryption_types.list()[0:1]) api.cinder.qos_spec_list(IsA(http.HttpRequest)).\ AndReturn(self.cinder_qos_specs.list()) api.cinder.qos_spec_delete(IsA(http.HttpRequest), str(qos_spec.id)) self.mox.ReplayAll() res = self.client.post( reverse('horizon:admin:volume_types:index'), formData) redirect = reverse('horizon:admin:volume_types:index') self.assertNoFormErrors(res) self.assertRedirectsNoFollow(res, redirect) self.assertMessageCount(success=1) @test.create_stubs({api.cinder: ('qos_spec_get', 'qos_spec_get_keys', 'qos_spec_set_keys',), }) def test_spec_edit(self): qos_spec = self.cinder_qos_specs.first() key = 'minIOPS' edit_url = reverse('horizon:admin:volume_types:qos_specs:edit', args=[qos_spec.id, key]) index_url = reverse('horizon:admin:volume_types:index') data = {'value': '9999'} qos_spec.specs[key] = data['value'] api.cinder.qos_spec_get(IsA(http.HttpRequest), qos_spec.id)\ .AndReturn(qos_spec) api.cinder.qos_spec_get_keys(IsA(http.HttpRequest), qos_spec.id, raw=True)\ .AndReturn(qos_spec) api.cinder.qos_spec_set_keys(IsA(http.HttpRequest), qos_spec.id, qos_spec.specs) self.mox.ReplayAll() resp = self.client.post(edit_url, data) self.assertEqual('admin/volume_types/qos_specs/edit.html', views.EditKeyValuePairView.template_name) self.assertEqual('horizon:admin:volume_types:qos_specs:edit', views.EditKeyValuePairView.submit_url) self.assertNoFormErrors(resp) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(resp, index_url) @test.create_stubs({api.cinder: ('qos_spec_get', 'qos_spec_set_keys',), }) def test_edit_consumer(self): qos_spec = self.cinder_qos_specs.first() # modify consumer to 'front-end' formData = {'consumer_choice': 'front-end'} edit_url = reverse( 'horizon:admin:volume_types:edit_qos_spec_consumer', args=[qos_spec.id]) api.cinder.qos_spec_get(IsA(http.HttpRequest), qos_spec.id).AndReturn(qos_spec) api.cinder.qos_spec_set_keys(IsA(http.HttpRequest), qos_spec.id, {'consumer': formData['consumer_choice']}) self.mox.ReplayAll() resp = self.client.post(edit_url, formData) redirect = reverse('horizon:admin:volume_types:index') self.assertNoFormErrors(resp) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(resp, redirect) @test.create_stubs({api.cinder: ('qos_spec_list', 'qos_spec_get', 'qos_spec_get_associations', 'volume_type_get', 'qos_spec_associate', 'qos_spec_disassociate'), }) def test_associate_qos_spec(self): volume_type = self.cinder_volume_types.first() volume_types = self.cinder_volume_types.list() qos_specs = self.cinder_qos_specs.list() # associate qos spec with volume type formData = {'qos_spec_choice': qos_specs[0].id} edit_url = reverse( 'horizon:admin:volume_types:manage_qos_spec_association', args=[volume_type.id]) # for maximum code coverage, this test swaps the QoS association # on one volume type moving the QoS assigned from 1 to 0 api.cinder.volume_type_get(IsA(http.HttpRequest), volume_type.id) \ .AndReturn(volume_type) api.cinder.qos_spec_list(IsA(http.HttpRequest)) \ .MultipleTimes().AndReturn(qos_specs) api.cinder.qos_spec_get_associations(IsA(http.HttpRequest), qos_specs[0].id) \ .AndReturn([]) api.cinder.qos_spec_get_associations(IsA(http.HttpRequest), qos_specs[1].id) \ .AndReturn(volume_types) api.cinder.qos_spec_get(IsA(http.HttpRequest), qos_specs[1].id).AndReturn(qos_specs[1]) api.cinder.qos_spec_disassociate(IsA(http.HttpRequest), qos_specs[1], volume_type.id) api.cinder.qos_spec_get(IsA(http.HttpRequest), qos_specs[0].id).AndReturn(qos_specs[0]) api.cinder.qos_spec_associate(IsA(http.HttpRequest), qos_specs[0], volume_type.id) self.mox.ReplayAll() resp = self.client.post(edit_url, formData) redirect = reverse('horizon:admin:volume_types:index') self.assertNoFormErrors(resp) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(resp, redirect)