From af7129cda33dcd4ac784097ddb4f119e145c40d5 Mon Sep 17 00:00:00 2001
From: guiyanxing <guiyanxing@cmss.chinamobile.com>
Date: Thu, 8 Dec 2016 15:17:53 +0800
Subject: Add '--type'and other options to network rbac list

This patch adds '--type','--action','--long' filtering options
to network rbac list command

Change-Id: I21846820ab223bb7832e89eb2d7658bbec271aec
Closes-Bug: #1648307
Partially-Implements: blueprint network-commands-options
---
 .../tests/unit/network/v2/test_network_rbac.py     | 64 +++++++++++++++++++++-
 1 file changed, 63 insertions(+), 1 deletion(-)

(limited to 'openstackclient/tests')

diff --git a/openstackclient/tests/unit/network/v2/test_network_rbac.py b/openstackclient/tests/unit/network/v2/test_network_rbac.py
index b884dbc0..935ce075 100644
--- a/openstackclient/tests/unit/network/v2/test_network_rbac.py
+++ b/openstackclient/tests/unit/network/v2/test_network_rbac.py
@@ -327,7 +327,12 @@ class TestListNetworkRABC(TestNetworkRBAC):
         'Object Type',
         'Object ID',
     )
-
+    columns_long = (
+        'ID',
+        'Object Type',
+        'Object ID',
+        'Action',
+    )
     data = []
     for r in rbac_policies:
         data.append((
@@ -335,6 +340,14 @@ class TestListNetworkRABC(TestNetworkRBAC):
             r.object_type,
             r.object_id,
         ))
+    data_long = []
+    for r in rbac_policies:
+        data_long.append((
+            r.id,
+            r.object_type,
+            r.object_id,
+            r.action,
+        ))
 
     def setUp(self):
         super(TestListNetworkRABC, self).setUp()
@@ -356,6 +369,55 @@ class TestListNetworkRABC(TestNetworkRBAC):
         self.assertEqual(self.columns, columns)
         self.assertEqual(self.data, list(data))
 
+    def test_network_rbac_list_type_opt(self):
+        arglist = [
+            '--type', self.rbac_policies[0].object_type, ]
+        verifylist = [
+            ('type', self.rbac_policies[0].object_type)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        self.network.rbac_policies.assert_called_with(**{
+            'object_type': self.rbac_policies[0].object_type
+        })
+        self.assertEqual(self.columns, columns)
+        self.assertEqual(self.data, list(data))
+
+    def test_network_rbac_list_action_opt(self):
+        arglist = [
+            '--action', self.rbac_policies[0].action, ]
+        verifylist = [
+            ('action', self.rbac_policies[0].action)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        self.network.rbac_policies.assert_called_with(**{
+            'action': self.rbac_policies[0].action
+        })
+        self.assertEqual(self.columns, columns)
+        self.assertEqual(self.data, list(data))
+
+    def test_network_rbac_list_with_long(self):
+        arglist = [
+            '--long',
+        ]
+
+        verifylist = [
+            ('long', True),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        columns, data = self.cmd.take_action(parsed_args)
+
+        self.network.rbac_policies.assert_called_with()
+        self.assertEqual(self.columns_long, columns)
+        self.assertEqual(self.data_long, list(data))
+
 
 class TestSetNetworkRBAC(TestNetworkRBAC):
 
-- 
cgit v1.2.1