tests: py2 cleanup - remove subclassing of object
[vpp.git] / src / plugins / nat / extras / nat_out2in_100ks.py
1 from trex_stl_lib.api import *
2
3 class STLS1:
4
5     def create_stream (self):
6         base_pkt = Ether()/IP(src="2.2.0.1")/UDP(sport=12)
7
8         pad = Padding()
9         if len(base_pkt) < 64:
10             pad_len = 64 - len(base_pkt)
11             pad.load = '\x00' * pad_len
12
13         vm = STLVM()
14
15         vm.tuple_var(name="tuple", ip_min="173.16.1.3", ip_max="173.16.4.234", port_min=1025, port_max=1124, limit_flows = 100000)
16
17         vm.write(fv_name="tuple.ip", pkt_offset="IP.dst")
18         vm.fix_chksum()
19
20         vm.write(fv_name="tuple.port", pkt_offset="UDP.dport")
21
22         pkt = STLPktBuilder(pkt=base_pkt/pad, vm=vm)
23
24         return STLStream(packet=pkt, mode=STLTXCont())
25
26     def get_streams (self, direction = 0, **kwargs):
27         return [self.create_stream()]
28
29
30 # dynamic load - used for trex console or simulator
31 def register():
32     return STLS1()
33
34
35