From ea0b8f91627f5ef4a89f23e391a5733032bccc0e Mon Sep 17 00:00:00 2001
From: Richard Theis <rtheis@us.ibm.com>
Date: Fri, 12 Feb 2016 12:48:33 -0600
Subject: [PATCH] Add quota functional tests

Add functional tests for "os quota" commands.

Change-Id: I0f5939bf4ce553174c9b7ce55bdb3dce0506c409
Related-Bug: #1528249
Partially-Implements: blueprint neutron-client
---
 functional/common/test.py             |  5 ++++
 functional/tests/common/test_quota.py | 38 +++++++++++++++++++++++++++
 2 files changed, 43 insertions(+)
 create mode 100644 functional/tests/common/test_quota.py

diff --git a/functional/common/test.py b/functional/common/test.py
index 2fc355f82f..1e767af886 100644
--- a/functional/common/test.py
+++ b/functional/common/test.py
@@ -51,6 +51,11 @@ class TestCase(testtools.TestCase):
         """Executes openstackclient command for the given action."""
         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 0000000000..7a0cb64026
--- /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)