8 from framework import VppTestCase, VppTestRunner, running_extended_tests, \
10 from vpp_ip_route import VppIpTable, VppIpRoute, VppRoutePath
13 class VCLAppWorker(Worker):
14 """ VCL Test Application Worker """
16 def __init__(self, build_dir, appname, args, logger, env={}):
17 vcl_lib_dir = "%s/vpp/.libs" % build_dir
18 if "iperf" in appname:
20 env.update({'LD_PRELOAD':
21 "%s/libvcl_ldpreload.so.0.0.0" % vcl_lib_dir})
23 app = "%s/%s" % (vcl_lib_dir, appname)
24 if not os.path.isfile(app):
25 app = "%s/vpp/%s" % (build_dir, appname)
26 env.update({'LD_PRELOAD':
27 "%s/libvcl_ldpreload.so.0.0.0" % vcl_lib_dir})
28 self.args = [app] + args
29 super(VCLAppWorker, self).__init__(self.args, logger, env)
32 class VCLTestCase(VppTestCase):
33 """ VCL Test Class """
35 def __init__(self, methodName):
36 var = "VPP_TEST_BUILD_DIR"
37 self.build_dir = os.getenv(var, None)
38 if self.build_dir is None:
39 raise Exception("Environment variable `%s' not set" % var)
40 self.vppDebug = 'vpp_debug' in self.build_dir
41 self.server_addr = "127.0.0.1"
42 self.server_port = "22000"
43 self.server_args = [self.server_port]
45 self.echo_phrase = "Hello, world! Jenny is a friend of mine."
47 super(VCLTestCase, self).__init__(methodName)
49 def cut_thru_setup(self):
50 self.vapi.session_enable_disable(is_enabled=1)
52 def cut_thru_tear_down(self):
53 self.vapi.session_enable_disable(is_enabled=0)
55 def cut_thru_test(self, server_app, server_args, client_app, client_args):
56 self.env = {'VCL_API_PREFIX': self.shm_prefix,
57 'VCL_APP_SCOPE_LOCAL': "true"}
59 worker_server = VCLAppWorker(self.build_dir, server_app, server_args,
60 self.logger, self.env)
63 worker_client = VCLAppWorker(self.build_dir, client_app, client_args,
64 self.logger, self.env)
66 worker_client.join(self.timeout)
68 self.validateResults(worker_client, worker_server, self.timeout)
69 except Exception, error:
70 self.fail("Failed with %s" % error)
72 def thru_host_stack_setup(self):
73 self.vapi.session_enable_disable(is_enabled=1)
74 self.create_loopback_interfaces(range(2))
78 for i in self.lo_interfaces:
82 tbl = VppIpTable(self, table_id)
85 i.set_table_ip4(table_id)
89 # Configure namespaces
90 self.vapi.app_namespace_add(namespace_id="0", secret=1234,
91 sw_if_index=self.loop0.sw_if_index)
92 self.vapi.app_namespace_add(namespace_id="1", secret=5678,
93 sw_if_index=self.loop1.sw_if_index)
95 # Add inter-table routes
96 ip_t01 = VppIpRoute(self, self.loop1.local_ip4, 32,
97 [VppRoutePath("0.0.0.0",
100 ip_t10 = VppIpRoute(self, self.loop0.local_ip4, 32,
101 [VppRoutePath("0.0.0.0",
103 nh_table_id=0)], table_id=1)
104 ip_t01.add_vpp_config()
105 ip_t10.add_vpp_config()
107 def thru_host_stack_tear_down(self):
108 for i in self.lo_interfaces:
113 self.vapi.session_enable_disable(is_enabled=0)
115 def thru_host_stack_test(self, server_app, server_args,
116 client_app, client_args):
117 self.env = {'VCL_API_PREFIX': self.shm_prefix,
118 'VCL_APP_SCOPE_GLOBAL': "true",
119 'VCL_APP_NAMESPACE_ID': "0",
120 'VCL_APP_NAMESPACE_SECRET': "1234"}
122 worker_server = VCLAppWorker(self.build_dir, server_app, server_args,
123 self.logger, self.env)
124 worker_server.start()
127 self.env.update({'VCL_APP_NAMESPACE_ID': "1",
128 'VCL_APP_NAMESPACE_SECRET': "5678"})
129 worker_client = VCLAppWorker(self.build_dir, client_app, client_args,
130 self.logger, self.env)
131 worker_client.start()
132 worker_client.join(self.timeout)
135 self.validateResults(worker_client, worker_server, self.timeout)
136 except Exception, error:
137 self.fail("Failed with %s" % error)
139 def validateResults(self, worker_client, worker_server, timeout):
140 if os.path.isdir('/proc/{}'.format(worker_server.process.pid)):
141 self.logger.info("Killing server worker process (pid %d)" %
142 worker_server.process.pid)
143 os.killpg(os.getpgid(worker_server.process.pid), signal.SIGTERM)
145 self.logger.info("Client worker result is `%s'" % worker_client.result)
147 if worker_client.result is None:
151 "Timeout: %ss! Killing client worker process (pid %d)" %
152 (timeout, worker_client.process.pid))
153 os.killpg(os.getpgid(worker_client.process.pid),
158 "Couldn't kill client worker process")
162 "Timeout! Client worker did not finish in %ss" % timeout)
163 self.assert_equal(worker_client.result, 0, "Binary test return code")
166 class VCLCutThruTestCase(VCLTestCase):
167 """ VCL Cut Thru Tests """
170 super(VCLCutThruTestCase, self).setUp()
172 self.cut_thru_setup()
173 self.client_echo_test_args = [self.server_addr, self.server_port,
174 "-E", self.echo_phrase, "-X"]
175 self.client_iperf3_timeout = 20
176 self.client_iperf3_args = ["-V4d", "-c", self.server_addr]
177 self.server_iperf3_args = ["-V4d", "-s"]
178 self.client_uni_dir_nsock_timeout = 60
179 self.client_uni_dir_nsock_test_args = [self.server_addr,
181 "-I", "5", "-U", "-X"]
182 self.client_bi_dir_nsock_timeout = 120
183 self.client_bi_dir_nsock_test_args = [self.server_addr,
185 "-I", "2", "-B", "-X"]
188 self.cut_thru_tear_down()
190 super(VCLCutThruTestCase, self).tearDown()
192 def test_ldp_cut_thru_echo(self):
193 """ run LDP cut thru echo test """
195 self.cut_thru_test("sock_test_server", self.server_args,
196 "sock_test_client", self.client_echo_test_args)
198 def test_ldp_cut_thru_iperf3(self):
199 """ run LDP cut thru iperf3 test """
202 subprocess.check_output(['iperf3', '-v'])
204 self.logger.error("WARNING: 'iperf3' is not installed,")
205 self.logger.error(" 'test_ldp_cut_thru_iperf3' not run!")
208 self.timeout = self.client_iperf3_timeout
209 self.cut_thru_test("iperf3", self.server_iperf3_args,
210 "iperf3", self.client_iperf3_args)
212 @unittest.skipUnless(running_extended_tests(), "part of extended tests")
213 def test_ldp_cut_thru_uni_dir_nsock(self):
214 """ run LDP cut thru uni-directional (multiple sockets) test """
216 self.timeout = self.client_uni_dir_nsock_timeout
217 self.cut_thru_test("sock_test_server", self.server_args,
219 self.client_uni_dir_nsock_test_args)
221 @unittest.skipUnless(running_extended_tests(), "part of extended tests")
222 def test_ldp_cut_thru_bi_dir_nsock(self):
223 """ run LDP cut thru bi-directional (multiple sockets) test """
225 self.timeout = self.client_bi_dir_nsock_timeout
226 self.cut_thru_test("sock_test_server", self.server_args,
228 self.client_bi_dir_nsock_test_args)
230 def test_vcl_cut_thru_echo(self):
231 """ run VCL cut thru echo test """
233 self.cut_thru_test("vcl_test_server", self.server_args,
234 "vcl_test_client", self.client_echo_test_args)
236 @unittest.skipUnless(running_extended_tests(), "part of extended tests")
237 def test_vcl_cut_thru_uni_dir_nsock(self):
238 """ run VCL cut thru uni-directional (multiple sockets) test """
240 self.timeout = self.client_uni_dir_nsock_timeout
241 self.cut_thru_test("vcl_test_server", self.server_args,
243 self.client_uni_dir_nsock_test_args)
245 @unittest.skipUnless(running_extended_tests(), "part of extended tests")
246 def test_vcl_cut_thru_bi_dir_nsock(self):
247 """ run VCL cut thru bi-directional (multiple sockets) test """
249 self.timeout = self.client_bi_dir_nsock_timeout
250 self.cut_thru_test("vcl_test_server", self.server_args,
252 self.client_bi_dir_nsock_test_args)
255 class VCLThruHostStackTestCase(VCLTestCase):
256 """ VCL Thru Host Stack Tests """
259 super(VCLThruHostStackTestCase, self).setUp()
261 self.thru_host_stack_setup()
262 self.client_echo_test_args = [self.loop0.local_ip4,
264 "-E", self.echo_phrase, "-X"]
267 self.thru_host_stack_tear_down()
269 super(VCLThruHostStackTestCase, self).tearDown()
271 def test_ldp_thru_host_stack_echo(self):
272 """ run LDP thru host stack echo test """
274 self.thru_host_stack_test("sock_test_server", self.server_args,
276 self.client_echo_test_args)
277 # TBD: Remove these when VPP thru host teardown config bug is fixed.
278 self.thru_host_stack_test("vcl_test_server", self.server_args,
280 self.client_echo_test_args)
282 def test_vcl_thru_host_stack_echo(self):
283 """ run VCL thru host stack echo test """
285 # TBD: Enable this when VPP thru host teardown config bug is fixed.
286 # self.thru_host_stack_test("vcl_test_server", self.server_args,
288 # self.client_echo_test_args)
290 # TBD: Remove VCLThruHostStackExtended*TestCase classes and move
291 # tests here when VPP thru host teardown/setup config bug
295 class VCLThruHostStackExtendedATestCase(VCLTestCase):
296 """ VCL Thru Host Stack Extended Tests """
299 super(VCLThruHostStackExtendedATestCase, self).setUp()
301 self.thru_host_stack_setup()
303 self.client_bi_dir_nsock_timeout = 120
305 self.client_bi_dir_nsock_timeout = 60
306 self.client_bi_dir_nsock_test_args = [self.loop0.local_ip4,
308 "-I", "2", "-B", "-X"]
311 self.thru_host_stack_tear_down()
313 super(VCLThruHostStackExtendedATestCase, self).tearDown()
315 @unittest.skipUnless(running_extended_tests(), "part of extended tests")
316 def test_vcl_thru_host_stack_bi_dir_nsock(self):
317 """ run VCL thru host stack bi-directional (multiple sockets) test """
319 self.timeout = self.client_bi_dir_nsock_timeout
320 self.thru_host_stack_test("vcl_test_server", self.server_args,
322 self.client_bi_dir_nsock_test_args)
325 class VCLThruHostStackExtendedBTestCase(VCLTestCase):
326 """ VCL Thru Host Stack Extended Tests """
329 super(VCLThruHostStackExtendedBTestCase, self).setUp()
331 self.thru_host_stack_setup()
333 self.client_bi_dir_nsock_timeout = 120
335 self.client_bi_dir_nsock_timeout = 60
336 self.client_bi_dir_nsock_test_args = [self.loop0.local_ip4,
338 "-I", "2", "-B", "-X"]
341 self.thru_host_stack_tear_down()
343 super(VCLThruHostStackExtendedBTestCase, self).tearDown()
345 @unittest.skipUnless(running_extended_tests(), "part of extended tests")
346 def test_ldp_thru_host_stack_bi_dir_nsock(self):
347 """ run LDP thru host stack bi-directional (multiple sockets) test """
349 self.timeout = self.client_bi_dir_nsock_timeout
350 self.thru_host_stack_test("sock_test_server", self.server_args,
352 self.client_bi_dir_nsock_test_args)
355 class VCLThruHostStackExtendedCTestCase(VCLTestCase):
356 """ VCL Thru Host Stack Extended Tests """
359 super(VCLThruHostStackExtendedCTestCase, self).setUp()
361 self.thru_host_stack_setup()
363 self.client_uni_dir_nsock_timeout = 120
364 self.numSockets = "2"
366 self.client_uni_dir_nsock_timeout = 120
367 self.numSockets = "5"
369 self.client_uni_dir_nsock_test_args = [self.loop0.local_ip4,
371 "-I", self.numSockets,
375 self.thru_host_stack_tear_down()
377 super(VCLThruHostStackExtendedCTestCase, self).tearDown()
379 @unittest.skipUnless(running_extended_tests(), "part of extended tests")
380 def test_ldp_thru_host_stack_uni_dir_nsock(self):
381 """ run LDP thru host stack uni-directional (multiple sockets) test """
383 self.timeout = self.client_uni_dir_nsock_timeout
384 self.thru_host_stack_test("sock_test_server", self.server_args,
386 self.client_uni_dir_nsock_test_args)
389 class VCLThruHostStackExtendedDTestCase(VCLTestCase):
390 """ VCL Thru Host Stack Extended Tests """
393 super(VCLThruHostStackExtendedDTestCase, self).setUp()
395 self.thru_host_stack_setup()
397 self.client_uni_dir_nsock_timeout = 120
398 self.numSockets = "2"
400 self.client_uni_dir_nsock_timeout = 120
401 self.numSockets = "5"
403 self.client_uni_dir_nsock_test_args = [self.loop0.local_ip4,
405 "-I", self.numSockets,
409 self.thru_host_stack_tear_down()
411 super(VCLThruHostStackExtendedDTestCase, self).tearDown()
413 @unittest.skipUnless(running_extended_tests(), "part of extended tests")
414 def test_vcl_thru_host_stack_uni_dir_nsock(self):
415 """ run VCL thru host stack uni-directional (multiple sockets) test """
417 self.timeout = self.client_uni_dir_nsock_timeout
418 self.thru_host_stack_test("vcl_test_server", self.server_args,
420 self.client_uni_dir_nsock_test_args)
423 class VCLThruHostStackIperfTestCase(VCLTestCase):
424 """ VCL Thru Host Stack Iperf Tests """
427 super(VCLThruHostStackIperfTestCase, self).setUp()
429 self.thru_host_stack_setup()
430 self.client_iperf3_timeout = 20
431 self.client_iperf3_args = ["-V4d", "-c", self.loop0.local_ip4]
432 self.server_iperf3_args = ["-V4d", "-s"]
435 self.thru_host_stack_tear_down()
437 super(VCLThruHostStackIperfTestCase, self).tearDown()
439 def test_ldp_thru_host_stack_iperf3(self):
440 """ run LDP thru host stack iperf3 test """
443 subprocess.check_output(['iperf3', '-v'])
445 self.logger.error("WARNING: 'iperf3' is not installed,")
447 " 'test_ldp_thru_host_stack_iperf3' not run!")
450 self.timeout = self.client_iperf3_timeout
451 self.thru_host_stack_test("iperf3", self.server_iperf3_args,
452 "iperf3", self.client_iperf3_args)
455 if __name__ == '__main__':
456 unittest.main(testRunner=VppTestRunner)