summaryrefslogtreecommitdiffstats
path: root/tests/topotests/lib/topolog.py
blob: b50167078982dc5ac038e5193f0719a4c5cce361 (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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
# SPDX-License-Identifier: ISC
#
# topolog.py
# Library of helper functions for NetDEF Topology Tests
#
# Copyright (c) 2017 by
# Network Device Education Foundation, Inc. ("NetDEF")
#

"""
Logging utilities for topology tests.

This file defines our logging abstraction.
"""

import logging
import os
import subprocess
import sys

if sys.version_info[0] > 2:
    pass
else:
    pass

try:
    from xdist import is_xdist_controller
except ImportError:

    def is_xdist_controller():
        return False


BASENAME = "topolog"

# Helper dictionary to convert Topogen logging levels to Python's logging.
DEBUG_TOPO2LOGGING = {
    "debug": logging.DEBUG,
    "info": logging.INFO,
    "output": logging.INFO,
    "warning": logging.WARNING,
    "error": logging.ERROR,
    "critical": logging.CRITICAL,
}
FORMAT = "%(asctime)s.%(msecs)03d %(levelname)s: %(name)s: %(message)s"

handlers = {}
logger = logging.getLogger("topolog")


def set_handler(l, target=None):
    if target is None:
        h = logging.NullHandler()
    else:
        if isinstance(target, str):
            h = logging.FileHandler(filename=target, mode="w")
        else:
            h = logging.StreamHandler(stream=target)
        h.setFormatter(logging.Formatter(fmt=FORMAT))
    # Don't filter anything at the handler level
    h.setLevel(logging.DEBUG)
    l.addHandler(h)
    return h


def set_log_level(l, level):
    "Set the logging level."
    # Messages sent to this logger only are created if this level or above.
    log_level = DEBUG_TOPO2LOGGING.get(level, level)
    l.setLevel(log_level)


def get_logger(name, log_level=None, target=None):
    l = logging.getLogger("{}.{}".format(BASENAME, name))

    if log_level is not None:
        set_log_level(l, log_level)

    if target is not None:
        set_handler(l, target)

    return l


# nodeid: all_protocol_startup/test_all_protocol_startup.py::test_router_running


def get_test_logdir(nodeid=None):
    """Get log directory relative pathname."""
    xdist_worker = os.getenv("PYTEST_XDIST_WORKER", "")
    mode = os.getenv("PYTEST_XDIST_MODE", "no")

    if not nodeid:
        nodeid = os.environ["PYTEST_CURRENT_TEST"].split(" ")[0]

    cur_test = nodeid.replace("[", "_").replace("]", "_")
    path, testname = cur_test.split("::")
    path = path[:-3].replace("/", ".")

    # We use different logdir paths based on how xdist is running.
    if mode == "each":
        return os.path.join(path, testname, xdist_worker)
    elif mode == "load":
        return os.path.join(path, testname)
    else:
        assert (
            mode == "no" or mode == "loadfile" or mode == "loadscope"
        ), "Unknown dist mode {}".format(mode)

        return path


def logstart(nodeid, location, rundir):
    """Called from pytest before module setup."""

    mode = os.getenv("PYTEST_XDIST_MODE", "no")
    worker = os.getenv("PYTEST_TOPOTEST_WORKER", "")

    # We only per-test log in the workers (or non-dist)
    if not worker and mode != "no":
        return

    handler_id = nodeid + worker
    assert handler_id not in handlers

    rel_log_dir = get_test_logdir(nodeid)
    exec_log_dir = os.path.join(rundir, rel_log_dir)
    subprocess.check_call(
        "mkdir -p {0} && chmod 1777 {0}".format(exec_log_dir), shell=True
    )
    exec_log_path = os.path.join(exec_log_dir, "exec.log")

    # Add test based exec log handler
    h = set_handler(logger, exec_log_path)
    handlers[handler_id] = h

    if worker:
        logger.info(
            "Logging on worker %s for %s into %s", worker, handler_id, exec_log_path
        )
    else:
        logger.info("Logging for %s into %s", handler_id, exec_log_path)


def logfinish(nodeid, location):
    """Called from pytest after module teardown."""
    # This function may not be called if pytest is interrupted.

    worker = os.getenv("PYTEST_TOPOTEST_WORKER", "")
    handler_id = nodeid + worker

    if handler_id in handlers:
        # Remove test based exec log handler
        if worker:
            logger.info("Closing logs for %s", handler_id)

        h = handlers[handler_id]
        logger.removeHandler(handlers[handler_id])
        h.flush()
        h.close()
        del handlers[handler_id]


console_handler = set_handler(logger, None)
set_log_level(logger, "debug")