d1442a6aec
Remove quota_class param from all of nova.quota resulting in no more quota_class code in nova.quota and no tests in test_quota. Change-Id: I1e788414f2f86d846612d3c4e909caae46273f4e
2255 lines
86 KiB
Python
2255 lines
86 KiB
Python
# Copyright 2010 United States Government as represented by the
|
|
# Administrator of the National Aeronautics and Space Administration.
|
|
# 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 datetime
|
|
|
|
from oslo.config import cfg
|
|
|
|
from nova import compute
|
|
from nova.compute import flavors
|
|
from nova import context
|
|
from nova import db
|
|
from nova.db.sqlalchemy import api as sqa_api
|
|
from nova.db.sqlalchemy import models as sqa_models
|
|
from nova import exception
|
|
from nova.openstack.common import timeutils
|
|
from nova import quota
|
|
from nova.scheduler import driver as scheduler_driver
|
|
from nova.scheduler import rpcapi as scheduler_rpcapi
|
|
from nova import test
|
|
import nova.tests.image.fake
|
|
|
|
CONF = cfg.CONF
|
|
CONF.import_opt('compute_driver', 'nova.virt.driver')
|
|
|
|
|
|
class QuotaIntegrationTestCase(test.TestCase):
|
|
|
|
def setUp(self):
|
|
super(QuotaIntegrationTestCase, self).setUp()
|
|
self.flags(compute_driver='nova.virt.fake.FakeDriver',
|
|
quota_instances=2,
|
|
quota_cores=4,
|
|
quota_floating_ips=1,
|
|
network_manager='nova.network.manager.FlatDHCPManager')
|
|
|
|
# Apparently needed by the RPC tests...
|
|
self.network = self.start_service('network')
|
|
|
|
self.user_id = 'admin'
|
|
self.project_id = 'admin'
|
|
self.context = context.RequestContext(self.user_id,
|
|
self.project_id,
|
|
is_admin=True)
|
|
|
|
nova.tests.image.fake.stub_out_image_service(self.stubs)
|
|
|
|
self.compute_api = compute.API()
|
|
|
|
def fake_run_instance(self, ctxt, request_spec, *args):
|
|
"""Stub out the scheduler creating the instance entry."""
|
|
instance = scheduler_driver.Scheduler().create_instance_db_entry(
|
|
context, request_spec, None)
|
|
return [scheduler_driver.encode_instance(instance)]
|
|
|
|
self.stubs.Set(scheduler_rpcapi.SchedulerAPI, 'run_instance',
|
|
fake_run_instance)
|
|
|
|
def tearDown(self):
|
|
super(QuotaIntegrationTestCase, self).tearDown()
|
|
nova.tests.image.fake.FakeImageService_reset()
|
|
|
|
def _create_instance(self, cores=2):
|
|
"""Create a test instance."""
|
|
inst = {}
|
|
inst['image_id'] = 'cedef40a-ed67-4d10-800e-17455edce175'
|
|
inst['reservation_id'] = 'r-fakeres'
|
|
inst['user_id'] = self.user_id
|
|
inst['project_id'] = self.project_id
|
|
inst['instance_type_id'] = '3' # m1.large
|
|
inst['vcpus'] = cores
|
|
return db.instance_create(self.context, inst)
|
|
|
|
def test_too_many_instances(self):
|
|
instance_uuids = []
|
|
for i in range(CONF.quota_instances):
|
|
instance = self._create_instance()
|
|
instance_uuids.append(instance['uuid'])
|
|
inst_type = flavors.get_flavor_by_name('m1.small')
|
|
image_uuid = 'cedef40a-ed67-4d10-800e-17455edce175'
|
|
try:
|
|
self.compute_api.create(self.context, min_count=1, max_count=1,
|
|
instance_type=inst_type,
|
|
image_href=image_uuid)
|
|
except exception.QuotaError as e:
|
|
expected_kwargs = {'code': 413, 'resource': 'cores', 'req': 1,
|
|
'used': 4, 'allowed': 4, 'overs': 'cores,instances'}
|
|
self.assertEqual(e.kwargs, expected_kwargs)
|
|
else:
|
|
self.fail('Expected QuotaError exception')
|
|
for instance_uuid in instance_uuids:
|
|
db.instance_destroy(self.context, instance_uuid)
|
|
|
|
def test_too_many_cores(self):
|
|
instance = self._create_instance(cores=4)
|
|
inst_type = flavors.get_flavor_by_name('m1.small')
|
|
image_uuid = 'cedef40a-ed67-4d10-800e-17455edce175'
|
|
try:
|
|
self.compute_api.create(self.context, min_count=1, max_count=1,
|
|
instance_type=inst_type,
|
|
image_href=image_uuid)
|
|
except exception.QuotaError as e:
|
|
expected_kwargs = {'code': 413, 'resource': 'cores', 'req': 1,
|
|
'used': 4, 'allowed': 4, 'overs': 'cores'}
|
|
self.assertEqual(e.kwargs, expected_kwargs)
|
|
else:
|
|
self.fail('Expected QuotaError exception')
|
|
db.instance_destroy(self.context, instance['uuid'])
|
|
|
|
def test_many_cores_with_unlimited_quota(self):
|
|
# Setting cores quota to unlimited:
|
|
self.flags(quota_cores=-1)
|
|
instance = self._create_instance(cores=4)
|
|
inst_type = flavors.get_flavor_by_name('m1.small')
|
|
image_uuid = 'cedef40a-ed67-4d10-800e-17455edce175'
|
|
db.instance_destroy(self.context, instance['uuid'])
|
|
|
|
def test_too_many_addresses(self):
|
|
address = '192.168.0.100'
|
|
db.floating_ip_create(context.get_admin_context(),
|
|
{'address': address,
|
|
'project_id': self.project_id})
|
|
self.assertRaises(exception.QuotaError,
|
|
self.network.allocate_floating_ip,
|
|
self.context,
|
|
self.project_id)
|
|
db.floating_ip_destroy(context.get_admin_context(), address)
|
|
|
|
def test_auto_assigned(self):
|
|
address = '192.168.0.100'
|
|
db.floating_ip_create(context.get_admin_context(),
|
|
{'address': address,
|
|
'project_id': self.project_id})
|
|
# auto allocated addresses should not be counted
|
|
self.assertRaises(exception.NoMoreFloatingIps,
|
|
self.network.allocate_floating_ip,
|
|
self.context,
|
|
self.project_id,
|
|
True)
|
|
db.floating_ip_destroy(context.get_admin_context(), address)
|
|
|
|
def test_too_many_metadata_items(self):
|
|
metadata = {}
|
|
for i in range(CONF.quota_metadata_items + 1):
|
|
metadata['key%s' % i] = 'value%s' % i
|
|
inst_type = flavors.get_flavor_by_name('m1.small')
|
|
image_uuid = 'cedef40a-ed67-4d10-800e-17455edce175'
|
|
self.assertRaises(exception.QuotaError, self.compute_api.create,
|
|
self.context,
|
|
min_count=1,
|
|
max_count=1,
|
|
instance_type=inst_type,
|
|
image_href=image_uuid,
|
|
metadata=metadata)
|
|
|
|
def _create_with_injected_files(self, files):
|
|
api = self.compute_api
|
|
inst_type = flavors.get_flavor_by_name('m1.small')
|
|
image_uuid = 'cedef40a-ed67-4d10-800e-17455edce175'
|
|
api.create(self.context, min_count=1, max_count=1,
|
|
instance_type=inst_type, image_href=image_uuid,
|
|
injected_files=files)
|
|
|
|
def test_no_injected_files(self):
|
|
api = self.compute_api
|
|
inst_type = flavors.get_flavor_by_name('m1.small')
|
|
image_uuid = 'cedef40a-ed67-4d10-800e-17455edce175'
|
|
api.create(self.context,
|
|
instance_type=inst_type,
|
|
image_href=image_uuid)
|
|
|
|
def test_max_injected_files(self):
|
|
files = []
|
|
for i in xrange(CONF.quota_injected_files):
|
|
files.append(('/my/path%d' % i, 'config = test\n'))
|
|
self._create_with_injected_files(files) # no QuotaError
|
|
|
|
def test_too_many_injected_files(self):
|
|
files = []
|
|
for i in xrange(CONF.quota_injected_files + 1):
|
|
files.append(('/my/path%d' % i, 'my\ncontent%d\n' % i))
|
|
self.assertRaises(exception.QuotaError,
|
|
self._create_with_injected_files, files)
|
|
|
|
def test_max_injected_file_content_bytes(self):
|
|
max = CONF.quota_injected_file_content_bytes
|
|
content = ''.join(['a' for i in xrange(max)])
|
|
files = [('/test/path', content)]
|
|
self._create_with_injected_files(files) # no QuotaError
|
|
|
|
def test_too_many_injected_file_content_bytes(self):
|
|
max = CONF.quota_injected_file_content_bytes
|
|
content = ''.join(['a' for i in xrange(max + 1)])
|
|
files = [('/test/path', content)]
|
|
self.assertRaises(exception.QuotaError,
|
|
self._create_with_injected_files, files)
|
|
|
|
def test_max_injected_file_path_bytes(self):
|
|
max = CONF.quota_injected_file_path_bytes
|
|
path = ''.join(['a' for i in xrange(max)])
|
|
files = [(path, 'config = quotatest')]
|
|
self._create_with_injected_files(files) # no QuotaError
|
|
|
|
def test_too_many_injected_file_path_bytes(self):
|
|
max = CONF.quota_injected_file_path_bytes
|
|
path = ''.join(['a' for i in xrange(max + 1)])
|
|
files = [(path, 'config = quotatest')]
|
|
self.assertRaises(exception.QuotaError,
|
|
self._create_with_injected_files, files)
|
|
|
|
def test_reservation_expire(self):
|
|
self.useFixture(test.TimeOverride())
|
|
|
|
def assertInstancesReserved(reserved):
|
|
result = quota.QUOTAS.get_project_quotas(self.context,
|
|
self.context.project_id)
|
|
self.assertEqual(result['instances']['reserved'], reserved)
|
|
|
|
quota.QUOTAS.reserve(self.context,
|
|
expire=60,
|
|
instances=2)
|
|
|
|
assertInstancesReserved(2)
|
|
|
|
timeutils.advance_time_seconds(80)
|
|
|
|
quota.QUOTAS.expire(self.context)
|
|
|
|
assertInstancesReserved(0)
|
|
|
|
|
|
class FakeContext(object):
|
|
def __init__(self, project_id):
|
|
self.is_admin = False
|
|
self.user_id = 'fake_user'
|
|
self.project_id = project_id
|
|
self.read_deleted = 'no'
|
|
|
|
def elevated(self):
|
|
elevated = self.__class__(self.project_id)
|
|
elevated.is_admin = True
|
|
return elevated
|
|
|
|
|
|
class FakeDriver(object):
|
|
def __init__(self, by_project=None, by_user=None, by_class=None,
|
|
reservations=None):
|
|
self.called = []
|
|
self.by_project = by_project or {}
|
|
self.by_user = by_user or {}
|
|
self.by_class = by_class or {}
|
|
self.reservations = reservations or []
|
|
|
|
def get_by_project_and_user(self, context, project_id, user_id, resource):
|
|
self.called.append(('get_by_project_and_user',
|
|
context, project_id, user_id, resource))
|
|
try:
|
|
return self.by_user[user_id][resource]
|
|
except KeyError:
|
|
raise exception.ProjectUserQuotaNotFound(project_id=project_id,
|
|
user_id=user_id)
|
|
|
|
def get_by_project(self, context, project_id, resource):
|
|
self.called.append(('get_by_project', context, project_id, resource))
|
|
try:
|
|
return self.by_project[project_id][resource]
|
|
except KeyError:
|
|
raise exception.ProjectQuotaNotFound(project_id=project_id)
|
|
|
|
def get_defaults(self, context, resources):
|
|
self.called.append(('get_defaults', context, resources))
|
|
return resources
|
|
|
|
def get_user_quotas(self, context, resources, project_id, user_id,
|
|
defaults=True, usages=True):
|
|
self.called.append(('get_user_quotas', context, resources,
|
|
project_id, user_id, defaults,
|
|
usages))
|
|
return resources
|
|
|
|
def get_project_quotas(self, context, resources, project_id,
|
|
defaults=True, usages=True, remains=False):
|
|
self.called.append(('get_project_quotas', context, resources,
|
|
project_id, defaults, usages,
|
|
remains))
|
|
return resources
|
|
|
|
def limit_check(self, context, resources, values, project_id=None,
|
|
user_id=None):
|
|
self.called.append(('limit_check', context, resources,
|
|
values, project_id, user_id))
|
|
|
|
def reserve(self, context, resources, deltas, expire=None,
|
|
project_id=None, user_id=None):
|
|
self.called.append(('reserve', context, resources, deltas,
|
|
expire, project_id, user_id))
|
|
return self.reservations
|
|
|
|
def commit(self, context, reservations, project_id=None, user_id=None):
|
|
self.called.append(('commit', context, reservations, project_id,
|
|
user_id))
|
|
|
|
def rollback(self, context, reservations, project_id=None, user_id=None):
|
|
self.called.append(('rollback', context, reservations, project_id,
|
|
user_id))
|
|
|
|
def usage_reset(self, context, resources):
|
|
self.called.append(('usage_reset', context, resources))
|
|
|
|
def destroy_all_by_project_and_user(self, context, project_id, user_id):
|
|
self.called.append(('destroy_all_by_project_and_user', context,
|
|
project_id, user_id))
|
|
|
|
def destroy_all_by_project(self, context, project_id):
|
|
self.called.append(('destroy_all_by_project', context, project_id))
|
|
|
|
def expire(self, context):
|
|
self.called.append(('expire', context))
|
|
|
|
|
|
class BaseResourceTestCase(test.TestCase):
|
|
def test_no_flag(self):
|
|
resource = quota.BaseResource('test_resource')
|
|
|
|
self.assertEqual(resource.name, 'test_resource')
|
|
self.assertIsNone(resource.flag)
|
|
self.assertEqual(resource.default, -1)
|
|
|
|
def test_with_flag(self):
|
|
# We know this flag exists, so use it...
|
|
self.flags(quota_instances=10)
|
|
resource = quota.BaseResource('test_resource', 'quota_instances')
|
|
|
|
self.assertEqual(resource.name, 'test_resource')
|
|
self.assertEqual(resource.flag, 'quota_instances')
|
|
self.assertEqual(resource.default, 10)
|
|
|
|
def test_with_flag_no_quota(self):
|
|
self.flags(quota_instances=-1)
|
|
resource = quota.BaseResource('test_resource', 'quota_instances')
|
|
|
|
self.assertEqual(resource.name, 'test_resource')
|
|
self.assertEqual(resource.flag, 'quota_instances')
|
|
self.assertEqual(resource.default, -1)
|
|
|
|
def test_quota_no_project(self):
|
|
self.flags(quota_instances=10)
|
|
resource = quota.BaseResource('test_resource', 'quota_instances')
|
|
driver = FakeDriver()
|
|
context = FakeContext(None)
|
|
quota_value = resource.quota(driver, context)
|
|
|
|
self.assertEqual(quota_value, 10)
|
|
|
|
def test_quota_with_project(self):
|
|
self.flags(quota_instances=10)
|
|
resource = quota.BaseResource('test_resource', 'quota_instances')
|
|
driver = FakeDriver(by_project=dict(
|
|
test_project=dict(test_resource=15),
|
|
))
|
|
context = FakeContext('test_project')
|
|
quota_value = resource.quota(driver, context)
|
|
|
|
self.assertEqual(quota_value, 15)
|
|
|
|
|
|
class QuotaEngineTestCase(test.TestCase):
|
|
def test_init(self):
|
|
quota_obj = quota.QuotaEngine()
|
|
|
|
self.assertEqual(quota_obj._resources, {})
|
|
self.assertIsInstance(quota_obj._driver, quota.DbQuotaDriver)
|
|
|
|
def test_init_override_string(self):
|
|
quota_obj = quota.QuotaEngine(
|
|
quota_driver_class='nova.tests.test_quota.FakeDriver')
|
|
|
|
self.assertEqual(quota_obj._resources, {})
|
|
self.assertIsInstance(quota_obj._driver, FakeDriver)
|
|
|
|
def test_init_override_obj(self):
|
|
quota_obj = quota.QuotaEngine(quota_driver_class=FakeDriver)
|
|
|
|
self.assertEqual(quota_obj._resources, {})
|
|
self.assertEqual(quota_obj._driver, FakeDriver)
|
|
|
|
def test_register_resource(self):
|
|
quota_obj = quota.QuotaEngine()
|
|
resource = quota.AbsoluteResource('test_resource')
|
|
quota_obj.register_resource(resource)
|
|
|
|
self.assertEqual(quota_obj._resources, dict(test_resource=resource))
|
|
|
|
def test_register_resources(self):
|
|
quota_obj = quota.QuotaEngine()
|
|
resources = [
|
|
quota.AbsoluteResource('test_resource1'),
|
|
quota.AbsoluteResource('test_resource2'),
|
|
quota.AbsoluteResource('test_resource3'),
|
|
]
|
|
quota_obj.register_resources(resources)
|
|
|
|
self.assertEqual(quota_obj._resources, dict(
|
|
test_resource1=resources[0],
|
|
test_resource2=resources[1],
|
|
test_resource3=resources[2],
|
|
))
|
|
|
|
def test_get_by_project_and_user(self):
|
|
context = FakeContext('test_project')
|
|
driver = FakeDriver(by_user=dict(
|
|
fake_user=dict(test_resource=42)))
|
|
quota_obj = quota.QuotaEngine(quota_driver_class=driver)
|
|
result = quota_obj.get_by_project_and_user(context, 'test_project',
|
|
'fake_user', 'test_resource')
|
|
|
|
self.assertEqual(driver.called, [
|
|
('get_by_project_and_user', context, 'test_project',
|
|
'fake_user', 'test_resource'),
|
|
])
|
|
self.assertEqual(result, 42)
|
|
|
|
def test_get_by_project(self):
|
|
context = FakeContext('test_project')
|
|
driver = FakeDriver(by_project=dict(
|
|
test_project=dict(test_resource=42)))
|
|
quota_obj = quota.QuotaEngine(quota_driver_class=driver)
|
|
result = quota_obj.get_by_project(context, 'test_project',
|
|
'test_resource')
|
|
|
|
self.assertEqual(driver.called, [
|
|
('get_by_project', context, 'test_project', 'test_resource'),
|
|
])
|
|
self.assertEqual(result, 42)
|
|
|
|
def _make_quota_obj(self, driver):
|
|
quota_obj = quota.QuotaEngine(quota_driver_class=driver)
|
|
resources = [
|
|
quota.AbsoluteResource('test_resource4'),
|
|
quota.AbsoluteResource('test_resource3'),
|
|
quota.AbsoluteResource('test_resource2'),
|
|
quota.AbsoluteResource('test_resource1'),
|
|
]
|
|
quota_obj.register_resources(resources)
|
|
|
|
return quota_obj
|
|
|
|
def test_get_defaults(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
result = quota_obj.get_defaults(context)
|
|
|
|
self.assertEqual(driver.called, [
|
|
('get_defaults', context, quota_obj._resources),
|
|
])
|
|
self.assertEqual(result, quota_obj._resources)
|
|
|
|
def test_get_user_quotas(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
result1 = quota_obj.get_user_quotas(context, 'test_project',
|
|
'fake_user')
|
|
result2 = quota_obj.get_user_quotas(context, 'test_project',
|
|
'fake_user',
|
|
defaults=False,
|
|
usages=False)
|
|
|
|
self.assertEqual(driver.called, [
|
|
('get_user_quotas', context, quota_obj._resources,
|
|
'test_project', 'fake_user', True, True),
|
|
('get_user_quotas', context, quota_obj._resources,
|
|
'test_project', 'fake_user', False, False),
|
|
])
|
|
self.assertEqual(result1, quota_obj._resources)
|
|
self.assertEqual(result2, quota_obj._resources)
|
|
|
|
def test_get_project_quotas(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
result1 = quota_obj.get_project_quotas(context, 'test_project')
|
|
result2 = quota_obj.get_project_quotas(context, 'test_project',
|
|
defaults=False,
|
|
usages=False)
|
|
|
|
self.assertEqual(driver.called, [
|
|
('get_project_quotas', context, quota_obj._resources,
|
|
'test_project', True, True, False),
|
|
('get_project_quotas', context, quota_obj._resources,
|
|
'test_project', False, False, False),
|
|
])
|
|
self.assertEqual(result1, quota_obj._resources)
|
|
self.assertEqual(result2, quota_obj._resources)
|
|
|
|
def test_count_no_resource(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
self.assertRaises(exception.QuotaResourceUnknown,
|
|
quota_obj.count, context, 'test_resource5',
|
|
True, foo='bar')
|
|
|
|
def test_count_wrong_resource(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
self.assertRaises(exception.QuotaResourceUnknown,
|
|
quota_obj.count, context, 'test_resource1',
|
|
True, foo='bar')
|
|
|
|
def test_count(self):
|
|
def fake_count(context, *args, **kwargs):
|
|
self.assertEqual(args, (True,))
|
|
self.assertEqual(kwargs, dict(foo='bar'))
|
|
return 5
|
|
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
quota_obj.register_resource(quota.CountableResource('test_resource5',
|
|
fake_count))
|
|
result = quota_obj.count(context, 'test_resource5', True, foo='bar')
|
|
|
|
self.assertEqual(result, 5)
|
|
|
|
def test_limit_check(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
quota_obj.limit_check(context, test_resource1=4, test_resource2=3,
|
|
test_resource3=2, test_resource4=1)
|
|
|
|
self.assertEqual(driver.called, [
|
|
('limit_check', context, quota_obj._resources, dict(
|
|
test_resource1=4,
|
|
test_resource2=3,
|
|
test_resource3=2,
|
|
test_resource4=1,
|
|
), None, None),
|
|
])
|
|
|
|
def test_reserve(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver(reservations=[
|
|
'resv-01', 'resv-02', 'resv-03', 'resv-04',
|
|
])
|
|
quota_obj = self._make_quota_obj(driver)
|
|
result1 = quota_obj.reserve(context, test_resource1=4,
|
|
test_resource2=3, test_resource3=2,
|
|
test_resource4=1)
|
|
result2 = quota_obj.reserve(context, expire=3600,
|
|
test_resource1=1, test_resource2=2,
|
|
test_resource3=3, test_resource4=4)
|
|
result3 = quota_obj.reserve(context, project_id='fake_project',
|
|
test_resource1=1, test_resource2=2,
|
|
test_resource3=3, test_resource4=4)
|
|
|
|
self.assertEqual(driver.called, [
|
|
('reserve', context, quota_obj._resources, dict(
|
|
test_resource1=4,
|
|
test_resource2=3,
|
|
test_resource3=2,
|
|
test_resource4=1,
|
|
), None, None, None),
|
|
('reserve', context, quota_obj._resources, dict(
|
|
test_resource1=1,
|
|
test_resource2=2,
|
|
test_resource3=3,
|
|
test_resource4=4,
|
|
), 3600, None, None),
|
|
('reserve', context, quota_obj._resources, dict(
|
|
test_resource1=1,
|
|
test_resource2=2,
|
|
test_resource3=3,
|
|
test_resource4=4,
|
|
), None, 'fake_project', None),
|
|
])
|
|
self.assertEqual(result1, [
|
|
'resv-01', 'resv-02', 'resv-03', 'resv-04',
|
|
])
|
|
self.assertEqual(result2, [
|
|
'resv-01', 'resv-02', 'resv-03', 'resv-04',
|
|
])
|
|
self.assertEqual(result3, [
|
|
'resv-01', 'resv-02', 'resv-03', 'resv-04',
|
|
])
|
|
|
|
def test_commit(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
quota_obj.commit(context, ['resv-01', 'resv-02', 'resv-03'])
|
|
|
|
self.assertEqual(driver.called, [
|
|
('commit', context, ['resv-01', 'resv-02', 'resv-03'], None,
|
|
None),
|
|
])
|
|
|
|
def test_rollback(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
quota_obj.rollback(context, ['resv-01', 'resv-02', 'resv-03'])
|
|
|
|
self.assertEqual(driver.called, [
|
|
('rollback', context, ['resv-01', 'resv-02', 'resv-03'], None,
|
|
None),
|
|
])
|
|
|
|
def test_usage_reset(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
quota_obj.usage_reset(context, ['res1', 'res2', 'res3'])
|
|
|
|
self.assertEqual(driver.called, [
|
|
('usage_reset', context, ['res1', 'res2', 'res3']),
|
|
])
|
|
|
|
def test_destroy_all_by_project_and_user(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
quota_obj.destroy_all_by_project_and_user(context,
|
|
'test_project', 'fake_user')
|
|
|
|
self.assertEqual(driver.called, [
|
|
('destroy_all_by_project_and_user', context, 'test_project',
|
|
'fake_user'),
|
|
])
|
|
|
|
def test_destroy_all_by_project(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
quota_obj.destroy_all_by_project(context, 'test_project')
|
|
|
|
self.assertEqual(driver.called, [
|
|
('destroy_all_by_project', context, 'test_project'),
|
|
])
|
|
|
|
def test_expire(self):
|
|
context = FakeContext(None)
|
|
driver = FakeDriver()
|
|
quota_obj = self._make_quota_obj(driver)
|
|
quota_obj.expire(context)
|
|
|
|
self.assertEqual(driver.called, [
|
|
('expire', context),
|
|
])
|
|
|
|
def test_resources(self):
|
|
quota_obj = self._make_quota_obj(None)
|
|
|
|
self.assertEqual(quota_obj.resources,
|
|
['test_resource1', 'test_resource2',
|
|
'test_resource3', 'test_resource4'])
|
|
|
|
|
|
class DbQuotaDriverTestCase(test.TestCase):
|
|
def setUp(self):
|
|
super(DbQuotaDriverTestCase, self).setUp()
|
|
|
|
self.flags(quota_instances=10,
|
|
quota_cores=20,
|
|
quota_ram=50 * 1024,
|
|
quota_floating_ips=10,
|
|
quota_fixed_ips=10,
|
|
quota_metadata_items=128,
|
|
quota_injected_files=5,
|
|
quota_injected_file_content_bytes=10 * 1024,
|
|
quota_injected_file_path_bytes=255,
|
|
quota_security_groups=10,
|
|
quota_security_group_rules=20,
|
|
reservation_expire=86400,
|
|
until_refresh=0,
|
|
max_age=0,
|
|
)
|
|
|
|
self.driver = quota.DbQuotaDriver()
|
|
|
|
self.calls = []
|
|
|
|
self.useFixture(test.TimeOverride())
|
|
|
|
def test_get_defaults(self):
|
|
# Use our pre-defined resources
|
|
result = self.driver.get_defaults(None, quota.QUOTAS._resources)
|
|
|
|
self.assertEqual(result, dict(
|
|
instances=10,
|
|
cores=20,
|
|
ram=50 * 1024,
|
|
floating_ips=10,
|
|
fixed_ips=10,
|
|
metadata_items=128,
|
|
injected_files=5,
|
|
injected_file_content_bytes=10 * 1024,
|
|
injected_file_path_bytes=255,
|
|
security_groups=10,
|
|
security_group_rules=20,
|
|
key_pairs=100,
|
|
))
|
|
|
|
def _stub_get_by_project_and_user(self):
|
|
def fake_qgabpau(context, project_id, user_id):
|
|
self.calls.append('quota_get_all_by_project_and_user')
|
|
self.assertEqual(project_id, 'test_project')
|
|
self.assertEqual(user_id, 'fake_user')
|
|
return dict(
|
|
cores=10,
|
|
injected_files=2,
|
|
injected_file_path_bytes=127,
|
|
)
|
|
|
|
def fake_qgabp(context, project_id):
|
|
self.calls.append('quota_get_all_by_project')
|
|
self.assertEqual(project_id, 'test_project')
|
|
return {
|
|
'cores': 10,
|
|
'injected_files': 2,
|
|
'injected_file_path_bytes': 127,
|
|
}
|
|
|
|
def fake_qugabpau(context, project_id, user_id):
|
|
self.calls.append('quota_usage_get_all_by_project_and_user')
|
|
self.assertEqual(project_id, 'test_project')
|
|
self.assertEqual(user_id, 'fake_user')
|
|
return dict(
|
|
instances=dict(in_use=2, reserved=2),
|
|
cores=dict(in_use=4, reserved=4),
|
|
ram=dict(in_use=10 * 1024, reserved=0),
|
|
floating_ips=dict(in_use=2, reserved=0),
|
|
metadata_items=dict(in_use=0, reserved=0),
|
|
injected_files=dict(in_use=0, reserved=0),
|
|
injected_file_content_bytes=dict(in_use=0, reserved=0),
|
|
injected_file_path_bytes=dict(in_use=0, reserved=0),
|
|
)
|
|
|
|
self.stubs.Set(db, 'quota_get_all_by_project_and_user', fake_qgabpau)
|
|
self.stubs.Set(db, 'quota_get_all_by_project', fake_qgabp)
|
|
self.stubs.Set(db, 'quota_usage_get_all_by_project_and_user',
|
|
fake_qugabpau)
|
|
|
|
def test_get_user_quotas(self):
|
|
self.maxDiff = None
|
|
self._stub_get_by_project_and_user()
|
|
result = self.driver.get_user_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project', 'fake_user')
|
|
|
|
self.assertEqual(self.calls, [
|
|
'quota_get_all_by_project_and_user',
|
|
'quota_get_all_by_project',
|
|
'quota_usage_get_all_by_project_and_user',
|
|
])
|
|
self.assertEqual(result, dict(
|
|
instances=dict(
|
|
limit=10,
|
|
in_use=2,
|
|
reserved=2,
|
|
),
|
|
cores=dict(
|
|
limit=10,
|
|
in_use=4,
|
|
reserved=4,
|
|
),
|
|
ram=dict(
|
|
limit=50 * 1024,
|
|
in_use=10 * 1024,
|
|
reserved=0,
|
|
),
|
|
floating_ips=dict(
|
|
limit=10,
|
|
in_use=2,
|
|
reserved=0,
|
|
),
|
|
fixed_ips=dict(
|
|
limit=10,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
metadata_items=dict(
|
|
limit=128,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_files=dict(
|
|
limit=2,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_content_bytes=dict(
|
|
limit=10 * 1024,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_path_bytes=dict(
|
|
limit=127,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
security_groups=dict(
|
|
limit=10,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
security_group_rules=dict(
|
|
limit=20,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
key_pairs=dict(
|
|
limit=100,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
))
|
|
|
|
def _stub_get_by_project_and_user_specific(self):
|
|
def fake_quota_get(context, project_id, resource, user_id=None):
|
|
self.calls.append('quota_get')
|
|
self.assertEqual(project_id, 'test_project')
|
|
self.assertEqual(user_id, 'fake_user')
|
|
self.assertEqual(resource, 'test_resource')
|
|
return dict(
|
|
test_resource=dict(in_use=20, reserved=10),
|
|
)
|
|
self.stubs.Set(db, 'quota_get', fake_quota_get)
|
|
|
|
def test_get_by_project_and_user(self):
|
|
self._stub_get_by_project_and_user_specific()
|
|
result = self.driver.get_by_project_and_user(
|
|
FakeContext('test_project'),
|
|
'test_project', 'fake_user', 'test_resource')
|
|
|
|
self.assertEqual(self.calls, ['quota_get'])
|
|
self.assertEqual(result, dict(
|
|
test_resource=dict(in_use=20, reserved=10),
|
|
))
|
|
|
|
def _stub_get_by_project(self):
|
|
def fake_qgabp(context, project_id):
|
|
self.calls.append('quota_get_all_by_project')
|
|
self.assertEqual(project_id, 'test_project')
|
|
return dict(
|
|
cores=10,
|
|
injected_files=2,
|
|
injected_file_path_bytes=127,
|
|
)
|
|
|
|
def fake_qugabp(context, project_id):
|
|
self.calls.append('quota_usage_get_all_by_project')
|
|
self.assertEqual(project_id, 'test_project')
|
|
return dict(
|
|
instances=dict(in_use=2, reserved=2),
|
|
cores=dict(in_use=4, reserved=4),
|
|
ram=dict(in_use=10 * 1024, reserved=0),
|
|
floating_ips=dict(in_use=2, reserved=0),
|
|
metadata_items=dict(in_use=0, reserved=0),
|
|
injected_files=dict(in_use=0, reserved=0),
|
|
injected_file_content_bytes=dict(in_use=0, reserved=0),
|
|
injected_file_path_bytes=dict(in_use=0, reserved=0),
|
|
)
|
|
|
|
self.stubs.Set(db, 'quota_get_all_by_project', fake_qgabp)
|
|
self.stubs.Set(db, 'quota_usage_get_all_by_project', fake_qugabp)
|
|
|
|
def test_get_project_quotas(self):
|
|
self.maxDiff = None
|
|
self._stub_get_by_project()
|
|
result = self.driver.get_project_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project')
|
|
|
|
self.assertEqual(self.calls, [
|
|
'quota_get_all_by_project',
|
|
'quota_usage_get_all_by_project',
|
|
])
|
|
self.assertEqual(result, dict(
|
|
instances=dict(
|
|
limit=10,
|
|
in_use=2,
|
|
reserved=2,
|
|
),
|
|
cores=dict(
|
|
limit=10,
|
|
in_use=4,
|
|
reserved=4,
|
|
),
|
|
ram=dict(
|
|
limit=50 * 1024,
|
|
in_use=10 * 1024,
|
|
reserved=0,
|
|
),
|
|
floating_ips=dict(
|
|
limit=10,
|
|
in_use=2,
|
|
reserved=0,
|
|
),
|
|
fixed_ips=dict(
|
|
limit=10,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
metadata_items=dict(
|
|
limit=128,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_files=dict(
|
|
limit=2,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_content_bytes=dict(
|
|
limit=10 * 1024,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_path_bytes=dict(
|
|
limit=127,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
security_groups=dict(
|
|
limit=10,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
security_group_rules=dict(
|
|
limit=20,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
key_pairs=dict(
|
|
limit=100,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
))
|
|
|
|
def test_get_user_quotas_alt_context(self):
|
|
self.maxDiff = None
|
|
self._stub_get_by_project_and_user()
|
|
result = self.driver.get_user_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project', 'fake_user')
|
|
|
|
self.assertEqual(self.calls, [
|
|
'quota_get_all_by_project_and_user',
|
|
'quota_get_all_by_project',
|
|
'quota_usage_get_all_by_project_and_user',
|
|
])
|
|
self.assertEqual(result, dict(
|
|
instances=dict(
|
|
limit=10,
|
|
in_use=2,
|
|
reserved=2,
|
|
),
|
|
cores=dict(
|
|
limit=10,
|
|
in_use=4,
|
|
reserved=4,
|
|
),
|
|
ram=dict(
|
|
limit=50 * 1024,
|
|
in_use=10 * 1024,
|
|
reserved=0,
|
|
),
|
|
floating_ips=dict(
|
|
limit=10,
|
|
in_use=2,
|
|
reserved=0,
|
|
),
|
|
fixed_ips=dict(
|
|
limit=10,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
metadata_items=dict(
|
|
limit=128,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_files=dict(
|
|
limit=2,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_content_bytes=dict(
|
|
limit=10 * 1024,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_path_bytes=dict(
|
|
limit=127,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
security_groups=dict(
|
|
limit=10,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
security_group_rules=dict(
|
|
limit=20,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
key_pairs=dict(
|
|
limit=100,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
))
|
|
|
|
def test_get_project_quotas_alt_context(self):
|
|
self.maxDiff = None
|
|
self._stub_get_by_project()
|
|
result = self.driver.get_project_quotas(
|
|
FakeContext('other_project'),
|
|
quota.QUOTAS._resources, 'test_project')
|
|
|
|
self.assertEqual(self.calls, [
|
|
'quota_get_all_by_project',
|
|
'quota_usage_get_all_by_project',
|
|
])
|
|
self.assertEqual(result, dict(
|
|
instances=dict(
|
|
limit=10,
|
|
in_use=2,
|
|
reserved=2,
|
|
),
|
|
cores=dict(
|
|
limit=10,
|
|
in_use=4,
|
|
reserved=4,
|
|
),
|
|
ram=dict(
|
|
limit=50 * 1024,
|
|
in_use=10 * 1024,
|
|
reserved=0,
|
|
),
|
|
floating_ips=dict(
|
|
limit=10,
|
|
in_use=2,
|
|
reserved=0,
|
|
),
|
|
fixed_ips=dict(
|
|
limit=10,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
metadata_items=dict(
|
|
limit=128,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_files=dict(
|
|
limit=2,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_content_bytes=dict(
|
|
limit=10 * 1024,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_path_bytes=dict(
|
|
limit=127,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
security_groups=dict(
|
|
limit=10,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
security_group_rules=dict(
|
|
limit=20,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
key_pairs=dict(
|
|
limit=100,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
))
|
|
|
|
def test_get_user_quotas_no_defaults(self):
|
|
self._stub_get_by_project_and_user()
|
|
result = self.driver.get_user_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project', 'fake_user',
|
|
defaults=False)
|
|
|
|
self.assertEqual(self.calls, [
|
|
'quota_get_all_by_project_and_user',
|
|
'quota_get_all_by_project',
|
|
'quota_usage_get_all_by_project_and_user',
|
|
])
|
|
self.assertEqual(result, dict(
|
|
cores=dict(
|
|
limit=10,
|
|
in_use=4,
|
|
reserved=4,
|
|
),
|
|
injected_files=dict(
|
|
limit=2,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_path_bytes=dict(
|
|
limit=127,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
))
|
|
|
|
def test_get_project_quotas_no_defaults(self):
|
|
self._stub_get_by_project()
|
|
result = self.driver.get_project_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project', defaults=False)
|
|
|
|
self.assertEqual(self.calls, [
|
|
'quota_get_all_by_project',
|
|
'quota_usage_get_all_by_project',
|
|
])
|
|
self.assertEqual(result, dict(
|
|
cores=dict(
|
|
limit=10,
|
|
in_use=4,
|
|
reserved=4,
|
|
),
|
|
injected_files=dict(
|
|
limit=2,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
injected_file_path_bytes=dict(
|
|
limit=127,
|
|
in_use=0,
|
|
reserved=0,
|
|
),
|
|
))
|
|
|
|
def test_get_user_quotas_no_usages(self):
|
|
self._stub_get_by_project_and_user()
|
|
result = self.driver.get_user_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project', 'fake_user', usages=False)
|
|
|
|
self.assertEqual(self.calls, [
|
|
'quota_get_all_by_project_and_user',
|
|
'quota_get_all_by_project',
|
|
])
|
|
self.assertEqual(result, dict(
|
|
instances=dict(
|
|
limit=10,
|
|
),
|
|
cores=dict(
|
|
limit=10,
|
|
),
|
|
ram=dict(
|
|
limit=50 * 1024,
|
|
),
|
|
floating_ips=dict(
|
|
limit=10,
|
|
),
|
|
fixed_ips=dict(
|
|
limit=10,
|
|
),
|
|
metadata_items=dict(
|
|
limit=128,
|
|
),
|
|
injected_files=dict(
|
|
limit=2,
|
|
),
|
|
injected_file_content_bytes=dict(
|
|
limit=10 * 1024,
|
|
),
|
|
injected_file_path_bytes=dict(
|
|
limit=127,
|
|
),
|
|
security_groups=dict(
|
|
limit=10,
|
|
),
|
|
security_group_rules=dict(
|
|
limit=20,
|
|
),
|
|
key_pairs=dict(
|
|
limit=100,
|
|
),
|
|
))
|
|
|
|
def test_get_project_quotas_no_usages(self):
|
|
self._stub_get_by_project()
|
|
result = self.driver.get_project_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project', usages=False)
|
|
|
|
self.assertEqual(self.calls, [
|
|
'quota_get_all_by_project',
|
|
])
|
|
self.assertEqual(result, dict(
|
|
instances=dict(
|
|
limit=10,
|
|
),
|
|
cores=dict(
|
|
limit=10,
|
|
),
|
|
ram=dict(
|
|
limit=50 * 1024,
|
|
),
|
|
floating_ips=dict(
|
|
limit=10,
|
|
),
|
|
fixed_ips=dict(
|
|
limit=10,
|
|
),
|
|
metadata_items=dict(
|
|
limit=128,
|
|
),
|
|
injected_files=dict(
|
|
limit=2,
|
|
),
|
|
injected_file_content_bytes=dict(
|
|
limit=10 * 1024,
|
|
),
|
|
injected_file_path_bytes=dict(
|
|
limit=127,
|
|
),
|
|
security_groups=dict(
|
|
limit=10,
|
|
),
|
|
security_group_rules=dict(
|
|
limit=20,
|
|
),
|
|
key_pairs=dict(
|
|
limit=100,
|
|
),
|
|
))
|
|
|
|
def _stub_get_settable_quotas(self):
|
|
def fake_get_project_quotas(context, resources, project_id,
|
|
defaults=True, usages=True, remains=False,
|
|
project_quotas=None):
|
|
self.calls.append('get_project_quotas')
|
|
result = {}
|
|
for k, v in resources.items():
|
|
limit = v.default
|
|
reserved = 0
|
|
if k == 'instances':
|
|
remains = v.default - 5
|
|
in_use = 1
|
|
elif k == 'cores':
|
|
remains = -1
|
|
in_use = 5
|
|
limit = -1
|
|
else:
|
|
remains = v.default
|
|
in_use = 0
|
|
result[k] = {'limit': limit, 'in_use': in_use,
|
|
'reserved': reserved, 'remains': remains}
|
|
return result
|
|
|
|
def fake_get_user_quotas(context, resources, project_id, user_id,
|
|
defaults=True, usages=True,
|
|
project_quotas=None, user_quotas=None):
|
|
self.calls.append('get_user_quotas')
|
|
result = {}
|
|
for k, v in resources.items():
|
|
reserved = 0
|
|
if k == 'instances':
|
|
in_use = 1
|
|
elif k == 'cores':
|
|
in_use = 5
|
|
reserved = 10
|
|
else:
|
|
in_use = 0
|
|
result[k] = {'limit': v.default,
|
|
'in_use': in_use, 'reserved': reserved}
|
|
return result
|
|
|
|
def fake_qgabpau(context, project_id, user_id):
|
|
self.calls.append('quota_get_all_by_project_and_user')
|
|
return {'instances': 2, 'cores': -1}
|
|
|
|
self.stubs.Set(self.driver, 'get_project_quotas',
|
|
fake_get_project_quotas)
|
|
self.stubs.Set(self.driver, 'get_user_quotas',
|
|
fake_get_user_quotas)
|
|
self.stubs.Set(db, 'quota_get_all_by_project_and_user',
|
|
fake_qgabpau)
|
|
|
|
def test_get_settable_quotas_with_user(self):
|
|
self._stub_get_settable_quotas()
|
|
result = self.driver.get_settable_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project', user_id='test_user')
|
|
|
|
self.assertEqual(self.calls, [
|
|
'get_project_quotas',
|
|
'quota_get_all_by_project_and_user',
|
|
'get_user_quotas',
|
|
])
|
|
self.assertEqual(result, {
|
|
'instances': {
|
|
'minimum': 1,
|
|
'maximum': 7,
|
|
},
|
|
'cores': {
|
|
'minimum': 15,
|
|
'maximum': -1,
|
|
},
|
|
'ram': {
|
|
'minimum': 0,
|
|
'maximum': 50 * 1024,
|
|
},
|
|
'floating_ips': {
|
|
'minimum': 0,
|
|
'maximum': 10,
|
|
},
|
|
'fixed_ips': {
|
|
'minimum': 0,
|
|
'maximum': 10,
|
|
},
|
|
'metadata_items': {
|
|
'minimum': 0,
|
|
'maximum': 128,
|
|
},
|
|
'injected_files': {
|
|
'minimum': 0,
|
|
'maximum': 5,
|
|
},
|
|
'injected_file_content_bytes': {
|
|
'minimum': 0,
|
|
'maximum': 10 * 1024,
|
|
},
|
|
'injected_file_path_bytes': {
|
|
'minimum': 0,
|
|
'maximum': 255,
|
|
},
|
|
'security_groups': {
|
|
'minimum': 0,
|
|
'maximum': 10,
|
|
},
|
|
'security_group_rules': {
|
|
'minimum': 0,
|
|
'maximum': 20,
|
|
},
|
|
'key_pairs': {
|
|
'minimum': 0,
|
|
'maximum': 100,
|
|
},
|
|
})
|
|
|
|
def test_get_settable_quotas_without_user(self):
|
|
self._stub_get_settable_quotas()
|
|
result = self.driver.get_settable_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project')
|
|
|
|
self.assertEqual(self.calls, [
|
|
'get_project_quotas',
|
|
])
|
|
self.assertEqual(result, {
|
|
'instances': {
|
|
'minimum': 5,
|
|
'maximum': -1,
|
|
},
|
|
'cores': {
|
|
'minimum': 5,
|
|
'maximum': -1,
|
|
},
|
|
'ram': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
'floating_ips': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
'fixed_ips': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
'metadata_items': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
'injected_files': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
'injected_file_content_bytes': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
'injected_file_path_bytes': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
'security_groups': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
'security_group_rules': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
'key_pairs': {
|
|
'minimum': 0,
|
|
'maximum': -1,
|
|
},
|
|
})
|
|
|
|
def test_get_settable_quotas_by_user_with_unlimited_value(self):
|
|
self._stub_get_settable_quotas()
|
|
result = self.driver.get_settable_quotas(
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources, 'test_project', user_id='test_user')
|
|
|
|
self.assertEqual(self.calls, [
|
|
'get_project_quotas',
|
|
'quota_get_all_by_project_and_user',
|
|
'get_user_quotas',
|
|
])
|
|
self.assertEqual(result, {
|
|
'instances': {
|
|
'minimum': 1,
|
|
'maximum': 7,
|
|
},
|
|
'cores': {
|
|
'minimum': 15,
|
|
'maximum': -1,
|
|
},
|
|
'ram': {
|
|
'minimum': 0,
|
|
'maximum': 50 * 1024,
|
|
},
|
|
'floating_ips': {
|
|
'minimum': 0,
|
|
'maximum': 10,
|
|
},
|
|
'fixed_ips': {
|
|
'minimum': 0,
|
|
'maximum': 10,
|
|
},
|
|
'metadata_items': {
|
|
'minimum': 0,
|
|
'maximum': 128,
|
|
},
|
|
'injected_files': {
|
|
'minimum': 0,
|
|
'maximum': 5,
|
|
},
|
|
'injected_file_content_bytes': {
|
|
'minimum': 0,
|
|
'maximum': 10 * 1024,
|
|
},
|
|
'injected_file_path_bytes': {
|
|
'minimum': 0,
|
|
'maximum': 255,
|
|
},
|
|
'security_groups': {
|
|
'minimum': 0,
|
|
'maximum': 10,
|
|
},
|
|
'security_group_rules': {
|
|
'minimum': 0,
|
|
'maximum': 20,
|
|
},
|
|
'key_pairs': {
|
|
'minimum': 0,
|
|
'maximum': 100,
|
|
},
|
|
})
|
|
|
|
def _stub_get_project_quotas(self):
|
|
def fake_get_project_quotas(context, resources, project_id,
|
|
defaults=True, usages=True, remains=False,
|
|
project_quotas=None):
|
|
self.calls.append('get_project_quotas')
|
|
return dict((k, dict(limit=v.default))
|
|
for k, v in resources.items())
|
|
|
|
self.stubs.Set(self.driver, 'get_project_quotas',
|
|
fake_get_project_quotas)
|
|
|
|
def test_get_quotas_has_sync_unknown(self):
|
|
self._stub_get_project_quotas()
|
|
self.assertRaises(exception.QuotaResourceUnknown,
|
|
self.driver._get_quotas,
|
|
None, quota.QUOTAS._resources,
|
|
['unknown'], True)
|
|
self.assertEqual(self.calls, [])
|
|
|
|
def test_get_quotas_no_sync_unknown(self):
|
|
self._stub_get_project_quotas()
|
|
self.assertRaises(exception.QuotaResourceUnknown,
|
|
self.driver._get_quotas,
|
|
None, quota.QUOTAS._resources,
|
|
['unknown'], False)
|
|
self.assertEqual(self.calls, [])
|
|
|
|
def test_get_quotas_has_sync_no_sync_resource(self):
|
|
self._stub_get_project_quotas()
|
|
self.assertRaises(exception.QuotaResourceUnknown,
|
|
self.driver._get_quotas,
|
|
None, quota.QUOTAS._resources,
|
|
['metadata_items'], True)
|
|
self.assertEqual(self.calls, [])
|
|
|
|
def test_get_quotas_no_sync_has_sync_resource(self):
|
|
self._stub_get_project_quotas()
|
|
self.assertRaises(exception.QuotaResourceUnknown,
|
|
self.driver._get_quotas,
|
|
None, quota.QUOTAS._resources,
|
|
['instances'], False)
|
|
self.assertEqual(self.calls, [])
|
|
|
|
def test_get_quotas_has_sync(self):
|
|
self._stub_get_project_quotas()
|
|
result = self.driver._get_quotas(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
['instances', 'cores', 'ram',
|
|
'floating_ips', 'security_groups'],
|
|
True,
|
|
project_id='test_project')
|
|
|
|
self.assertEqual(self.calls, ['get_project_quotas'])
|
|
self.assertEqual(result, dict(
|
|
instances=10,
|
|
cores=20,
|
|
ram=50 * 1024,
|
|
floating_ips=10,
|
|
security_groups=10,
|
|
))
|
|
|
|
def test_get_quotas_no_sync(self):
|
|
self._stub_get_project_quotas()
|
|
result = self.driver._get_quotas(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
['metadata_items', 'injected_files',
|
|
'injected_file_content_bytes',
|
|
'injected_file_path_bytes',
|
|
'security_group_rules'], False,
|
|
project_id='test_project')
|
|
|
|
self.assertEqual(self.calls, ['get_project_quotas'])
|
|
self.assertEqual(result, dict(
|
|
metadata_items=128,
|
|
injected_files=5,
|
|
injected_file_content_bytes=10 * 1024,
|
|
injected_file_path_bytes=255,
|
|
security_group_rules=20,
|
|
))
|
|
|
|
def test_limit_check_under(self):
|
|
self._stub_get_project_quotas()
|
|
self.assertRaises(exception.InvalidQuotaValue,
|
|
self.driver.limit_check,
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(metadata_items=-1))
|
|
|
|
def test_limit_check_over(self):
|
|
self._stub_get_project_quotas()
|
|
self.assertRaises(exception.OverQuota,
|
|
self.driver.limit_check,
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(metadata_items=129))
|
|
|
|
def test_limit_check_unlimited(self):
|
|
self.flags(quota_metadata_items=-1)
|
|
self._stub_get_project_quotas()
|
|
self.driver.limit_check(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(metadata_items=32767))
|
|
|
|
def test_limit_check(self):
|
|
self._stub_get_project_quotas()
|
|
self.driver.limit_check(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(metadata_items=128))
|
|
|
|
def _stub_quota_reserve(self):
|
|
def fake_quota_reserve(context, resources, quotas, user_quotas, deltas,
|
|
expire, until_refresh, max_age, project_id=None,
|
|
user_id=None):
|
|
self.calls.append(('quota_reserve', expire, until_refresh,
|
|
max_age))
|
|
return ['resv-1', 'resv-2', 'resv-3']
|
|
self.stubs.Set(db, 'quota_reserve', fake_quota_reserve)
|
|
|
|
def test_reserve_bad_expire(self):
|
|
self._stub_get_project_quotas()
|
|
self._stub_quota_reserve()
|
|
self.assertRaises(exception.InvalidReservationExpiration,
|
|
self.driver.reserve,
|
|
FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(instances=2), expire='invalid')
|
|
self.assertEqual(self.calls, [])
|
|
|
|
def test_reserve_default_expire(self):
|
|
self._stub_get_project_quotas()
|
|
self._stub_quota_reserve()
|
|
result = self.driver.reserve(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(instances=2))
|
|
|
|
expire = timeutils.utcnow() + datetime.timedelta(seconds=86400)
|
|
self.assertEqual(self.calls, [
|
|
'get_project_quotas',
|
|
('quota_reserve', expire, 0, 0),
|
|
])
|
|
self.assertEqual(result, ['resv-1', 'resv-2', 'resv-3'])
|
|
|
|
def test_reserve_int_expire(self):
|
|
self._stub_get_project_quotas()
|
|
self._stub_quota_reserve()
|
|
result = self.driver.reserve(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(instances=2), expire=3600)
|
|
|
|
expire = timeutils.utcnow() + datetime.timedelta(seconds=3600)
|
|
self.assertEqual(self.calls, [
|
|
'get_project_quotas',
|
|
('quota_reserve', expire, 0, 0),
|
|
])
|
|
self.assertEqual(result, ['resv-1', 'resv-2', 'resv-3'])
|
|
|
|
def test_reserve_timedelta_expire(self):
|
|
self._stub_get_project_quotas()
|
|
self._stub_quota_reserve()
|
|
expire_delta = datetime.timedelta(seconds=60)
|
|
result = self.driver.reserve(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(instances=2), expire=expire_delta)
|
|
|
|
expire = timeutils.utcnow() + expire_delta
|
|
self.assertEqual(self.calls, [
|
|
'get_project_quotas',
|
|
('quota_reserve', expire, 0, 0),
|
|
])
|
|
self.assertEqual(result, ['resv-1', 'resv-2', 'resv-3'])
|
|
|
|
def test_reserve_datetime_expire(self):
|
|
self._stub_get_project_quotas()
|
|
self._stub_quota_reserve()
|
|
expire = timeutils.utcnow() + datetime.timedelta(seconds=120)
|
|
result = self.driver.reserve(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(instances=2), expire=expire)
|
|
|
|
self.assertEqual(self.calls, [
|
|
'get_project_quotas',
|
|
('quota_reserve', expire, 0, 0),
|
|
])
|
|
self.assertEqual(result, ['resv-1', 'resv-2', 'resv-3'])
|
|
|
|
def test_reserve_until_refresh(self):
|
|
self._stub_get_project_quotas()
|
|
self._stub_quota_reserve()
|
|
self.flags(until_refresh=500)
|
|
expire = timeutils.utcnow() + datetime.timedelta(seconds=120)
|
|
result = self.driver.reserve(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(instances=2), expire=expire)
|
|
|
|
self.assertEqual(self.calls, [
|
|
'get_project_quotas',
|
|
('quota_reserve', expire, 500, 0),
|
|
])
|
|
self.assertEqual(result, ['resv-1', 'resv-2', 'resv-3'])
|
|
|
|
def test_reserve_max_age(self):
|
|
self._stub_get_project_quotas()
|
|
self._stub_quota_reserve()
|
|
self.flags(max_age=86400)
|
|
expire = timeutils.utcnow() + datetime.timedelta(seconds=120)
|
|
result = self.driver.reserve(FakeContext('test_project'),
|
|
quota.QUOTAS._resources,
|
|
dict(instances=2), expire=expire)
|
|
|
|
self.assertEqual(self.calls, [
|
|
'get_project_quotas',
|
|
('quota_reserve', expire, 0, 86400),
|
|
])
|
|
self.assertEqual(result, ['resv-1', 'resv-2', 'resv-3'])
|
|
|
|
def test_usage_reset(self):
|
|
calls = []
|
|
|
|
def fake_quota_usage_update(context, project_id, user_id, resource,
|
|
**kwargs):
|
|
calls.append(('quota_usage_update', context, project_id, user_id,
|
|
resource, kwargs))
|
|
if resource == 'nonexist':
|
|
raise exception.QuotaUsageNotFound(project_id=project_id)
|
|
self.stubs.Set(db, 'quota_usage_update', fake_quota_usage_update)
|
|
|
|
ctx = FakeContext('test_project')
|
|
resources = ['res1', 'res2', 'nonexist', 'res4']
|
|
self.driver.usage_reset(ctx, resources)
|
|
|
|
# Make sure we had some calls
|
|
self.assertEqual(len(calls), len(resources))
|
|
|
|
# Extract the elevated context that was used and do some
|
|
# sanity checks
|
|
elevated = calls[0][1]
|
|
self.assertEqual(elevated.project_id, ctx.project_id)
|
|
self.assertEqual(elevated.is_admin, True)
|
|
|
|
# Now check that all the expected calls were made
|
|
exemplar = [('quota_usage_update', elevated, 'test_project',
|
|
'fake_user', res, dict(in_use=-1)) for res in resources]
|
|
self.assertEqual(calls, exemplar)
|
|
|
|
|
|
class FakeSession(object):
|
|
def begin(self):
|
|
return self
|
|
|
|
def add(self, instance):
|
|
pass
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_value, exc_traceback):
|
|
return False
|
|
|
|
|
|
class FakeUsage(sqa_models.QuotaUsage):
|
|
def save(self, *args, **kwargs):
|
|
pass
|
|
|
|
|
|
class QuotaReserveSqlAlchemyTestCase(test.TestCase):
|
|
# nova.db.sqlalchemy.api.quota_reserve is so complex it needs its
|
|
# own test case, and since it's a quota manipulator, this is the
|
|
# best place to put it...
|
|
|
|
def setUp(self):
|
|
super(QuotaReserveSqlAlchemyTestCase, self).setUp()
|
|
self.sync_called = set()
|
|
self.quotas = dict(
|
|
instances=5,
|
|
cores=10,
|
|
ram=10 * 1024,
|
|
fixed_ips=5,
|
|
)
|
|
self.deltas = dict(
|
|
instances=2,
|
|
cores=4,
|
|
ram=2 * 1024,
|
|
fixed_ips=2,
|
|
)
|
|
|
|
def make_sync(res_name):
|
|
def sync(context, project_id, user_id, session):
|
|
self.sync_called.add(res_name)
|
|
if res_name in self.usages:
|
|
if self.usages[res_name].in_use < 0:
|
|
return {res_name: 2}
|
|
else:
|
|
return {res_name: self.usages[res_name].in_use - 1}
|
|
return {res_name: 0}
|
|
return sync
|
|
self.resources = {}
|
|
|
|
for res_name in ('instances', 'cores', 'ram', 'fixed_ips'):
|
|
method_name = '_sync_%s' % res_name
|
|
sqa_api.QUOTA_SYNC_FUNCTIONS[method_name] = make_sync(res_name)
|
|
res = quota.ReservableResource(res_name, '_sync_%s' % res_name)
|
|
self.resources[res_name] = res
|
|
|
|
self.expire = timeutils.utcnow() + datetime.timedelta(seconds=3600)
|
|
self.usages = {}
|
|
self.usages_created = {}
|
|
self.reservations_created = {}
|
|
self.usages_list = [
|
|
dict(resource='instances',
|
|
project_id='test_project',
|
|
user_id='fake_user',
|
|
in_use=2,
|
|
reserved=2,
|
|
until_refresh=None),
|
|
dict(resource='cores',
|
|
project_id='test_project',
|
|
user_id='fake_user',
|
|
in_use=2,
|
|
reserved=4,
|
|
until_refresh=None),
|
|
dict(resource='ram',
|
|
project_id='test_project',
|
|
user_id='fake_user',
|
|
in_use=2,
|
|
reserved=2 * 1024,
|
|
until_refresh=None),
|
|
dict(resource='fixed_ips',
|
|
project_id='test_project',
|
|
user_id=None,
|
|
in_use=2,
|
|
reserved=2,
|
|
until_refresh=None),
|
|
]
|
|
|
|
def fake_get_session():
|
|
return FakeSession()
|
|
|
|
def fake_get_project_user_quota_usages(context, session, project_id,
|
|
user_id):
|
|
return self.usages.copy(), self.usages.copy()
|
|
|
|
def fake_quota_usage_create(context, project_id, user_id, resource,
|
|
in_use, reserved, until_refresh,
|
|
session=None, save=True):
|
|
quota_usage_ref = self._make_quota_usage(
|
|
project_id, user_id, resource, in_use, reserved, until_refresh,
|
|
timeutils.utcnow(), timeutils.utcnow())
|
|
|
|
self.usages_created[resource] = quota_usage_ref
|
|
|
|
return quota_usage_ref
|
|
|
|
def fake_reservation_create(context, uuid, usage_id, project_id,
|
|
user_id, resource, delta, expire,
|
|
session=None):
|
|
reservation_ref = self._make_reservation(
|
|
uuid, usage_id, project_id, user_id, resource, delta, expire,
|
|
timeutils.utcnow(), timeutils.utcnow())
|
|
|
|
self.reservations_created[resource] = reservation_ref
|
|
|
|
return reservation_ref
|
|
|
|
self.stubs.Set(sqa_api, 'get_session', fake_get_session)
|
|
self.stubs.Set(sqa_api, '_get_project_user_quota_usages',
|
|
fake_get_project_user_quota_usages)
|
|
self.stubs.Set(sqa_api, '_quota_usage_create', fake_quota_usage_create)
|
|
self.stubs.Set(sqa_api, '_reservation_create', fake_reservation_create)
|
|
|
|
self.useFixture(test.TimeOverride())
|
|
|
|
def _make_quota_usage(self, project_id, user_id, resource, in_use,
|
|
reserved, until_refresh, created_at, updated_at):
|
|
quota_usage_ref = FakeUsage()
|
|
quota_usage_ref.id = len(self.usages) + len(self.usages_created)
|
|
quota_usage_ref.project_id = project_id
|
|
quota_usage_ref.user_id = user_id
|
|
quota_usage_ref.resource = resource
|
|
quota_usage_ref.in_use = in_use
|
|
quota_usage_ref.reserved = reserved
|
|
quota_usage_ref.until_refresh = until_refresh
|
|
quota_usage_ref.created_at = created_at
|
|
quota_usage_ref.updated_at = updated_at
|
|
quota_usage_ref.deleted_at = None
|
|
quota_usage_ref.deleted = False
|
|
|
|
return quota_usage_ref
|
|
|
|
def init_usage(self, project_id, user_id, resource, in_use, reserved=0,
|
|
until_refresh=None, created_at=None, updated_at=None):
|
|
if created_at is None:
|
|
created_at = timeutils.utcnow()
|
|
if updated_at is None:
|
|
updated_at = timeutils.utcnow()
|
|
if resource == 'fixed_ips':
|
|
user_id = None
|
|
|
|
quota_usage_ref = self._make_quota_usage(project_id, user_id, resource,
|
|
in_use, reserved,
|
|
until_refresh,
|
|
created_at, updated_at)
|
|
|
|
self.usages[resource] = quota_usage_ref
|
|
|
|
def compare_usage(self, usage_dict, expected):
|
|
for usage in expected:
|
|
resource = usage['resource']
|
|
for key, value in usage.items():
|
|
actual = getattr(usage_dict[resource], key)
|
|
self.assertEqual(actual, value,
|
|
"%s != %s on usage for resource %s" %
|
|
(actual, value, resource))
|
|
|
|
def _make_reservation(self, uuid, usage_id, project_id, user_id, resource,
|
|
delta, expire, created_at, updated_at):
|
|
reservation_ref = sqa_models.Reservation()
|
|
reservation_ref.id = len(self.reservations_created)
|
|
reservation_ref.uuid = uuid
|
|
reservation_ref.usage_id = usage_id
|
|
reservation_ref.project_id = project_id
|
|
reservation_ref.user_id = user_id
|
|
reservation_ref.resource = resource
|
|
reservation_ref.delta = delta
|
|
reservation_ref.expire = expire
|
|
reservation_ref.created_at = created_at
|
|
reservation_ref.updated_at = updated_at
|
|
reservation_ref.deleted_at = None
|
|
reservation_ref.deleted = False
|
|
|
|
return reservation_ref
|
|
|
|
def compare_reservation(self, reservations, expected):
|
|
reservations = set(reservations)
|
|
for resv in expected:
|
|
resource = resv['resource']
|
|
resv_obj = self.reservations_created[resource]
|
|
|
|
self.assertIn(resv_obj.uuid, reservations)
|
|
reservations.discard(resv_obj.uuid)
|
|
|
|
for key, value in resv.items():
|
|
actual = getattr(resv_obj, key)
|
|
self.assertEqual(actual, value,
|
|
"%s != %s on reservation for resource %s" %
|
|
(actual, value, resource))
|
|
|
|
self.assertEqual(len(reservations), 0)
|
|
|
|
def _update_reservations_list(self, usage_id_change=False,
|
|
delta_change=False):
|
|
reservations_list = [
|
|
dict(resource='instances',
|
|
project_id='test_project',
|
|
delta=2),
|
|
dict(resource='cores',
|
|
project_id='test_project',
|
|
delta=4),
|
|
dict(resource='ram',
|
|
delta=2 * 1024),
|
|
dict(resource='fixed_ips',
|
|
project_id='test_project',
|
|
delta=2),
|
|
]
|
|
if usage_id_change:
|
|
reservations_list[0]["usage_id"] = self.usages_created['instances']
|
|
reservations_list[1]["usage_id"] = self.usages_created['cores']
|
|
reservations_list[2]["usage_id"] = self.usages_created['ram']
|
|
reservations_list[3]["usage_id"] = self.usages_created['fixed_ips']
|
|
else:
|
|
reservations_list[0]["usage_id"] = self.usages['instances']
|
|
reservations_list[1]["usage_id"] = self.usages['cores']
|
|
reservations_list[2]["usage_id"] = self.usages['ram']
|
|
reservations_list[3]["usage_id"] = self.usages['fixed_ips']
|
|
if delta_change:
|
|
reservations_list[0]["delta"] = -2
|
|
reservations_list[1]["delta"] = -4
|
|
reservations_list[2]["delta"] = -2 * 1024
|
|
reservations_list[3]["delta"] = -2
|
|
return reservations_list
|
|
|
|
def _init_usages(self, *in_use, **kwargs):
|
|
for i, option in enumerate(('instances', 'cores', 'ram', 'fixed_ips')):
|
|
self.init_usage('test_project', 'fake_user',
|
|
option, in_use[i], **kwargs)
|
|
return FakeContext('test_project')
|
|
|
|
def test_quota_reserve_create_usages(self):
|
|
context = FakeContext('test_project')
|
|
result = sqa_api.quota_reserve(context, self.resources, self.quotas,
|
|
self.quotas, self.deltas, self.expire,
|
|
0, 0)
|
|
|
|
self.assertEqual(self.sync_called, set(['instances', 'cores',
|
|
'ram', 'fixed_ips']))
|
|
self.usages_list[0]["in_use"] = 0
|
|
self.usages_list[1]["in_use"] = 0
|
|
self.usages_list[2]["in_use"] = 0
|
|
self.usages_list[3]["in_use"] = 0
|
|
self.compare_usage(self.usages_created, self.usages_list)
|
|
reservations_list = self._update_reservations_list(True)
|
|
self.compare_reservation(result, reservations_list)
|
|
|
|
def test_quota_reserve_negative_in_use(self):
|
|
context = self._init_usages(-1, -1, -1, -1, until_refresh=1)
|
|
result = sqa_api.quota_reserve(context, self.resources, self.quotas,
|
|
self.quotas, self.deltas, self.expire,
|
|
5, 0)
|
|
|
|
self.assertEqual(self.sync_called, set(['instances', 'cores',
|
|
'ram', 'fixed_ips']))
|
|
self.usages_list[0]["until_refresh"] = 5
|
|
self.usages_list[1]["until_refresh"] = 5
|
|
self.usages_list[2]["until_refresh"] = 5
|
|
self.usages_list[3]["until_refresh"] = 5
|
|
self.compare_usage(self.usages, self.usages_list)
|
|
self.assertEqual(self.usages_created, {})
|
|
self.compare_reservation(result, self._update_reservations_list())
|
|
|
|
def test_quota_reserve_until_refresh(self):
|
|
context = self._init_usages(3, 3, 3, 3, until_refresh=1)
|
|
result = sqa_api.quota_reserve(context, self.resources, self.quotas,
|
|
self.quotas, self.deltas, self.expire,
|
|
5, 0)
|
|
|
|
self.assertEqual(self.sync_called, set(['instances', 'cores',
|
|
'ram', 'fixed_ips']))
|
|
self.usages_list[0]["until_refresh"] = 5
|
|
self.usages_list[1]["until_refresh"] = 5
|
|
self.usages_list[2]["until_refresh"] = 5
|
|
self.usages_list[3]["until_refresh"] = 5
|
|
self.compare_usage(self.usages, self.usages_list)
|
|
self.assertEqual(self.usages_created, {})
|
|
self.compare_reservation(result, self._update_reservations_list())
|
|
|
|
def test_quota_reserve_max_age(self):
|
|
max_age = 3600
|
|
record_created = (timeutils.utcnow() -
|
|
datetime.timedelta(seconds=max_age))
|
|
context = self._init_usages(3, 3, 3, 3, created_at=record_created,
|
|
updated_at=record_created)
|
|
result = sqa_api.quota_reserve(context, self.resources, self.quotas,
|
|
self.quotas, self.deltas, self.expire,
|
|
0, max_age)
|
|
|
|
self.assertEqual(self.sync_called, set(['instances', 'cores',
|
|
'ram', 'fixed_ips']))
|
|
self.compare_usage(self.usages, self.usages_list)
|
|
self.assertEqual(self.usages_created, {})
|
|
self.compare_reservation(result, self._update_reservations_list())
|
|
|
|
def test_quota_reserve_no_refresh(self):
|
|
context = self._init_usages(3, 3, 3, 3)
|
|
result = sqa_api.quota_reserve(context, self.resources, self.quotas,
|
|
self.quotas, self.deltas, self.expire,
|
|
0, 0)
|
|
|
|
self.assertEqual(self.sync_called, set([]))
|
|
self.usages_list[0]["in_use"] = 3
|
|
self.usages_list[1]["in_use"] = 3
|
|
self.usages_list[2]["in_use"] = 3
|
|
self.usages_list[3]["in_use"] = 3
|
|
self.compare_usage(self.usages, self.usages_list)
|
|
self.assertEqual(self.usages_created, {})
|
|
self.compare_reservation(result, self._update_reservations_list())
|
|
|
|
def test_quota_reserve_unders(self):
|
|
context = self._init_usages(1, 3, 1 * 1024, 1)
|
|
self.deltas["instances"] = -2
|
|
self.deltas["cores"] = -4
|
|
self.deltas["ram"] = -2 * 1024
|
|
self.deltas["fixed_ips"] = -2
|
|
result = sqa_api.quota_reserve(context, self.resources, self.quotas,
|
|
self.quotas, self.deltas, self.expire,
|
|
0, 0)
|
|
|
|
self.assertEqual(self.sync_called, set([]))
|
|
self.usages_list[0]["in_use"] = 1
|
|
self.usages_list[0]["reserved"] = 0
|
|
self.usages_list[1]["in_use"] = 3
|
|
self.usages_list[1]["reserved"] = 0
|
|
self.usages_list[2]["in_use"] = 1 * 1024
|
|
self.usages_list[2]["reserved"] = 0
|
|
self.usages_list[3]["in_use"] = 1
|
|
self.usages_list[3]["reserved"] = 0
|
|
self.compare_usage(self.usages, self.usages_list)
|
|
self.assertEqual(self.usages_created, {})
|
|
reservations_list = self._update_reservations_list(False, True)
|
|
self.compare_reservation(result, reservations_list)
|
|
|
|
def test_quota_reserve_overs(self):
|
|
context = self._init_usages(4, 8, 10 * 1024, 4)
|
|
try:
|
|
sqa_api.quota_reserve(context, self.resources, self.quotas,
|
|
self.quotas, self.deltas, self.expire, 0, 0)
|
|
except exception.OverQuota as e:
|
|
expected_kwargs = {'code': 500,
|
|
'usages': {'instances': {'reserved': 0, 'in_use': 4},
|
|
'ram': {'reserved': 0, 'in_use': 10240},
|
|
'fixed_ips': {'reserved': 0, 'in_use': 4},
|
|
'cores': {'reserved': 0, 'in_use': 8}},
|
|
'headroom': {'cores': 2, 'ram': 0, 'fixed_ips': 1,
|
|
'instances': 1},
|
|
'overs': ['cores', 'fixed_ips', 'instances', 'ram'],
|
|
'quotas': {'cores': 10, 'ram': 10240,
|
|
'fixed_ips': 5, 'instances': 5}}
|
|
self.assertEqual(e.kwargs, expected_kwargs)
|
|
else:
|
|
self.fail('Expected OverQuota failure')
|
|
self.assertEqual(self.sync_called, set([]))
|
|
self.usages_list[0]["in_use"] = 4
|
|
self.usages_list[0]["reserved"] = 0
|
|
self.usages_list[1]["in_use"] = 8
|
|
self.usages_list[1]["reserved"] = 0
|
|
self.usages_list[2]["in_use"] = 10 * 1024
|
|
self.usages_list[2]["reserved"] = 0
|
|
self.usages_list[3]["in_use"] = 4
|
|
self.usages_list[3]["reserved"] = 0
|
|
self.compare_usage(self.usages, self.usages_list)
|
|
self.assertEqual(self.usages_created, {})
|
|
self.assertEqual(self.reservations_created, {})
|
|
|
|
def test_quota_reserve_cores_unlimited(self):
|
|
# Requesting 8 cores, quota_cores set to unlimited:
|
|
self.flags(quota_cores=-1)
|
|
context = self._init_usages(1, 8, 1 * 1024, 1)
|
|
self.assertEqual(self.sync_called, set([]))
|
|
self.usages_list[0]["in_use"] = 1
|
|
self.usages_list[0]["reserved"] = 0
|
|
self.usages_list[1]["in_use"] = 8
|
|
self.usages_list[1]["reserved"] = 0
|
|
self.usages_list[2]["in_use"] = 1 * 1024
|
|
self.usages_list[2]["reserved"] = 0
|
|
self.usages_list[3]["in_use"] = 1
|
|
self.usages_list[3]["reserved"] = 0
|
|
self.compare_usage(self.usages, self.usages_list)
|
|
self.assertEqual(self.usages_created, {})
|
|
self.assertEqual(self.reservations_created, {})
|
|
|
|
def test_quota_reserve_ram_unlimited(self):
|
|
# Requesting 10*1024 ram, quota_ram set to unlimited:
|
|
self.flags(quota_ram=-1)
|
|
context = self._init_usages(1, 1, 10 * 1024, 1)
|
|
self.assertEqual(self.sync_called, set([]))
|
|
self.usages_list[0]["in_use"] = 1
|
|
self.usages_list[0]["reserved"] = 0
|
|
self.usages_list[1]["in_use"] = 1
|
|
self.usages_list[1]["reserved"] = 0
|
|
self.usages_list[2]["in_use"] = 10 * 1024
|
|
self.usages_list[2]["reserved"] = 0
|
|
self.usages_list[3]["in_use"] = 1
|
|
self.usages_list[3]["reserved"] = 0
|
|
self.compare_usage(self.usages, self.usages_list)
|
|
self.assertEqual(self.usages_created, {})
|
|
self.assertEqual(self.reservations_created, {})
|
|
|
|
def test_quota_reserve_reduction(self):
|
|
context = self._init_usages(10, 20, 20 * 1024, 10)
|
|
self.deltas["instances"] = -2
|
|
self.deltas["cores"] = -4
|
|
self.deltas["ram"] = -2 * 1024
|
|
self.deltas["fixed_ips"] = -2
|
|
result = sqa_api.quota_reserve(context, self.resources, self.quotas,
|
|
self.quotas, self.deltas, self.expire,
|
|
0, 0)
|
|
|
|
self.assertEqual(self.sync_called, set([]))
|
|
self.usages_list[0]["in_use"] = 10
|
|
self.usages_list[0]["reserved"] = 0
|
|
self.usages_list[1]["in_use"] = 20
|
|
self.usages_list[1]["reserved"] = 0
|
|
self.usages_list[2]["in_use"] = 20 * 1024
|
|
self.usages_list[2]["reserved"] = 0
|
|
self.usages_list[3]["in_use"] = 10
|
|
self.usages_list[3]["reserved"] = 0
|
|
self.compare_usage(self.usages, self.usages_list)
|
|
self.assertEqual(self.usages_created, {})
|
|
reservations_list = self._update_reservations_list(False, True)
|
|
self.compare_reservation(result, reservations_list)
|
|
|
|
|
|
class NoopQuotaDriverTestCase(test.TestCase):
|
|
def setUp(self):
|
|
super(NoopQuotaDriverTestCase, self).setUp()
|
|
|
|
self.flags(quota_instances=10,
|
|
quota_cores=20,
|
|
quota_ram=50 * 1024,
|
|
quota_floating_ips=10,
|
|
quota_metadata_items=128,
|
|
quota_injected_files=5,
|
|
quota_injected_file_content_bytes=10 * 1024,
|
|
quota_injected_file_path_bytes=255,
|
|
quota_security_groups=10,
|
|
quota_security_group_rules=20,
|
|
reservation_expire=86400,
|
|
until_refresh=0,
|
|
max_age=0,
|
|
)
|
|
|
|
self.expected_with_usages = {}
|
|
self.expected_without_usages = {}
|
|
self.expected_without_dict = {}
|
|
for r in quota.QUOTAS._resources:
|
|
self.expected_with_usages[r] = dict(limit=-1,
|
|
in_use=-1,
|
|
reserved=-1)
|
|
self.expected_without_usages[r] = dict(limit=-1)
|
|
self.expected_without_dict[r] = -1
|
|
|
|
self.driver = quota.NoopQuotaDriver()
|
|
|
|
def test_get_defaults(self):
|
|
# Use our pre-defined resources
|
|
result = self.driver.get_defaults(None, quota.QUOTAS._resources)
|
|
self.assertEqual(self.expected_without_dict, result)
|
|
|
|
def test_get_project_quotas(self):
|
|
result = self.driver.get_project_quotas(None,
|
|
quota.QUOTAS._resources,
|
|
'test_project')
|
|
self.assertEqual(self.expected_with_usages, result)
|
|
|
|
def test_get_user_quotas(self):
|
|
result = self.driver.get_user_quotas(None,
|
|
quota.QUOTAS._resources,
|
|
'test_project',
|
|
'fake_user')
|
|
self.assertEqual(self.expected_with_usages, result)
|
|
|
|
def test_get_project_quotas_no_defaults(self):
|
|
result = self.driver.get_project_quotas(None,
|
|
quota.QUOTAS._resources,
|
|
'test_project',
|
|
defaults=False)
|
|
self.assertEqual(self.expected_with_usages, result)
|
|
|
|
def test_get_user_quotas_no_defaults(self):
|
|
result = self.driver.get_user_quotas(None,
|
|
quota.QUOTAS._resources,
|
|
'test_project',
|
|
'fake_user',
|
|
defaults=False)
|
|
self.assertEqual(self.expected_with_usages, result)
|
|
|
|
def test_get_project_quotas_no_usages(self):
|
|
result = self.driver.get_project_quotas(None,
|
|
quota.QUOTAS._resources,
|
|
'test_project',
|
|
usages=False)
|
|
self.assertEqual(self.expected_without_usages, result)
|
|
|
|
def test_get_user_quotas_no_usages(self):
|
|
result = self.driver.get_user_quotas(None,
|
|
quota.QUOTAS._resources,
|
|
'test_project',
|
|
'fake_user',
|
|
usages=False)
|
|
self.assertEqual(self.expected_without_usages, result)
|