Tests: Refactor tearDown show command logging, add lifecycle markers.
[vpp.git] / test / test_vcl.py
1 #!/usr/bin/env python
2 """ Vpp VCL tests """
3
4 import unittest
5 import os
6 import subprocess
7 import signal
8 from framework import VppTestCase, VppTestRunner, running_extended_tests, \
9     Worker
10 from vpp_ip_route import VppIpTable, VppIpRoute, VppRoutePath, DpoProto
11
12
13 class VCLAppWorker(Worker):
14     """ VCL Test Application Worker """
15
16     def __init__(self, build_dir, appname, args, logger, env={}):
17         vcl_lib_dir = "%s/vpp/lib" % build_dir
18         if "iperf" in appname:
19             app = appname
20             env.update({'LD_PRELOAD':
21                         "%s/libvcl_ldpreload.so" % vcl_lib_dir})
22         elif "sock" in appname:
23             app = "%s/vpp/bin/%s" % (build_dir, appname)
24             env.update({'LD_PRELOAD':
25                         "%s/libvcl_ldpreload.so" % vcl_lib_dir})
26         else:
27             app = "%s/vpp/bin/%s" % (build_dir, appname)
28         self.args = [app] + args
29         super(VCLAppWorker, self).__init__(self.args, logger, env)
30
31
32 class VCLTestCase(VppTestCase):
33     """ VCL Test Class """
34
35     @classmethod
36     def setUpClass(cls):
37         super(VCLTestCase, cls).setUpClass()
38
39     @classmethod
40     def tearDownClass(cls):
41         super(VCLTestCase, cls).tearDownClass()
42
43     def setUp(self):
44         var = "VPP_BUILD_DIR"
45         self.build_dir = os.getenv(var, None)
46         if self.build_dir is None:
47             raise Exception("Environment variable `%s' not set" % var)
48         self.vppDebug = 'vpp_debug' in self.build_dir
49         self.server_addr = "127.0.0.1"
50         self.server_port = "22000"
51         self.server_args = [self.server_port]
52         self.server_ipv6_addr = "::1"
53         self.server_ipv6_args = ["-6", self.server_port]
54         self.timeout = 20
55         self.echo_phrase = "Hello, world! Jenny is a friend of mine."
56         self.pre_test_sleep = 0.3
57         self.post_test_sleep = 0.2
58
59         if os.path.isfile("/tmp/ldp_server_af_unix_socket"):
60             os.remove("/tmp/ldp_server_af_unix_socket")
61
62         super(VCLTestCase, self).setUp()
63
64     def cut_thru_setup(self):
65         self.vapi.session_enable_disable(is_enabled=1)
66
67     def cut_thru_tear_down(self):
68         self.vapi.session_enable_disable(is_enabled=0)
69
70     def cut_thru_test(self, server_app, server_args, client_app, client_args):
71         self.env = {'VCL_API_PREFIX': self.shm_prefix,
72                     'VCL_APP_SCOPE_LOCAL': "true"}
73
74         worker_server = VCLAppWorker(self.build_dir, server_app, server_args,
75                                      self.logger, self.env)
76         worker_server.start()
77         self.sleep(self.pre_test_sleep)
78         worker_client = VCLAppWorker(self.build_dir, client_app, client_args,
79                                      self.logger, self.env)
80         worker_client.start()
81         worker_client.join(self.timeout)
82         try:
83             self.validateResults(worker_client, worker_server, self.timeout)
84         except Exception as error:
85             self.fail("Failed with %s" % error)
86         self.sleep(self.post_test_sleep)
87
88     def thru_host_stack_setup(self):
89         self.vapi.session_enable_disable(is_enabled=1)
90         self.create_loopback_interfaces(2)
91
92         table_id = 1
93
94         for i in self.lo_interfaces:
95             i.admin_up()
96
97             if table_id != 0:
98                 tbl = VppIpTable(self, table_id)
99                 tbl.add_vpp_config()
100
101             i.set_table_ip4(table_id)
102             i.config_ip4()
103             table_id += 1
104
105         # Configure namespaces
106         self.vapi.app_namespace_add_del(namespace_id=b"1", secret=1234,
107                                         sw_if_index=self.loop0.sw_if_index)
108         self.vapi.app_namespace_add_del(namespace_id=b"2", secret=5678,
109                                         sw_if_index=self.loop1.sw_if_index)
110
111         # Add inter-table routes
112         ip_t01 = VppIpRoute(self, self.loop1.local_ip4, 32,
113                             [VppRoutePath("0.0.0.0",
114                                           0xffffffff,
115                                           nh_table_id=2)], table_id=1)
116         ip_t10 = VppIpRoute(self, self.loop0.local_ip4, 32,
117                             [VppRoutePath("0.0.0.0",
118                                           0xffffffff,
119                                           nh_table_id=1)], table_id=2)
120         ip_t01.add_vpp_config()
121         ip_t10.add_vpp_config()
122         self.logger.debug(self.vapi.cli("show ip fib"))
123
124     def thru_host_stack_tear_down(self):
125         for i in self.lo_interfaces:
126             i.unconfig_ip4()
127             i.set_table_ip4(0)
128             i.admin_down()
129
130         self.vapi.session_enable_disable(is_enabled=0)
131
132     def thru_host_stack_ipv6_setup(self):
133         self.vapi.session_enable_disable(is_enabled=1)
134         self.create_loopback_interfaces(2)
135
136         table_id = 1
137
138         for i in self.lo_interfaces:
139             i.admin_up()
140
141             tbl = VppIpTable(self, table_id, is_ip6=1)
142             tbl.add_vpp_config()
143
144             i.set_table_ip6(table_id)
145             i.config_ip6()
146             table_id += 1
147
148         # Configure namespaces
149         self.vapi.app_namespace_add_del(namespace_id=b"1", secret=1234,
150                                         sw_if_index=self.loop0.sw_if_index)
151         self.vapi.app_namespace_add_del(namespace_id=b"2", secret=5678,
152                                         sw_if_index=self.loop1.sw_if_index)
153
154         # Add inter-table routes
155         ip_t01 = VppIpRoute(self, self.loop1.local_ip6, 128,
156                             [VppRoutePath("::0", 0xffffffff,
157                                           nh_table_id=2,
158                                           proto=DpoProto.DPO_PROTO_IP6)],
159                             table_id=1, is_ip6=1)
160         ip_t10 = VppIpRoute(self, self.loop0.local_ip6, 128,
161                             [VppRoutePath("::0", 0xffffffff,
162                                           nh_table_id=1,
163                                           proto=DpoProto.DPO_PROTO_IP6)],
164                             table_id=2, is_ip6=1)
165         ip_t01.add_vpp_config()
166         ip_t10.add_vpp_config()
167         self.logger.debug(self.vapi.cli("show interface addr"))
168         self.logger.debug(self.vapi.cli("show ip6 fib"))
169
170     def thru_host_stack_ipv6_tear_down(self):
171         for i in self.lo_interfaces:
172             i.unconfig_ip6()
173             i.set_table_ip6(0)
174             i.admin_down()
175
176         self.vapi.session_enable_disable(is_enabled=0)
177
178     def thru_host_stack_test(self, server_app, server_args,
179                              client_app, client_args):
180         self.env = {'VCL_API_PREFIX': self.shm_prefix,
181                     'VCL_APP_SCOPE_GLOBAL': "true",
182                     'VCL_APP_NAMESPACE_ID': "1",
183                     'VCL_APP_NAMESPACE_SECRET': "1234"}
184
185         worker_server = VCLAppWorker(self.build_dir, server_app, server_args,
186                                      self.logger, self.env)
187         worker_server.start()
188         self.sleep(self.pre_test_sleep)
189
190         self.env.update({'VCL_APP_NAMESPACE_ID': "2",
191                          'VCL_APP_NAMESPACE_SECRET': "5678"})
192         worker_client = VCLAppWorker(self.build_dir, client_app, client_args,
193                                      self.logger, self.env)
194         worker_client.start()
195         worker_client.join(self.timeout)
196
197         try:
198             self.validateResults(worker_client, worker_server, self.timeout)
199         except Exception as error:
200             self.fail("Failed with %s" % error)
201         self.sleep(self.post_test_sleep)
202
203     def validateResults(self, worker_client, worker_server, timeout):
204         if os.path.isdir('/proc/{}'.format(worker_server.process.pid)):
205             self.logger.info("Killing server worker process (pid %d)" %
206                              worker_server.process.pid)
207             os.killpg(os.getpgid(worker_server.process.pid), signal.SIGKILL)
208             worker_server.join()
209         self.logger.info("Client worker result is `%s'" % worker_client.result)
210         error = False
211         if worker_client.result is None:
212             try:
213                 error = True
214                 self.logger.error(
215                     "Timeout: %ss! Killing client worker process (pid %d)" %
216                     (timeout, worker_client.process.pid))
217                 os.killpg(os.getpgid(worker_client.process.pid),
218                           signal.SIGKILL)
219                 worker_client.join()
220             except:
221                 self.logger.debug(
222                     "Couldn't kill client worker process")
223                 raise
224         if error:
225             raise Exception(
226                 "Timeout! Client worker did not finish in %ss" % timeout)
227         self.assert_equal(worker_client.result, 0, "Binary test return code")
228
229
230 class LDPCutThruTestCase(VCLTestCase):
231     """ LDP Cut Thru Tests """
232
233     @classmethod
234     def setUpClass(cls):
235         super(LDPCutThruTestCase, cls).setUpClass()
236
237     @classmethod
238     def tearDownClass(cls):
239         super(LDPCutThruTestCase, cls).tearDownClass()
240
241     def setUp(self):
242         super(LDPCutThruTestCase, self).setUp()
243
244         self.cut_thru_setup()
245         self.client_echo_test_args = ["-E", self.echo_phrase, "-X",
246                                       self.server_addr, self.server_port]
247         self.client_iperf3_timeout = 20
248         self.client_iperf3_args = ["-V4d", "-t 2", "-c", self.server_addr]
249         self.server_iperf3_args = ["-V4d", "-s"]
250         self.client_uni_dir_nsock_timeout = 20
251         self.client_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
252                                                "-I", "2",
253                                                self.server_addr,
254                                                self.server_port]
255         self.client_bi_dir_nsock_timeout = 20
256         self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
257                                               "-I", "2",
258                                               self.server_addr,
259                                               self.server_port]
260
261     def tearDown(self):
262         self.cut_thru_tear_down()
263         super(LDPCutThruTestCase, self).tearDown()
264
265     def show_commands_at_teardown(self):
266         self.logger.debug(self.vapi.cli("show session verbose 2"))
267
268     @unittest.skipUnless(running_extended_tests, "part of extended tests")
269     def test_ldp_cut_thru_echo(self):
270         """ run LDP cut thru echo test """
271
272         self.cut_thru_test("sock_test_server", self.server_args,
273                            "sock_test_client", self.client_echo_test_args)
274
275     def test_ldp_cut_thru_iperf3(self):
276         """ run LDP cut thru iperf3 test """
277
278         try:
279             subprocess.check_output(['iperf3', '-v'])
280         except subprocess.CalledProcessError:
281             self.logger.error(
282                 "WARNING: Subprocess returned non-0 running 'iperf3 -v")
283             self.logger.error("         'test_ldp_cut_thru_iperf3' not run!")
284             return
285         except OSError as e:
286             self.logger.error(
287                 "WARNING: Subprocess returned with OS error (%s) %s\n"
288                 "         'iperf3' is likely not installed,",
289                 e.errno, e.strerror)
290             self.logger.error("         'test_ldp_cut_thru_iperf3' not run!")
291             return
292         except Exception:
293             self.logger.exception(
294                 "Subprocess returned non-0 running 'iperf3 -v")
295
296         self.timeout = self.client_iperf3_timeout
297         self.cut_thru_test("iperf3", self.server_iperf3_args,
298                            "iperf3", self.client_iperf3_args)
299
300     @unittest.skipUnless(running_extended_tests, "part of extended tests")
301     def test_ldp_cut_thru_uni_dir_nsock(self):
302         """ run LDP cut thru uni-directional (multiple sockets) test """
303
304         self.timeout = self.client_uni_dir_nsock_timeout
305         self.cut_thru_test("sock_test_server", self.server_args,
306                            "sock_test_client",
307                            self.client_uni_dir_nsock_test_args)
308
309     @unittest.skipUnless(running_extended_tests, "part of extended tests")
310     def test_ldp_cut_thru_bi_dir_nsock(self):
311         """ run LDP cut thru bi-directional (multiple sockets) test """
312
313         self.timeout = self.client_bi_dir_nsock_timeout
314         self.cut_thru_test("sock_test_server", self.server_args,
315                            "sock_test_client",
316                            self.client_bi_dir_nsock_test_args)
317
318
319 class VCLCutThruTestCase(VCLTestCase):
320     """ VCL Cut Thru Tests """
321
322     @classmethod
323     def setUpClass(cls):
324         super(VCLCutThruTestCase, cls).setUpClass()
325
326     @classmethod
327     def tearDownClass(cls):
328         super(VCLCutThruTestCase, cls).tearDownClass()
329
330     def setUp(self):
331         super(VCLCutThruTestCase, self).setUp()
332
333         self.cut_thru_setup()
334         self.client_echo_test_args = ["-E", self.echo_phrase, "-X",
335                                       self.server_addr, self.server_port]
336
337         self.client_uni_dir_nsock_timeout = 20
338         self.client_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
339                                                "-I", "2",
340                                                self.server_addr,
341                                                self.server_port]
342         self.client_bi_dir_nsock_timeout = 20
343         self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
344                                               "-I", "2",
345                                               self.server_addr,
346                                               self.server_port]
347
348     def tearDown(self):
349         self.cut_thru_tear_down()
350
351         super(VCLCutThruTestCase, self).tearDown()
352
353     def test_vcl_cut_thru_echo(self):
354         """ run VCL cut thru echo test """
355
356         self.cut_thru_test("vcl_test_server", self.server_args,
357                            "vcl_test_client", self.client_echo_test_args)
358
359     def test_vcl_cut_thru_uni_dir_nsock(self):
360         """ run VCL cut thru uni-directional (multiple sockets) test """
361
362         self.timeout = self.client_uni_dir_nsock_timeout
363         self.cut_thru_test("vcl_test_server", self.server_args,
364                            "vcl_test_client",
365                            self.client_uni_dir_nsock_test_args)
366
367     def test_vcl_cut_thru_bi_dir_nsock(self):
368         """ run VCL cut thru bi-directional (multiple sockets) test """
369
370         self.timeout = self.client_bi_dir_nsock_timeout
371         self.cut_thru_test("vcl_test_server", self.server_args,
372                            "vcl_test_client",
373                            self.client_bi_dir_nsock_test_args)
374
375
376 class VCLThruHostStackEcho(VCLTestCase):
377     """ VCL Thru Host Stack Echo """
378
379     @classmethod
380     def setUpClass(cls):
381         super(VCLThruHostStackEcho, cls).setUpClass()
382
383     @classmethod
384     def tearDownClass(cls):
385         super(VCLThruHostStackEcho, cls).tearDownClass()
386
387     def setUp(self):
388         super(VCLThruHostStackEcho, self).setUp()
389
390         self.thru_host_stack_setup()
391         self.client_bi_dir_nsock_timeout = 20
392         self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
393                                               "-I", "2",
394                                               self.loop0.local_ip4,
395                                               self.server_port]
396         self.client_echo_test_args = ["-E", self.echo_phrase, "-X",
397                                       self.loop0.local_ip4,
398                                       self.server_port]
399
400     def tearDown(self):
401         self.thru_host_stack_tear_down()
402         super(VCLThruHostStackEcho, self).tearDown()
403
404     def show_commands_at_teardown(self):
405         self.logger.debug(self.vapi.cli("show app server"))
406         self.logger.debug(self.vapi.cli("show session verbose"))
407
408
409 class VCLThruHostStackTLS(VCLTestCase):
410     """ VCL Thru Host Stack TLS """
411
412     @classmethod
413     def setUpClass(cls):
414         super(VCLThruHostStackTLS, cls).setUpClass()
415
416     @classmethod
417     def tearDownClass(cls):
418         super(VCLThruHostStackTLS, cls).tearDownClass()
419
420     def setUp(self):
421         super(VCLThruHostStackTLS, self).setUp()
422
423         self.thru_host_stack_setup()
424         self.client_uni_dir_tls_timeout = 20
425         self.server_tls_args = ["-L", self.server_port]
426         self.client_uni_dir_tls_test_args = ["-N", "1000", "-U", "-X", "-L",
427                                              self.loop0.local_ip4,
428                                              self.server_port]
429
430     def test_vcl_thru_host_stack_tls_uni_dir(self):
431         """ run VCL thru host stack uni-directional TLS test """
432
433         self.timeout = self.client_uni_dir_tls_timeout
434         self.thru_host_stack_test("vcl_test_server", self.server_tls_args,
435                                   "vcl_test_client",
436                                   self.client_uni_dir_tls_test_args)
437
438     def tearDown(self):
439         self.thru_host_stack_tear_down()
440         super(VCLThruHostStackTLS, self).tearDown()
441
442     def show_commands_at_teardown(self):
443         self.logger.debug(self.vapi.cli("show app server"))
444         self.logger.debug(self.vapi.cli("show session verbose 2"))
445
446
447 class VCLThruHostStackBidirNsock(VCLTestCase):
448     """ VCL Thru Host Stack Bidir Nsock """
449
450     @classmethod
451     def setUpClass(cls):
452         super(VCLThruHostStackBidirNsock, cls).setUpClass()
453
454     @classmethod
455     def tearDownClass(cls):
456         super(VCLThruHostStackBidirNsock, cls).tearDownClass()
457
458     def setUp(self):
459         super(VCLThruHostStackBidirNsock, self).setUp()
460
461         self.thru_host_stack_setup()
462         self.client_bi_dir_nsock_timeout = 20
463         self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
464                                               "-I", "2",
465                                               self.loop0.local_ip4,
466                                               self.server_port]
467         self.client_echo_test_args = ["-E", self.echo_phrase, "-X",
468                                       self.loop0.local_ip4,
469                                       self.server_port]
470
471     def tearDown(self):
472         self.thru_host_stack_tear_down()
473         super(VCLThruHostStackBidirNsock, self).tearDown()
474
475     def show_commands_at_teardown(self):
476         self.logger.debug(self.vapi.cli("show session verbose 2"))
477
478     def test_vcl_thru_host_stack_bi_dir_nsock(self):
479         """ run VCL thru host stack bi-directional (multiple sockets) test """
480
481         self.timeout = self.client_bi_dir_nsock_timeout
482         self.thru_host_stack_test("vcl_test_server", self.server_args,
483                                   "vcl_test_client",
484                                   self.client_bi_dir_nsock_test_args)
485
486
487 class LDPThruHostStackBidirNsock(VCLTestCase):
488     """ LDP Thru Host Stack Bidir Nsock """
489
490     @classmethod
491     def setUpClass(cls):
492         super(LDPThruHostStackBidirNsock, cls).setUpClass()
493
494     @classmethod
495     def tearDownClass(cls):
496         super(LDPThruHostStackBidirNsock, cls).tearDownClass()
497
498     def setUp(self):
499         super(LDPThruHostStackBidirNsock, self).setUp()
500
501         self.thru_host_stack_setup()
502         if self.vppDebug:
503             self.client_bi_dir_nsock_timeout = 20
504             self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
505                                                   # OUCH! Host Stack Bug?
506                                                   # "-I", "2",
507                                                   self.loop0.local_ip4,
508                                                   self.server_port]
509         else:
510             self.client_bi_dir_nsock_timeout = 20
511             self.client_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
512                                                   # OUCH! Host Stack Bug?
513                                                   # "-I", "2",
514                                                   self.loop0.local_ip4,
515                                                   self.server_port]
516
517     def tearDown(self):
518         self.thru_host_stack_tear_down()
519         super(LDPThruHostStackBidirNsock, self).tearDown()
520
521     def show_commands_at_teardown(self):
522         self.logger.debug(self.vapi.cli("show session verbose 2"))
523
524     def test_ldp_thru_host_stack_bi_dir_nsock(self):
525         """ run LDP thru host stack bi-directional (multiple sockets) test """
526
527         self.timeout = self.client_bi_dir_nsock_timeout
528         self.thru_host_stack_test("sock_test_server", self.server_args,
529                                   "sock_test_client",
530                                   self.client_bi_dir_nsock_test_args)
531
532
533 class LDPThruHostStackNsock(VCLTestCase):
534     """ LDP Thru Host Stack Nsock """
535
536     @classmethod
537     def setUpClass(cls):
538         super(LDPThruHostStackNsock, cls).setUpClass()
539
540     @classmethod
541     def tearDownClass(cls):
542         super(LDPThruHostStackNsock, cls).tearDownClass()
543
544     def setUp(self):
545         super(LDPThruHostStackNsock, self).setUp()
546
547         self.thru_host_stack_setup()
548         if self.vppDebug:
549             self.client_uni_dir_nsock_timeout = 20
550             self.numSockets = "2"
551         else:
552             self.client_uni_dir_nsock_timeout = 20
553             self.numSockets = "5"
554
555         self.client_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
556                                                "-I", self.numSockets,
557                                                self.loop0.local_ip4,
558                                                self.server_port]
559
560     def tearDown(self):
561         self.thru_host_stack_tear_down()
562         super(LDPThruHostStackNsock, self).tearDown()
563
564     def test_ldp_thru_host_stack_uni_dir_nsock(self):
565         """ run LDP thru host stack uni-directional (multiple sockets) test """
566
567         self.timeout = self.client_uni_dir_nsock_timeout
568         self.thru_host_stack_test("sock_test_server", self.server_args,
569                                   "sock_test_client",
570                                   self.client_uni_dir_nsock_test_args)
571
572
573 class VCLThruHostStackNsock(VCLTestCase):
574     """ VCL Thru Host Stack Nsock """
575
576     @classmethod
577     def setUpClass(cls):
578         super(VCLThruHostStackNsock, cls).setUpClass()
579
580     @classmethod
581     def tearDownClass(cls):
582         super(VCLThruHostStackNsock, cls).tearDownClass()
583
584     def setUp(self):
585         super(VCLThruHostStackNsock, self).setUp()
586
587         self.thru_host_stack_setup()
588         if self.vppDebug:
589             self.client_uni_dir_nsock_timeout = 20
590             self.numSockets = "2"
591         else:
592             self.client_uni_dir_nsock_timeout = 20
593             self.numSockets = "5"
594
595         self.client_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
596                                                "-I", self.numSockets,
597                                                self.loop0.local_ip4,
598                                                self.server_port]
599
600     def tearDown(self):
601         self.thru_host_stack_tear_down()
602         super(VCLThruHostStackNsock, self).tearDown()
603
604     def test_vcl_thru_host_stack_uni_dir_nsock(self):
605         """ run VCL thru host stack uni-directional (multiple sockets) test """
606
607         self.timeout = self.client_uni_dir_nsock_timeout
608         self.thru_host_stack_test("vcl_test_server", self.server_args,
609                                   "vcl_test_client",
610                                   self.client_uni_dir_nsock_test_args)
611
612
613 class LDPThruHostStackIperf(VCLTestCase):
614     """ LDP Thru Host Stack Iperf  """
615
616     @classmethod
617     def setUpClass(cls):
618         super(LDPThruHostStackIperf, cls).setUpClass()
619
620     @classmethod
621     def tearDownClass(cls):
622         super(LDPThruHostStackIperf, cls).tearDownClass()
623
624     def setUp(self):
625         super(LDPThruHostStackIperf, self).setUp()
626
627         self.thru_host_stack_setup()
628         self.client_iperf3_timeout = 20
629         self.client_iperf3_args = ["-V4d", "-t 2", "-c", self.loop0.local_ip4]
630         self.server_iperf3_args = ["-V4d", "-s"]
631
632     def tearDown(self):
633         self.thru_host_stack_tear_down()
634         super(LDPThruHostStackIperf, self).tearDown()
635
636     def show_commands_at_teardown(self):
637         self.logger.debug(self.vapi.cli("show session verbose 2"))
638
639     def test_ldp_thru_host_stack_iperf3(self):
640         """ run LDP thru host stack iperf3 test """
641
642         try:
643             subprocess.check_output(['iperf3', '-v'])
644         except subprocess.CalledProcessError:
645             self.logger.error("WARNING: 'iperf3' is not installed,")
646             self.logger.error(
647                 "         'test_ldp_thru_host_stack_iperf3' not run!")
648             return
649         except OSError as e:
650             self.logger.error("WARNING: 'iperf3' is not installed,")
651             self.logger.error("         'test' not run!")
652             return
653         except Exception as e:
654             self.logger.error("WARNING: 'iperf3' unexpected error,")
655             self.logger.error("         'test' not run!")
656             return
657
658         self.timeout = self.client_iperf3_timeout
659         self.thru_host_stack_test("iperf3", self.server_iperf3_args,
660                                   "iperf3", self.client_iperf3_args)
661
662
663 class LDPIpv6CutThruTestCase(VCLTestCase):
664     """ LDP IPv6 Cut Thru Tests """
665
666     @classmethod
667     def setUpClass(cls):
668         super(LDPIpv6CutThruTestCase, cls).setUpClass()
669
670     @classmethod
671     def tearDownClass(cls):
672         super(LDPIpv6CutThruTestCase, cls).tearDownClass()
673
674     def setUp(self):
675         super(LDPIpv6CutThruTestCase, self).setUp()
676
677         self.cut_thru_setup()
678         self.client_iperf3_timeout = 20
679         self.client_uni_dir_nsock_timeout = 20
680         self.client_bi_dir_nsock_timeout = 20
681         self.client_ipv6_echo_test_args = ["-6", "-E", self.echo_phrase, "-X",
682                                            self.server_ipv6_addr,
683                                            self.server_port]
684         self.client_ipv6_iperf3_args = ["-V6d", "-t 2", "-c",
685                                         self.server_ipv6_addr]
686         self.server_ipv6_iperf3_args = ["-V6d", "-s"]
687         self.client_ipv6_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
688                                                     "-6",
689                                                     "-I", "2",
690                                                     self.server_ipv6_addr,
691                                                     self.server_port]
692         self.client_ipv6_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
693                                                    "-6",
694                                                    "-I", "2",
695                                                    self.server_ipv6_addr,
696                                                    self.server_port]
697
698     def tearDown(self):
699         self.cut_thru_tear_down()
700
701         super(LDPIpv6CutThruTestCase, self).tearDown()
702
703     def test_ldp_ipv6_cut_thru_echo(self):
704         """ run LDP IPv6 cut thru echo test """
705
706         self.cut_thru_test("sock_test_server",
707                            self.server_ipv6_args,
708                            "sock_test_client",
709                            self.client_ipv6_echo_test_args)
710
711     @unittest.skipUnless(running_extended_tests, "part of extended tests")
712     def test_ldp_ipv6_cut_thru_iperf3(self):
713         """ run LDP IPv6 cut thru iperf3 test """
714
715         try:
716             subprocess.check_output(['iperf3', '-v'])
717         except:
718             self.logger.error("WARNING: 'iperf3' is not installed,")
719             self.logger.error(
720                 "         'test_ldp_ipv6_cut_thru_iperf3' not run!")
721             return
722
723         self.timeout = self.client_iperf3_timeout
724         self.cut_thru_test("iperf3", self.server_ipv6_iperf3_args,
725                            "iperf3", self.client_ipv6_iperf3_args)
726
727     @unittest.skipUnless(running_extended_tests, "part of extended tests")
728     def test_ldp_ipv6_cut_thru_uni_dir_nsock(self):
729         """ run LDP IPv6 cut thru uni-directional (multiple sockets) test """
730
731         self.timeout = self.client_uni_dir_nsock_timeout
732         self.cut_thru_test("sock_test_server", self.server_ipv6_args,
733                            "sock_test_client",
734                            self.client_ipv6_uni_dir_nsock_test_args)
735
736     @unittest.skipUnless(running_extended_tests, "part of extended tests")
737     def test_ldp_ipv6_cut_thru_bi_dir_nsock(self):
738         """ run LDP IPv6 cut thru bi-directional (multiple sockets) test """
739
740         self.timeout = self.client_bi_dir_nsock_timeout
741         self.cut_thru_test("sock_test_server", self.server_ipv6_args,
742                            "sock_test_client",
743                            self.client_ipv6_bi_dir_nsock_test_args)
744
745
746 class VCLIpv6CutThruTestCase(VCLTestCase):
747     """ VCL IPv6 Cut Thru Tests """
748
749     @classmethod
750     def setUpClass(cls):
751         super(VCLIpv6CutThruTestCase, cls).setUpClass()
752
753     @classmethod
754     def tearDownClass(cls):
755         super(VCLIpv6CutThruTestCase, cls).tearDownClass()
756
757     def setUp(self):
758         super(VCLIpv6CutThruTestCase, self).setUp()
759
760         self.cut_thru_setup()
761         self.client_uni_dir_nsock_timeout = 20
762         self.client_bi_dir_nsock_timeout = 20
763         self.client_ipv6_echo_test_args = ["-6", "-E", self.echo_phrase, "-X",
764                                            self.server_ipv6_addr,
765                                            self.server_port]
766         self.client_ipv6_uni_dir_nsock_test_args = ["-N", "1000", "-U", "-X",
767                                                     "-6",
768                                                     "-I", "2",
769                                                     self.server_ipv6_addr,
770                                                     self.server_port]
771         self.client_ipv6_bi_dir_nsock_test_args = ["-N", "1000", "-B", "-X",
772                                                    "-6",
773                                                    "-I", "2",
774                                                    self.server_ipv6_addr,
775                                                    self.server_port]
776
777     def tearDown(self):
778         self.cut_thru_tear_down()
779
780         super(VCLIpv6CutThruTestCase, self).tearDown()
781
782     def test_vcl_ipv6_cut_thru_echo(self):
783         """ run VCL IPv6 cut thru echo test """
784
785         self.cut_thru_test("vcl_test_server",
786                            self.server_ipv6_args,
787                            "vcl_test_client",
788                            self.client_ipv6_echo_test_args)
789
790     @unittest.skipUnless(running_extended_tests, "part of extended tests")
791     def test_vcl_ipv6_cut_thru_uni_dir_nsock(self):
792         """ run VCL IPv6 cut thru uni-directional (multiple sockets) test """
793
794         self.timeout = self.client_uni_dir_nsock_timeout
795         self.cut_thru_test("vcl_test_server", self.server_ipv6_args,
796                            "vcl_test_client",
797                            self.client_ipv6_uni_dir_nsock_test_args)
798
799     @unittest.skipUnless(running_extended_tests, "part of extended tests")
800     def test_vcl_ipv6_cut_thru_bi_dir_nsock(self):
801         """ run VCL IPv6 cut thru bi-directional (multiple sockets) test """
802
803         self.timeout = self.client_bi_dir_nsock_timeout
804         self.cut_thru_test("vcl_test_server", self.server_ipv6_args,
805                            "vcl_test_client",
806                            self.client_ipv6_bi_dir_nsock_test_args)
807
808
809 class VCLIpv6ThruHostStackEcho(VCLTestCase):
810     """ VCL IPv6 Thru Host Stack Echo """
811
812     @classmethod
813     def setUpClass(cls):
814         super(VCLIpv6ThruHostStackEcho, cls).setUpClass()
815
816     @classmethod
817     def tearDownClass(cls):
818         super(VCLIpv6ThruHostStackEcho, cls).tearDownClass()
819
820     def setUp(self):
821         super(VCLIpv6ThruHostStackEcho, self).setUp()
822
823         self.thru_host_stack_ipv6_setup()
824         self.client_ipv6_echo_test_args = ["-6", "-E", self.echo_phrase, "-X",
825                                            self.loop0.local_ip6,
826                                            self.server_port]
827
828     def tearDown(self):
829         self.thru_host_stack_ipv6_tear_down()
830         super(VCLIpv6ThruHostStackEcho, self).tearDown()
831
832     def test_vcl_ipv6_thru_host_stack_echo(self):
833         """ run VCL IPv6 thru host stack echo test """
834
835         self.thru_host_stack_test("vcl_test_server",
836                                   self.server_ipv6_args,
837                                   "vcl_test_client",
838                                   self.client_ipv6_echo_test_args)
839
840
841 if __name__ == '__main__':
842     unittest.main(testRunner=VppTestRunner)