@@ -870,7 +870,7 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork):
870870 _security_group_rules = [_security_group_rule_tcp ,
871871 _security_group_rule_icmp ]
872872
873- expected_columns_with_group_and_long = (
873+ expected_columns_with_group = (
874874 'ID' ,
875875 'IP Protocol' ,
876876 'Ethertype' ,
@@ -885,14 +885,15 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork):
885885 'Ethertype' ,
886886 'IP Range' ,
887887 'Port Range' ,
888+ 'Direction' ,
888889 'Remote Security Group' ,
889890 'Security Group' ,
890891 )
891892
892- expected_data_with_group_and_long = []
893+ expected_data_with_group = []
893894 expected_data_no_group = []
894895 for _security_group_rule in _security_group_rules :
895- expected_data_with_group_and_long .append ((
896+ expected_data_with_group .append ((
896897 _security_group_rule .id ,
897898 _security_group_rule .protocol ,
898899 _security_group_rule .ether_type ,
@@ -909,6 +910,7 @@ class TestListSecurityGroupRuleNetwork(TestSecurityGroupRuleNetwork):
909910 _security_group_rule .remote_ip_prefix ,
910911 security_group_rule ._format_network_port_range (
911912 _security_group_rule ),
913+ _security_group_rule .direction ,
912914 _security_group_rule .remote_group_id ,
913915 _security_group_rule .security_group_id ,
914916 ))
@@ -935,14 +937,12 @@ def test_list_default(self):
935937 self .assertEqual (self .expected_columns_no_group , columns )
936938 self .assertEqual (self .expected_data_no_group , list (data ))
937939
938- def test_list_with_group_and_long (self ):
940+ def test_list_with_group (self ):
939941 self ._security_group_rule_tcp .port_range_min = 80
940942 arglist = [
941- '--long' ,
942943 self ._security_group .id ,
943944 ]
944945 verifylist = [
945- ('long' , True ),
946946 ('group' , self ._security_group .id ),
947947 ]
948948 parsed_args = self .check_parser (self .cmd , arglist , verifylist )
@@ -952,8 +952,8 @@ def test_list_with_group_and_long(self):
952952 self .network .security_group_rules .assert_called_once_with (** {
953953 'security_group_id' : self ._security_group .id ,
954954 })
955- self .assertEqual (self .expected_columns_with_group_and_long , columns )
956- self .assertEqual (self .expected_data_with_group_and_long , list (data ))
955+ self .assertEqual (self .expected_columns_with_group , columns )
956+ self .assertEqual (self .expected_data_with_group , list (data ))
957957
958958 def test_list_with_ignored_options (self ):
959959 self ._security_group_rule_tcp .port_range_min = 80
0 commit comments