quota: remove default kwarg on get_class_quotas()
No code was calling the QuotaDriver.get_class_quotas() method with anything other the default=True kwarg value (other than unit tests) so this patch removes that complexity. Change-Id: If330320e92ba9249ccdad14582119923ac57f885
This commit is contained in:
parent
8153cddad1
commit
a0f64b80be
|
@ -61,8 +61,7 @@ class DbQuotaDriver(object):
|
||||||
|
|
||||||
return quotas
|
return quotas
|
||||||
|
|
||||||
def get_class_quotas(self, context, resources, quota_class,
|
def get_class_quotas(self, context, resources, quota_class):
|
||||||
defaults=True):
|
|
||||||
"""Given a list of resources, retrieve the quotas for the given
|
"""Given a list of resources, retrieve the quotas for the given
|
||||||
quota class.
|
quota class.
|
||||||
|
|
||||||
|
@ -70,18 +69,14 @@ class DbQuotaDriver(object):
|
||||||
:param resources: A dictionary of the registered resources.
|
:param resources: A dictionary of the registered resources.
|
||||||
:param quota_class: The name of the quota class to return
|
:param quota_class: The name of the quota class to return
|
||||||
quotas for.
|
quotas for.
|
||||||
:param defaults: If True, the default value will be reported
|
|
||||||
if there is no specific value for the
|
|
||||||
resource.
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
quotas = {}
|
quotas = {}
|
||||||
class_quotas = objects.Quotas.get_all_class_by_name(context,
|
class_quotas = objects.Quotas.get_all_class_by_name(context,
|
||||||
quota_class)
|
quota_class)
|
||||||
for resource in resources.values():
|
for resource in resources.values():
|
||||||
if defaults or resource.name in class_quotas:
|
quotas[resource.name] = class_quotas.get(resource.name,
|
||||||
quotas[resource.name] = class_quotas.get(resource.name,
|
resource.default)
|
||||||
resource.default)
|
|
||||||
|
|
||||||
return quotas
|
return quotas
|
||||||
|
|
||||||
|
@ -632,8 +627,7 @@ class NoopQuotaDriver(object):
|
||||||
quotas[resource.name] = -1
|
quotas[resource.name] = -1
|
||||||
return quotas
|
return quotas
|
||||||
|
|
||||||
def get_class_quotas(self, context, resources, quota_class,
|
def get_class_quotas(self, context, resources, quota_class):
|
||||||
defaults=True):
|
|
||||||
"""Given a list of resources, retrieve the quotas for the given
|
"""Given a list of resources, retrieve the quotas for the given
|
||||||
quota class.
|
quota class.
|
||||||
|
|
||||||
|
@ -641,9 +635,6 @@ class NoopQuotaDriver(object):
|
||||||
:param resources: A dictionary of the registered resources.
|
:param resources: A dictionary of the registered resources.
|
||||||
:param quota_class: The name of the quota class to return
|
:param quota_class: The name of the quota class to return
|
||||||
quotas for.
|
quotas for.
|
||||||
:param defaults: If True, the default value will be reported
|
|
||||||
if there is no specific value for the
|
|
||||||
resource.
|
|
||||||
"""
|
"""
|
||||||
quotas = {}
|
quotas = {}
|
||||||
for resource in resources.values():
|
for resource in resources.values():
|
||||||
|
@ -913,19 +904,16 @@ class QuotaEngine(object):
|
||||||
|
|
||||||
return self._driver.get_defaults(context, self._resources)
|
return self._driver.get_defaults(context, self._resources)
|
||||||
|
|
||||||
def get_class_quotas(self, context, quota_class, defaults=True):
|
def get_class_quotas(self, context, quota_class):
|
||||||
"""Retrieve the quotas for the given quota class.
|
"""Retrieve the quotas for the given quota class.
|
||||||
|
|
||||||
:param context: The request context, for access checks.
|
:param context: The request context, for access checks.
|
||||||
:param quota_class: The name of the quota class to return
|
:param quota_class: The name of the quota class to return
|
||||||
quotas for.
|
quotas for.
|
||||||
:param defaults: If True, the default value will be reported
|
|
||||||
if there is no specific value for the
|
|
||||||
resource.
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
return self._driver.get_class_quotas(context, self._resources,
|
return self._driver.get_class_quotas(context, self._resources,
|
||||||
quota_class, defaults=defaults)
|
quota_class)
|
||||||
|
|
||||||
def get_user_quotas(self, context, project_id, user_id, quota_class=None,
|
def get_user_quotas(self, context, project_id, user_id, quota_class=None,
|
||||||
defaults=True, usages=True):
|
defaults=True, usages=True):
|
||||||
|
|
|
@ -285,10 +285,9 @@ class FakeDriver(object):
|
||||||
self.called.append(('get_defaults', context, resources))
|
self.called.append(('get_defaults', context, resources))
|
||||||
return resources
|
return resources
|
||||||
|
|
||||||
def get_class_quotas(self, context, resources, quota_class,
|
def get_class_quotas(self, context, resources, quota_class):
|
||||||
defaults=True):
|
|
||||||
self.called.append(('get_class_quotas', context, resources,
|
self.called.append(('get_class_quotas', context, resources,
|
||||||
quota_class, defaults))
|
quota_class))
|
||||||
return resources
|
return resources
|
||||||
|
|
||||||
def get_user_quotas(self, context, resources, project_id, user_id,
|
def get_user_quotas(self, context, resources, project_id, user_id,
|
||||||
|
@ -444,16 +443,12 @@ class QuotaEngineTestCase(test.TestCase):
|
||||||
driver = FakeDriver()
|
driver = FakeDriver()
|
||||||
quota_obj = self._make_quota_obj(driver)
|
quota_obj = self._make_quota_obj(driver)
|
||||||
result1 = quota_obj.get_class_quotas(context, 'test_class')
|
result1 = quota_obj.get_class_quotas(context, 'test_class')
|
||||||
result2 = quota_obj.get_class_quotas(context, 'test_class', False)
|
|
||||||
|
|
||||||
self.assertEqual(driver.called, [
|
self.assertEqual(driver.called, [
|
||||||
('get_class_quotas', context, quota_obj._resources,
|
('get_class_quotas', context, quota_obj._resources,
|
||||||
'test_class', True),
|
'test_class'),
|
||||||
('get_class_quotas', context, quota_obj._resources,
|
|
||||||
'test_class', False),
|
|
||||||
])
|
])
|
||||||
self.assertEqual(result1, quota_obj._resources)
|
self.assertEqual(result1, quota_obj._resources)
|
||||||
self.assertEqual(result2, quota_obj._resources)
|
|
||||||
|
|
||||||
def test_get_user_quotas(self):
|
def test_get_user_quotas(self):
|
||||||
context = FakeContext(None, None)
|
context = FakeContext(None, None)
|
||||||
|
@ -667,19 +662,6 @@ class DbQuotaDriverTestCase(test.TestCase):
|
||||||
server_group_members=10,
|
server_group_members=10,
|
||||||
))
|
))
|
||||||
|
|
||||||
def test_get_class_quotas_no_defaults(self):
|
|
||||||
self._stub_quota_class_get_all_by_name()
|
|
||||||
result = self.driver.get_class_quotas(None, quota.QUOTAS._resources,
|
|
||||||
'test_class', False)
|
|
||||||
|
|
||||||
self.assertEqual(self.calls, ['quota_class_get_all_by_name'])
|
|
||||||
self.assertEqual(result, dict(
|
|
||||||
instances=5,
|
|
||||||
ram=25 * 1024,
|
|
||||||
metadata_items=64,
|
|
||||||
injected_file_content_bytes=5 * 1024,
|
|
||||||
))
|
|
||||||
|
|
||||||
def _stub_get_by_project_and_user(self):
|
def _stub_get_by_project_and_user(self):
|
||||||
def fake_qgabpau(context, project_id, user_id):
|
def fake_qgabpau(context, project_id, user_id):
|
||||||
self.calls.append('quota_get_all_by_project_and_user')
|
self.calls.append('quota_get_all_by_project_and_user')
|
||||||
|
@ -2020,13 +2002,6 @@ class NoopQuotaDriverTestCase(test.TestCase):
|
||||||
'test_class')
|
'test_class')
|
||||||
self.assertEqual(self.expected_without_dict, result)
|
self.assertEqual(self.expected_without_dict, result)
|
||||||
|
|
||||||
def test_get_class_quotas_no_defaults(self):
|
|
||||||
result = self.driver.get_class_quotas(None,
|
|
||||||
quota.QUOTAS._resources,
|
|
||||||
'test_class',
|
|
||||||
False)
|
|
||||||
self.assertEqual(self.expected_without_dict, result)
|
|
||||||
|
|
||||||
def test_get_project_quotas(self):
|
def test_get_project_quotas(self):
|
||||||
result = self.driver.get_project_quotas(None,
|
result = self.driver.get_project_quotas(None,
|
||||||
quota.QUOTAS._resources,
|
quota.QUOTAS._resources,
|
||||||
|
|
Loading…
Reference in New Issue