From b92cf77fb58b4aafcf4b3a1ef6b89fd15166505b Mon Sep 17 00:00:00 2001
From: Fang Zhen <zhen.fang@easystack.cn>
Date: Mon, 14 Mar 2016 19:27:36 +0800
Subject: Add network support for "quota set"

The "quota set" command support compute and volume quotas previously.
This patch add support network.

Partially-implements: blueprint neutron-client-quota
Closes-bug: 1489441

Change-Id: I9d297f52bc30614b3493f09ed15f8f1d3f8ff952
---
 openstackclient/tests/common/test_quota.py | 59 ++++++++++++++++++++++++++++++
 openstackclient/tests/network/v2/fakes.py  |  3 ++
 2 files changed, 62 insertions(+)

(limited to 'openstackclient/tests')

diff --git a/openstackclient/tests/common/test_quota.py b/openstackclient/tests/common/test_quota.py
index ba7ee469..c9ec5599 100644
--- a/openstackclient/tests/common/test_quota.py
+++ b/openstackclient/tests/common/test_quota.py
@@ -97,6 +97,9 @@ class TestQuotaSet(TestQuota):
             loaded=True,
         )
 
+        self.network_mock = self.app.client_manager.network
+        self.network_mock.update_quota = mock.Mock()
+
         self.cmd = quota.SetQuota(self.app, None)
 
     def test_quota_set(self):
@@ -132,6 +135,7 @@ class TestQuotaSet(TestQuota):
             ('project', identity_fakes.project_name),
         ]
 
+        self.app.client_manager.network_endpoint_enabled = False
         parsed_args = self.check_parser(self.cmd, arglist, verifylist)
 
         self.cmd.take_action(parsed_args)
@@ -185,6 +189,61 @@ class TestQuotaSet(TestQuota):
             **kwargs
         )
 
+    def test_quota_set_network(self):
+        arglist = [
+            '--subnets', str(network_fakes.QUOTA['subnet']),
+            '--networks', str(network_fakes.QUOTA['network']),
+            '--floating-ips', str(network_fakes.QUOTA['floatingip']),
+            '--subnetpools', str(network_fakes.QUOTA['subnetpool']),
+            '--secgroup-rules',
+            str(network_fakes.QUOTA['security_group_rule']),
+            '--secgroups', str(network_fakes.QUOTA['security_group']),
+            '--routers', str(network_fakes.QUOTA['router']),
+            '--rbac-policies', str(network_fakes.QUOTA['rbac_policy']),
+            '--ports', str(network_fakes.QUOTA['port']),
+            '--vips', str(network_fakes.QUOTA['vip']),
+            '--members', str(network_fakes.QUOTA['member']),
+            '--health-monitors', str(network_fakes.QUOTA['health_monitor']),
+            identity_fakes.project_name,
+        ]
+        verifylist = [
+            ('subnet', network_fakes.QUOTA['subnet']),
+            ('network', network_fakes.QUOTA['network']),
+            ('floatingip', network_fakes.QUOTA['floatingip']),
+            ('subnetpool', network_fakes.QUOTA['subnetpool']),
+            ('security_group_rule',
+             network_fakes.QUOTA['security_group_rule']),
+            ('security_group', network_fakes.QUOTA['security_group']),
+            ('router', network_fakes.QUOTA['router']),
+            ('rbac_policy', network_fakes.QUOTA['rbac_policy']),
+            ('port', network_fakes.QUOTA['port']),
+            ('vip', network_fakes.QUOTA['vip']),
+            ('member', network_fakes.QUOTA['member']),
+            ('health_monitor', network_fakes.QUOTA['health_monitor']),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        self.cmd.take_action(parsed_args)
+        kwargs = {
+            'subnet': network_fakes.QUOTA['subnet'],
+            'network': network_fakes.QUOTA['network'],
+            'floatingip': network_fakes.QUOTA['floatingip'],
+            'subnetpool': network_fakes.QUOTA['subnetpool'],
+            'security_group_rule':
+                network_fakes.QUOTA['security_group_rule'],
+            'security_group': network_fakes.QUOTA['security_group'],
+            'router': network_fakes.QUOTA['router'],
+            'rbac_policy': network_fakes.QUOTA['rbac_policy'],
+            'port': network_fakes.QUOTA['port'],
+            'vip': network_fakes.QUOTA['vip'],
+            'member': network_fakes.QUOTA['member'],
+            'health_monitor': network_fakes.QUOTA['health_monitor'],
+        }
+        self.network_mock.update_quota.assert_called_with(
+            identity_fakes.project_id,
+            **kwargs
+        )
+
 
 class TestQuotaShow(TestQuota):
 
diff --git a/openstackclient/tests/network/v2/fakes.py b/openstackclient/tests/network/v2/fakes.py
index 84ede381..417cf26e 100644
--- a/openstackclient/tests/network/v2/fakes.py
+++ b/openstackclient/tests/network/v2/fakes.py
@@ -36,6 +36,9 @@ QUOTA = {
     "router": 10,
     "rbac_policy": -1,
     "port": 50,
+    "vip": 10,
+    "member": 10,
+    "health_monitor": 10,
 }
 
 
-- 
cgit v1.2.1