at master 55 lines 2.4 kB view raw
1# SPDX-License-Identifier: GPL-2.0 2 3""" 4Driver test environment. 5NetDrvEnv and NetDrvEpEnv are the main environment classes. 6Former is for local host only tests, latter creates / connects 7to a remote endpoint. See NIPA wiki for more information about 8running and writing driver tests. 9""" 10 11import sys 12from pathlib import Path 13 14KSFT_DIR = (Path(__file__).parent / "../../../..").resolve() 15 16try: 17 sys.path.append(KSFT_DIR.as_posix()) 18 19 # Import one by one to avoid pylint false positives 20 from net.lib.py import NetNS, NetNSEnter, NetdevSimDev 21 from net.lib.py import EthtoolFamily, NetdevFamily, NetshaperFamily, \ 22 NlError, RtnlFamily, DevlinkFamily, PSPFamily 23 from net.lib.py import CmdExitFailure 24 from net.lib.py import bkg, cmd, bpftool, bpftrace, defer, ethtool, \ 25 fd_read_timeout, ip, rand_port, wait_port_listen, wait_file 26 from net.lib.py import KsftSkipEx, KsftFailEx, KsftXfailEx 27 from net.lib.py import ksft_disruptive, ksft_exit, ksft_pr, ksft_run, \ 28 ksft_setup, ksft_variants, KsftNamedVariant 29 from net.lib.py import ksft_eq, ksft_ge, ksft_in, ksft_is, ksft_lt, \ 30 ksft_ne, ksft_not_in, ksft_raises, ksft_true, ksft_gt, ksft_not_none 31 32 __all__ = ["NetNS", "NetNSEnter", "NetdevSimDev", 33 "EthtoolFamily", "NetdevFamily", "NetshaperFamily", 34 "NlError", "RtnlFamily", "DevlinkFamily", "PSPFamily", 35 "CmdExitFailure", 36 "bkg", "cmd", "bpftool", "bpftrace", "defer", "ethtool", 37 "fd_read_timeout", "ip", "rand_port", 38 "wait_port_listen", "wait_file", 39 "KsftSkipEx", "KsftFailEx", "KsftXfailEx", 40 "ksft_disruptive", "ksft_exit", "ksft_pr", "ksft_run", 41 "ksft_setup", "ksft_variants", "KsftNamedVariant", 42 "ksft_eq", "ksft_ge", "ksft_in", "ksft_is", "ksft_lt", 43 "ksft_ne", "ksft_not_in", "ksft_raises", "ksft_true", "ksft_gt", 44 "ksft_not_none", "ksft_not_none"] 45 46 from .env import NetDrvEnv, NetDrvEpEnv 47 from .load import GenerateTraffic, Iperf3Runner 48 from .remote import Remote 49 50 __all__ += ["NetDrvEnv", "NetDrvEpEnv", "GenerateTraffic", "Remote", 51 "Iperf3Runner"] 52except ModuleNotFoundError as e: 53 print("Failed importing `net` library from kernel sources") 54 print(str(e)) 55 sys.exit(4)