diff --git a/cinderclient/tests/v1/test_quota_classes.py b/cinderclient/tests/v1/test_quota_classes.py index 33b390df0..83e297fd4 100644 --- a/cinderclient/tests/v1/test_quota_classes.py +++ b/cinderclient/tests/v1/test_quota_classes.py @@ -29,7 +29,7 @@ class QuotaClassSetsTest(utils.TestCase): def test_update_quota(self): q = cs.quota_classes.get('test') - q.update(volumes=2) + q.update(volumes=2, snapshots=2) cs.assert_called('PUT', '/os-quota-class-sets/test') def test_refresh_quota(self): diff --git a/cinderclient/tests/v2/test_quota_classes.py b/cinderclient/tests/v2/test_quota_classes.py index ce7646c99..83cc71022 100644 --- a/cinderclient/tests/v2/test_quota_classes.py +++ b/cinderclient/tests/v2/test_quota_classes.py @@ -29,7 +29,7 @@ class QuotaClassSetsTest(utils.TestCase): def test_update_quota(self): q = cs.quota_classes.get('test') - q.update(volumes=2) + q.update(volumes=2, snapshots=2) cs.assert_called('PUT', '/os-quota-class-sets/test') def test_refresh_quota(self): diff --git a/cinderclient/v1/quota_classes.py b/cinderclient/v1/quota_classes.py index da96e5c46..df637eaa7 100644 --- a/cinderclient/v1/quota_classes.py +++ b/cinderclient/v1/quota_classes.py @@ -39,11 +39,13 @@ class QuotaClassSetManager(base.Manager): def update(self, class_name, volumes=None, + snapshots=None, gigabytes=None): body = {'quota_class_set': { 'class_name': class_name, 'volumes': volumes, + 'snapshots': snapshots, 'gigabytes': gigabytes}} for key in list(body['quota_class_set'].keys()): diff --git a/cinderclient/v2/quota_classes.py b/cinderclient/v2/quota_classes.py index eeda85ccd..a4e2043d1 100644 --- a/cinderclient/v2/quota_classes.py +++ b/cinderclient/v2/quota_classes.py @@ -37,11 +37,13 @@ class QuotaClassSetManager(base.Manager): def update(self, class_name, volumes=None, + snapshots=None, gigabytes=None): body = {'quota_class_set': { 'class_name': class_name, 'volumes': volumes, + 'snapshots': snapshots, 'gigabytes': gigabytes}} for key in list(body['quota_class_set'].keys()):