From 5988ee61d894417813450e5ecc8fdf704b015240 Mon Sep 17 00:00:00 2001
From: Dean Troyer <dtroyer@gmail.com>
Date: Fri, 13 Jan 2017 11:56:34 -0600
Subject: [PATCH] Fix security group rule list for NEtwork v2

Fix the formatting of Port Range in the security group rule list command
for Network v2 to handle SDK changes.

Change-Id: Id954cbfaedbb74f60125ebda91f80db751759933
---
 .../network/v2/security_group_rule.py         | 45 +++++++++++--------
 .../network/v2/test_security_group_rule.py    |  3 --
 2 files changed, 26 insertions(+), 22 deletions(-)

diff --git a/openstackclient/network/v2/security_group_rule.py b/openstackclient/network/v2/security_group_rule.py
index b878d8751b..4fb62c7bff 100644
--- a/openstackclient/network/v2/security_group_rule.py
+++ b/openstackclient/network/v2/security_group_rule.py
@@ -51,17 +51,17 @@ def _format_network_port_range(rule):
     # - Single port: '80:80'
     # - No port range: ''
     port_range = ''
-    if _is_icmp_protocol(rule.protocol):
-        if rule.port_range_min:
-            port_range += 'type=' + str(rule.port_range_min)
-        if rule.port_range_max:
-            port_range += ':code=' + str(rule.port_range_max)
-    elif rule.port_range_min or rule.port_range_max:
-        port_range_min = str(rule.port_range_min)
-        port_range_max = str(rule.port_range_max)
-        if rule.port_range_min is None:
+    if _is_icmp_protocol(rule['protocol']):
+        if rule['port_range_min']:
+            port_range += 'type=' + str(rule['port_range_min'])
+        if rule['port_range_max']:
+            port_range += ':code=' + str(rule['port_range_max'])
+    elif rule['port_range_min'] or rule['port_range_max']:
+        port_range_min = str(rule['port_range_min'])
+        port_range_max = str(rule['port_range_max'])
+        if rule['port_range_min'] is None:
             port_range_min = port_range_max
-        if rule.port_range_max is None:
+        if rule['port_range_max'] is None:
             port_range_max = port_range_min
         port_range = port_range_min + ':' + port_range_max
     return port_range
@@ -423,6 +423,16 @@ class DeleteSecurityGroupRule(common.NetworkAndComputeDelete):
 class ListSecurityGroupRule(common.NetworkAndComputeLister):
     _description = _("List security group rules")
 
+    def _format_network_security_group_rule(self, rule):
+        """Transform the SDK SecurityGroupRule object to a dict
+
+        The SDK object gets in the way of reformatting columns...
+        Create port_range column from port_range_min and port_range_max
+        """
+        rule = rule.to_dict()
+        rule['port_range'] = _format_network_port_range(rule)
+        return rule
+
     def update_parser_common(self, parser):
         parser.add_argument(
             'group',
@@ -508,7 +518,7 @@ class ListSecurityGroupRule(common.NetworkAndComputeLister):
             'id',
             'protocol',
             'remote_ip_prefix',
-            'port_range_min',
+            'port_range',
         )
         if parsed_args.long:
             columns = columns + ('direction', 'ethertype',)
@@ -535,16 +545,13 @@ class ListSecurityGroupRule(common.NetworkAndComputeLister):
         if parsed_args.protocol is not None:
             query['protocol'] = parsed_args.protocol
 
-        rules = list(client.security_group_rules(**query))
-
-        # Reformat the rules to display a port range instead
-        # of just the port range minimum. This maintains
-        # output compatibility with compute.
-        for rule in rules:
-            rule.port_range_min = _format_network_port_range(rule)
+        rules = [
+            self._format_network_security_group_rule(r)
+            for r in client.security_group_rules(**query)
+        ]
 
         return (column_headers,
-                (utils.get_item_properties(
+                (utils.get_dict_properties(
                     s, columns,
                 ) for s in rules))
 
diff --git a/openstackclient/tests/functional/network/v2/test_security_group_rule.py b/openstackclient/tests/functional/network/v2/test_security_group_rule.py
index ec3731eb9a..c91de1a570 100644
--- a/openstackclient/tests/functional/network/v2/test_security_group_rule.py
+++ b/openstackclient/tests/functional/network/v2/test_security_group_rule.py
@@ -12,8 +12,6 @@
 
 import uuid
 
-import testtools
-
 from openstackclient.tests.functional import base
 
 
@@ -54,7 +52,6 @@ class SecurityGroupRuleTests(base.TestCase):
                                    cls.SECURITY_GROUP_NAME)
         cls.assertOutput('', raw_output)
 
-    @testtools.skip('broken SDK testing')
     def test_security_group_rule_list(self):
         opts = self.get_opts(self.ID_HEADER)
         raw_output = self.openstack('security group rule list ' +