1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
#!/router/bin/python
from .trex_general_test import CTRexGeneral_Test, CTRexScenario
from .tests_exceptions import *
import time
from CPlatform import CStaticRouteConfig
class CTRexVlan_Test(CTRexGeneral_Test):#(unittest.TestCase):
"""This class defines test for vlan platform configutation file"""
def __init__(self, *args, **kwargs):
super(CTRexVlan_Test, self).__init__(*args, **kwargs)
self.unsupported_modes = ['loopback'] # We test on routers
def setUp(self):
super(CTRexVlan_Test, self).setUp() # launch super test class setUp process
def test_platform_cfg_vlan(self):
if self.is_loopback:
self.skip('Not relevant on loopback')
if not CTRexScenario.router_cfg['no_dut_config']:
self.router.configure_basic_interfaces(vlan = True)
self.router.config_pbr(mode = "config", vlan = True)
mult = self.get_benchmark_param('multiplier')
core = self.get_benchmark_param('cores')
self.create_vlan_conf_file()
ret = self.trex.start_trex (
c = core,
m = mult,
d = 60,
f = 'cap2/dns.yaml',
cfg = '/tmp/trex_files/trex_cfg_vlan.yaml',
l = 100,
limit_ports = 4)
trex_res = self.trex.sample_to_run_finish()
print("\nLATEST RESULT OBJECT:")
print(trex_res)
self.check_general_scenario_results(trex_res, check_latency = True)
def tearDown(self):
pass
# Add vlan to platform config file
# write the result to new config file in /tmp, to be used by the test
def create_vlan_conf_file(self):
remote_cfg_file = self.trex.get_trex_config()
out = open("/tmp/trex_cfg_vlan.yaml", "w")
insert_vlan = False
vlan_val_1 = "100"
vlan_val_2 = "200"
vlan_val = vlan_val_1
for line in remote_cfg_file.split("\n"):
out.write (line + "\n")
# when we see "port_info" start inserting, until port_info section end
if "port_info" in line.lstrip():
num_space_port_info = len(line) - len(line.lstrip())
insert_vlan = True
continue
if insert_vlan:
num_space = len(line) - len(line.lstrip())
if num_space == num_space_port_info:
insert_vlan = False
continue
if len(line.lstrip()) > 0 and line.lstrip()[0] == '-':
# insert vlan for each port (every time a section start with '-'
num_space = len(line) - len(line.lstrip(' -'))
# need to insert in the same indentation as other lines in the section
out.write(num_space* " " + "vlan: " + vlan_val + "\n")
if vlan_val == vlan_val_1:
vlan_val = vlan_val_2
else:
vlan_val = vlan_val_1
out.close()
self.trex.push_files("/tmp/trex_cfg_vlan.yaml");
if __name__ == "__main__":
pass
|
