def test_build_acl_tcp_range_dst_eq_src(self):
""" Build flows from TCP Range Dst and Eq Src ACL."""
type_ = 'range_dst_eq_src'
input_ = self.tcp_input_path.format(type_)
data = self.load_json_file(input_)
flow_builder = AclFlowBuilder(data)
flow_builder.ALLOWED_FLOWS_SIZE = 5
output_path = self.tcp_output_path
generator = flow_builder.build()
limit_ = 1
for i in xrange(1, limit_ + 1):
output_ = output_path.format(type_, i)
self.compare_json(output_, generator.next())
self.assertRaises(StopIteration, generator.next)
def test_build_2ip_1udp_3tcp_eq_src_eq_dst_1udp_double_range(self):
""" Build 2 IP, 1 UDP, 3 TCP Eq Src Eq Dst and 1 UDP Double Range."""
type_ = '2ip+1udp+3tcp_eq_src_eq_dst+1udp_double_range'
input_ = self.mixed_input_path.format(type_)
data = self.load_json_file(input_)
flow_builder = AclFlowBuilder(data)
flow_builder.ALLOWED_FLOWS_SIZE = 5
output_path = self.mixed_output_path
generator = flow_builder.build()
limit_ = 5
for i in xrange(1, limit_ + 1):
output_ = output_path.format(type_, i)
self.compare_json(output_, generator.next())
self.assertRaises(StopIteration, generator.next)
def test_build_1tcp_range_src_eq_dst_4icmp_1ip_1udp_double_range(self):
""" Build 1 TCP Range Src and Eq Dst, 4 ICMP, 1 IP and 1 UDP Double Range."""
type_ = '1tcp_range_src_eq_dst+4icmp+1ip+1udp_double_range'
input_ = self.mixed_input_path.format(type_)
data = self.load_json_file(input_)
flow_builder = AclFlowBuilder(data)
flow_builder.ALLOWED_FLOWS_SIZE = 5
output_path = self.mixed_output_path
generator = flow_builder.build()
limit_ = 5
for i in xrange(1, limit_ + 1):
output_ = output_path.format(type_, i)
self.compare_json(output_, generator.next())
self.assertRaises(StopIteration, generator.next)
def test_build_1udp_1udp_range_src_2icmp_1tcp_eq_src(self):
""" Build 1 UDP, 1 UDP Range Src, 2 ICMP and 1 TCP Eq Src."""
type_ = '1udp+1udp_range_src+2icmp+1tcp_eq_src'
input_ = self.mixed_input_path.format(type_)
data = self.load_json_file(input_)
flow_builder = AclFlowBuilder(data)
flow_builder.ALLOWED_FLOWS_SIZE = 5
output_path = self.mixed_output_path
generator = flow_builder.build()
limit_ = 2
for i in xrange(1, limit_ + 1):
output_ = output_path.format(type_, i)
self.compare_json(output_, generator.next())
self.assertRaises(StopIteration, generator.next)
def test_build_acl_udp_double_range(self):
""" Build flows from UDP Double Range ACL."""
type_ = 'double_range'
input_ = self.udp_input_path.format(type_)
data = self.load_json_file(input_)
flow_builder = AclFlowBuilder(data)
flow_builder.ALLOWED_FLOWS_SIZE = 5
output_path = self.udp_output_path
generator = flow_builder.build()
limit_ = 3
for i in xrange(1, limit_ + 1):
output_ = output_path.format(type_, i)
self.compare_json(output_, generator.next())
self.assertRaises(StopIteration, generator.next)
