__init__.py

 1# SPDX-License-Identifier: GPL-2.0
 2
 3"""
 4Driver test environment (hardware-only tests).
 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, tool
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    from drivers.net.lib.py import GenerateTraffic, Remote, Iperf3Runner
32    from drivers.net.lib.py import NetDrvEnv, NetDrvEpEnv
33
34    __all__ = ["NetNS", "NetNSEnter", "NetdevSimDev",
35               "EthtoolFamily", "NetdevFamily", "NetshaperFamily",
36               "NlError", "RtnlFamily", "DevlinkFamily", "PSPFamily",
37               "CmdExitFailure",
38               "bkg", "cmd", "bpftool", "bpftrace", "defer", "ethtool",
39               "fd_read_timeout", "ip", "rand_port",
40               "wait_port_listen", "wait_file", "tool",
41               "KsftSkipEx", "KsftFailEx", "KsftXfailEx",
42               "ksft_disruptive", "ksft_exit", "ksft_pr", "ksft_run",
43               "ksft_setup", "ksft_variants", "KsftNamedVariant",
44               "ksft_eq", "ksft_ge", "ksft_in", "ksft_is", "ksft_lt",
45               "ksft_ne", "ksft_not_in", "ksft_raises", "ksft_true", "ksft_gt",
46               "ksft_not_none", "ksft_not_none",
47               "NetDrvEnv", "NetDrvEpEnv", "GenerateTraffic", "Remote",
48               "Iperf3Runner"]
49except ModuleNotFoundError as e:
50    print("Failed importing `net` library from kernel sources")
51    print(str(e))
52    sys.exit(4)