limit=limit)
context.session.add(tenant_quota)
- def _get_quotas(self, context, tenant_id, resources, keys):
+ def _get_quotas(self, context, tenant_id, resources):
"""Retrieves the quotas for specific resources.
A helper method which retrieves the quotas for the specific
:param context: The request context, for access checks.
:param tenant_id: the tenant_id to check quota.
:param resources: A dictionary of the registered resources.
- :param keys: A list of the desired quotas to retrieve.
-
"""
- desired = set(keys)
- sub_resources = dict((k, v) for k, v in resources.items()
- if k in desired)
-
- # Make sure we accounted for all of them...
- if len(keys) != len(sub_resources):
- unknown = desired - set(sub_resources.keys())
- raise exceptions.QuotaResourceUnknown(unknown=sorted(unknown))
-
# Grab and return the quotas (without usages)
quotas = DbQuotaDriver.get_tenant_quotas(
- context, sub_resources, tenant_id)
+ context, resources, tenant_id)
return dict((k, v) for k, v in quotas.items())
synchronization function--this method checks that a set of
proposed values are permitted by the limit restriction.
- This method will raise a QuotaResourceUnknown exception if a
- given resource is unknown or if it is not a simple limit
- resource.
-
If any of the proposed values is over the defined quota, an
OverQuota exception will be raised with the sorted list of the
resources which are too high. Otherwise, the method returns
raise exceptions.InvalidQuotaValue(unders=sorted(unders))
# Get the applicable quotas
- quotas = self._get_quotas(context, tenant_id, resources, values.keys())
+ quotas = self._get_quotas(context, tenant_id, resources)
# Check the quotas and construct a list of the resources that
# would be put over limit by the desired values
in neutron.conf.
"""
- def _get_quotas(self, context, resources, keys):
+ def _get_quotas(self, context, resources):
"""Get quotas.
A helper method which retrieves the quotas for the specific
:param context: The request context, for access checks.
:param resources: A dictionary of the registered resources.
- :param keys: A list of the desired quotas to retrieve.
"""
- # Filter resources
- desired = set(keys)
- sub_resources = dict((k, v) for k, v in resources.items()
- if k in desired)
-
- # Make sure we accounted for all of them...
- if len(keys) != len(sub_resources):
- unknown = desired - set(sub_resources.keys())
- raise exceptions.QuotaResourceUnknown(unknown=sorted(unknown))
quotas = {}
- for resource in sub_resources.values():
+ for resource in resources.values():
quotas[resource.name] = resource.default
return quotas
synchronization function--this method checks that a set of
proposed values are permitted by the limit restriction.
- This method will raise a QuotaResourceUnknown exception if a
- given resource is unknown or if it is not a simple limit
- resource.
-
If any of the proposed values is over the defined quota, an
OverQuota exception will be raised with the sorted list of the
resources which are too high. Otherwise, the method returns
:param values: A dictionary of the values to check against the
quota.
"""
-
# Ensure no value is less than zero
unders = [key for key, val in values.items() if val < 0]
if unders:
raise exceptions.InvalidQuotaValue(unders=sorted(unders))
# Get the applicable quotas
- quotas = self._get_quotas(context, resources, values.keys())
+ quotas = self._get_quotas(context, resources)
# Check the quotas and construct a list of the resources that
# would be put over limit by the desired values
the proposed value.
This method will raise a QuotaResourceUnknown exception if a
- given resource is unknown or if it is not a simple limit
- resource.
+ given resource is unknown or if it is not a countable resource.
- If any of the proposed values is over the defined quota, an
- OverQuota exception will be raised with the sorted list of the
- resources which are too high. Otherwise, the method returns
- nothing.
+ If any of the proposed values exceeds the respective quota defined
+ for the tenant, an OverQuota exception will be raised.
+ The exception will include a sorted list with the resources
+ which exceed the quota limit. Otherwise, the method returns nothing.
- :param context: The request context, for access checks.
+ :param context: Request context
+ :param tenant_id: Tenant for which the quota limit is being checked
+ :param values: Dict specifying requested deltas for each resource
"""
+ # Verify that resources are managed by the quota engine
+ requested_resources = set(values.keys())
+ managed_resources = set([res for res in self._resources.keys()
+ if res in requested_resources])
+
+ # Make sure we accounted for all of them...
+ unknown_resources = requested_resources - managed_resources
+ if unknown_resources:
+ raise exceptions.QuotaResourceUnknown(
+ unknown=sorted(unknown_resources))
return self.get_driver().limit_check(context, tenant_id,
self._resources, values)
self.assertRaises(exceptions.InvalidQuotaValue,
self.plugin.limit_check, context.get_admin_context(),
PROJECT, resources, values)
-
- def test_limit_check_wrong_values_size(self):
- resource_1 = 'res_test_1'
- resource_2 = 'res_test_2'
-
- resources = {resource_1: TestResource(resource_1, 2)}
- values = {resource_1: 1, resource_2: 1}
-
- self.plugin.update_quota_limit(self.context, PROJECT, resource_1, 2)
- self.assertRaises(exceptions.QuotaResourceUnknown,
- self.plugin.limit_check, context.get_admin_context(),
- PROJECT, resources, values)
tenant_id,
network=-2)
+ def test_quotas_limit_check_with_not_registered_resource_fails(self):
+ tenant_id = 'tenant_id1'
+ self.assertRaises(exceptions.QuotaResourceUnknown,
+ quota.QUOTAS.limit_check,
+ context.get_admin_context(load_admin_roles=False),
+ tenant_id,
+ foobar=1)
+
def test_quotas_get_tenant_from_request_context(self):
tenant_id = 'tenant_id1'
env = {'neutron.context': context.Context('', tenant_id,
quotas = driver._get_quotas(ctx,
target_tenant,
- default_quotas,
- ['network'])
+ default_quotas)
self.assertEqual(quotas, foo_quotas)
get_tenant_quotas.assert_called_once_with(ctx,
Code Review / openstack-build / neutron-build.git / commitdiff