415 lines
20 KiB
Python
415 lines
20 KiB
Python
# Copyright 2012 United States Government as represented by the
|
|
# Administrator of the National Aeronautics and Space Administration.
|
|
# All Rights Reserved.
|
|
#
|
|
# Copyright 2012 Nebula, Inc.
|
|
# Copyright (c) 2012 X.commerce, a business unit of eBay Inc.
|
|
#
|
|
# 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 __future__ import absolute_import
|
|
|
|
from django import http
|
|
from django.test.utils import override_settings
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from mox3.mox import IsA
|
|
|
|
from horizon import exceptions
|
|
from openstack_dashboard import api
|
|
from openstack_dashboard.api import cinder
|
|
from openstack_dashboard.test import helpers as test
|
|
from openstack_dashboard.usage import quotas
|
|
|
|
|
|
class QuotaTests(test.APITestCase):
|
|
|
|
def get_usages(self, with_volume=True, with_compute=True,
|
|
nova_quotas_enabled=True):
|
|
usages = {}
|
|
if with_compute:
|
|
# These are all nova fields; the neutron ones are named slightly
|
|
# differently and aren't included in here yet
|
|
if nova_quotas_enabled:
|
|
usages.update({
|
|
'injected_file_content_bytes': {'quota': 1},
|
|
'metadata_items': {'quota': 1},
|
|
'injected_files': {'quota': 1},
|
|
'security_groups': {'quota': 10},
|
|
'security_group_rules': {'quota': 20},
|
|
'fixed_ips': {'quota': 10},
|
|
'ram': {'available': 8976, 'used': 1024, 'quota': 10000},
|
|
'floating_ips': {'available': 0, 'used': 2, 'quota': 1},
|
|
'instances': {'available': 8, 'used': 2, 'quota': 10},
|
|
'cores': {'available': 8, 'used': 2, 'quota': 10},
|
|
'key_pairs': {'quota': 100},
|
|
'injected_file_path_bytes': {'quota': 255}
|
|
})
|
|
|
|
if with_volume:
|
|
usages.update({'volumes': {'available': 0, 'used': 4, 'quota': 1},
|
|
'snapshots': {'available': 0, 'used': 3,
|
|
'quota': 1},
|
|
'gigabytes': {'available': 600, 'used': 400,
|
|
'quota': 1000}})
|
|
return usages
|
|
|
|
def assertAvailableQuotasEqual(self, expected_usages, actual_usages):
|
|
expected_available = {key: value['available'] for key, value in
|
|
expected_usages.items() if 'available' in value}
|
|
actual_available = {key: value['available'] for key, value in
|
|
actual_usages.items() if 'available' in value}
|
|
self.assertEqual(expected_available, actual_available)
|
|
|
|
@test.create_stubs({api.nova: ('server_list',
|
|
'flavor_list',
|
|
'tenant_quota_get',),
|
|
api.neutron: ('tenant_floating_ip_list',
|
|
'floating_ip_supported'),
|
|
api.base: ('is_service_enabled',),
|
|
cinder: ('volume_list', 'volume_snapshot_list',
|
|
'tenant_quota_get',
|
|
'is_volume_service_enabled')})
|
|
def _test_tenant_quota_usages(self, nova_quotas_enabled=True,
|
|
with_compute=True, with_volume=True):
|
|
|
|
cinder.is_volume_service_enabled(IsA(http.HttpRequest)).AndReturn(
|
|
with_volume)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'network').AndReturn(False)
|
|
api.base.is_service_enabled(
|
|
IsA(http.HttpRequest), 'compute'
|
|
).MultipleTimes().AndReturn(with_compute)
|
|
if with_compute:
|
|
if nova_quotas_enabled:
|
|
servers = [s for s in self.servers.list()
|
|
if s.tenant_id == self.request.user.tenant_id]
|
|
api.nova.flavor_list(IsA(http.HttpRequest)) \
|
|
.AndReturn(self.flavors.list())
|
|
api.neutron.floating_ip_supported(IsA(http.HttpRequest)) \
|
|
.AndReturn(True)
|
|
api.neutron.tenant_floating_ip_list(IsA(http.HttpRequest)) \
|
|
.AndReturn(self.floating_ips.list())
|
|
search_opts = {'tenant_id': self.request.user.tenant_id}
|
|
api.nova.server_list(IsA(http.HttpRequest),
|
|
search_opts=search_opts) \
|
|
.AndReturn([servers, False])
|
|
api.nova.tenant_quota_get(IsA(http.HttpRequest), '1') \
|
|
.AndReturn(self.quotas.first())
|
|
|
|
if with_volume:
|
|
opts = {'all_tenants': 1,
|
|
'project_id': self.request.user.tenant_id}
|
|
cinder.volume_list(IsA(http.HttpRequest), opts) \
|
|
.AndReturn(self.volumes.list())
|
|
cinder.volume_snapshot_list(IsA(http.HttpRequest), opts) \
|
|
.AndReturn(self.cinder_volume_snapshots.list())
|
|
cinder.tenant_quota_get(IsA(http.HttpRequest), '1') \
|
|
.AndReturn(self.cinder_quotas.first())
|
|
|
|
self.mox.ReplayAll()
|
|
|
|
quota_usages = quotas.tenant_quota_usages(self.request)
|
|
expected_output = self.get_usages(
|
|
nova_quotas_enabled=nova_quotas_enabled, with_volume=with_volume,
|
|
with_compute=with_compute)
|
|
|
|
# Compare internal structure of usages to expected.
|
|
self.assertItemsEqual(expected_output, quota_usages.usages)
|
|
# Compare available resources
|
|
self.assertAvailableQuotasEqual(expected_output, quota_usages.usages)
|
|
|
|
def test_tenant_quota_usages(self):
|
|
self._test_tenant_quota_usages()
|
|
|
|
@override_settings(OPENSTACK_HYPERVISOR_FEATURES={'enable_quotas': False})
|
|
def test_tenant_quota_usages_wo_nova_quotas(self):
|
|
self._test_tenant_quota_usages(nova_quotas_enabled=False,
|
|
with_compute=True,
|
|
with_volume=False)
|
|
|
|
@override_settings(OPENSTACK_HYPERVISOR_FEATURES={'enable_quotas': False})
|
|
@test.create_stubs({api.base: ('is_service_enabled',),
|
|
cinder: ('is_volume_service_enabled',)})
|
|
def test_get_all_disabled_quotas(self):
|
|
cinder.is_volume_service_enabled(IsA(http.HttpRequest)).AndReturn(
|
|
False)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'network').AndReturn(False)
|
|
# Nova enabled but quotas disabled
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'compute').AndReturn(True)
|
|
self.mox.ReplayAll()
|
|
|
|
result_quotas = quotas.get_disabled_quotas(self.request)
|
|
expected_quotas = (quotas.CINDER_QUOTA_FIELDS |
|
|
quotas.NEUTRON_QUOTA_FIELDS |
|
|
quotas.NOVA_QUOTA_FIELDS)
|
|
self.assertItemsEqual(result_quotas, expected_quotas)
|
|
|
|
@test.create_stubs({api.nova: ('server_list',
|
|
'flavor_list',
|
|
'tenant_quota_get',),
|
|
api.neutron: ('tenant_floating_ip_list',
|
|
'floating_ip_supported'),
|
|
api.base: ('is_service_enabled',),
|
|
api.cinder: ('is_volume_service_enabled',)})
|
|
def test_tenant_quota_usages_without_volume(self):
|
|
servers = [s for s in self.servers.list()
|
|
if s.tenant_id == self.request.user.tenant_id]
|
|
|
|
api.cinder.is_volume_service_enabled(
|
|
IsA(http.HttpRequest)
|
|
).AndReturn(False)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'network').AndReturn(False)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'compute').MultipleTimes().AndReturn(True)
|
|
api.nova.flavor_list(IsA(http.HttpRequest)) \
|
|
.AndReturn(self.flavors.list())
|
|
api.nova.tenant_quota_get(IsA(http.HttpRequest), '1') \
|
|
.AndReturn(self.quotas.first())
|
|
api.neutron.floating_ip_supported(IsA(http.HttpRequest)) \
|
|
.AndReturn(True)
|
|
api.neutron.tenant_floating_ip_list(IsA(http.HttpRequest)) \
|
|
.AndReturn(self.floating_ips.list())
|
|
search_opts = {'tenant_id': self.request.user.tenant_id}
|
|
api.nova.server_list(IsA(http.HttpRequest), search_opts=search_opts) \
|
|
.AndReturn([servers, False])
|
|
|
|
self.mox.ReplayAll()
|
|
|
|
quota_usages = quotas.tenant_quota_usages(self.request)
|
|
expected_output = self.get_usages(with_volume=False)
|
|
|
|
# Compare internal structure of usages to expected.
|
|
self.assertItemsEqual(expected_output, quota_usages.usages)
|
|
|
|
# Make sure that the `in` operator and the `.get()` method
|
|
# behave as expected
|
|
self.assertIn('ram', quota_usages)
|
|
self.assertIsNotNone(quota_usages.get('ram'))
|
|
|
|
@test.create_stubs({api.nova: ('server_list',
|
|
'flavor_list',
|
|
'tenant_quota_get',),
|
|
api.neutron: ('tenant_floating_ip_list',
|
|
'floating_ip_supported'),
|
|
api.base: ('is_service_enabled',),
|
|
api.cinder: ('is_volume_service_enabled',)})
|
|
def test_tenant_quota_usages_no_instances_running(self):
|
|
api.cinder.is_volume_service_enabled(
|
|
IsA(http.HttpRequest)
|
|
).AndReturn(False)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'network').AndReturn(False)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'compute').MultipleTimes().AndReturn(True)
|
|
api.nova.flavor_list(IsA(http.HttpRequest)) \
|
|
.AndReturn(self.flavors.list())
|
|
api.nova.tenant_quota_get(IsA(http.HttpRequest), '1') \
|
|
.AndReturn(self.quotas.first())
|
|
api.neutron.floating_ip_supported(IsA(http.HttpRequest)) \
|
|
.AndReturn(True)
|
|
api.neutron.tenant_floating_ip_list(IsA(http.HttpRequest)) \
|
|
.AndReturn([])
|
|
search_opts = {'tenant_id': self.request.user.tenant_id}
|
|
api.nova.server_list(IsA(http.HttpRequest), search_opts=search_opts) \
|
|
.AndReturn([[], False])
|
|
|
|
self.mox.ReplayAll()
|
|
|
|
quota_usages = quotas.tenant_quota_usages(self.request)
|
|
expected_output = self.get_usages(with_volume=False)
|
|
|
|
expected_output.update({
|
|
'ram': {'available': 10000, 'used': 0, 'quota': 10000},
|
|
'floating_ips': {'available': 1, 'used': 0, 'quota': 1},
|
|
'instances': {'available': 10, 'used': 0, 'quota': 10},
|
|
'cores': {'available': 10, 'used': 0, 'quota': 10}})
|
|
|
|
# Compare internal structure of usages to expected.
|
|
self.assertItemsEqual(expected_output, quota_usages.usages)
|
|
|
|
@test.create_stubs({api.nova: ('server_list',
|
|
'flavor_list',
|
|
'tenant_quota_get',),
|
|
api.neutron: ('tenant_floating_ip_list',
|
|
'floating_ip_supported'),
|
|
api.base: ('is_service_enabled',),
|
|
cinder: ('volume_list', 'volume_snapshot_list',
|
|
'tenant_quota_get',
|
|
'is_volume_service_enabled')})
|
|
def test_tenant_quota_usages_unlimited_quota(self):
|
|
inf_quota = self.quotas.first()
|
|
inf_quota['ram'] = -1
|
|
servers = [s for s in self.servers.list()
|
|
if s.tenant_id == self.request.user.tenant_id]
|
|
|
|
cinder.is_volume_service_enabled(
|
|
IsA(http.HttpRequest)
|
|
).AndReturn(True)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'network').AndReturn(False)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'compute').MultipleTimes().AndReturn(True)
|
|
api.nova.flavor_list(IsA(http.HttpRequest)) \
|
|
.AndReturn(self.flavors.list())
|
|
api.nova.tenant_quota_get(IsA(http.HttpRequest), '1') \
|
|
.AndReturn(inf_quota)
|
|
api.neutron.floating_ip_supported(IsA(http.HttpRequest)) \
|
|
.AndReturn(True)
|
|
api.neutron.tenant_floating_ip_list(IsA(http.HttpRequest)) \
|
|
.AndReturn(self.floating_ips.list())
|
|
search_opts = {'tenant_id': self.request.user.tenant_id}
|
|
api.nova.server_list(IsA(http.HttpRequest), search_opts=search_opts) \
|
|
.AndReturn([servers, False])
|
|
opts = {'all_tenants': 1, 'project_id': self.request.user.tenant_id}
|
|
cinder.volume_list(IsA(http.HttpRequest), opts) \
|
|
.AndReturn(self.volumes.list())
|
|
cinder.volume_snapshot_list(IsA(http.HttpRequest), opts) \
|
|
.AndReturn(self.cinder_volume_snapshots.list())
|
|
cinder.tenant_quota_get(IsA(http.HttpRequest), '1') \
|
|
.AndReturn(self.cinder_quotas.first())
|
|
|
|
self.mox.ReplayAll()
|
|
|
|
quota_usages = quotas.tenant_quota_usages(self.request)
|
|
expected_output = self.get_usages()
|
|
expected_output.update({'ram': {'available': float("inf"),
|
|
'used': 1024,
|
|
'quota': float("inf")}})
|
|
|
|
# Compare internal structure of usages to expected.
|
|
self.assertItemsEqual(expected_output, quota_usages.usages)
|
|
|
|
@test.create_stubs({api.nova: ('server_list',
|
|
'flavor_list',
|
|
'tenant_quota_get',),
|
|
api.neutron: ('tenant_floating_ip_list',
|
|
'floating_ip_supported'),
|
|
api.base: ('is_service_enabled',),
|
|
cinder: ('volume_list', 'volume_snapshot_list',
|
|
'tenant_quota_get',
|
|
'is_volume_service_enabled')})
|
|
def test_tenant_quota_usages_neutron_fip_disabled(self):
|
|
servers = [s for s in self.servers.list()
|
|
if s.tenant_id == self.request.user.tenant_id]
|
|
|
|
cinder.is_volume_service_enabled(
|
|
IsA(http.HttpRequest)
|
|
).AndReturn(True)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'network').AndReturn(False)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'compute').MultipleTimes().AndReturn(True)
|
|
api.nova.flavor_list(IsA(http.HttpRequest)) \
|
|
.AndReturn(self.flavors.list())
|
|
api.nova.tenant_quota_get(IsA(http.HttpRequest), '1') \
|
|
.AndReturn(self.quotas.first())
|
|
api.neutron.floating_ip_supported(IsA(http.HttpRequest)) \
|
|
.AndReturn(False)
|
|
search_opts = {'tenant_id': self.request.user.tenant_id}
|
|
api.nova.server_list(IsA(http.HttpRequest), search_opts=search_opts) \
|
|
.AndReturn([servers, False])
|
|
opts = {'all_tenants': 1, 'project_id': self.request.user.tenant_id}
|
|
cinder.volume_list(IsA(http.HttpRequest), opts) \
|
|
.AndReturn(self.volumes.list())
|
|
cinder.volume_snapshot_list(IsA(http.HttpRequest), opts) \
|
|
.AndReturn(self.cinder_volume_snapshots.list())
|
|
cinder.tenant_quota_get(IsA(http.HttpRequest), '1') \
|
|
.AndReturn(self.cinder_quotas.first())
|
|
|
|
self.mox.ReplayAll()
|
|
|
|
quota_usages = quotas.tenant_quota_usages(self.request)
|
|
expected_output = self.get_usages()
|
|
expected_output['floating_ips']['used'] = 0
|
|
expected_output['floating_ips']['available'] = 1
|
|
|
|
# Compare internal structure of usages to expected.
|
|
self.assertItemsEqual(expected_output, quota_usages.usages)
|
|
|
|
@test.create_stubs({cinder: ('volume_list',),
|
|
exceptions: ('handle',)})
|
|
def test_get_tenant_volume_usages_cinder_exception(self):
|
|
cinder.volume_list(IsA(http.HttpRequest)) \
|
|
.AndRaise(cinder.cinder_exception.ClientException('test'))
|
|
exceptions.handle(IsA(http.HttpRequest),
|
|
_("Unable to retrieve volume limit information."))
|
|
self.mox.ReplayAll()
|
|
|
|
quotas._get_tenant_volume_usages(self.request, {}, set(), None)
|
|
|
|
@test.create_stubs({api.base: ('is_service_enabled',),
|
|
api.cinder: ('tenant_quota_get',
|
|
'is_volume_service_enabled'),
|
|
exceptions: ('handle',)})
|
|
def test_get_quota_data_cinder_exception(self):
|
|
api.cinder.is_volume_service_enabled(
|
|
IsA(http.HttpRequest)
|
|
).AndReturn(True)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'network').AndReturn(False)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'compute').AndReturn(False)
|
|
api.cinder.tenant_quota_get(IsA(http.HttpRequest), '1') \
|
|
.AndRaise(cinder.cinder_exception.ClientException('test'))
|
|
exceptions.handle(IsA(http.HttpRequest),
|
|
_("Unable to retrieve volume limit information."))
|
|
self.mox.ReplayAll()
|
|
|
|
quotas._get_quota_data(self.request, 'tenant_quota_get')
|
|
|
|
@test.create_stubs({api.base: ('is_service_enabled',),
|
|
api.cinder: ('tenant_absolute_limits',
|
|
'is_volume_service_enabled'),
|
|
exceptions: ('handle',)})
|
|
def test_tenant_limit_usages_cinder_exception(self):
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'compute').AndReturn(False)
|
|
api.cinder.is_volume_service_enabled(
|
|
IsA(http.HttpRequest)
|
|
).AndReturn(True)
|
|
api.cinder.tenant_absolute_limits(IsA(http.HttpRequest)) \
|
|
.AndRaise(cinder.cinder_exception.ClientException('test'))
|
|
exceptions.handle(IsA(http.HttpRequest),
|
|
_("Unable to retrieve volume limit information."))
|
|
self.mox.ReplayAll()
|
|
quotas.tenant_limit_usages(self.request)
|
|
|
|
@test.create_stubs({api.neutron: ('is_router_enabled',
|
|
'is_extension_supported',
|
|
'is_quotas_extension_supported',),
|
|
api.cinder: ('is_volume_service_enabled',),
|
|
api.base: ('is_service_enabled',)})
|
|
def test_get_disabled_quotas_router_disabled(self):
|
|
api.cinder.is_volume_service_enabled(
|
|
IsA(http.HttpRequest)
|
|
).AndReturn(True)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'network').AndReturn(True)
|
|
api.neutron.is_extension_supported(IsA(http.HttpRequest),
|
|
'security-group').AndReturn(True)
|
|
api.neutron.is_router_enabled(IsA(http.HttpRequest)).AndReturn(False)
|
|
api.neutron.is_quotas_extension_supported(IsA(http.HttpRequest)) \
|
|
.AndReturn(True)
|
|
api.base.is_service_enabled(IsA(http.HttpRequest),
|
|
'compute').AndReturn(True)
|
|
|
|
self.mox.ReplayAll()
|
|
|
|
disabled_quotas = quotas.get_disabled_quotas(self.request)
|
|
expected = set(['floating_ips', 'fixed_ips', 'security_groups',
|
|
'security_group_rules', 'router', 'floatingip'])
|
|
self.assertEqual(expected, disabled_quotas)
|