Merge "tests: Add functional tests for block storage quotas"

openstack/tests/functional/block_storage/v3/test_quota_set.py

@@ -0,0 +1,53 @@
+# 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 openstack.block_storage.v3 import quota_set as _quota_set
+from openstack.tests.functional import base
+
+
+class TestQuotaSet(base.BaseFunctionalTest):
+    def setUp(self):
+        super().setUp()
+
+        if not self.operator_cloud:
+            self.skipTest("Operator cloud required for this test")
+
+        self.project = self.create_temporary_project()
+
+    def test_quota_set(self):
+        # update quota
+
+        quota_set = self.operator_cloud.block_storage.update_quota_set(
+            self.project.id, volumes=123
+        )
+        self.assertIsInstance(quota_set, _quota_set.QuotaSet)
+        self.assertEqual(quota_set.volumes, 123)
+
+        # retrieve details of the (updated) quota
+
+        quota_set = self.operator_cloud.block_storage.get_quota_set(
+            self.project.id
+        )
+        self.assertIsInstance(quota_set, _quota_set.QuotaSet)
+        self.assertEqual(quota_set.volumes, 123)
+
+        # retrieve quota defaults
+
+        defaults = self.operator_cloud.block_storage.get_quota_set_defaults(
+            self.project.id
+        )
+        self.assertIsInstance(defaults, _quota_set.QuotaSet)
+        self.assertNotEqual(defaults.volumes, 123)
+
+        # revert quota
+
+        self.operator_cloud.block_storage.revert_quota_set(self.project.id)