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)