blob: 2b197326532f3172d879ba68e2602d20d2ceba1e [file] [log] [blame]
Renato Botelho do Coutoead1e532019-10-31 13:31:07 -05001#!/usr/bin/env python3
Damjan Marionf56b77a2016-10-03 19:44:57 +02002
Klement Sekeraacb9b8e2017-02-14 02:55:31 +01003from __future__ import print_function
Paul Vinciguerra9beabd82019-12-01 22:24:28 -05004import logging
Paul Vinciguerra72f00042018-11-25 11:05:13 -08005import sys
Ole Trøan162989e2018-11-26 10:27:50 +00006import os
7import select
Paul Vinciguerra496b0de2019-06-20 12:24:12 -04008import signal
Paul Vinciguerrad6f22172020-12-05 22:39:14 +00009import subprocess
Ole Trøan162989e2018-11-26 10:27:50 +000010import unittest
Klement Sekerab23ffd72021-05-31 16:08:53 +020011import re
Klement Sekera277b89c2016-10-28 13:20:27 +020012import time
Paul Vinciguerra72f00042018-11-25 11:05:13 -080013import faulthandler
Ole Trøan162989e2018-11-26 10:27:50 +000014import random
15import copy
juraj.linkes68ebc832018-11-29 09:37:08 +010016import platform
Klement Sekerab23ffd72021-05-31 16:08:53 +020017import shutil
Ole Trøan162989e2018-11-26 10:27:50 +000018from collections import deque
19from threading import Thread, Event
20from inspect import getdoc, isclass
21from traceback import format_exception
22from logging import FileHandler, DEBUG, Formatter
Andrew Yourtchenko06f32812021-01-14 10:19:08 +000023from enum import Enum
Klement Sekera558ceab2021-04-08 19:37:41 +020024from abc import ABC, abstractmethod
Ray Kinsellab8165b92021-09-22 11:24:06 +010025from struct import pack, unpack
Paul Vinciguerraa7427ec2019-03-10 10:04:23 -070026
27import scapy.compat
Klement Sekera56c492a2022-01-10 21:57:27 +000028from scapy.packet import Raw, Packet
Klement Sekerab23ffd72021-05-31 16:08:53 +020029from config import config, available_cpus, num_cpus, max_vpp_cpus
Paul Vinciguerra496b0de2019-06-20 12:24:12 -040030import hook as hookmodule
Paul Vinciguerra919efad2018-12-17 21:43:43 -080031from vpp_pg_interface import VppPGInterface
Ole Troana45dc072018-12-21 16:04:22 +010032from vpp_sub_interface import VppSubInterface
Ole Trøan162989e2018-11-26 10:27:50 +000033from vpp_lo_interface import VppLoInterface
Neale Ranns192b13f2019-03-15 02:16:20 -070034from vpp_bvi_interface import VppBviInterface
Ole Trøan162989e2018-11-26 10:27:50 +000035from vpp_papi_provider import VppPapiProvider
Neale Ranns6197cb72021-06-03 14:43:21 +000036from vpp_papi import VppEnum
Paul Vinciguerra1043fd32019-12-02 21:42:28 -050037import vpp_papi
Ole Trøan162989e2018-11-26 10:27:50 +000038from vpp_papi.vpp_stats import VPPStats
Ole Troan4376ab22021-03-03 10:40:05 +010039from vpp_papi.vpp_transport_socket import VppTransportSocketIOError
Klement Sekerad9b0c6f2022-04-26 19:02:15 +020040from log import (
41 RED,
42 GREEN,
43 YELLOW,
44 double_line_delim,
45 single_line_delim,
46 get_logger,
47 colorize,
48)
Ole Trøan162989e2018-11-26 10:27:50 +000049from vpp_object import VppObjectRegistry
50from util import ppp, is_core_present
Klement Sekerad81ae412018-05-16 10:52:54 +020051from scapy.layers.inet import IPerror, TCPerror, UDPerror, ICMPerror
52from scapy.layers.inet6 import ICMPv6DestUnreach, ICMPv6EchoRequest
53from scapy.layers.inet6 import ICMPv6EchoReply
Naveen Joyc872cec2022-08-30 13:59:03 -070054from vpp_running import use_running
Paul Vinciguerra6919b0d2018-12-09 15:37:04 -080055
Klement Sekera558ceab2021-04-08 19:37:41 +020056
Paul Vinciguerra9beabd82019-12-01 22:24:28 -050057logger = logging.getLogger(__name__)
58
59# Set up an empty logger for the testcase that can be overridden as necessary
Klement Sekerad9b0c6f2022-04-26 19:02:15 +020060null_logger = logging.getLogger("VppTestCase")
Paul Vinciguerra9beabd82019-12-01 22:24:28 -050061null_logger.addHandler(logging.NullHandler())
62
juraj.linkescae64f82018-09-19 15:01:47 +020063PASS = 0
64FAIL = 1
65ERROR = 2
66SKIP = 3
67TEST_RUN = 4
Klement Sekera558ceab2021-04-08 19:37:41 +020068SKIP_CPU_SHORTAGE = 5
juraj.linkescae64f82018-09-19 15:01:47 +020069
Paul Vinciguerra0cbc71d2019-07-03 08:38:38 -040070
Klement Sekerab23ffd72021-05-31 16:08:53 +020071if config.debug_framework:
Klement Sekeraebbaf552018-02-17 13:41:33 +010072 import debug_internal
73
Klement Sekeraf62ae122016-10-11 11:47:09 +020074"""
75 Test framework module.
76
77 The module provides a set of tools for constructing and running tests and
78 representing the results.
79"""
80
Klement Sekeraf62ae122016-10-11 11:47:09 +020081
Paul Vinciguerra496b0de2019-06-20 12:24:12 -040082class VppDiedError(Exception):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +020083 """exception for reporting that the subprocess has died."""
Paul Vinciguerra496b0de2019-06-20 12:24:12 -040084
Klement Sekerad9b0c6f2022-04-26 19:02:15 +020085 signals_by_value = {
86 v: k
87 for k, v in signal.__dict__.items()
88 if k.startswith("SIG") and not k.startswith("SIG_")
89 }
Paul Vinciguerra496b0de2019-06-20 12:24:12 -040090
Paul Vinciguerra5dd6a7b2019-06-19 10:29:24 -040091 def __init__(self, rv=None, testcase=None, method_name=None):
Paul Vinciguerra496b0de2019-06-20 12:24:12 -040092 self.rv = rv
93 self.signal_name = None
Paul Vinciguerra5dd6a7b2019-06-19 10:29:24 -040094 self.testcase = testcase
95 self.method_name = method_name
96
Paul Vinciguerra496b0de2019-06-20 12:24:12 -040097 try:
98 self.signal_name = VppDiedError.signals_by_value[-rv]
Paul Vinciguerrafea82602019-06-26 20:45:08 -040099 except (KeyError, TypeError):
Paul Vinciguerra496b0de2019-06-20 12:24:12 -0400100 pass
101
Paul Vinciguerra5dd6a7b2019-06-19 10:29:24 -0400102 if testcase is None and method_name is None:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200103 in_msg = ""
Paul Vinciguerra5dd6a7b2019-06-19 10:29:24 -0400104 else:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200105 in_msg = " while running %s.%s" % (testcase, method_name)
Paul Vinciguerra5dd6a7b2019-06-19 10:29:24 -0400106
Klement Sekera79a31db2021-03-12 18:16:10 +0100107 if self.rv:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200108 msg = "VPP subprocess died unexpectedly%s with return code: %d%s." % (
109 in_msg,
110 self.rv,
111 " [%s]" % (self.signal_name if self.signal_name is not None else ""),
112 )
Klement Sekera79a31db2021-03-12 18:16:10 +0100113 else:
114 msg = "VPP subprocess died unexpectedly%s." % in_msg
115
Paul Vinciguerra496b0de2019-06-20 12:24:12 -0400116 super(VppDiedError, self).__init__(msg)
117
118
Damjan Marionf56b77a2016-10-03 19:44:57 +0200119class _PacketInfo(object):
Klement Sekeraf62ae122016-10-11 11:47:09 +0200120 """Private class to create packet info object.
121
122 Help process information about the next packet.
123 Set variables to default values.
Klement Sekeraf62ae122016-10-11 11:47:09 +0200124 """
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200125
Matej Klotton86d87c42016-11-11 11:38:55 +0100126 #: Store the index of the packet.
Damjan Marionf56b77a2016-10-03 19:44:57 +0200127 index = -1
Matej Klotton86d87c42016-11-11 11:38:55 +0100128 #: Store the index of the source packet generator interface of the packet.
Damjan Marionf56b77a2016-10-03 19:44:57 +0200129 src = -1
Matej Klotton86d87c42016-11-11 11:38:55 +0100130 #: Store the index of the destination packet generator interface
131 #: of the packet.
Damjan Marionf56b77a2016-10-03 19:44:57 +0200132 dst = -1
Pavel Kotucek59dda062017-03-02 15:22:47 +0100133 #: Store expected ip version
134 ip = -1
135 #: Store expected upper protocol
136 proto = -1
Matej Klotton86d87c42016-11-11 11:38:55 +0100137 #: Store the copy of the former packet.
Damjan Marionf56b77a2016-10-03 19:44:57 +0200138 data = None
Damjan Marionf56b77a2016-10-03 19:44:57 +0200139
Matej Klotton16a14cd2016-12-07 15:09:13 +0100140 def __eq__(self, other):
141 index = self.index == other.index
142 src = self.src == other.src
143 dst = self.dst == other.dst
144 data = self.data == other.data
145 return index and src and dst and data
146
Klement Sekeraf62ae122016-10-11 11:47:09 +0200147
Klement Sekeraacb9b8e2017-02-14 02:55:31 +0100148def pump_output(testclass):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200149 """pump output from vpp stdout/stderr to proper queues"""
Dave Wallace670724c2022-09-20 21:52:18 -0400150 if not hasattr(testclass, "vpp"):
151 return
Klement Sekera6a6f4f72017-11-09 09:16:39 +0100152 stdout_fragment = ""
153 stderr_fragment = ""
Neale Ranns16782362018-07-23 05:35:56 -0400154 while not testclass.pump_thread_stop_flag.is_set():
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200155 readable = select.select(
156 [
157 testclass.vpp.stdout.fileno(),
158 testclass.vpp.stderr.fileno(),
159 testclass.pump_thread_wakeup_pipe[0],
160 ],
161 [],
162 [],
163 )[0]
Klement Sekeraacb9b8e2017-02-14 02:55:31 +0100164 if testclass.vpp.stdout.fileno() in readable:
Klement Sekera6a6f4f72017-11-09 09:16:39 +0100165 read = os.read(testclass.vpp.stdout.fileno(), 102400)
166 if len(read) > 0:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200167 split = read.decode("ascii", errors="backslashreplace").splitlines(True)
Klement Sekera6a6f4f72017-11-09 09:16:39 +0100168 if len(stdout_fragment) > 0:
169 split[0] = "%s%s" % (stdout_fragment, split[0])
170 if len(split) > 0 and split[-1].endswith("\n"):
171 limit = None
172 else:
173 limit = -1
174 stdout_fragment = split[-1]
175 testclass.vpp_stdout_deque.extend(split[:limit])
Klement Sekerab23ffd72021-05-31 16:08:53 +0200176 if not config.cache_vpp_output:
Klement Sekera6a6f4f72017-11-09 09:16:39 +0100177 for line in split[:limit]:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200178 testclass.logger.info("VPP STDOUT: %s" % line.rstrip("\n"))
Klement Sekeraacb9b8e2017-02-14 02:55:31 +0100179 if testclass.vpp.stderr.fileno() in readable:
Klement Sekera6a6f4f72017-11-09 09:16:39 +0100180 read = os.read(testclass.vpp.stderr.fileno(), 102400)
181 if len(read) > 0:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200182 split = read.decode("ascii", errors="backslashreplace").splitlines(True)
Klement Sekera6a6f4f72017-11-09 09:16:39 +0100183 if len(stderr_fragment) > 0:
184 split[0] = "%s%s" % (stderr_fragment, split[0])
Ole Troan6ed154f2019-10-15 19:31:55 +0200185 if len(split) > 0 and split[-1].endswith("\n"):
Klement Sekera6a6f4f72017-11-09 09:16:39 +0100186 limit = None
187 else:
188 limit = -1
189 stderr_fragment = split[-1]
Ole Troan6ed154f2019-10-15 19:31:55 +0200190
Klement Sekera6a6f4f72017-11-09 09:16:39 +0100191 testclass.vpp_stderr_deque.extend(split[:limit])
Klement Sekerab23ffd72021-05-31 16:08:53 +0200192 if not config.cache_vpp_output:
Klement Sekera6a6f4f72017-11-09 09:16:39 +0100193 for line in split[:limit]:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200194 testclass.logger.error("VPP STDERR: %s" % line.rstrip("\n"))
Paul Vinciguerra6919b0d2018-12-09 15:37:04 -0800195 # ignoring the dummy pipe here intentionally - the
196 # flag will take care of properly terminating the loop
Klement Sekera277b89c2016-10-28 13:20:27 +0200197
198
Paul Vinciguerradefde0f2018-12-06 07:46:13 -0800199def _is_platform_aarch64():
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200200 return platform.machine() == "aarch64"
juraj.linkes68ebc832018-11-29 09:37:08 +0100201
Klement Sekera6aa58b72019-05-16 14:34:55 +0200202
Paul Vinciguerradefde0f2018-12-06 07:46:13 -0800203is_platform_aarch64 = _is_platform_aarch64()
juraj.linkes68ebc832018-11-29 09:37:08 +0100204
Paul Vinciguerradefde0f2018-12-06 07:46:13 -0800205
Dave Wallacee95b2462022-09-18 22:28:44 -0400206def _is_distro_ubuntu2204():
207 with open("/etc/os-release") as f:
208 for line in f.readlines():
209 if "jammy" in line:
210 return True
211 return False
212
213
214is_distro_ubuntu2204 = _is_distro_ubuntu2204()
215
216
Dave Wallace670724c2022-09-20 21:52:18 -0400217def _is_distro_debian11():
218 with open("/etc/os-release") as f:
219 for line in f.readlines():
220 if "bullseye" in line:
221 return True
222 return False
223
224
225is_distro_debian11 = _is_distro_debian11()
226
227
Klement Sekera909a6a12017-08-08 04:33:53 +0200228class KeepAliveReporter(object):
229 """
230 Singleton object which reports test start to parent process
231 """
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200232
Klement Sekera909a6a12017-08-08 04:33:53 +0200233 _shared_state = {}
234
235 def __init__(self):
236 self.__dict__ = self._shared_state
Paul Vinciguerrac7b03fe2018-11-18 08:17:34 -0800237 self._pipe = None
Klement Sekera909a6a12017-08-08 04:33:53 +0200238
239 @property
240 def pipe(self):
241 return self._pipe
242
243 @pipe.setter
244 def pipe(self, pipe):
Paul Vinciguerrac7b03fe2018-11-18 08:17:34 -0800245 if self._pipe is not None:
Klement Sekera909a6a12017-08-08 04:33:53 +0200246 raise Exception("Internal error - pipe should only be set once.")
247 self._pipe = pipe
248
juraj.linkes40dd73b2018-09-21 13:55:16 +0200249 def send_keep_alive(self, test, desc=None):
Klement Sekera909a6a12017-08-08 04:33:53 +0200250 """
251 Write current test tmpdir & desc to keep-alive pipe to signal liveness
252 """
Dave Wallace670724c2022-09-20 21:52:18 -0400253 if not hasattr(test, "vpp") or self.pipe is None:
Klement Sekera3f6ff192017-08-11 06:56:05 +0200254 # if not running forked..
255 return
256
Klement Sekera909a6a12017-08-08 04:33:53 +0200257 if isclass(test):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200258 desc = "%s (%s)" % (desc, unittest.util.strclass(test))
Klement Sekera909a6a12017-08-08 04:33:53 +0200259 else:
juraj.linkes40dd73b2018-09-21 13:55:16 +0200260 desc = test.id()
Klement Sekera909a6a12017-08-08 04:33:53 +0200261
Klement Sekerab23ffd72021-05-31 16:08:53 +0200262 self.pipe.send((desc, config.vpp, test.tempdir, test.vpp.pid))
Klement Sekera909a6a12017-08-08 04:33:53 +0200263
264
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000265class TestCaseTag(Enum):
Andrew Yourtchenkoc8eae8d2021-01-20 20:30:36 +0000266 # marks the suites that must run at the end
267 # using only a single test runner
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000268 RUN_SOLO = 1
Andrew Yourtchenkoc8eae8d2021-01-20 20:30:36 +0000269 # marks the suites broken on VPP multi-worker
270 FIXME_VPP_WORKERS = 2
Naveen Joy6eaeea92021-09-09 17:57:02 -0700271 # marks the suites broken when ASan is enabled
272 FIXME_ASAN = 3
Dave Wallacee95b2462022-09-18 22:28:44 -0400273 # marks suites broken on Ubuntu-22.04
274 FIXME_UBUNTU2204 = 4
Dave Wallace670724c2022-09-20 21:52:18 -0400275 # marks suites broken on Debian-11
276 FIXME_DEBIAN11 = 5
Dave Wallace8a0a9d22022-10-04 22:02:49 -0400277 # marks suites broken on debug vpp image
278 FIXME_VPP_DEBUG = 6
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000279
280
281def create_tag_decorator(e):
282 def decorator(cls):
283 try:
284 cls.test_tags.append(e)
285 except AttributeError:
286 cls.test_tags = [e]
287 return cls
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200288
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000289 return decorator
290
Andrew Yourtchenkoc8eae8d2021-01-20 20:30:36 +0000291
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000292tag_run_solo = create_tag_decorator(TestCaseTag.RUN_SOLO)
Andrew Yourtchenkoc8eae8d2021-01-20 20:30:36 +0000293tag_fixme_vpp_workers = create_tag_decorator(TestCaseTag.FIXME_VPP_WORKERS)
Naveen Joy6eaeea92021-09-09 17:57:02 -0700294tag_fixme_asan = create_tag_decorator(TestCaseTag.FIXME_ASAN)
Dave Wallacee95b2462022-09-18 22:28:44 -0400295tag_fixme_ubuntu2204 = create_tag_decorator(TestCaseTag.FIXME_UBUNTU2204)
Dave Wallace670724c2022-09-20 21:52:18 -0400296tag_fixme_debian11 = create_tag_decorator(TestCaseTag.FIXME_DEBIAN11)
Dave Wallace8a0a9d22022-10-04 22:02:49 -0400297tag_fixme_vpp_debug = create_tag_decorator(TestCaseTag.FIXME_VPP_DEBUG)
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000298
299
Klement Sekerae2636852021-03-16 12:52:12 +0100300class DummyVpp:
301 returncode = None
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200302 pid = 0xCAFEBAFE
Klement Sekerae2636852021-03-16 12:52:12 +0100303
304 def poll(self):
305 pass
306
307 def terminate(self):
308 pass
309
310
Klement Sekera558ceab2021-04-08 19:37:41 +0200311class CPUInterface(ABC):
312 cpus = []
313 skipped_due_to_cpu_lack = False
314
315 @classmethod
316 @abstractmethod
317 def get_cpus_required(cls):
318 pass
319
320 @classmethod
321 def assign_cpus(cls, cpus):
322 cls.cpus = cpus
323
324
Naveen Joyc872cec2022-08-30 13:59:03 -0700325@use_running
Klement Sekera558ceab2021-04-08 19:37:41 +0200326class VppTestCase(CPUInterface, unittest.TestCase):
Matej Klotton86d87c42016-11-11 11:38:55 +0100327 """This subclass is a base class for VPP test cases that are implemented as
328 classes. It provides methods to create and run test case.
Klement Sekeraf62ae122016-10-11 11:47:09 +0200329 """
330
Arthur de Kerhordb023802021-03-11 10:26:54 -0800331 extra_vpp_statseg_config = ""
Ole Troana45dc072018-12-21 16:04:22 +0100332 extra_vpp_punt_config = []
333 extra_vpp_plugin_config = []
Paul Vinciguerra9beabd82019-12-01 22:24:28 -0500334 logger = null_logger
Paul Vinciguerrabfd7d292019-10-26 22:25:49 -0400335 vapi_response_timeout = 5
Klement Sekera140af152022-02-18 10:30:51 +0000336 remove_configured_vpp_objects_on_tear_down = True
Pavel Kotuceke88865d2018-11-28 07:42:11 +0100337
Klement Sekeraf62ae122016-10-11 11:47:09 +0200338 @property
339 def packet_infos(self):
340 """List of packet infos"""
341 return self._packet_infos
342
Klement Sekeradab231a2016-12-21 08:50:14 +0100343 @classmethod
344 def get_packet_count_for_if_idx(cls, dst_if_index):
345 """Get the number of packet info for specified destination if index"""
346 if dst_if_index in cls._packet_count_for_dst_if_idx:
347 return cls._packet_count_for_dst_if_idx[dst_if_index]
348 else:
349 return 0
Klement Sekeraf62ae122016-10-11 11:47:09 +0200350
351 @classmethod
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000352 def has_tag(cls, tag):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200353 """if the test case has a given tag - return true"""
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000354 try:
355 return tag in cls.test_tags
356 except AttributeError:
357 pass
Andrew Yourtchenkoa3b7c552020-08-26 14:33:54 +0000358 return False
359
360 @classmethod
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000361 def is_tagged_run_solo(cls):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200362 """if the test case class is timing-sensitive - return true"""
Andrew Yourtchenko06f32812021-01-14 10:19:08 +0000363 return cls.has_tag(TestCaseTag.RUN_SOLO)
364
365 @classmethod
Naveen Joy6eaeea92021-09-09 17:57:02 -0700366 def skip_fixme_asan(cls):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200367 """if @tag_fixme_asan & ASan is enabled - mark for skip"""
Naveen Joy6eaeea92021-09-09 17:57:02 -0700368 if cls.has_tag(TestCaseTag.FIXME_ASAN):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200369 vpp_extra_cmake_args = os.environ.get("VPP_EXTRA_CMAKE_ARGS", "")
370 if "DVPP_ENABLE_SANITIZE_ADDR=ON" in vpp_extra_cmake_args:
Naveen Joy6eaeea92021-09-09 17:57:02 -0700371 cls = unittest.skip("Skipping @tag_fixme_asan tests")(cls)
372
373 @classmethod
Dave Wallacee95b2462022-09-18 22:28:44 -0400374 def skip_fixme_ubuntu2204(cls):
375 """if distro is ubuntu 22.04 and @tag_fixme_ubuntu2204 mark for skip"""
376 if cls.has_tag(TestCaseTag.FIXME_UBUNTU2204):
377 cls = unittest.skip("Skipping @tag_fixme_ubuntu2204 tests")(cls)
378
379 @classmethod
Dave Wallace670724c2022-09-20 21:52:18 -0400380 def skip_fixme_debian11(cls):
381 """if distro is Debian-11 and @tag_fixme_debian11 mark for skip"""
382 if cls.has_tag(TestCaseTag.FIXME_DEBIAN11):
383 cls = unittest.skip("Skipping @tag_fixme_debian11 tests")(cls)
384
385 @classmethod
Dave Wallace8a0a9d22022-10-04 22:02:49 -0400386 def skip_fixme_vpp_debug(cls):
387 cls = unittest.skip("Skipping @tag_fixme_vpp_debug tests")(cls)
388
389 @classmethod
Klement Sekeraf62ae122016-10-11 11:47:09 +0200390 def instance(cls):
391 """Return the instance of this testcase"""
392 return cls.test_instance
393
Damjan Marionf56b77a2016-10-03 19:44:57 +0200394 @classmethod
Klement Sekera277b89c2016-10-28 13:20:27 +0200395 def set_debug_flags(cls, d):
Dave Wallace24564332019-10-21 02:53:14 +0000396 cls.gdbserver_port = 7777
Klement Sekera277b89c2016-10-28 13:20:27 +0200397 cls.debug_core = False
398 cls.debug_gdb = False
399 cls.debug_gdbserver = False
Dave Wallace24564332019-10-21 02:53:14 +0000400 cls.debug_all = False
Klement Sekerae2636852021-03-16 12:52:12 +0100401 cls.debug_attach = False
Klement Sekera277b89c2016-10-28 13:20:27 +0200402 if d is None:
403 return
404 dl = d.lower()
405 if dl == "core":
Klement Sekera277b89c2016-10-28 13:20:27 +0200406 cls.debug_core = True
Dave Wallace24564332019-10-21 02:53:14 +0000407 elif dl == "gdb" or dl == "gdb-all":
Klement Sekera277b89c2016-10-28 13:20:27 +0200408 cls.debug_gdb = True
Dave Wallace24564332019-10-21 02:53:14 +0000409 elif dl == "gdbserver" or dl == "gdbserver-all":
Klement Sekera277b89c2016-10-28 13:20:27 +0200410 cls.debug_gdbserver = True
Klement Sekerae2636852021-03-16 12:52:12 +0100411 elif dl == "attach":
412 cls.debug_attach = True
Klement Sekera277b89c2016-10-28 13:20:27 +0200413 else:
414 raise Exception("Unrecognized DEBUG option: '%s'" % d)
Dave Wallace24564332019-10-21 02:53:14 +0000415 if dl == "gdb-all" or dl == "gdbserver-all":
416 cls.debug_all = True
Klement Sekera277b89c2016-10-28 13:20:27 +0200417
Klement Sekera558ceab2021-04-08 19:37:41 +0200418 @classmethod
419 def get_vpp_worker_count(cls):
420 if not hasattr(cls, "vpp_worker_count"):
421 if cls.has_tag(TestCaseTag.FIXME_VPP_WORKERS):
422 cls.vpp_worker_count = 0
423 else:
Klement Sekerab23ffd72021-05-31 16:08:53 +0200424 cls.vpp_worker_count = config.vpp_worker_count
Klement Sekera558ceab2021-04-08 19:37:41 +0200425 return cls.vpp_worker_count
juraj.linkes184870a2018-07-16 14:22:01 +0200426
Klement Sekera558ceab2021-04-08 19:37:41 +0200427 @classmethod
428 def get_cpus_required(cls):
429 return 1 + cls.get_vpp_worker_count()
juraj.linkes184870a2018-07-16 14:22:01 +0200430
Paul Vinciguerra86ebba62018-11-21 09:28:32 -0800431 @classmethod
Damjan Marionf56b77a2016-10-03 19:44:57 +0200432 def setUpConstants(cls):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200433 """Set-up the test case class based on environment variables"""
Klement Sekerab23ffd72021-05-31 16:08:53 +0200434 cls.step = config.step
435 cls.plugin_path = ":".join(config.vpp_plugin_dir)
436 cls.test_plugin_path = ":".join(config.vpp_test_plugin_dir)
437 cls.extern_plugin_path = ":".join(config.extern_plugin_dir)
Ole Troana45dc072018-12-21 16:04:22 +0100438 debug_cli = ""
Klement Sekera01bbbe92016-11-02 09:25:05 +0100439 if cls.step or cls.debug_gdb or cls.debug_gdbserver:
Ole Troana45dc072018-12-21 16:04:22 +0100440 debug_cli = "cli-listen localhost:5002"
Klement Sekerab23ffd72021-05-31 16:08:53 +0200441 size = re.search(r"\d+[gG]", config.coredump_size)
442 if size:
443 coredump_size = f"coredump-size {config.coredump_size}".lower()
444 else:
Ole Troana45dc072018-12-21 16:04:22 +0100445 coredump_size = "coredump-size unlimited"
Klement Sekerab23ffd72021-05-31 16:08:53 +0200446 default_variant = config.variant
Ray Kinsella4830e4f2020-03-10 14:35:32 +0000447 if default_variant is not None:
Benoît Ganne09ef5922022-07-29 10:52:34 +0200448 default_variant = "default { variant %s 100 }" % default_variant
Ray Kinsella4830e4f2020-03-10 14:35:32 +0000449 else:
450 default_variant = ""
451
Klement Sekerab23ffd72021-05-31 16:08:53 +0200452 api_fuzzing = config.api_fuzz
Dave Barach77841402020-04-29 17:04:10 -0400453 if api_fuzzing is None:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200454 api_fuzzing = "off"
Dave Barach77841402020-04-29 17:04:10 -0400455
Klement Sekera8d815022021-03-15 16:58:10 +0100456 cls.vpp_cmdline = [
Klement Sekerab23ffd72021-05-31 16:08:53 +0200457 config.vpp,
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200458 "unix",
459 "{",
460 "nodaemon",
461 debug_cli,
462 "full-coredump",
463 coredump_size,
464 "runtime-dir",
465 cls.tempdir,
Klement Sekera8d815022021-03-15 16:58:10 +0100466 "}",
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200467 "api-trace",
468 "{",
469 "on",
470 "}",
471 "api-segment",
472 "{",
473 "prefix",
474 cls.get_api_segment_prefix(),
475 "}",
476 "cpu",
477 "{",
478 "main-core",
479 str(cls.cpus[0]),
480 ]
481 if cls.extern_plugin_path not in (None, ""):
482 cls.extra_vpp_plugin_config.append("add-path %s" % cls.extern_plugin_path)
483 if cls.get_vpp_worker_count():
484 cls.vpp_cmdline.extend(
485 ["corelist-workers", ",".join([str(x) for x in cls.cpus[1:]])]
486 )
487 cls.vpp_cmdline.extend(
488 [
489 "}",
490 "physmem",
491 "{",
492 "max-size",
493 "32m",
494 "}",
495 "statseg",
496 "{",
497 "socket-name",
498 cls.get_stats_sock_path(),
499 cls.extra_vpp_statseg_config,
500 "}",
501 "socksvr",
502 "{",
503 "socket-name",
504 cls.get_api_sock_path(),
505 "}",
506 "node { ",
507 default_variant,
508 "}",
509 "api-fuzz {",
510 api_fuzzing,
511 "}",
512 "plugins",
513 "{",
514 "plugin",
515 "dpdk_plugin.so",
516 "{",
517 "disable",
518 "}",
519 "plugin",
520 "rdma_plugin.so",
521 "{",
522 "disable",
523 "}",
524 "plugin",
525 "lisp_unittest_plugin.so",
526 "{",
527 "enable",
528 "}",
529 "plugin",
530 "unittest_plugin.so",
531 "{",
532 "enable",
533 "}",
534 ]
535 + cls.extra_vpp_plugin_config
536 + [
537 "}",
538 ]
539 )
Ray Kinsella4830e4f2020-03-10 14:35:32 +0000540
Ole Troana45dc072018-12-21 16:04:22 +0100541 if cls.extra_vpp_punt_config is not None:
542 cls.vpp_cmdline.extend(cls.extra_vpp_punt_config)
Dave Barach7d31ab22019-05-08 19:18:18 -0400543
Klement Sekerae2636852021-03-16 12:52:12 +0100544 if not cls.debug_attach:
545 cls.logger.info("vpp_cmdline args: %s" % cls.vpp_cmdline)
546 cls.logger.info("vpp_cmdline: %s" % " ".join(cls.vpp_cmdline))
Klement Sekera277b89c2016-10-28 13:20:27 +0200547
548 @classmethod
549 def wait_for_enter(cls):
550 if cls.debug_gdbserver:
551 print(double_line_delim)
552 print("Spawned GDB server with PID: %d" % cls.vpp.pid)
553 elif cls.debug_gdb:
554 print(double_line_delim)
555 print("Spawned VPP with PID: %d" % cls.vpp.pid)
556 else:
557 cls.logger.debug("Spawned VPP with PID: %d" % cls.vpp.pid)
558 return
559 print(single_line_delim)
Dave Wallace24564332019-10-21 02:53:14 +0000560 print("You can debug VPP using:")
Klement Sekera277b89c2016-10-28 13:20:27 +0200561 if cls.debug_gdbserver:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200562 print(
563 f"sudo gdb {config.vpp} "
564 f"-ex 'target remote localhost:{cls.gdbserver_port}'"
565 )
566 print(
567 "Now is the time to attach gdb by running the above "
568 "command, set up breakpoints etc., then resume VPP from "
569 "within gdb by issuing the 'continue' command"
570 )
Dave Wallace24564332019-10-21 02:53:14 +0000571 cls.gdbserver_port += 1
Klement Sekera277b89c2016-10-28 13:20:27 +0200572 elif cls.debug_gdb:
Klement Sekerab23ffd72021-05-31 16:08:53 +0200573 print(f"sudo gdb {config.vpp} -ex 'attach {cls.vpp.pid}'")
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200574 print(
575 "Now is the time to attach gdb by running the above "
576 "command and set up breakpoints etc., then resume VPP from"
577 " within gdb by issuing the 'continue' command"
578 )
Klement Sekera277b89c2016-10-28 13:20:27 +0200579 print(single_line_delim)
Paul Vinciguerra852f5ef2018-12-15 10:16:35 -0800580 input("Press ENTER to continue running the testcase...")
Klement Sekera277b89c2016-10-28 13:20:27 +0200581
582 @classmethod
Klement Sekerae2636852021-03-16 12:52:12 +0100583 def attach_vpp(cls):
584 cls.vpp = DummyVpp()
585
586 @classmethod
Klement Sekera277b89c2016-10-28 13:20:27 +0200587 def run_vpp(cls):
Dave Wallace670724c2022-09-20 21:52:18 -0400588 if (
589 is_distro_ubuntu2204 == True and cls.has_tag(TestCaseTag.FIXME_UBUNTU2204)
590 ) or (is_distro_debian11 == True and cls.has_tag(TestCaseTag.FIXME_DEBIAN11)):
591 return
Klement Sekera558ceab2021-04-08 19:37:41 +0200592 cls.logger.debug(f"Assigned cpus: {cls.cpus}")
Klement Sekera277b89c2016-10-28 13:20:27 +0200593 cmdline = cls.vpp_cmdline
594
595 if cls.debug_gdbserver:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200596 gdbserver = "/usr/bin/gdbserver"
597 if not os.path.isfile(gdbserver) or not os.access(gdbserver, os.X_OK):
598 raise Exception(
599 "gdbserver binary '%s' does not exist or is "
600 "not executable" % gdbserver
601 )
Klement Sekera931be3a2016-11-03 05:36:01 +0100602
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200603 cmdline = [
604 gdbserver,
605 "localhost:{port}".format(port=cls.gdbserver_port),
606 ] + cls.vpp_cmdline
Klement Sekera277b89c2016-10-28 13:20:27 +0200607 cls.logger.info("Gdbserver cmdline is %s", " ".join(cmdline))
608
Klement Sekera931be3a2016-11-03 05:36:01 +0100609 try:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200610 cls.vpp = subprocess.Popen(
611 cmdline, stdout=subprocess.PIPE, stderr=subprocess.PIPE
612 )
Paul Vinciguerra61e63bf2018-11-24 21:19:38 -0800613 except subprocess.CalledProcessError as e:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200614 cls.logger.critical(
615 "Subprocess returned with non-0 return code: (%s)", e.returncode
616 )
Paul Vinciguerra38a4ec72018-11-28 11:34:21 -0800617 raise
618 except OSError as e:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200619 cls.logger.critical(
620 "Subprocess returned with OS error: (%s) %s", e.errno, e.strerror
621 )
Paul Vinciguerra38a4ec72018-11-28 11:34:21 -0800622 raise
623 except Exception as e:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200624 cls.logger.exception("Subprocess returned unexpected from %s:", cmdline)
Klement Sekera931be3a2016-11-03 05:36:01 +0100625 raise
626
Klement Sekera277b89c2016-10-28 13:20:27 +0200627 cls.wait_for_enter()
Pierre Pfistercd8e3182016-10-07 16:30:03 +0100628
Damjan Marionf56b77a2016-10-03 19:44:57 +0200629 @classmethod
Andrew Yourtchenkob31d3932019-10-31 08:34:22 +0000630 def wait_for_coredump(cls):
631 corefile = cls.tempdir + "/core"
Dave Wallace3e9b7a22019-04-09 20:19:51 -0400632 if os.path.isfile(corefile):
Andrew Yourtchenkob31d3932019-10-31 08:34:22 +0000633 cls.logger.error("Waiting for coredump to complete: %s", corefile)
Dave Wallace3e9b7a22019-04-09 20:19:51 -0400634 curr_size = os.path.getsize(corefile)
Andrew Yourtchenkob31d3932019-10-31 08:34:22 +0000635 deadline = time.time() + 60
Dave Wallace3e9b7a22019-04-09 20:19:51 -0400636 ok = False
637 while time.time() < deadline:
638 cls.sleep(1)
639 size = curr_size
640 curr_size = os.path.getsize(corefile)
641 if size == curr_size:
642 ok = True
643 break
644 if not ok:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200645 cls.logger.error(
646 "Timed out waiting for coredump to complete: %s", corefile
647 )
Dave Wallace3e9b7a22019-04-09 20:19:51 -0400648 else:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200649 cls.logger.error("Coredump complete: %s, size %d", corefile, curr_size)
Dave Wallace3e9b7a22019-04-09 20:19:51 -0400650
651 @classmethod
Klement Sekerae2636852021-03-16 12:52:12 +0100652 def get_stats_sock_path(cls):
653 return "%s/stats.sock" % cls.tempdir
654
655 @classmethod
656 def get_api_sock_path(cls):
657 return "%s/api.sock" % cls.tempdir
658
659 @classmethod
660 def get_api_segment_prefix(cls):
661 return os.path.basename(cls.tempdir) # Only used for VAPI
662
663 @classmethod
664 def get_tempdir(cls):
Klement Sekerab3fc6582022-03-10 11:47:45 +0100665 if cls.debug_attach:
666 tmpdir = f"{config.tmp_dir}/unittest-attach-gdb"
667 else:
668 tmpdir = f"{config.tmp_dir}/vpp-unittest-{cls.__name__}"
669 if config.wipe_tmp_dir:
670 shutil.rmtree(tmpdir, ignore_errors=True)
671 os.mkdir(tmpdir)
Klement Sekerab23ffd72021-05-31 16:08:53 +0200672 return tmpdir
673
674 @classmethod
675 def create_file_handler(cls):
676 if config.log_dir is None:
677 cls.file_handler = FileHandler(f"{cls.tempdir}/log.txt")
678 return
679
680 logdir = f"{config.log_dir}/vpp-unittest-{cls.__name__}"
681 if config.wipe_tmp_dir:
682 shutil.rmtree(logdir, ignore_errors=True)
683 os.mkdir(logdir)
684 cls.file_handler = FileHandler(f"{logdir}/log.txt")
Klement Sekerae2636852021-03-16 12:52:12 +0100685
686 @classmethod
Damjan Marionf56b77a2016-10-03 19:44:57 +0200687 def setUpClass(cls):
Klement Sekeraf62ae122016-10-11 11:47:09 +0200688 """
689 Perform class setup before running the testcase
690 Remove shared memory files, start vpp and connect the vpp-api
691 """
Paul Vinciguerra8d991d92019-01-25 14:05:48 -0800692 super(VppTestCase, cls).setUpClass()
juraj.linkesdfb5f2a2018-11-09 11:58:54 +0100693 cls.logger = get_logger(cls.__name__)
Klement Sekerab23ffd72021-05-31 16:08:53 +0200694 random.seed(config.rnd_seed)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200695 if hasattr(cls, "parallel_handler"):
juraj.linkesdfb5f2a2018-11-09 11:58:54 +0100696 cls.logger.addHandler(cls.parallel_handler)
juraj.linkes3d9b92a2018-11-21 13:13:39 +0100697 cls.logger.propagate = False
Klement Sekerab23ffd72021-05-31 16:08:53 +0200698 cls.set_debug_flags(config.debug)
Klement Sekerae2636852021-03-16 12:52:12 +0100699 cls.tempdir = cls.get_tempdir()
Klement Sekerab23ffd72021-05-31 16:08:53 +0200700 cls.create_file_handler()
Klement Sekera027dbd52017-04-11 06:01:53 +0200701 cls.file_handler.setFormatter(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200702 Formatter(fmt="%(asctime)s,%(msecs)03d %(message)s", datefmt="%H:%M:%S")
703 )
Klement Sekera027dbd52017-04-11 06:01:53 +0200704 cls.file_handler.setLevel(DEBUG)
705 cls.logger.addHandler(cls.file_handler)
Klement Sekerae2636852021-03-16 12:52:12 +0100706 cls.logger.debug("--- setUpClass() for %s called ---" % cls.__name__)
Klement Sekeraf62ae122016-10-11 11:47:09 +0200707 os.chdir(cls.tempdir)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200708 cls.logger.info(
709 "Temporary dir is %s, api socket is %s",
710 cls.tempdir,
711 cls.get_api_sock_path(),
712 )
Klement Sekerab23ffd72021-05-31 16:08:53 +0200713 cls.logger.debug("Random seed is %s", config.rnd_seed)
Damjan Marionf56b77a2016-10-03 19:44:57 +0200714 cls.setUpConstants()
Klement Sekeradab231a2016-12-21 08:50:14 +0100715 cls.reset_packet_infos()
Klement Sekera3ff6ffc2021-04-01 18:19:29 +0200716 cls._pcaps = []
717 cls._old_pcaps = []
Klement Sekeraf62ae122016-10-11 11:47:09 +0200718 cls.verbose = 0
Klement Sekera085f5c02016-11-24 01:59:16 +0100719 cls.vpp_dead = False
Klement Sekera10db26f2017-01-11 08:16:53 +0100720 cls.registry = VppObjectRegistry()
Klement Sekera3747c752017-04-10 06:30:17 +0200721 cls.vpp_startup_failed = False
Klement Sekera909a6a12017-08-08 04:33:53 +0200722 cls.reporter = KeepAliveReporter()
Klement Sekeraf62ae122016-10-11 11:47:09 +0200723 # need to catch exceptions here because if we raise, then the cleanup
724 # doesn't get called and we might end with a zombie vpp
725 try:
Klement Sekerae2636852021-03-16 12:52:12 +0100726 if cls.debug_attach:
727 cls.attach_vpp()
728 else:
729 cls.run_vpp()
Dave Wallace670724c2022-09-20 21:52:18 -0400730 if not hasattr(cls, "vpp"):
731 return
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200732 cls.reporter.send_keep_alive(cls, "setUpClass")
juraj.linkes40dd73b2018-09-21 13:55:16 +0200733 VppTestResult.current_test_case_info = TestCaseInfo(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200734 cls.logger, cls.tempdir, cls.vpp.pid, config.vpp
735 )
Klement Sekerae4504c62016-12-08 10:16:41 +0100736 cls.vpp_stdout_deque = deque()
Klement Sekerae4504c62016-12-08 10:16:41 +0100737 cls.vpp_stderr_deque = deque()
Naveen Joyc872cec2022-08-30 13:59:03 -0700738 # Pump thread in a non-debug-attached & not running-vpp
739 if not cls.debug_attach and not hasattr(cls, "running_vpp"):
Klement Sekerae2636852021-03-16 12:52:12 +0100740 cls.pump_thread_stop_flag = Event()
741 cls.pump_thread_wakeup_pipe = os.pipe()
742 cls.pump_thread = Thread(target=pump_output, args=(cls,))
743 cls.pump_thread.daemon = True
744 cls.pump_thread.start()
745 if cls.debug_gdb or cls.debug_gdbserver or cls.debug_attach:
Paul Vinciguerrabfd7d292019-10-26 22:25:49 -0400746 cls.vapi_response_timeout = 0
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200747 cls.vapi = VppPapiProvider(cls.__name__, cls, cls.vapi_response_timeout)
Klement Sekera085f5c02016-11-24 01:59:16 +0100748 if cls.step:
Paul Vinciguerra496b0de2019-06-20 12:24:12 -0400749 hook = hookmodule.StepHook(cls)
Klement Sekera085f5c02016-11-24 01:59:16 +0100750 else:
Paul Vinciguerra496b0de2019-06-20 12:24:12 -0400751 hook = hookmodule.PollHook(cls)
Klement Sekera085f5c02016-11-24 01:59:16 +0100752 cls.vapi.register_hook(hook)
Klement Sekerae2636852021-03-16 12:52:12 +0100753 cls.statistics = VPPStats(socketname=cls.get_stats_sock_path())
Klement Sekera3747c752017-04-10 06:30:17 +0200754 try:
755 hook.poll_vpp()
Klement Sekera13a83ef2018-03-21 12:35:51 +0100756 except VppDiedError:
Klement Sekera3747c752017-04-10 06:30:17 +0200757 cls.vpp_startup_failed = True
758 cls.logger.critical(
759 "VPP died shortly after startup, check the"
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200760 " output to standard error for possible cause"
761 )
Klement Sekera3747c752017-04-10 06:30:17 +0200762 raise
Klement Sekera085f5c02016-11-24 01:59:16 +0100763 try:
764 cls.vapi.connect()
Ole Troan4376ab22021-03-03 10:40:05 +0100765 except (vpp_papi.VPPIOError, Exception) as e:
Paul Vinciguerra1043fd32019-12-02 21:42:28 -0500766 cls.logger.debug("Exception connecting to vapi: %s" % e)
767 cls.vapi.disconnect()
768
Klement Sekera085f5c02016-11-24 01:59:16 +0100769 if cls.debug_gdbserver:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200770 print(
771 colorize(
772 "You're running VPP inside gdbserver but "
773 "VPP-API connection failed, did you forget "
774 "to 'continue' VPP from within gdb?",
775 RED,
776 )
777 )
Ole Troan4376ab22021-03-03 10:40:05 +0100778 raise e
Klement Sekerae2636852021-03-16 12:52:12 +0100779 if cls.debug_attach:
780 last_line = cls.vapi.cli("show thread").split("\n")[-2]
781 cls.vpp_worker_count = int(last_line.split(" ")[0])
782 print("Detected VPP with %s workers." % cls.vpp_worker_count)
Paul Vinciguerrae64e5ff2020-04-28 00:27:38 -0400783 except vpp_papi.VPPRuntimeError as e:
784 cls.logger.debug("%s" % e)
785 cls.quit()
Ole Troan4376ab22021-03-03 10:40:05 +0100786 raise e
Andrew Yourtchenko4f05a8e2019-10-13 10:06:46 +0000787 except Exception as e:
788 cls.logger.debug("Exception connecting to VPP: %s" % e)
Paul Vinciguerra496b0de2019-06-20 12:24:12 -0400789 cls.quit()
Ole Troan4376ab22021-03-03 10:40:05 +0100790 raise e
Damjan Marionf56b77a2016-10-03 19:44:57 +0200791
Damjan Marionf56b77a2016-10-03 19:44:57 +0200792 @classmethod
Paul Vinciguerrac701e572019-12-19 16:09:43 -0500793 def _debug_quit(cls):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200794 if cls.debug_gdbserver or cls.debug_gdb:
Paul Vinciguerrac701e572019-12-19 16:09:43 -0500795 try:
796 cls.vpp.poll()
797
798 if cls.vpp.returncode is None:
799 print()
800 print(double_line_delim)
801 print("VPP or GDB server is still running")
802 print(single_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200803 input(
804 "When done debugging, press ENTER to kill the "
805 "process and finish running the testcase..."
806 )
Paul Vinciguerrac701e572019-12-19 16:09:43 -0500807 except AttributeError:
808 pass
809
810 @classmethod
Damjan Marionf56b77a2016-10-03 19:44:57 +0200811 def quit(cls):
Klement Sekeraf62ae122016-10-11 11:47:09 +0200812 """
813 Disconnect vpp-api, kill vpp and cleanup shared memory files
814 """
Paul Vinciguerrac701e572019-12-19 16:09:43 -0500815 cls._debug_quit()
Naveen Joyc872cec2022-08-30 13:59:03 -0700816 if hasattr(cls, "running_vpp"):
817 cls.vpp.quit_vpp()
Klement Sekera277b89c2016-10-28 13:20:27 +0200818
juraj.linkes184870a2018-07-16 14:22:01 +0200819 # first signal that we want to stop the pump thread, then wake it up
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200820 if hasattr(cls, "pump_thread_stop_flag"):
juraj.linkes184870a2018-07-16 14:22:01 +0200821 cls.pump_thread_stop_flag.set()
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200822 if hasattr(cls, "pump_thread_wakeup_pipe"):
823 os.write(cls.pump_thread_wakeup_pipe[1], b"ding dong wake up")
824 if hasattr(cls, "pump_thread"):
Klement Sekeraacb9b8e2017-02-14 02:55:31 +0100825 cls.logger.debug("Waiting for pump thread to stop")
826 cls.pump_thread.join()
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200827 if hasattr(cls, "vpp_stderr_reader_thread"):
Paul Vinciguerrac701e572019-12-19 16:09:43 -0500828 cls.logger.debug("Waiting for stderr pump to stop")
Klement Sekeraacb9b8e2017-02-14 02:55:31 +0100829 cls.vpp_stderr_reader_thread.join()
830
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200831 if hasattr(cls, "vpp"):
832 if hasattr(cls, "vapi"):
Ole Troanfd574082019-11-27 23:12:48 +0100833 cls.logger.debug(cls.vapi.vpp.get_stats())
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200834 cls.logger.debug("Disconnecting class vapi client on %s", cls.__name__)
Klement Sekera0529a742016-12-02 07:05:24 +0100835 cls.vapi.disconnect()
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200836 cls.logger.debug("Deleting class vapi attribute on %s", cls.__name__)
Klement Sekeraacb9b8e2017-02-14 02:55:31 +0100837 del cls.vapi
Klement Sekeraf62ae122016-10-11 11:47:09 +0200838 cls.vpp.poll()
Klement Sekerae2636852021-03-16 12:52:12 +0100839 if not cls.debug_attach and cls.vpp.returncode is None:
Andrew Yourtchenkob31d3932019-10-31 08:34:22 +0000840 cls.wait_for_coredump()
Klement Sekeraacb9b8e2017-02-14 02:55:31 +0100841 cls.logger.debug("Sending TERM to vpp")
Dave Barachad646872019-05-06 10:49:41 -0400842 cls.vpp.terminate()
Klement Sekeraacb9b8e2017-02-14 02:55:31 +0100843 cls.logger.debug("Waiting for vpp to die")
Ole Troan4376ab22021-03-03 10:40:05 +0100844 try:
845 outs, errs = cls.vpp.communicate(timeout=5)
846 except subprocess.TimeoutExpired:
847 cls.vpp.kill()
848 outs, errs = cls.vpp.communicate()
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200849 cls.logger.debug("Deleting class vpp attribute on %s", cls.__name__)
Naveen Joyc872cec2022-08-30 13:59:03 -0700850 if not cls.debug_attach and not hasattr(cls, "running_vpp"):
Klement Sekerae2636852021-03-16 12:52:12 +0100851 cls.vpp.stdout.close()
852 cls.vpp.stderr.close()
Naveen Joyc872cec2022-08-30 13:59:03 -0700853 # If vpp is a dynamic attribute set by the func use_running,
854 # deletion will result in an AttributeError that we can
855 # safetly pass.
856 try:
857 del cls.vpp
858 except AttributeError:
859 pass
Damjan Marionf56b77a2016-10-03 19:44:57 +0200860
Klement Sekera3747c752017-04-10 06:30:17 +0200861 if cls.vpp_startup_failed:
862 stdout_log = cls.logger.info
863 stderr_log = cls.logger.critical
864 else:
865 stdout_log = cls.logger.info
866 stderr_log = cls.logger.info
867
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200868 if hasattr(cls, "vpp_stdout_deque"):
Klement Sekera3747c752017-04-10 06:30:17 +0200869 stdout_log(single_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200870 stdout_log("VPP output to stdout while running %s:", cls.__name__)
Klement Sekera3747c752017-04-10 06:30:17 +0200871 stdout_log(single_line_delim)
Klement Sekerae4504c62016-12-08 10:16:41 +0100872 vpp_output = "".join(cls.vpp_stdout_deque)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200873 with open(cls.tempdir + "/vpp_stdout.txt", "w") as f:
Klement Sekera027dbd52017-04-11 06:01:53 +0200874 f.write(vpp_output)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200875 stdout_log("\n%s", vpp_output)
Klement Sekera3747c752017-04-10 06:30:17 +0200876 stdout_log(single_line_delim)
Klement Sekera277b89c2016-10-28 13:20:27 +0200877
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200878 if hasattr(cls, "vpp_stderr_deque"):
Klement Sekera3747c752017-04-10 06:30:17 +0200879 stderr_log(single_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200880 stderr_log("VPP output to stderr while running %s:", cls.__name__)
Klement Sekera3747c752017-04-10 06:30:17 +0200881 stderr_log(single_line_delim)
Klement Sekera0ede47a2019-01-29 11:49:09 +0100882 vpp_output = "".join(cls.vpp_stderr_deque)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200883 with open(cls.tempdir + "/vpp_stderr.txt", "w") as f:
Klement Sekera027dbd52017-04-11 06:01:53 +0200884 f.write(vpp_output)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200885 stderr_log("\n%s", vpp_output)
Klement Sekera3747c752017-04-10 06:30:17 +0200886 stderr_log(single_line_delim)
Klement Sekera277b89c2016-10-28 13:20:27 +0200887
Damjan Marionf56b77a2016-10-03 19:44:57 +0200888 @classmethod
889 def tearDownClass(cls):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200890 """Perform final cleanup after running all tests in this test-case"""
891 cls.logger.debug("--- tearDownClass() for %s called ---" % cls.__name__)
Dave Wallace670724c2022-09-20 21:52:18 -0400892 if not hasattr(cls, "vpp"):
893 return
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200894 cls.reporter.send_keep_alive(cls, "tearDownClass")
Damjan Marionf56b77a2016-10-03 19:44:57 +0200895 cls.quit()
Klement Sekera027dbd52017-04-11 06:01:53 +0200896 cls.file_handler.close()
Klement Sekeraebbaf552018-02-17 13:41:33 +0100897 cls.reset_packet_infos()
Klement Sekerab23ffd72021-05-31 16:08:53 +0200898 if config.debug_framework:
Klement Sekeraebbaf552018-02-17 13:41:33 +0100899 debug_internal.on_tear_down_class(cls)
Damjan Marionf56b77a2016-10-03 19:44:57 +0200900
Paul Vinciguerra90cf21b2019-03-13 09:23:05 -0700901 def show_commands_at_teardown(self):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200902 """Allow subclass specific teardown logging additions."""
Paul Vinciguerra90cf21b2019-03-13 09:23:05 -0700903 self.logger.info("--- No test specific show commands provided. ---")
904
Damjan Marionf56b77a2016-10-03 19:44:57 +0200905 def tearDown(self):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200906 """Show various debug prints after each test"""
907 self.logger.debug(
908 "--- tearDown() for %s.%s(%s) called ---"
909 % (self.__class__.__name__, self._testMethodName, self._testMethodDoc)
910 )
Dave Wallace670724c2022-09-20 21:52:18 -0400911 if not hasattr(self, "vpp"):
912 return
Paul Vinciguerra499ea642019-03-15 09:39:19 -0700913
914 try:
915 if not self.vpp_dead:
916 self.logger.debug(self.vapi.cli("show trace max 1000"))
917 self.logger.info(self.vapi.ppcli("show interface"))
918 self.logger.info(self.vapi.ppcli("show hardware"))
919 self.logger.info(self.statistics.set_errors_str())
920 self.logger.info(self.vapi.ppcli("show run"))
921 self.logger.info(self.vapi.ppcli("show log"))
Dave Barach32dcd3b2019-07-08 12:25:38 -0400922 self.logger.info(self.vapi.ppcli("show bihash"))
Paul Vinciguerra499ea642019-03-15 09:39:19 -0700923 self.logger.info("Logging testcase specific show commands.")
924 self.show_commands_at_teardown()
Klement Sekera140af152022-02-18 10:30:51 +0000925 if self.remove_configured_vpp_objects_on_tear_down:
926 self.registry.remove_vpp_config(self.logger)
Dave Wallace90c55722017-02-16 11:25:26 -0500927 # Save/Dump VPP api trace log
Andrew Yourtchenko586d3ed2019-10-21 12:55:48 +0000928 m = self._testMethodName
929 api_trace = "vpp_api_trace.%s.%d.log" % (m, self.vpp.pid)
Dave Wallace90c55722017-02-16 11:25:26 -0500930 tmp_api_trace = "/tmp/%s" % api_trace
931 vpp_api_trace_log = "%s/%s" % (self.tempdir, api_trace)
932 self.logger.info(self.vapi.ppcli("api trace save %s" % api_trace))
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200933 self.logger.info("Moving %s to %s\n" % (tmp_api_trace, vpp_api_trace_log))
Dave Wallace90c55722017-02-16 11:25:26 -0500934 os.rename(tmp_api_trace, vpp_api_trace_log)
Ole Troan4376ab22021-03-03 10:40:05 +0100935 except VppTransportSocketIOError:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200936 self.logger.debug(
937 "VppTransportSocketIOError: Vpp dead. Cannot log show commands."
938 )
Paul Vinciguerra499ea642019-03-15 09:39:19 -0700939 self.vpp_dead = True
Klement Sekera1b686402017-03-02 11:29:19 +0100940 else:
941 self.registry.unregister_all(self.logger)
Damjan Marionf56b77a2016-10-03 19:44:57 +0200942
Damjan Marionf56b77a2016-10-03 19:44:57 +0200943 def setUp(self):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200944 """Clear trace before running each test"""
Paul Vinciguerra8d991d92019-01-25 14:05:48 -0800945 super(VppTestCase, self).setUp()
Dave Wallace670724c2022-09-20 21:52:18 -0400946 if not hasattr(self, "vpp"):
947 return
Klement Sekera909a6a12017-08-08 04:33:53 +0200948 self.reporter.send_keep_alive(self)
Klement Sekera0c1519b2016-12-08 05:03:32 +0100949 if self.vpp_dead:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200950 raise VppDiedError(
951 rv=None,
952 testcase=self.__class__.__name__,
953 method_name=self._testMethodName,
954 )
955 self.sleep(0.1, "during setUp")
Klement Sekerae4504c62016-12-08 10:16:41 +0100956 self.vpp_stdout_deque.append(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200957 "--- test setUp() for %s.%s(%s) starts here ---\n"
958 % (self.__class__.__name__, self._testMethodName, self._testMethodDoc)
959 )
Klement Sekerae4504c62016-12-08 10:16:41 +0100960 self.vpp_stderr_deque.append(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200961 "--- test setUp() for %s.%s(%s) starts here ---\n"
962 % (self.__class__.__name__, self._testMethodName, self._testMethodDoc)
963 )
Klement Sekeraf62ae122016-10-11 11:47:09 +0200964 self.vapi.cli("clear trace")
965 # store the test instance inside the test class - so that objects
966 # holding the class can access instance methods (like assertEqual)
967 type(self).test_instance = self
Damjan Marionf56b77a2016-10-03 19:44:57 +0200968
Damjan Marionf56b77a2016-10-03 19:44:57 +0200969 @classmethod
Klement Sekera75e7d132017-09-20 08:26:30 +0200970 def pg_enable_capture(cls, interfaces=None):
Klement Sekeraf62ae122016-10-11 11:47:09 +0200971 """
972 Enable capture on packet-generator interfaces
Damjan Marionf56b77a2016-10-03 19:44:57 +0200973
Klement Sekera75e7d132017-09-20 08:26:30 +0200974 :param interfaces: iterable interface indexes (if None,
975 use self.pg_interfaces)
Damjan Marionf56b77a2016-10-03 19:44:57 +0200976
Klement Sekeraf62ae122016-10-11 11:47:09 +0200977 """
Klement Sekera75e7d132017-09-20 08:26:30 +0200978 if interfaces is None:
979 interfaces = cls.pg_interfaces
Klement Sekeraf62ae122016-10-11 11:47:09 +0200980 for i in interfaces:
981 i.enable_capture()
Damjan Marionf56b77a2016-10-03 19:44:57 +0200982
Damjan Marionf56b77a2016-10-03 19:44:57 +0200983 @classmethod
Klement Sekera3ff6ffc2021-04-01 18:19:29 +0200984 def register_pcap(cls, intf, worker):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200985 """Register a pcap in the testclass"""
Klement Sekera9225dee2016-12-12 08:36:58 +0100986 # add to the list of captures with current timestamp
Klement Sekera3ff6ffc2021-04-01 18:19:29 +0200987 cls._pcaps.append((intf, worker))
Klement Sekera9225dee2016-12-12 08:36:58 +0100988
989 @classmethod
Andrew Yourtchenko63cb8822019-10-13 18:56:03 +0000990 def get_vpp_time(cls):
Dave Barach19718002020-03-11 10:31:36 -0400991 # processes e.g. "Time now 2.190522, Wed, 11 Mar 2020 17:29:54 GMT"
992 # returns float("2.190522")
Klement Sekerad9b0c6f2022-04-26 19:02:15 +0200993 timestr = cls.vapi.cli("show clock")
994 head, sep, tail = timestr.partition(",")
995 head, sep, tail = head.partition("Time now")
Dave Barach19718002020-03-11 10:31:36 -0400996 return float(tail)
Andrew Yourtchenko63cb8822019-10-13 18:56:03 +0000997
998 @classmethod
999 def sleep_on_vpp_time(cls, sec):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001000 """Sleep according to time in VPP world"""
Andrew Yourtchenko63cb8822019-10-13 18:56:03 +00001001 # On a busy system with many processes
1002 # we might end up with VPP time being slower than real world
1003 # So take that into account when waiting for VPP to do something
1004 start_time = cls.get_vpp_time()
1005 while cls.get_vpp_time() - start_time < sec:
1006 cls.sleep(0.1)
1007
1008 @classmethod
Benoît Ganne8c45e512021-02-19 16:39:13 +01001009 def pg_start(cls, trace=True):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001010 """Enable the PG, wait till it is done, then clean up"""
Klement Sekera3ff6ffc2021-04-01 18:19:29 +02001011 for (intf, worker) in cls._old_pcaps:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001012 intf.handle_old_pcap_file(intf.get_in_path(worker), intf.in_history_counter)
Klement Sekera3ff6ffc2021-04-01 18:19:29 +02001013 cls._old_pcaps = []
Benoît Ganne8c45e512021-02-19 16:39:13 +01001014 if trace:
1015 cls.vapi.cli("clear trace")
1016 cls.vapi.cli("trace add pg-input 1000")
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001017 cls.vapi.cli("packet-generator enable")
Andrew Yourtchenko8d829f62019-10-13 10:09:50 +00001018 # PG, when starts, runs to completion -
1019 # so let's avoid a race condition,
1020 # and wait a little till it's done.
1021 # Then clean it up - and then be gone.
1022 deadline = time.time() + 300
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001023 while cls.vapi.cli("show packet-generator").find("Yes") != -1:
Andrew Yourtchenko8d829f62019-10-13 10:09:50 +00001024 cls.sleep(0.01) # yield
1025 if time.time() > deadline:
1026 cls.logger.error("Timeout waiting for pg to stop")
1027 break
Klement Sekera3ff6ffc2021-04-01 18:19:29 +02001028 for intf, worker in cls._pcaps:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001029 cls.vapi.cli("packet-generator delete %s" % intf.get_cap_name(worker))
Klement Sekera3ff6ffc2021-04-01 18:19:29 +02001030 cls._old_pcaps = cls._pcaps
1031 cls._pcaps = []
Damjan Marionf56b77a2016-10-03 19:44:57 +02001032
Damjan Marionf56b77a2016-10-03 19:44:57 +02001033 @classmethod
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001034 def create_pg_interfaces_internal(cls, interfaces, gso=0, gso_size=0, mode=None):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001035 """
Matej Klotton8d8a1da2016-12-22 11:06:56 +01001036 Create packet-generator interfaces.
Damjan Marionf56b77a2016-10-03 19:44:57 +02001037
Matej Klotton8d8a1da2016-12-22 11:06:56 +01001038 :param interfaces: iterable indexes of the interfaces.
1039 :returns: List of created interfaces.
Damjan Marionf56b77a2016-10-03 19:44:57 +02001040
Klement Sekeraf62ae122016-10-11 11:47:09 +02001041 """
1042 result = []
1043 for i in interfaces:
Neale Ranns6197cb72021-06-03 14:43:21 +00001044 intf = VppPGInterface(cls, i, gso, gso_size, mode)
Klement Sekeraf62ae122016-10-11 11:47:09 +02001045 setattr(cls, intf.name, intf)
1046 result.append(intf)
1047 cls.pg_interfaces = result
1048 return result
1049
Matej Klotton0178d522016-11-04 11:11:44 +01001050 @classmethod
Neale Ranns6197cb72021-06-03 14:43:21 +00001051 def create_pg_ip4_interfaces(cls, interfaces, gso=0, gso_size=0):
Dave Wallace670724c2022-09-20 21:52:18 -04001052 if not hasattr(cls, "vpp"):
1053 cls.pg_interfaces = []
1054 return cls.pg_interfaces
Neale Ranns6197cb72021-06-03 14:43:21 +00001055 pgmode = VppEnum.vl_api_pg_interface_mode_t
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001056 return cls.create_pg_interfaces_internal(
1057 interfaces, gso, gso_size, pgmode.PG_API_MODE_IP4
1058 )
Neale Ranns6197cb72021-06-03 14:43:21 +00001059
1060 @classmethod
1061 def create_pg_ip6_interfaces(cls, interfaces, gso=0, gso_size=0):
Dave Wallace670724c2022-09-20 21:52:18 -04001062 if not hasattr(cls, "vpp"):
1063 cls.pg_interfaces = []
1064 return cls.pg_interfaces
Neale Ranns6197cb72021-06-03 14:43:21 +00001065 pgmode = VppEnum.vl_api_pg_interface_mode_t
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001066 return cls.create_pg_interfaces_internal(
1067 interfaces, gso, gso_size, pgmode.PG_API_MODE_IP6
1068 )
Neale Ranns6197cb72021-06-03 14:43:21 +00001069
1070 @classmethod
1071 def create_pg_interfaces(cls, interfaces, gso=0, gso_size=0):
Dave Wallace670724c2022-09-20 21:52:18 -04001072 if not hasattr(cls, "vpp"):
1073 cls.pg_interfaces = []
1074 return cls.pg_interfaces
Neale Ranns6197cb72021-06-03 14:43:21 +00001075 pgmode = VppEnum.vl_api_pg_interface_mode_t
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001076 return cls.create_pg_interfaces_internal(
1077 interfaces, gso, gso_size, pgmode.PG_API_MODE_ETHERNET
1078 )
Neale Ranns6197cb72021-06-03 14:43:21 +00001079
1080 @classmethod
1081 def create_pg_ethernet_interfaces(cls, interfaces, gso=0, gso_size=0):
Dave Wallace670724c2022-09-20 21:52:18 -04001082 if not hasattr(cls, "vpp"):
1083 cls.pg_interfaces = []
1084 return cls.pg_interfaces
Neale Ranns6197cb72021-06-03 14:43:21 +00001085 pgmode = VppEnum.vl_api_pg_interface_mode_t
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001086 return cls.create_pg_interfaces_internal(
1087 interfaces, gso, gso_size, pgmode.PG_API_MODE_ETHERNET
1088 )
Neale Ranns6197cb72021-06-03 14:43:21 +00001089
1090 @classmethod
Klement Sekerab9ef2732018-06-24 22:49:33 +02001091 def create_loopback_interfaces(cls, count):
Matej Klotton0178d522016-11-04 11:11:44 +01001092 """
Matej Klotton8d8a1da2016-12-22 11:06:56 +01001093 Create loopback interfaces.
Matej Klotton0178d522016-11-04 11:11:44 +01001094
Klement Sekerab9ef2732018-06-24 22:49:33 +02001095 :param count: number of interfaces created.
Matej Klotton8d8a1da2016-12-22 11:06:56 +01001096 :returns: List of created interfaces.
Matej Klotton0178d522016-11-04 11:11:44 +01001097 """
Dave Wallace670724c2022-09-20 21:52:18 -04001098 if not hasattr(cls, "vpp"):
1099 cls.lo_interfaces = []
1100 return cls.lo_interfaces
Klement Sekerab9ef2732018-06-24 22:49:33 +02001101 result = [VppLoInterface(cls) for i in range(count)]
1102 for intf in result:
Matej Klotton0178d522016-11-04 11:11:44 +01001103 setattr(cls, intf.name, intf)
Matej Klotton0178d522016-11-04 11:11:44 +01001104 cls.lo_interfaces = result
1105 return result
1106
Neale Ranns192b13f2019-03-15 02:16:20 -07001107 @classmethod
1108 def create_bvi_interfaces(cls, count):
1109 """
1110 Create BVI interfaces.
1111
1112 :param count: number of interfaces created.
1113 :returns: List of created interfaces.
1114 """
Dave Wallace670724c2022-09-20 21:52:18 -04001115 if not hasattr(cls, "vpp"):
1116 cls.bvi_interfaces = []
1117 return cls.bvi_interfaces
Neale Ranns192b13f2019-03-15 02:16:20 -07001118 result = [VppBviInterface(cls) for i in range(count)]
1119 for intf in result:
1120 setattr(cls, intf.name, intf)
1121 cls.bvi_interfaces = result
1122 return result
1123
Damjan Marionf56b77a2016-10-03 19:44:57 +02001124 @staticmethod
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001125 def extend_packet(packet, size, padding=" "):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001126 """
Klement Sekera75e7d132017-09-20 08:26:30 +02001127 Extend packet to given size by padding with spaces or custom padding
Klement Sekeraf62ae122016-10-11 11:47:09 +02001128 NOTE: Currently works only when Raw layer is present.
1129
1130 :param packet: packet
1131 :param size: target size
Klement Sekera75e7d132017-09-20 08:26:30 +02001132 :param padding: padding used to extend the payload
Klement Sekeraf62ae122016-10-11 11:47:09 +02001133
1134 """
Damjan Marionf56b77a2016-10-03 19:44:57 +02001135 packet_len = len(packet) + 4
1136 extend = size - packet_len
1137 if extend > 0:
Alexandre Poirriercc991492019-05-07 10:39:57 +02001138 num = (extend // len(padding)) + 1
1139 packet[Raw].load += (padding * num)[:extend].encode("ascii")
Damjan Marionf56b77a2016-10-03 19:44:57 +02001140
Klement Sekeradab231a2016-12-21 08:50:14 +01001141 @classmethod
1142 def reset_packet_infos(cls):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001143 """Reset the list of packet info objects and packet counts to zero"""
Klement Sekeradab231a2016-12-21 08:50:14 +01001144 cls._packet_infos = {}
1145 cls._packet_count_for_dst_if_idx = {}
Klement Sekeraf62ae122016-10-11 11:47:09 +02001146
Klement Sekeradab231a2016-12-21 08:50:14 +01001147 @classmethod
1148 def create_packet_info(cls, src_if, dst_if):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001149 """
1150 Create packet info object containing the source and destination indexes
1151 and add it to the testcase's packet info list
1152
Klement Sekeradab231a2016-12-21 08:50:14 +01001153 :param VppInterface src_if: source interface
1154 :param VppInterface dst_if: destination interface
Klement Sekeraf62ae122016-10-11 11:47:09 +02001155
1156 :returns: _PacketInfo object
1157
1158 """
Damjan Marionf56b77a2016-10-03 19:44:57 +02001159 info = _PacketInfo()
Klement Sekeradab231a2016-12-21 08:50:14 +01001160 info.index = len(cls._packet_infos)
1161 info.src = src_if.sw_if_index
1162 info.dst = dst_if.sw_if_index
1163 if isinstance(dst_if, VppSubInterface):
1164 dst_idx = dst_if.parent.sw_if_index
1165 else:
1166 dst_idx = dst_if.sw_if_index
1167 if dst_idx in cls._packet_count_for_dst_if_idx:
1168 cls._packet_count_for_dst_if_idx[dst_idx] += 1
1169 else:
1170 cls._packet_count_for_dst_if_idx[dst_idx] = 1
1171 cls._packet_infos[info.index] = info
Damjan Marionf56b77a2016-10-03 19:44:57 +02001172 return info
Damjan Marionf56b77a2016-10-03 19:44:57 +02001173
Damjan Marionf56b77a2016-10-03 19:44:57 +02001174 @staticmethod
1175 def info_to_payload(info):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001176 """
1177 Convert _PacketInfo object to packet payload
1178
1179 :param info: _PacketInfo object
1180
1181 :returns: string containing serialized data from packet info
1182 """
Ray Kinsellab8165b92021-09-22 11:24:06 +01001183
1184 # retrieve payload, currently 18 bytes (4 x ints + 1 short)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001185 return pack("iiiih", info.index, info.src, info.dst, info.ip, info.proto)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001186
Damjan Marionf56b77a2016-10-03 19:44:57 +02001187 @staticmethod
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001188 def payload_to_info(payload, payload_field="load"):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001189 """
1190 Convert packet payload to _PacketInfo object
1191
1192 :param payload: packet payload
Paul Vinciguerra22364e42019-03-12 20:04:56 -07001193 :type payload: <class 'scapy.packet.Raw'>
1194 :param payload_field: packet fieldname of payload "load" for
Paul Vinciguerraeaea4212019-03-06 11:58:06 -08001195 <class 'scapy.packet.Raw'>
Paul Vinciguerra22364e42019-03-12 20:04:56 -07001196 :type payload_field: str
Klement Sekeraf62ae122016-10-11 11:47:09 +02001197 :returns: _PacketInfo object containing de-serialized data from payload
1198
1199 """
Ray Kinsellab8165b92021-09-22 11:24:06 +01001200
1201 # retrieve payload, currently 18 bytes (4 x ints + 1 short)
1202 payload_b = getattr(payload, payload_field)[:18]
1203
Damjan Marionf56b77a2016-10-03 19:44:57 +02001204 info = _PacketInfo()
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001205 info.index, info.src, info.dst, info.ip, info.proto = unpack("iiiih", payload_b)
Ray Kinsellab8165b92021-09-22 11:24:06 +01001206
1207 # some SRv6 TCs depend on get an exception if bad values are detected
1208 if info.index > 0x4000:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001209 raise ValueError("Index value is invalid")
Ray Kinsellab8165b92021-09-22 11:24:06 +01001210
Damjan Marionf56b77a2016-10-03 19:44:57 +02001211 return info
Damjan Marionf56b77a2016-10-03 19:44:57 +02001212
Damjan Marionf56b77a2016-10-03 19:44:57 +02001213 def get_next_packet_info(self, info):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001214 """
1215 Iterate over the packet info list stored in the testcase
1216 Start iteration with first element if info is None
1217 Continue based on index in info if info is specified
1218
1219 :param info: info or None
1220 :returns: next info in list or None if no more infos
1221 """
Damjan Marionf56b77a2016-10-03 19:44:57 +02001222 if info is None:
1223 next_index = 0
1224 else:
1225 next_index = info.index + 1
Klement Sekeradab231a2016-12-21 08:50:14 +01001226 if next_index == len(self._packet_infos):
Damjan Marionf56b77a2016-10-03 19:44:57 +02001227 return None
1228 else:
Klement Sekeradab231a2016-12-21 08:50:14 +01001229 return self._packet_infos[next_index]
Damjan Marionf56b77a2016-10-03 19:44:57 +02001230
Klement Sekeraf62ae122016-10-11 11:47:09 +02001231 def get_next_packet_info_for_interface(self, src_index, info):
1232 """
1233 Search the packet info list for the next packet info with same source
1234 interface index
1235
1236 :param src_index: source interface index to search for
1237 :param info: packet info - where to start the search
1238 :returns: packet info or None
1239
1240 """
Damjan Marionf56b77a2016-10-03 19:44:57 +02001241 while True:
1242 info = self.get_next_packet_info(info)
1243 if info is None:
1244 return None
Klement Sekeraf62ae122016-10-11 11:47:09 +02001245 if info.src == src_index:
Damjan Marionf56b77a2016-10-03 19:44:57 +02001246 return info
Damjan Marionf56b77a2016-10-03 19:44:57 +02001247
Klement Sekeraf62ae122016-10-11 11:47:09 +02001248 def get_next_packet_info_for_interface2(self, src_index, dst_index, info):
1249 """
1250 Search the packet info list for the next packet info with same source
1251 and destination interface indexes
1252
1253 :param src_index: source interface index to search for
1254 :param dst_index: destination interface index to search for
1255 :param info: packet info - where to start the search
1256 :returns: packet info or None
1257
1258 """
Damjan Marionf56b77a2016-10-03 19:44:57 +02001259 while True:
Klement Sekeraf62ae122016-10-11 11:47:09 +02001260 info = self.get_next_packet_info_for_interface(src_index, info)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001261 if info is None:
1262 return None
Klement Sekeraf62ae122016-10-11 11:47:09 +02001263 if info.dst == dst_index:
Damjan Marionf56b77a2016-10-03 19:44:57 +02001264 return info
Damjan Marionf56b77a2016-10-03 19:44:57 +02001265
Klement Sekera0e3c0de2016-09-29 14:43:44 +02001266 def assert_equal(self, real_value, expected_value, name_or_class=None):
1267 if name_or_class is None:
Klement Sekera239790f2017-02-16 10:53:53 +01001268 self.assertEqual(real_value, expected_value)
Klement Sekera0e3c0de2016-09-29 14:43:44 +02001269 return
1270 try:
1271 msg = "Invalid %s: %d('%s') does not match expected value %d('%s')"
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001272 msg = msg % (
1273 getdoc(name_or_class).strip(),
1274 real_value,
1275 str(name_or_class(real_value)),
1276 expected_value,
1277 str(name_or_class(expected_value)),
1278 )
Klement Sekera13a83ef2018-03-21 12:35:51 +01001279 except Exception:
Klement Sekera0e3c0de2016-09-29 14:43:44 +02001280 msg = "Invalid %s: %s does not match expected value %s" % (
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001281 name_or_class,
1282 real_value,
1283 expected_value,
1284 )
Klement Sekera0e3c0de2016-09-29 14:43:44 +02001285
1286 self.assertEqual(real_value, expected_value, msg)
1287
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001288 def assert_in_range(self, real_value, expected_min, expected_max, name=None):
Klement Sekera0e3c0de2016-09-29 14:43:44 +02001289 if name is None:
1290 msg = None
1291 else:
1292 msg = "Invalid %s: %s out of range <%s,%s>" % (
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001293 name,
1294 real_value,
1295 expected_min,
1296 expected_max,
1297 )
Klement Sekera0e3c0de2016-09-29 14:43:44 +02001298 self.assertTrue(expected_min <= real_value <= expected_max, msg)
1299
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001300 def assert_packet_checksums_valid(self, packet, ignore_zero_udp_checksums=True):
Paul Vinciguerraa7427ec2019-03-10 10:04:23 -07001301 received = packet.__class__(scapy.compat.raw(packet))
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001302 udp_layers = ["UDP", "UDPerror"]
1303 checksum_fields = ["cksum", "chksum"]
Klement Sekerad81ae412018-05-16 10:52:54 +02001304 checksums = []
1305 counter = 0
Paul Vinciguerraa7427ec2019-03-10 10:04:23 -07001306 temp = received.__class__(scapy.compat.raw(received))
Klement Sekerad81ae412018-05-16 10:52:54 +02001307 while True:
1308 layer = temp.getlayer(counter)
1309 if layer:
Klement Sekera66cea092019-12-05 13:13:21 +00001310 layer = layer.copy()
1311 layer.remove_payload()
Klement Sekerad81ae412018-05-16 10:52:54 +02001312 for cf in checksum_fields:
1313 if hasattr(layer, cf):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001314 if (
1315 ignore_zero_udp_checksums
1316 and 0 == getattr(layer, cf)
1317 and layer.name in udp_layers
1318 ):
Klement Sekerad81ae412018-05-16 10:52:54 +02001319 continue
Klement Sekera66cea092019-12-05 13:13:21 +00001320 delattr(temp.getlayer(counter), cf)
Klement Sekerad81ae412018-05-16 10:52:54 +02001321 checksums.append((counter, cf))
1322 else:
1323 break
1324 counter = counter + 1
Klement Sekera31da2e32018-06-24 22:49:55 +02001325 if 0 == len(checksums):
1326 return
Paul Vinciguerraa7427ec2019-03-10 10:04:23 -07001327 temp = temp.__class__(scapy.compat.raw(temp))
Klement Sekerad81ae412018-05-16 10:52:54 +02001328 for layer, cf in checksums:
Klement Sekera31da2e32018-06-24 22:49:55 +02001329 calc_sum = getattr(temp[layer], cf)
1330 self.assert_equal(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001331 getattr(received[layer], cf),
1332 calc_sum,
1333 "packet checksum on layer #%d: %s" % (layer, temp[layer].name),
1334 )
Klement Sekera31da2e32018-06-24 22:49:55 +02001335 self.logger.debug(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001336 "Checksum field `%s` on `%s` layer has correct value `%s`"
1337 % (cf, temp[layer].name, calc_sum)
1338 )
Klement Sekerad81ae412018-05-16 10:52:54 +02001339
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001340 def assert_checksum_valid(
1341 self, received_packet, layer, field_name="chksum", ignore_zero_checksum=False
1342 ):
1343 """Check checksum of received packet on given layer"""
Klement Sekerad81ae412018-05-16 10:52:54 +02001344 received_packet_checksum = getattr(received_packet[layer], field_name)
1345 if ignore_zero_checksum and 0 == received_packet_checksum:
1346 return
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001347 recalculated = received_packet.__class__(scapy.compat.raw(received_packet))
Klement Sekerad81ae412018-05-16 10:52:54 +02001348 delattr(recalculated[layer], field_name)
Paul Vinciguerraa7427ec2019-03-10 10:04:23 -07001349 recalculated = recalculated.__class__(scapy.compat.raw(recalculated))
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001350 self.assert_equal(
1351 received_packet_checksum,
1352 getattr(recalculated[layer], field_name),
1353 "packet checksum on layer: %s" % layer,
1354 )
Klement Sekerad81ae412018-05-16 10:52:54 +02001355
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001356 def assert_ip_checksum_valid(self, received_packet, ignore_zero_checksum=False):
1357 self.assert_checksum_valid(
1358 received_packet, "IP", ignore_zero_checksum=ignore_zero_checksum
1359 )
Klement Sekerad81ae412018-05-16 10:52:54 +02001360
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001361 def assert_tcp_checksum_valid(self, received_packet, ignore_zero_checksum=False):
1362 self.assert_checksum_valid(
1363 received_packet, "TCP", ignore_zero_checksum=ignore_zero_checksum
1364 )
Klement Sekerad81ae412018-05-16 10:52:54 +02001365
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001366 def assert_udp_checksum_valid(self, received_packet, ignore_zero_checksum=True):
1367 self.assert_checksum_valid(
1368 received_packet, "UDP", ignore_zero_checksum=ignore_zero_checksum
1369 )
Klement Sekerad81ae412018-05-16 10:52:54 +02001370
1371 def assert_embedded_icmp_checksum_valid(self, received_packet):
1372 if received_packet.haslayer(IPerror):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001373 self.assert_checksum_valid(received_packet, "IPerror")
Klement Sekerad81ae412018-05-16 10:52:54 +02001374 if received_packet.haslayer(TCPerror):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001375 self.assert_checksum_valid(received_packet, "TCPerror")
Klement Sekerad81ae412018-05-16 10:52:54 +02001376 if received_packet.haslayer(UDPerror):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001377 self.assert_checksum_valid(
1378 received_packet, "UDPerror", ignore_zero_checksum=True
1379 )
Klement Sekerad81ae412018-05-16 10:52:54 +02001380 if received_packet.haslayer(ICMPerror):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001381 self.assert_checksum_valid(received_packet, "ICMPerror")
Klement Sekerad81ae412018-05-16 10:52:54 +02001382
1383 def assert_icmp_checksum_valid(self, received_packet):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001384 self.assert_checksum_valid(received_packet, "ICMP")
Klement Sekerad81ae412018-05-16 10:52:54 +02001385 self.assert_embedded_icmp_checksum_valid(received_packet)
1386
1387 def assert_icmpv6_checksum_valid(self, pkt):
1388 if pkt.haslayer(ICMPv6DestUnreach):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001389 self.assert_checksum_valid(pkt, "ICMPv6DestUnreach", "cksum")
Klement Sekerad81ae412018-05-16 10:52:54 +02001390 self.assert_embedded_icmp_checksum_valid(pkt)
1391 if pkt.haslayer(ICMPv6EchoRequest):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001392 self.assert_checksum_valid(pkt, "ICMPv6EchoRequest", "cksum")
Klement Sekerad81ae412018-05-16 10:52:54 +02001393 if pkt.haslayer(ICMPv6EchoReply):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001394 self.assert_checksum_valid(pkt, "ICMPv6EchoReply", "cksum")
Klement Sekerad81ae412018-05-16 10:52:54 +02001395
Klement Sekera107ad732022-02-18 10:32:08 +00001396 def get_counter(self, counter):
Klement Sekera3a343d42019-05-16 14:35:46 +02001397 if counter.startswith("/"):
1398 counter_value = self.statistics.get_counter(counter)
1399 else:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001400 counters = self.vapi.cli("sh errors").split("\n")
Klement Sekera6aa58b72019-05-16 14:34:55 +02001401 counter_value = 0
Klement Sekera3a343d42019-05-16 14:35:46 +02001402 for i in range(1, len(counters) - 1):
1403 results = counters[i].split()
1404 if results[1] == counter:
1405 counter_value = int(results[0])
1406 break
1407 return counter_value
1408
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001409 def assert_counter_equal(self, counter, expected_value, thread=None, index=0):
Klement Sekera107ad732022-02-18 10:32:08 +00001410 c = self.get_counter(counter)
1411 if thread is not None:
1412 c = c[thread][index]
1413 else:
1414 c = sum(x[index] for x in c)
1415 self.assert_equal(c, expected_value, "counter `%s'" % counter)
1416
Klement Sekeraf37c3ba2018-11-08 11:24:34 +01001417 def assert_packet_counter_equal(self, counter, expected_value):
Klement Sekera107ad732022-02-18 10:32:08 +00001418 counter_value = self.get_counter(counter)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001419 self.assert_equal(
1420 counter_value, expected_value, "packet counter `%s'" % counter
1421 )
Klement Sekeraf37c3ba2018-11-08 11:24:34 +01001422
Ole Troan233e4682019-05-16 15:01:34 +02001423 def assert_error_counter_equal(self, counter, expected_value):
Ole Troane66443c2021-03-18 11:12:01 +01001424 counter_value = self.statistics[counter].sum()
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001425 self.assert_equal(counter_value, expected_value, "error counter `%s'" % counter)
Ole Troan233e4682019-05-16 15:01:34 +02001426
Klement Sekeraacb9b8e2017-02-14 02:55:31 +01001427 @classmethod
1428 def sleep(cls, timeout, remark=None):
Paul Vinciguerra0f6602c2019-03-10 09:10:54 -07001429
1430 # /* Allow sleep(0) to maintain win32 semantics, and as decreed
1431 # * by Guido, only the main thread can be interrupted.
1432 # */
1433 # https://github.com/python/cpython/blob/6673decfa0fb078f60587f5cb5e98460eea137c2/Modules/timemodule.c#L1892 # noqa
1434 if timeout == 0:
1435 # yield quantum
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001436 if hasattr(os, "sched_yield"):
Paul Vinciguerra0f6602c2019-03-10 09:10:54 -07001437 os.sched_yield()
1438 else:
1439 time.sleep(0)
1440 return
1441
Paul Vinciguerra9beabd82019-12-01 22:24:28 -05001442 cls.logger.debug("Starting sleep for %es (%s)", timeout, remark)
Klement Sekera3cfa5582017-04-19 07:10:58 +00001443 before = time.time()
Klement Sekeraa57a9702017-02-02 06:58:07 +01001444 time.sleep(timeout)
Klement Sekera3cfa5582017-04-19 07:10:58 +00001445 after = time.time()
Paul Vinciguerra9beabd82019-12-01 22:24:28 -05001446 if after - before > 2 * timeout:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001447 cls.logger.error(
1448 "unexpected self.sleep() result - slept for %es instead of ~%es!",
1449 after - before,
1450 timeout,
1451 )
Paul Vinciguerra9beabd82019-12-01 22:24:28 -05001452
1453 cls.logger.debug(
Klement Sekera79a31db2021-03-12 18:16:10 +01001454 "Finished sleep (%s) - slept %es (wanted %es)",
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001455 remark,
1456 after - before,
1457 timeout,
1458 )
Klement Sekeraa57a9702017-02-02 06:58:07 +01001459
Benoît Ganne56eccdb2021-08-20 09:18:31 +02001460 def virtual_sleep(self, timeout, remark=None):
1461 self.logger.debug("Moving VPP time by %s (%s)", timeout, remark)
1462 self.vapi.cli("set clock adjust %s" % timeout)
1463
Benoît Ganne8c45e512021-02-19 16:39:13 +01001464 def pg_send(self, intf, pkts, worker=None, trace=True):
Neale Ranns4a56f4e2019-12-23 04:10:25 +00001465 intf.add_stream(pkts, worker=worker)
Neale Ranns52fae862018-01-08 04:41:42 -08001466 self.pg_enable_capture(self.pg_interfaces)
Benoît Ganne8c45e512021-02-19 16:39:13 +01001467 self.pg_start(trace=trace)
Paul Vinciguerraeb414432019-02-20 09:01:14 -08001468
Klement Sekeraad3187f2022-02-18 10:34:35 +00001469 def snapshot_stats(self, stats_diff):
1470 """Return snapshot of interesting stats based on diff dictionary."""
1471 stats_snapshot = {}
1472 for sw_if_index in stats_diff:
1473 for counter in stats_diff[sw_if_index]:
1474 stats_snapshot[counter] = self.statistics[counter]
1475 self.logger.debug(f"Took statistics stats_snapshot: {stats_snapshot}")
1476 return stats_snapshot
1477
1478 def compare_stats_with_snapshot(self, stats_diff, stats_snapshot):
1479 """Assert appropriate difference between current stats and snapshot."""
1480 for sw_if_index in stats_diff:
1481 for cntr, diff in stats_diff[sw_if_index].items():
1482 if sw_if_index == "err":
1483 self.assert_equal(
1484 self.statistics[cntr].sum(),
1485 stats_snapshot[cntr].sum() + diff,
1486 f"'{cntr}' counter value (previous value: "
1487 f"{stats_snapshot[cntr].sum()}, "
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001488 f"expected diff: {diff})",
1489 )
Klement Sekeraad3187f2022-02-18 10:34:35 +00001490 else:
1491 try:
1492 self.assert_equal(
1493 self.statistics[cntr][:, sw_if_index].sum(),
1494 stats_snapshot[cntr][:, sw_if_index].sum() + diff,
1495 f"'{cntr}' counter value (previous value: "
1496 f"{stats_snapshot[cntr][:, sw_if_index].sum()}, "
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001497 f"expected diff: {diff})",
1498 )
Klement Sekeraad3187f2022-02-18 10:34:35 +00001499 except IndexError:
1500 # if diff is 0, then this most probably a case where
1501 # test declares multiple interfaces but traffic hasn't
1502 # passed through this one yet - which means the counter
1503 # value is 0 and can be ignored
1504 if 0 != diff:
1505 raise
1506
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001507 def send_and_assert_no_replies(
1508 self, intf, pkts, remark="", timeout=None, stats_diff=None, trace=True, msg=None
1509 ):
Klement Sekeraad3187f2022-02-18 10:34:35 +00001510 if stats_diff:
1511 stats_snapshot = self.snapshot_stats(stats_diff)
1512
Paul Vinciguerraeb414432019-02-20 09:01:14 -08001513 self.pg_send(intf, pkts)
Klement Sekeraad3187f2022-02-18 10:34:35 +00001514
1515 try:
1516 if not timeout:
1517 timeout = 1
1518 for i in self.pg_interfaces:
Klement Sekera26cd0242022-02-18 10:35:08 +00001519 i.assert_nothing_captured(timeout=timeout, remark=remark)
Klement Sekeraad3187f2022-02-18 10:34:35 +00001520 timeout = 0.1
1521 finally:
1522 if trace:
1523 if msg:
1524 self.logger.debug(f"send_and_assert_no_replies: {msg}")
1525 self.logger.debug(self.vapi.cli("show trace"))
1526
1527 if stats_diff:
1528 self.compare_stats_with_snapshot(stats_diff, stats_snapshot)
Neale Ranns52fae862018-01-08 04:41:42 -08001529
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001530 def send_and_expect(
1531 self,
1532 intf,
1533 pkts,
1534 output,
1535 n_rx=None,
1536 worker=None,
1537 trace=True,
1538 msg=None,
1539 stats_diff=None,
1540 ):
Klement Sekeraad3187f2022-02-18 10:34:35 +00001541 if stats_diff:
1542 stats_snapshot = self.snapshot_stats(stats_diff)
1543
Neale Rannsd7603d92019-03-28 08:56:10 +00001544 if not n_rx:
Klement Sekera56c492a2022-01-10 21:57:27 +00001545 n_rx = 1 if isinstance(pkts, Packet) else len(pkts)
Benoît Ganne8c45e512021-02-19 16:39:13 +01001546 self.pg_send(intf, pkts, worker=worker, trace=trace)
Neale Rannsd7603d92019-03-28 08:56:10 +00001547 rx = output.get_capture(n_rx)
Klement Sekeraff334db2021-05-26 13:02:35 +02001548 if trace:
Klement Sekeraad3187f2022-02-18 10:34:35 +00001549 if msg:
1550 self.logger.debug(f"send_and_expect: {msg}")
Klement Sekeraff334db2021-05-26 13:02:35 +02001551 self.logger.debug(self.vapi.cli("show trace"))
Klement Sekeraad3187f2022-02-18 10:34:35 +00001552
1553 if stats_diff:
1554 self.compare_stats_with_snapshot(stats_diff, stats_snapshot)
1555
Neale Ranns93cc3ee2018-10-10 07:22:51 -07001556 return rx
1557
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001558 def send_and_expect_load_balancing(
1559 self, input, pkts, outputs, worker=None, trace=True
1560 ):
Neale Ranns699bea22022-02-17 09:22:16 +00001561 self.pg_send(input, pkts, worker=worker, trace=trace)
1562 rxs = []
1563 for oo in outputs:
1564 rx = oo._get_capture(1)
1565 self.assertNotEqual(0, len(rx))
1566 rxs.append(rx)
1567 if trace:
1568 self.logger.debug(self.vapi.cli("show trace"))
1569 return rxs
1570
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001571 def send_and_expect_some(self, intf, pkts, output, worker=None, trace=True):
Neale Ranns5c6dd172022-02-17 09:08:47 +00001572 self.pg_send(intf, pkts, worker=worker, trace=trace)
1573 rx = output._get_capture(1)
1574 if trace:
1575 self.logger.debug(self.vapi.cli("show trace"))
1576 self.assertTrue(len(rx) > 0)
1577 self.assertTrue(len(rx) < len(pkts))
1578 return rx
1579
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001580 def send_and_expect_only(self, intf, pkts, output, timeout=None, stats_diff=None):
Klement Sekeraad3187f2022-02-18 10:34:35 +00001581 if stats_diff:
1582 stats_snapshot = self.snapshot_stats(stats_diff)
1583
Paul Vinciguerraeb414432019-02-20 09:01:14 -08001584 self.pg_send(intf, pkts)
Paul Vinciguerra8aeb2202019-01-07 16:29:26 -08001585 rx = output.get_capture(len(pkts))
1586 outputs = [output]
Neale Ranns93cc3ee2018-10-10 07:22:51 -07001587 if not timeout:
1588 timeout = 1
1589 for i in self.pg_interfaces:
1590 if i not in outputs:
Klement Sekera26cd0242022-02-18 10:35:08 +00001591 i.assert_nothing_captured(timeout=timeout)
Neale Ranns93cc3ee2018-10-10 07:22:51 -07001592 timeout = 0.1
1593
Klement Sekeraad3187f2022-02-18 10:34:35 +00001594 if stats_diff:
1595 self.compare_stats_with_snapshot(stats_diff, stats_snapshot)
1596
Neale Ranns52fae862018-01-08 04:41:42 -08001597 return rx
1598
Damjan Marionf56b77a2016-10-03 19:44:57 +02001599
juraj.linkes184870a2018-07-16 14:22:01 +02001600def get_testcase_doc_name(test):
1601 return getdoc(test.__class__).splitlines()[0]
1602
1603
Ole Trøan5ba91592018-11-22 10:01:09 +00001604def get_test_description(descriptions, test):
1605 short_description = test.shortDescription()
1606 if descriptions and short_description:
1607 return short_description
1608 else:
1609 return str(test)
1610
1611
juraj.linkes40dd73b2018-09-21 13:55:16 +02001612class TestCaseInfo(object):
1613 def __init__(self, logger, tempdir, vpp_pid, vpp_bin_path):
1614 self.logger = logger
1615 self.tempdir = tempdir
1616 self.vpp_pid = vpp_pid
1617 self.vpp_bin_path = vpp_bin_path
1618 self.core_crash_test = None
Klement Sekera87134932017-03-07 11:39:27 +01001619
1620
Damjan Marionf56b77a2016-10-03 19:44:57 +02001621class VppTestResult(unittest.TestResult):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001622 """
1623 @property result_string
1624 String variable to store the test case result string.
1625 @property errors
1626 List variable containing 2-tuples of TestCase instances and strings
1627 holding formatted tracebacks. Each tuple represents a test which
1628 raised an unexpected exception.
1629 @property failures
1630 List variable containing 2-tuples of TestCase instances and strings
1631 holding formatted tracebacks. Each tuple represents a test where
1632 a failure was explicitly signalled using the TestCase.assert*()
1633 methods.
1634 """
1635
juraj.linkes40dd73b2018-09-21 13:55:16 +02001636 failed_test_cases_info = set()
1637 core_crash_test_cases_info = set()
1638 current_test_case_info = None
1639
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001640 def __init__(self, stream=None, descriptions=None, verbosity=None, runner=None):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001641 """
Klement Sekerada505f62017-01-04 12:58:53 +01001642 :param stream File descriptor to store where to report test results.
1643 Set to the standard error stream by default.
1644 :param descriptions Boolean variable to store information if to use
1645 test case descriptions.
Klement Sekeraf62ae122016-10-11 11:47:09 +02001646 :param verbosity Integer variable to store required verbosity level.
1647 """
Paul Vinciguerradd3c5d22019-01-13 16:09:10 -08001648 super(VppTestResult, self).__init__(stream, descriptions, verbosity)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001649 self.stream = stream
1650 self.descriptions = descriptions
1651 self.verbosity = verbosity
1652 self.result_string = None
juraj.linkesabec0122018-11-16 17:28:56 +01001653 self.runner = runner
Klement Sekera558ceab2021-04-08 19:37:41 +02001654 self.printed = []
Damjan Marionf56b77a2016-10-03 19:44:57 +02001655
Damjan Marionf56b77a2016-10-03 19:44:57 +02001656 def addSuccess(self, test):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001657 """
1658 Record a test succeeded result
1659
1660 :param test:
1661
1662 """
juraj.linkes40dd73b2018-09-21 13:55:16 +02001663 if self.current_test_case_info:
1664 self.current_test_case_info.logger.debug(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001665 "--- addSuccess() %s.%s(%s) called"
1666 % (test.__class__.__name__, test._testMethodName, test._testMethodDoc)
1667 )
Damjan Marionf56b77a2016-10-03 19:44:57 +02001668 unittest.TestResult.addSuccess(self, test)
Klement Sekera277b89c2016-10-28 13:20:27 +02001669 self.result_string = colorize("OK", GREEN)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001670
juraj.linkescae64f82018-09-19 15:01:47 +02001671 self.send_result_through_pipe(test, PASS)
1672
Klement Sekeraf62ae122016-10-11 11:47:09 +02001673 def addSkip(self, test, reason):
1674 """
1675 Record a test skipped.
1676
1677 :param test:
1678 :param reason:
1679
1680 """
juraj.linkes40dd73b2018-09-21 13:55:16 +02001681 if self.current_test_case_info:
1682 self.current_test_case_info.logger.debug(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001683 "--- addSkip() %s.%s(%s) called, reason is %s"
1684 % (
1685 test.__class__.__name__,
1686 test._testMethodName,
1687 test._testMethodDoc,
1688 reason,
1689 )
1690 )
Klement Sekeraf62ae122016-10-11 11:47:09 +02001691 unittest.TestResult.addSkip(self, test, reason)
Klement Sekera277b89c2016-10-28 13:20:27 +02001692 self.result_string = colorize("SKIP", YELLOW)
Klement Sekeraf62ae122016-10-11 11:47:09 +02001693
Klement Sekera558ceab2021-04-08 19:37:41 +02001694 if reason == "not enough cpus":
1695 self.send_result_through_pipe(test, SKIP_CPU_SHORTAGE)
1696 else:
1697 self.send_result_through_pipe(test, SKIP)
juraj.linkescae64f82018-09-19 15:01:47 +02001698
juraj.linkes40dd73b2018-09-21 13:55:16 +02001699 def symlink_failed(self):
1700 if self.current_test_case_info:
Klement Sekeraf413bef2017-08-15 07:09:02 +02001701 try:
Klement Sekerab23ffd72021-05-31 16:08:53 +02001702 failed_dir = config.failed_dir
juraj.linkes40dd73b2018-09-21 13:55:16 +02001703 link_path = os.path.join(
1704 failed_dir,
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001705 "%s-FAILED" % os.path.basename(self.current_test_case_info.tempdir),
1706 )
Paul Vinciguerra9beabd82019-12-01 22:24:28 -05001707
1708 self.current_test_case_info.logger.debug(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001709 "creating a link to the failed test"
1710 )
Paul Vinciguerra9beabd82019-12-01 22:24:28 -05001711 self.current_test_case_info.logger.debug(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001712 "os.symlink(%s, %s)"
1713 % (self.current_test_case_info.tempdir, link_path)
1714 )
juraj.linkes184870a2018-07-16 14:22:01 +02001715 if os.path.exists(link_path):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001716 self.current_test_case_info.logger.debug("symlink already exists")
juraj.linkes184870a2018-07-16 14:22:01 +02001717 else:
juraj.linkes40dd73b2018-09-21 13:55:16 +02001718 os.symlink(self.current_test_case_info.tempdir, link_path)
juraj.linkes184870a2018-07-16 14:22:01 +02001719
Klement Sekeraf413bef2017-08-15 07:09:02 +02001720 except Exception as e:
Paul Vinciguerra9beabd82019-12-01 22:24:28 -05001721 self.current_test_case_info.logger.error(e)
Klement Sekeraf413bef2017-08-15 07:09:02 +02001722
juraj.linkescae64f82018-09-19 15:01:47 +02001723 def send_result_through_pipe(self, test, result):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001724 if hasattr(self, "test_framework_result_pipe"):
juraj.linkescae64f82018-09-19 15:01:47 +02001725 pipe = self.test_framework_result_pipe
juraj.linkes5e2c54d2018-08-30 10:51:45 +02001726 if pipe:
juraj.linkescae64f82018-09-19 15:01:47 +02001727 pipe.send((test.id(), result))
juraj.linkes5e2c54d2018-08-30 10:51:45 +02001728
juraj.linkes40dd73b2018-09-21 13:55:16 +02001729 def log_error(self, test, err, fn_name):
1730 if self.current_test_case_info:
1731 if isinstance(test, unittest.suite._ErrorHolder):
1732 test_name = test.description
1733 else:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001734 test_name = "%s.%s(%s)" % (
1735 test.__class__.__name__,
1736 test._testMethodName,
1737 test._testMethodDoc,
1738 )
juraj.linkes40dd73b2018-09-21 13:55:16 +02001739 self.current_test_case_info.logger.debug(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001740 "--- %s() %s called, err is %s" % (fn_name, test_name, err)
1741 )
juraj.linkes40dd73b2018-09-21 13:55:16 +02001742 self.current_test_case_info.logger.debug(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001743 "formatted exception is:\n%s" % "".join(format_exception(*err))
1744 )
juraj.linkes40dd73b2018-09-21 13:55:16 +02001745
1746 def add_error(self, test, err, unittest_fn, error_type):
1747 if error_type == FAIL:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001748 self.log_error(test, err, "addFailure")
juraj.linkes40dd73b2018-09-21 13:55:16 +02001749 error_type_str = colorize("FAIL", RED)
1750 elif error_type == ERROR:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001751 self.log_error(test, err, "addError")
juraj.linkes40dd73b2018-09-21 13:55:16 +02001752 error_type_str = colorize("ERROR", RED)
1753 else:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001754 raise Exception(
1755 "Error type %s cannot be used to record an "
1756 "error or a failure" % error_type
1757 )
juraj.linkes40dd73b2018-09-21 13:55:16 +02001758
1759 unittest_fn(self, test, err)
1760 if self.current_test_case_info:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001761 self.result_string = "%s [ temp dir used by test case: %s ]" % (
1762 error_type_str,
1763 self.current_test_case_info.tempdir,
1764 )
juraj.linkes40dd73b2018-09-21 13:55:16 +02001765 self.symlink_failed()
1766 self.failed_test_cases_info.add(self.current_test_case_info)
1767 if is_core_present(self.current_test_case_info.tempdir):
1768 if not self.current_test_case_info.core_crash_test:
1769 if isinstance(test, unittest.suite._ErrorHolder):
1770 test_name = str(test)
1771 else:
Paul Vinciguerraea2450f2019-03-06 08:23:58 -08001772 test_name = "'{!s}' ({!s})".format(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001773 get_testcase_doc_name(test), test.id()
1774 )
juraj.linkes40dd73b2018-09-21 13:55:16 +02001775 self.current_test_case_info.core_crash_test = test_name
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001776 self.core_crash_test_cases_info.add(self.current_test_case_info)
juraj.linkes40dd73b2018-09-21 13:55:16 +02001777 else:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001778 self.result_string = "%s [no temp dir]" % error_type_str
juraj.linkes40dd73b2018-09-21 13:55:16 +02001779
1780 self.send_result_through_pipe(test, error_type)
1781
Damjan Marionf56b77a2016-10-03 19:44:57 +02001782 def addFailure(self, test, err):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001783 """
1784 Record a test failed result
1785
1786 :param test:
1787 :param err: error message
1788
1789 """
juraj.linkes40dd73b2018-09-21 13:55:16 +02001790 self.add_error(test, err, unittest.TestResult.addFailure, FAIL)
juraj.linkescae64f82018-09-19 15:01:47 +02001791
Damjan Marionf56b77a2016-10-03 19:44:57 +02001792 def addError(self, test, err):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001793 """
1794 Record a test error result
Damjan Marionf56b77a2016-10-03 19:44:57 +02001795
Klement Sekeraf62ae122016-10-11 11:47:09 +02001796 :param test:
1797 :param err: error message
1798
1799 """
juraj.linkes40dd73b2018-09-21 13:55:16 +02001800 self.add_error(test, err, unittest.TestResult.addError, ERROR)
juraj.linkescae64f82018-09-19 15:01:47 +02001801
Damjan Marionf56b77a2016-10-03 19:44:57 +02001802 def getDescription(self, test):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001803 """
1804 Get test description
1805
1806 :param test:
1807 :returns: test description
1808
1809 """
Ole Trøan5ba91592018-11-22 10:01:09 +00001810 return get_test_description(self.descriptions, test)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001811
Damjan Marionf56b77a2016-10-03 19:44:57 +02001812 def startTest(self, test):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001813 """
1814 Start a test
1815
1816 :param test:
1817
1818 """
Paul Vinciguerradd3c5d22019-01-13 16:09:10 -08001819
1820 def print_header(test):
Klement Sekera558ceab2021-04-08 19:37:41 +02001821 if test.__class__ in self.printed:
1822 return
1823
Andrew Yourtchenkoa3b7c552020-08-26 14:33:54 +00001824 test_doc = getdoc(test)
1825 if not test_doc:
1826 raise Exception("No doc string for test '%s'" % test.id())
Klement Sekera558ceab2021-04-08 19:37:41 +02001827
Klement Sekeraea6236b2021-03-25 14:03:44 +01001828 test_title = test_doc.splitlines()[0].rstrip()
Klement Sekera558ceab2021-04-08 19:37:41 +02001829 test_title = colorize(test_title, GREEN)
Andrew Yourtchenko06f32812021-01-14 10:19:08 +00001830 if test.is_tagged_run_solo():
Klement Sekera558ceab2021-04-08 19:37:41 +02001831 test_title = colorize(f"SOLO RUN: {test_title}", YELLOW)
Andrew Yourtchenkoa3b7c552020-08-26 14:33:54 +00001832
Andrew Yourtchenkoc8eae8d2021-01-20 20:30:36 +00001833 # This block may overwrite the colorized title above,
1834 # but we want this to stand out and be fixed
1835 if test.has_tag(TestCaseTag.FIXME_VPP_WORKERS):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001836 test_title = colorize(f"FIXME with VPP workers: {test_title}", RED)
Andrew Yourtchenkoc8eae8d2021-01-20 20:30:36 +00001837
Naveen Joy6eaeea92021-09-09 17:57:02 -07001838 if test.has_tag(TestCaseTag.FIXME_ASAN):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001839 test_title = colorize(f"FIXME with ASAN: {test_title}", RED)
Naveen Joy6eaeea92021-09-09 17:57:02 -07001840 test.skip_fixme_asan()
1841
Dave Wallacee95b2462022-09-18 22:28:44 -04001842 if is_distro_ubuntu2204 == True and test.has_tag(
1843 TestCaseTag.FIXME_UBUNTU2204
1844 ):
1845 test_title = colorize(f"FIXME on Ubuntu-22.04: {test_title}", RED)
1846 test.skip_fixme_ubuntu2204()
1847
Dave Wallace670724c2022-09-20 21:52:18 -04001848 if is_distro_debian11 == True and test.has_tag(TestCaseTag.FIXME_DEBIAN11):
1849 test_title = colorize(f"FIXME on Debian-11: {test_title}", RED)
1850 test.skip_fixme_debian11()
1851
Dave Wallace8a0a9d22022-10-04 22:02:49 -04001852 if "debug" in config.vpp_tag and test.has_tag(TestCaseTag.FIXME_VPP_DEBUG):
1853 test_title = colorize(f"FIXME on VPP Debug: {test_title}", RED)
1854 test.skip_fixme_vpp_debug()
1855
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001856 if hasattr(test, "vpp_worker_count"):
Klement Sekeraea6236b2021-03-25 14:03:44 +01001857 if test.vpp_worker_count == 0:
1858 test_title += " [main thread only]"
1859 elif test.vpp_worker_count == 1:
1860 test_title += " [1 worker thread]"
1861 else:
1862 test_title += f" [{test.vpp_worker_count} worker threads]"
1863
Klement Sekera558ceab2021-04-08 19:37:41 +02001864 if test.__class__.skipped_due_to_cpu_lack:
1865 test_title = colorize(
1866 f"{test_title} [skipped - not enough cpus, "
1867 f"required={test.__class__.get_cpus_required()}, "
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001868 f"available={max_vpp_cpus}]",
1869 YELLOW,
1870 )
Klement Sekera558ceab2021-04-08 19:37:41 +02001871
1872 print(double_line_delim)
1873 print(test_title)
1874 print(double_line_delim)
1875 self.printed.append(test.__class__)
Paul Vinciguerradd3c5d22019-01-13 16:09:10 -08001876
1877 print_header(test)
Ole Troan0c629322019-11-28 14:48:44 +01001878 self.start_test = time.time()
Damjan Marionf56b77a2016-10-03 19:44:57 +02001879 unittest.TestResult.startTest(self, test)
1880 if self.verbosity > 0:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001881 self.stream.writeln("Starting " + self.getDescription(test) + " ...")
Klement Sekeraf62ae122016-10-11 11:47:09 +02001882 self.stream.writeln(single_line_delim)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001883
Damjan Marionf56b77a2016-10-03 19:44:57 +02001884 def stopTest(self, test):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001885 """
juraj.linkes5e2c54d2018-08-30 10:51:45 +02001886 Called when the given test has been run
Klement Sekeraf62ae122016-10-11 11:47:09 +02001887
1888 :param test:
1889
1890 """
Damjan Marionf56b77a2016-10-03 19:44:57 +02001891 unittest.TestResult.stopTest(self, test)
Ole Troan0c629322019-11-28 14:48:44 +01001892
Damjan Marionf56b77a2016-10-03 19:44:57 +02001893 if self.verbosity > 0:
Klement Sekeraf62ae122016-10-11 11:47:09 +02001894 self.stream.writeln(single_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001895 self.stream.writeln(
1896 "%-73s%s" % (self.getDescription(test), self.result_string)
1897 )
Klement Sekeraf62ae122016-10-11 11:47:09 +02001898 self.stream.writeln(single_line_delim)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001899 else:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001900 self.stream.writeln(
1901 "%-68s %4.2f %s"
1902 % (
1903 self.getDescription(test),
1904 time.time() - self.start_test,
1905 self.result_string,
1906 )
1907 )
juraj.linkescae64f82018-09-19 15:01:47 +02001908
1909 self.send_result_through_pipe(test, TEST_RUN)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001910
Damjan Marionf56b77a2016-10-03 19:44:57 +02001911 def printErrors(self):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001912 """
1913 Print errors from running the test case
1914 """
juraj.linkesabec0122018-11-16 17:28:56 +01001915 if len(self.errors) > 0 or len(self.failures) > 0:
1916 self.stream.writeln()
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001917 self.printErrorList("ERROR", self.errors)
1918 self.printErrorList("FAIL", self.failures)
juraj.linkesabec0122018-11-16 17:28:56 +01001919
1920 # ^^ that is the last output from unittest before summary
1921 if not self.runner.print_summary:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001922 devnull = unittest.runner._WritelnDecorator(open(os.devnull, "w"))
juraj.linkesabec0122018-11-16 17:28:56 +01001923 self.stream = devnull
1924 self.runner.stream = devnull
Damjan Marionf56b77a2016-10-03 19:44:57 +02001925
Damjan Marionf56b77a2016-10-03 19:44:57 +02001926 def printErrorList(self, flavour, errors):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001927 """
1928 Print error list to the output stream together with error type
1929 and test case description.
1930
1931 :param flavour: error type
1932 :param errors: iterable errors
1933
1934 """
Damjan Marionf56b77a2016-10-03 19:44:57 +02001935 for test, err in errors:
Klement Sekeraf62ae122016-10-11 11:47:09 +02001936 self.stream.writeln(double_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001937 self.stream.writeln("%s: %s" % (flavour, self.getDescription(test)))
Klement Sekeraf62ae122016-10-11 11:47:09 +02001938 self.stream.writeln(single_line_delim)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001939 self.stream.writeln("%s" % err)
Damjan Marionf56b77a2016-10-03 19:44:57 +02001940
1941
Damjan Marionf56b77a2016-10-03 19:44:57 +02001942class VppTestRunner(unittest.TextTestRunner):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001943 """
Klement Sekera104543f2017-02-03 07:29:43 +01001944 A basic test runner implementation which prints results to standard error.
Klement Sekeraf62ae122016-10-11 11:47:09 +02001945 """
Paul Vinciguerra6919b0d2018-12-09 15:37:04 -08001946
Klement Sekeraf62ae122016-10-11 11:47:09 +02001947 @property
1948 def resultclass(self):
1949 """Class maintaining the results of the tests"""
1950 return VppTestResult
Damjan Marionf56b77a2016-10-03 19:44:57 +02001951
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001952 def __init__(
1953 self,
1954 keep_alive_pipe=None,
1955 descriptions=True,
1956 verbosity=1,
1957 result_pipe=None,
1958 failfast=False,
1959 buffer=False,
1960 resultclass=None,
1961 print_summary=True,
1962 **kwargs,
1963 ):
Klement Sekera7a161da2017-01-17 13:42:48 +01001964 # ignore stream setting here, use hard-coded stdout to be in sync
1965 # with prints from VppTestCase methods ...
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001966 super(VppTestRunner, self).__init__(
1967 sys.stdout, descriptions, verbosity, failfast, buffer, resultclass, **kwargs
1968 )
juraj.linkesccfead62018-11-21 13:20:43 +01001969 KeepAliveReporter.pipe = keep_alive_pipe
Klement Sekera104543f2017-02-03 07:29:43 +01001970
juraj.linkesabec0122018-11-16 17:28:56 +01001971 self.orig_stream = self.stream
1972 self.resultclass.test_framework_result_pipe = result_pipe
1973
1974 self.print_summary = print_summary
1975
1976 def _makeResult(self):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02001977 return self.resultclass(self.stream, self.descriptions, self.verbosity, self)
juraj.linkes5e2c54d2018-08-30 10:51:45 +02001978
Damjan Marionf56b77a2016-10-03 19:44:57 +02001979 def run(self, test):
Klement Sekeraf62ae122016-10-11 11:47:09 +02001980 """
1981 Run the tests
1982
1983 :param test:
1984
1985 """
Klement Sekera3658adc2017-06-07 08:19:47 +02001986 faulthandler.enable() # emit stack trace to stderr if killed by signal
juraj.linkes184870a2018-07-16 14:22:01 +02001987
1988 result = super(VppTestRunner, self).run(test)
juraj.linkesabec0122018-11-16 17:28:56 +01001989 if not self.print_summary:
1990 self.stream = self.orig_stream
1991 result.stream = self.orig_stream
juraj.linkes184870a2018-07-16 14:22:01 +02001992 return result
Neale Ranns812ed392017-10-16 04:20:13 -07001993
1994
1995class Worker(Thread):
Paul Vinciguerra48bdbcd2019-12-04 19:43:53 -05001996 def __init__(self, executable_args, logger, env=None, *args, **kwargs):
1997 super(Worker, self).__init__(*args, **kwargs)
Neale Ranns812ed392017-10-16 04:20:13 -07001998 self.logger = logger
Paul Vinciguerra48bdbcd2019-12-04 19:43:53 -05001999 self.args = executable_args
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002000 if hasattr(self, "testcase") and self.testcase.debug_all:
Dave Wallace24564332019-10-21 02:53:14 +00002001 if self.testcase.debug_gdbserver:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002002 self.args = [
2003 "/usr/bin/gdbserver",
2004 "localhost:{port}".format(port=self.testcase.gdbserver_port),
2005 ] + args
2006 elif self.testcase.debug_gdb and hasattr(self, "wait_for_gdb"):
Dave Wallace24564332019-10-21 02:53:14 +00002007 self.args.append(self.wait_for_gdb)
Paul Vinciguerra48bdbcd2019-12-04 19:43:53 -05002008 self.app_bin = executable_args[0]
Dave Wallace24564332019-10-21 02:53:14 +00002009 self.app_name = os.path.basename(self.app_bin)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002010 if hasattr(self, "role"):
2011 self.app_name += " {role}".format(role=self.role)
Paul Vinciguerra063366e2019-06-30 15:38:55 -04002012 self.process = None
Neale Ranns812ed392017-10-16 04:20:13 -07002013 self.result = None
Paul Vinciguerra063366e2019-06-30 15:38:55 -04002014 env = {} if env is None else env
Dave Wallace42996c02018-02-26 14:40:13 -05002015 self.env = copy.deepcopy(env)
Neale Ranns812ed392017-10-16 04:20:13 -07002016
Dave Wallace24564332019-10-21 02:53:14 +00002017 def wait_for_enter(self):
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002018 if not hasattr(self, "testcase"):
Dave Wallace24564332019-10-21 02:53:14 +00002019 return
2020 if self.testcase.debug_all and self.testcase.debug_gdbserver:
2021 print()
2022 print(double_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002023 print(
2024 "Spawned GDB Server for '{app}' with PID: {pid}".format(
2025 app=self.app_name, pid=self.process.pid
2026 )
2027 )
Dave Wallace24564332019-10-21 02:53:14 +00002028 elif self.testcase.debug_all and self.testcase.debug_gdb:
2029 print()
2030 print(double_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002031 print(
2032 "Spawned '{app}' with PID: {pid}".format(
2033 app=self.app_name, pid=self.process.pid
2034 )
2035 )
Dave Wallace24564332019-10-21 02:53:14 +00002036 else:
2037 return
2038 print(single_line_delim)
2039 print("You can debug '{app}' using:".format(app=self.app_name))
2040 if self.testcase.debug_gdbserver:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002041 print(
2042 "sudo gdb "
2043 + self.app_bin
2044 + " -ex 'target remote localhost:{port}'".format(
2045 port=self.testcase.gdbserver_port
2046 )
2047 )
2048 print(
2049 "Now is the time to attach gdb by running the above "
2050 "command, set up breakpoints etc., then resume from "
2051 "within gdb by issuing the 'continue' command"
2052 )
Dave Wallace24564332019-10-21 02:53:14 +00002053 self.testcase.gdbserver_port += 1
2054 elif self.testcase.debug_gdb:
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002055 print(
2056 "sudo gdb "
2057 + self.app_bin
2058 + " -ex 'attach {pid}'".format(pid=self.process.pid)
2059 )
2060 print(
2061 "Now is the time to attach gdb by running the above "
2062 "command and set up breakpoints etc., then resume from"
2063 " within gdb by issuing the 'continue' command"
2064 )
Dave Wallace24564332019-10-21 02:53:14 +00002065 print(single_line_delim)
2066 input("Press ENTER to continue running the testcase...")
2067
Neale Ranns812ed392017-10-16 04:20:13 -07002068 def run(self):
2069 executable = self.args[0]
Paul Vinciguerra063366e2019-06-30 15:38:55 -04002070 if not os.path.exists(executable) or not os.access(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002071 executable, os.F_OK | os.X_OK
2072 ):
Paul Vinciguerra063366e2019-06-30 15:38:55 -04002073 # Exit code that means some system file did not exist,
2074 # could not be opened, or had some other kind of error.
2075 self.result = os.EX_OSFILE
2076 raise EnvironmentError(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002077 "executable '%s' is not found or executable." % executable
2078 )
2079 self.logger.debug(
2080 "Running executable '{app}': '{cmd}'".format(
2081 app=self.app_name, cmd=" ".join(self.args)
2082 )
2083 )
Neale Ranns812ed392017-10-16 04:20:13 -07002084 env = os.environ.copy()
Dave Wallacecfcf2f42018-02-16 18:31:56 -05002085 env.update(self.env)
Neale Ranns812ed392017-10-16 04:20:13 -07002086 env["CK_LOG_FILE_NAME"] = "-"
2087 self.process = subprocess.Popen(
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002088 ["stdbuf", "-o0", "-e0"] + self.args,
2089 shell=False,
2090 env=env,
2091 preexec_fn=os.setpgrp,
2092 stdout=subprocess.PIPE,
2093 stderr=subprocess.PIPE,
2094 )
Dave Wallace24564332019-10-21 02:53:14 +00002095 self.wait_for_enter()
Neale Ranns812ed392017-10-16 04:20:13 -07002096 out, err = self.process.communicate()
Dave Wallace24564332019-10-21 02:53:14 +00002097 self.logger.debug("Finished running `{app}'".format(app=self.app_name))
Neale Ranns812ed392017-10-16 04:20:13 -07002098 self.logger.info("Return code is `%s'" % self.process.returncode)
2099 self.logger.info(single_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002100 self.logger.info(
2101 "Executable `{app}' wrote to stdout:".format(app=self.app_name)
2102 )
Neale Ranns812ed392017-10-16 04:20:13 -07002103 self.logger.info(single_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002104 self.logger.info(out.decode("utf-8"))
Neale Ranns812ed392017-10-16 04:20:13 -07002105 self.logger.info(single_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002106 self.logger.info(
2107 "Executable `{app}' wrote to stderr:".format(app=self.app_name)
2108 )
Neale Ranns812ed392017-10-16 04:20:13 -07002109 self.logger.info(single_line_delim)
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002110 self.logger.info(err.decode("utf-8"))
Neale Ranns812ed392017-10-16 04:20:13 -07002111 self.logger.info(single_line_delim)
2112 self.result = self.process.returncode
Paul Vinciguerradd3c5d22019-01-13 16:09:10 -08002113
Klement Sekera6aa58b72019-05-16 14:34:55 +02002114
Klement Sekerad9b0c6f2022-04-26 19:02:15 +02002115if __name__ == "__main__":
Paul Vinciguerradd3c5d22019-01-13 16:09:10 -08002116 pass