__init__.py

 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)