Commit b86761ff authored by Jakub Kicinski's avatar Jakub Kicinski Committed by David S. Miller
Browse files

selftests: net: add scaffolding for Netlink tests in Python



Add glue code for accessing the YNL library which lives under
tools/net and YAML spec files from under Documentation/.
Automatically figure out if tests are run in tree or not.
Since we'll want to use this library both from net and
drivers/net test targets make the library a target as well,
and automatically include it when net or drivers/net are
included. Making net/lib a target ensures that we end up
with only one copy of it, and saves us some path guessing.

Add a tiny bit of formatting support to be able to output KTAP
from the start.

Reviewed-by: default avatarPetr Machata <petrm@nvidia.com>
Signed-off-by: default avatarJakub Kicinski <kuba@kernel.org>
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent d7d6e470
Loading
Loading
Loading
Loading
+8 −1
Original line number Diff line number Diff line
@@ -116,6 +116,13 @@ TARGETS += zram
TARGETS_HOTPLUG = cpu-hotplug
TARGETS_HOTPLUG += memory-hotplug

# Networking tests want the net/lib target, include it automatically
ifneq ($(filter net,$(TARGETS)),)
ifeq ($(filter net/lib,$(TARGETS)),)
	INSTALL_DEP_TARGETS := net/lib
endif
endif

# User can optionally provide a TARGETS skiplist.  By default we skip
# BPF since it has cutting edge build time dependencies which require
# more effort to install.
@@ -245,7 +252,7 @@ ifdef INSTALL_PATH
	install -m 744 run_kselftest.sh $(INSTALL_PATH)/
	rm -f $(TEST_LIST)
	@ret=1;	\
	for TARGET in $(TARGETS); do \
	for TARGET in $(TARGETS) $(INSTALL_DEP_TARGETS); do \
		BUILD_TARGET=$$BUILD/$$TARGET;	\
		$(MAKE) OUTPUT=$$BUILD_TARGET -C $$TARGET install \
				INSTALL_PATH=$(INSTALL_PATH)/$$TARGET \
+8 −0
Original line number Diff line number Diff line
# SPDX-License-Identifier: GPL-2.0

TEST_FILES := ../../../../../Documentation/netlink/specs
TEST_FILES += ../../../../net/ynl

TEST_INCLUDES := $(wildcard py/*.py)

include ../../lib.mk
+6 −0
Original line number Diff line number Diff line
# SPDX-License-Identifier: GPL-2.0

from .consts import KSRC
from .ksft import *
from .utils import *
from .ynl import NlError, YnlFamily, EthtoolFamily, NetdevFamily, RtnlFamily
+9 −0
Original line number Diff line number Diff line
# SPDX-License-Identifier: GPL-2.0

import sys
from pathlib import Path

KSFT_DIR = (Path(__file__).parent / "../../..").resolve()
KSRC = (Path(__file__).parent / "../../../../../..").resolve()

KSFT_MAIN_NAME = Path(sys.argv[0]).with_suffix("").name
+96 −0
Original line number Diff line number Diff line
# SPDX-License-Identifier: GPL-2.0

import builtins
from .consts import KSFT_MAIN_NAME

KSFT_RESULT = None


class KsftSkipEx(Exception):
    pass


class KsftXfailEx(Exception):
    pass


def ksft_pr(*objs, **kwargs):
    print("#", *objs, **kwargs)


def ksft_eq(a, b, comment=""):
    global KSFT_RESULT
    if a != b:
        KSFT_RESULT = False
        ksft_pr("Check failed", a, "!=", b, comment)


def ksft_true(a, comment=""):
    global KSFT_RESULT
    if not a:
        KSFT_RESULT = False
        ksft_pr("Check failed", a, "does not eval to True", comment)


def ksft_in(a, b, comment=""):
    global KSFT_RESULT
    if a not in b:
        KSFT_RESULT = False
        ksft_pr("Check failed", a, "not in", b, comment)


def ksft_ge(a, b, comment=""):
    global KSFT_RESULT
    if a < b:
        KSFT_RESULT = False
        ksft_pr("Check failed", a, "<", b, comment)


def ktap_result(ok, cnt=1, case="", comment=""):
    res = ""
    if not ok:
        res += "not "
    res += "ok "
    res += str(cnt) + " "
    res += KSFT_MAIN_NAME
    if case:
        res += "." + str(case.__name__)
    if comment:
        res += " # " + comment
    print(res)


def ksft_run(cases, args=()):
    totals = {"pass": 0, "fail": 0, "skip": 0, "xfail": 0}

    print("KTAP version 1")
    print("1.." + str(len(cases)))

    global KSFT_RESULT
    cnt = 0
    for case in cases:
        KSFT_RESULT = True
        cnt += 1
        try:
            case(*args)
        except KsftSkipEx as e:
            ktap_result(True, cnt, case, comment="SKIP " + str(e))
            totals['skip'] += 1
            continue
        except KsftXfailEx as e:
            ktap_result(True, cnt, case, comment="XFAIL " + str(e))
            totals['xfail'] += 1
            continue
        except Exception as e:
            for line in str(e).split('\n'):
                ksft_pr("Exception|", line)
            ktap_result(False, cnt, case)
            totals['fail'] += 1
            continue

        ktap_result(KSFT_RESULT, cnt, case)
        totals['pass'] += 1

    print(
        f"# Totals: pass:{totals['pass']} fail:{totals['fail']} xfail:{totals['xfail']} xpass:0 skip:{totals['skip']} error:0"
    )
Loading