d2f4e9717d
Move all modules under the next structure: - rally_openstack.common - rally_openstack.enviromnet - rally_openstack.task - rally_openstack.verification Change-Id: I41702d017cd49b117da3b8e12b19c7327229ae32
282 lines
11 KiB
Python
282 lines
11 KiB
Python
# Copyright 2014: Dassault Systemes
|
|
# 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 copy
|
|
from unittest import mock
|
|
|
|
import ddt
|
|
from rally.common import logging
|
|
from rally.task import context
|
|
|
|
from rally_openstack.task.contexts.quotas import quotas
|
|
from tests.unit import test
|
|
|
|
QUOTAS_PATH = "rally_openstack.task.contexts.quotas"
|
|
|
|
|
|
@ddt.ddt
|
|
class QuotasTestCase(test.TestCase):
|
|
|
|
def setUp(self):
|
|
super(QuotasTestCase, self).setUp()
|
|
self.unlimited = -1
|
|
self.context = {
|
|
"config": {
|
|
},
|
|
"tenants": {
|
|
"t1": {"credential": mock.MagicMock()},
|
|
"t2": {"credential": mock.MagicMock()}},
|
|
"admin": {"credential": mock.MagicMock()},
|
|
"task": mock.MagicMock()
|
|
}
|
|
|
|
@ddt.data(("cinder", "backup_gigabytes"),
|
|
("cinder", "backups"),
|
|
("cinder", "gigabytes"),
|
|
("cinder", "snapshots"),
|
|
("cinder", "volumes"),
|
|
("manila", "gigabytes"),
|
|
("manila", "share_networks"),
|
|
("manila", "shares"),
|
|
("manila", "snapshot_gigabytes"),
|
|
("manila", "snapshots"),
|
|
("neutron", "floatingip"),
|
|
("neutron", "health_monitor"),
|
|
("neutron", "network"),
|
|
("neutron", "pool"),
|
|
("neutron", "port"),
|
|
("neutron", "router"),
|
|
("neutron", "security_group"),
|
|
("neutron", "security_group_rule"),
|
|
("neutron", "subnet"),
|
|
("neutron", "vip"),
|
|
("nova", "cores"),
|
|
("nova", "fixed_ips"),
|
|
("nova", "floating_ips"),
|
|
("nova", "injected_file_content_bytes"),
|
|
("nova", "injected_file_path_bytes"),
|
|
("nova", "injected_files"),
|
|
("nova", "instances"),
|
|
("nova", "key_pairs"),
|
|
("nova", "metadata_items"),
|
|
("nova", "ram"),
|
|
("nova", "security_group_rules"),
|
|
("nova", "security_groups"),
|
|
("nova", "server_group_members"),
|
|
("nova", "server_groups"))
|
|
@ddt.unpack
|
|
def test_validate(self, group, parameter):
|
|
configs = [
|
|
({group: {parameter: self.unlimited}}, True),
|
|
({group: {parameter: 0}}, True),
|
|
({group: {parameter: 10000}}, True),
|
|
({group: {parameter: 2.5}}, False),
|
|
({group: {parameter: "-1"}}, False),
|
|
({group: {parameter: -2}}, False),
|
|
]
|
|
for config, valid in configs:
|
|
results = context.Context.validate(
|
|
"quotas", None, None, config, vtype="syntax")
|
|
if valid:
|
|
self.assertEqual([], results)
|
|
else:
|
|
self.assertGreater(len(results), 0)
|
|
|
|
@mock.patch("%s.quotas.osclients.Clients" % QUOTAS_PATH)
|
|
@mock.patch("%s.cinder_quotas.CinderQuotas" % QUOTAS_PATH)
|
|
@ddt.data(True, False)
|
|
def test_cinder_quotas(self, ex_users, mock_cinder_quotas, mock_clients):
|
|
cinder_quo = mock_cinder_quotas.return_value
|
|
ctx = copy.deepcopy(self.context)
|
|
if ex_users:
|
|
ctx["existing_users"] = None
|
|
ctx["config"]["quotas"] = {
|
|
"cinder": {
|
|
"volumes": self.unlimited,
|
|
"snapshots": self.unlimited,
|
|
"gigabytes": self.unlimited
|
|
}
|
|
}
|
|
|
|
tenants = ctx["tenants"]
|
|
cinder_quotas = ctx["config"]["quotas"]["cinder"]
|
|
cinder_quo.get.return_value = cinder_quotas
|
|
with quotas.Quotas(ctx) as quotas_ctx:
|
|
quotas_ctx.setup()
|
|
if ex_users:
|
|
self.assertEqual([mock.call(tenant) for tenant in tenants],
|
|
cinder_quo.get.call_args_list)
|
|
self.assertEqual([mock.call(tenant, **cinder_quotas)
|
|
for tenant in tenants],
|
|
cinder_quo.update.call_args_list)
|
|
mock_cinder_quotas.reset_mock()
|
|
|
|
if ex_users:
|
|
self.assertEqual([mock.call(tenant, **cinder_quotas)
|
|
for tenant in tenants],
|
|
cinder_quo.update.call_args_list)
|
|
else:
|
|
self.assertEqual([mock.call(tenant) for tenant in tenants],
|
|
cinder_quo.delete.call_args_list)
|
|
|
|
@mock.patch("%s.quotas.osclients.Clients" % QUOTAS_PATH)
|
|
@mock.patch("%s.nova_quotas.NovaQuotas" % QUOTAS_PATH)
|
|
@ddt.data(True, False)
|
|
def test_nova_quotas(self, ex_users, mock_nova_quotas, mock_clients):
|
|
nova_quo = mock_nova_quotas.return_value
|
|
ctx = copy.deepcopy(self.context)
|
|
if ex_users:
|
|
ctx["existing_users"] = None
|
|
|
|
ctx["config"]["quotas"] = {
|
|
"nova": {
|
|
"instances": self.unlimited,
|
|
"cores": self.unlimited,
|
|
"ram": self.unlimited,
|
|
"floating-ips": self.unlimited,
|
|
"fixed-ips": self.unlimited,
|
|
"metadata_items": self.unlimited,
|
|
"injected_files": self.unlimited,
|
|
"injected_file_content_bytes": self.unlimited,
|
|
"injected_file_path_bytes": self.unlimited,
|
|
"key_pairs": self.unlimited,
|
|
"security_groups": self.unlimited,
|
|
"security_group_rules": self.unlimited,
|
|
}
|
|
}
|
|
|
|
tenants = ctx["tenants"]
|
|
nova_quotas = ctx["config"]["quotas"]["nova"]
|
|
nova_quo.get.return_value = nova_quotas
|
|
with quotas.Quotas(ctx) as quotas_ctx:
|
|
quotas_ctx.setup()
|
|
if ex_users:
|
|
self.assertEqual([mock.call(tenant) for tenant in tenants],
|
|
nova_quo.get.call_args_list)
|
|
self.assertEqual([mock.call(tenant, **nova_quotas)
|
|
for tenant in tenants],
|
|
nova_quo.update.call_args_list)
|
|
mock_nova_quotas.reset_mock()
|
|
|
|
if ex_users:
|
|
self.assertEqual([mock.call(tenant, **nova_quotas)
|
|
for tenant in tenants],
|
|
nova_quo.update.call_args_list)
|
|
else:
|
|
self.assertEqual([mock.call(tenant) for tenant in tenants],
|
|
nova_quo.delete.call_args_list)
|
|
|
|
@mock.patch("%s.quotas.osclients.Clients" % QUOTAS_PATH)
|
|
@mock.patch("%s.neutron_quotas.NeutronQuotas" % QUOTAS_PATH)
|
|
@ddt.data(True, False)
|
|
def test_neutron_quotas(self, ex_users, mock_neutron_quotas, mock_clients):
|
|
neutron_quo = mock_neutron_quotas.return_value
|
|
ctx = copy.deepcopy(self.context)
|
|
if ex_users:
|
|
ctx["existing_users"] = None
|
|
|
|
ctx["config"]["quotas"] = {
|
|
"neutron": {
|
|
"network": self.unlimited,
|
|
"subnet": self.unlimited,
|
|
"port": self.unlimited,
|
|
"router": self.unlimited,
|
|
"floatingip": self.unlimited,
|
|
"security_group": self.unlimited,
|
|
"security_group_rule": self.unlimited
|
|
}
|
|
}
|
|
|
|
tenants = ctx["tenants"]
|
|
neutron_quotas = ctx["config"]["quotas"]["neutron"]
|
|
neutron_quo.get.return_value = neutron_quotas
|
|
with quotas.Quotas(ctx) as quotas_ctx:
|
|
quotas_ctx.setup()
|
|
if ex_users:
|
|
self.assertEqual([mock.call(tenant) for tenant in tenants],
|
|
neutron_quo.get.call_args_list)
|
|
self.assertEqual([mock.call(tenant, **neutron_quotas)
|
|
for tenant in tenants],
|
|
neutron_quo.update.call_args_list)
|
|
neutron_quo.reset_mock()
|
|
|
|
if ex_users:
|
|
self.assertEqual([mock.call(tenant, **neutron_quotas)
|
|
for tenant in tenants],
|
|
neutron_quo.update.call_args_list)
|
|
else:
|
|
self.assertEqual([mock.call(tenant) for tenant in tenants],
|
|
neutron_quo.delete.call_args_list)
|
|
|
|
@mock.patch("rally_openstack.task.contexts."
|
|
"quotas.quotas.osclients.Clients")
|
|
@mock.patch("rally_openstack.task.contexts."
|
|
"quotas.nova_quotas.NovaQuotas")
|
|
@mock.patch("rally_openstack.task.contexts."
|
|
"quotas.cinder_quotas.CinderQuotas")
|
|
@mock.patch("rally_openstack.task.contexts."
|
|
"quotas.neutron_quotas.NeutronQuotas")
|
|
def test_no_quotas(self, mock_neutron_quotas, mock_cinder_quotas,
|
|
mock_nova_quotas, mock_clients):
|
|
ctx = copy.deepcopy(self.context)
|
|
if "quotas" in ctx["config"]:
|
|
del ctx["config"]["quotas"]
|
|
|
|
with quotas.Quotas(ctx) as quotas_ctx:
|
|
quotas_ctx.setup()
|
|
self.assertFalse(mock_cinder_quotas.update.called)
|
|
self.assertFalse(mock_nova_quotas.update.called)
|
|
self.assertFalse(mock_neutron_quotas.update.called)
|
|
|
|
self.assertFalse(mock_cinder_quotas.delete.called)
|
|
self.assertFalse(mock_nova_quotas.delete.called)
|
|
self.assertFalse(mock_neutron_quotas.delete.called)
|
|
|
|
@ddt.data(
|
|
{"quotas_ctxt": {"nova": {"cpu": 1}},
|
|
"quotas_class_path": "nova_quotas.NovaQuotas"},
|
|
{"quotas_ctxt": {"neutron": {"network": 2}},
|
|
"quotas_class_path": "neutron_quotas.NeutronQuotas"},
|
|
{"quotas_ctxt": {"cinder": {"volumes": 3}},
|
|
"quotas_class_path": "cinder_quotas.CinderQuotas"},
|
|
{"quotas_ctxt": {"manila": {"shares": 4}},
|
|
"quotas_class_path": "manila_quotas.ManilaQuotas"},
|
|
{"quotas_ctxt": {"designate": {"domains": 5}},
|
|
"quotas_class_path": "designate_quotas.DesignateQuotas"},
|
|
)
|
|
@ddt.unpack
|
|
def test_exception_during_cleanup(self, quotas_ctxt, quotas_class_path):
|
|
quotas_path = "%s.%s" % (QUOTAS_PATH, quotas_class_path)
|
|
with mock.patch(quotas_path) as mock_quotas:
|
|
mock_quotas.return_value.update.side_effect = Exception
|
|
|
|
ctx = copy.deepcopy(self.context)
|
|
ctx["config"]["quotas"] = quotas_ctxt
|
|
|
|
quotas_instance = quotas.Quotas(ctx)
|
|
quotas_instance.original_quotas = []
|
|
for service in quotas_ctxt:
|
|
for tenant in self.context["tenants"]:
|
|
quotas_instance.original_quotas.append(
|
|
(service, tenant, quotas_ctxt[service]))
|
|
# NOTE(boris-42): ensure that cleanup didn't raise exceptions.
|
|
with logging.LogCatcher(quotas.LOG) as log:
|
|
quotas_instance.cleanup()
|
|
|
|
log.assertInLogs("Failed to restore quotas for tenant")
|
|
|
|
self.assertEqual(mock_quotas.return_value.update.call_count,
|
|
len(self.context["tenants"]))
|