Merge "Add quota functional tests"

2 files changed, 43 insertions, 0 deletions

diff --git a/functional/common/test.py b/functional/common/test.py
index 2fc355f8..1e767af8 100644
--- a/functional/common/test.py
+++ b/functional/common/test.py
@@ -52,6 +52,11 @@ class TestCase(testtools.TestCase):
         return execute('openstack ' + cmd, fail_ok=fail_ok)
 
     @classmethod
+    def get_openstack_configuration_value(cls, configuration):
+        opts = cls.get_show_opts([configuration])
+        return cls.openstack('configuration show ' + opts)
+
+    @classmethod
     def get_show_opts(cls, fields=[]):
         return ' -f value ' + ' '.join(['-c ' + it for it in fields])
 
diff --git a/functional/tests/common/test_quota.py b/functional/tests/common/test_quota.py
new file mode 100644
index 00000000..7a0cb640
--- /dev/null
+++ b/functional/tests/common/test_quota.py
@@ -0,0 +1,38 @@
+#    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 functional.common import test
+
+
+class QuotaTests(test.TestCase):
+    """Functional tests for quota. """
+    # Test quota information for compute, network and volume.
+    EXPECTED_FIELDS = ['instances', 'network', 'volumes']
+    PROJECT_NAME = None
+
+    @classmethod
+    def setUpClass(cls):
+        cls.PROJECT_NAME =\
+            cls.get_openstack_configuration_value('auth.project_name')
+
+    def test_quota_set(self):
+        # TODO(rtheis): Add --network option once supported on set.
+        self.openstack('quota set --instances 11 --volumes 11 ' +
+                       self.PROJECT_NAME)
+        opts = self.get_show_opts(self.EXPECTED_FIELDS)
+        raw_output = self.openstack('quota show ' + self.PROJECT_NAME + opts)
+        self.assertEqual("11\n10\n11\n", raw_output)
+
+    def test_quota_show(self):
+        raw_output = self.openstack('quota show ' + self.PROJECT_NAME)
+        for expected_field in self.EXPECTED_FIELDS:
+            self.assertInOutput(expected_field, raw_output)