def test_invalid_priority(self): self.assertRaises(InvalidSecurityGroupRule, SecurityGroupRuleFactory.create, SecurityGroupRuleFactory.PROTOCOL_UDP, -1) self.assertRaises(InvalidSecurityGroupRule, SecurityGroupRuleFactory.create, SecurityGroupRuleFactory.PROTOCOL_UDP, 101) self.assertRaises(InvalidSecurityGroupRule, SecurityGroupRuleFactory.create, SecurityGroupRuleFactory.PROTOCOL_UDP, '10') rule = SecurityGroupRuleFactory.create(SecurityGroupRuleFactory.PROTOCOL_UDP, 0) self.assertTrue(rule) rule = SecurityGroupRuleFactory.create(SecurityGroupRuleFactory.PROTOCOL_UDP, 100) self.assertTrue(rule)
def test_rule_with_existing_id(self): rule = SecurityGroupRuleFactory.create('gre', 0, security_group_rule_id='fakeid') json_data = rule.to_json() self.assertEqual(json_data['security_group_rule_id'], 'fakeid')
def test_gre_rule_structure(self): rule = SecurityGroupRuleFactory.create( SecurityGroupRuleFactory.PROTOCOL_GRE, 0, direction=1, action='drop', security_group_rule_name='unittest', ip_network=self.ip_network) json_data = rule.to_json() self.assertEqual(json_data['val3'], self.ip_network)
def test_invalid_priority(self): self.assertRaises(InvalidSecurityGroupRule, SecurityGroupRuleFactory.create, SecurityGroupRuleFactory.PROTOCOL_UDP, -1) self.assertRaises(InvalidSecurityGroupRule, SecurityGroupRuleFactory.create, SecurityGroupRuleFactory.PROTOCOL_UDP, 101) self.assertRaises(InvalidSecurityGroupRule, SecurityGroupRuleFactory.create, SecurityGroupRuleFactory.PROTOCOL_UDP, '10') rule = SecurityGroupRuleFactory.create( SecurityGroupRuleFactory.PROTOCOL_UDP, 0) self.assertTrue(rule) rule = SecurityGroupRuleFactory.create( SecurityGroupRuleFactory.PROTOCOL_UDP, 100) self.assertTrue(rule)
def test_icmp_rule_structure(self): rule = SecurityGroupRuleFactory.create( SecurityGroupRuleFactory.PROTOCOL_ICMP, 0, icmp_type=self.icmp_type, icmp_code=self.icmp_code, ip_network=self.ip_network) json_data = rule.to_json() self.assertEqual(json_data['val1'], self.icmp_type) self.assertEqual(json_data['val2'], self.icmp_code) self.assertEqual(json_data['val3'], self.ip_network)
def test_udp_rule_structure(self): rule = SecurityGroupRuleFactory.create( SecurityGroupRuleFactory.PROTOCOL_UDP, 0, start_port=self.start_port, end_port=self.end_port, ip_network=self.ip_network) json_data = rule.to_json() self.assertEqual(json_data['val1'], self.start_port) self.assertEqual(json_data['val2'], self.end_port) self.assertEqual(json_data['val3'], self.ip_network)
def test_gre_rule_structure(self): rule = SecurityGroupRuleFactory.create( SecurityGroupRuleFactory.PROTOCOL_GRE, 0, direction=1, action='drop', security_group_rule_name='unittest', ip_network=self.ip_network) json_data = rule.to_json() self.assertEqual(json_data['val3'], self.ip_network)
def test_icmp_rule_structure(self): rule = SecurityGroupRuleFactory.create( SecurityGroupRuleFactory.PROTOCOL_ICMP, 0, icmp_type=self.icmp_type, icmp_code=self.icmp_code, ip_network=self.ip_network) json_data = rule.to_json() self.assertEqual(json_data['val1'], self.icmp_type) self.assertEqual(json_data['val2'], self.icmp_code) self.assertEqual(json_data['val3'], self.ip_network)
def test_udp_rule_structure(self): rule = SecurityGroupRuleFactory.create( SecurityGroupRuleFactory.PROTOCOL_UDP, 0, start_port=self.start_port, end_port=self.end_port, ip_network=self.ip_network) json_data = rule.to_json() self.assertEqual(json_data['val1'], self.start_port) self.assertEqual(json_data['val2'], self.end_port) self.assertEqual(json_data['val3'], self.ip_network)
def test_rule_with_existing_id(self): rule = SecurityGroupRuleFactory.create('gre', 0, security_group_rule_id='fakeid') json_data = rule.to_json() self.assertEqual(json_data['security_group_rule_id'], 'fakeid')