@@ -56,7 +56,7 @@ def test_rules(cloud_name: str):
5656 # we allow ingress from the same security group
5757 # but only for the default security group
5858 r_group_id = rule .remote_group_id
59- if ( r_group_id == "PARENT" and not r_custom_sg ) :
59+ if r_group_id == "PARENT" and not r_custom_sg :
6060 ingress_from_same_sg += 1
6161 elif direction == "egress" and ethertype == "IPv4" :
6262 egress_rules += 1
@@ -82,34 +82,39 @@ def test_rules(cloud_name: str):
8282 f"Expected only ingress rules for default security groups, "
8383 f"that allow ingress traffic from the same group. "
8484 f"But there are more - in total { ingress_rules } ingress rules. "
85- f"There should be only { ingress_from_same_sg } ingress rules." )
86- assert egress_rules > 0 , (
87- f"Expected to have more than { egress_rules } egress rules present." )
88- var_list = [egress_ipv4_default_sg , egress_ipv4_custom_sg ,
89- egress_ipv6_default_sg , egress_ipv6_custom_sg ]
85+ f"There should be only { ingress_from_same_sg } ingress rules."
86+ )
87+ assert (
88+ egress_rules > 0
89+ ), f"Expected to have more than { egress_rules } egress rules present."
90+ var_list = [
91+ egress_ipv4_default_sg ,
92+ egress_ipv4_custom_sg ,
93+ egress_ipv6_default_sg ,
94+ egress_ipv6_custom_sg ,
95+ ]
9096 assert all ([var > 0 for var in var_list ]), (
9197 "Not all expected egress rules are present. "
9298 "Expected rules for egress for IPv4 and IPv6 "
93- "both for default and custom security groups." )
99+ "both for default and custom security groups."
100+ )
94101
95- result_dict = {
96- "Ingress Rules" : ingress_rules ,
97- "Egress Rules" : egress_rules
98- }
102+ result_dict = {"Ingress Rules" : ingress_rules , "Egress Rules" : egress_rules }
99103 return result_dict
100104
101105
102106def main ():
103107 parser = argparse .ArgumentParser (
104- description = "SCS Default Security Group Rules Checker" )
108+ description = "SCS Default Security Group Rules Checker"
109+ )
105110 parser .add_argument (
106- "--os-cloud" , type = str ,
111+ "--os-cloud" ,
112+ type = str ,
107113 help = "Name of the cloud from clouds.yaml, alternative "
108- "to the OS_CLOUD environment variable"
114+ "to the OS_CLOUD environment variable" ,
109115 )
110116 parser .add_argument (
111- "--debug" , action = "store_true" ,
112- help = "Enable OpenStack SDK debug logging"
117+ "--debug" , action = "store_true" , help = "Enable OpenStack SDK debug logging"
113118 )
114119 args = parser .parse_args ()
115120 openstack .enable_logging (debug = args .debug )
0 commit comments