summaryrefslogtreecommitdiff
path: root/tools/testing/selftests/net/lib/py/ynl.py
blob: 1ace58370c06380f518b74a1e079a7b20ff0ac34 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# SPDX-License-Identifier: GPL-2.0

import sys
from pathlib import Path
from .consts import KSRC, KSFT_DIR
from .ksft import ksft_pr, ktap_result

# Resolve paths
try:
    if (KSFT_DIR / "kselftest-list.txt").exists():
        # Running in "installed" selftests
        tools_full_path = KSFT_DIR
        SPEC_PATH = KSFT_DIR / "net/lib/specs"

        sys.path.append(tools_full_path.as_posix())
        from net.lib.ynl.lib import YnlFamily, NlError
    else:
        # Running in tree
        tools_full_path = KSRC / "tools"
        SPEC_PATH = KSRC / "Documentation/netlink/specs"

        sys.path.append(tools_full_path.as_posix())
        from net.ynl.lib import YnlFamily, NlError
except ModuleNotFoundError as e:
    ksft_pr("Failed importing `ynl` library from kernel sources")
    ksft_pr(str(e))
    ktap_result(True, comment="SKIP")
    sys.exit(4)

#
# Wrapper classes, loading the right specs
# Set schema='' to avoid jsonschema validation, it's slow
#
class EthtoolFamily(YnlFamily):
    def __init__(self):
        super().__init__((SPEC_PATH / Path('ethtool.yaml')).as_posix(),
                         schema='')


class RtnlFamily(YnlFamily):
    def __init__(self):
        super().__init__((SPEC_PATH / Path('rt_link.yaml')).as_posix(),
                         schema='')


class NetdevFamily(YnlFamily):
    def __init__(self):
        super().__init__((SPEC_PATH / Path('netdev.yaml')).as_posix(),
                         schema='')