mirror of
https://github.com/ilyakooo0/nixpkgs.git
synced 2024-12-29 06:45:54 +03:00
Merge pull request #96254 from Mic92/logging
This commit is contained in:
commit
4fc708567f
@ -1,19 +1,13 @@
|
|||||||
#! /somewhere/python3
|
#! /somewhere/python3
|
||||||
from contextlib import contextmanager, _GeneratorContextManager
|
|
||||||
from queue import Queue, Empty
|
|
||||||
from typing import Tuple, Any, Callable, Dict, Iterator, Optional, List
|
|
||||||
from xml.sax.saxutils import XMLGenerator
|
|
||||||
import queue
|
|
||||||
import io
|
|
||||||
import _thread
|
|
||||||
import argparse
|
import argparse
|
||||||
import atexit
|
import atexit
|
||||||
import base64
|
import base64
|
||||||
import codecs
|
import io
|
||||||
|
import logging
|
||||||
import os
|
import os
|
||||||
import pathlib
|
import pathlib
|
||||||
import ptpython.repl
|
|
||||||
import pty
|
import pty
|
||||||
|
import queue
|
||||||
import re
|
import re
|
||||||
import shlex
|
import shlex
|
||||||
import shutil
|
import shutil
|
||||||
@ -21,9 +15,12 @@ import socket
|
|||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
import tempfile
|
import tempfile
|
||||||
|
import _thread
|
||||||
import time
|
import time
|
||||||
import traceback
|
from contextlib import contextmanager
|
||||||
import unicodedata
|
from typing import Any, Callable, Dict, Iterator, List, Optional, Tuple
|
||||||
|
|
||||||
|
import ptpython.repl
|
||||||
|
|
||||||
CHAR_TO_KEY = {
|
CHAR_TO_KEY = {
|
||||||
"A": "shift-a",
|
"A": "shift-a",
|
||||||
@ -88,13 +85,17 @@ CHAR_TO_KEY = {
|
|||||||
")": "shift-0x0B",
|
")": "shift-0x0B",
|
||||||
}
|
}
|
||||||
|
|
||||||
# Forward references
|
# Forward reference
|
||||||
log: "Logger"
|
|
||||||
machines: "List[Machine]"
|
machines: "List[Machine]"
|
||||||
|
|
||||||
|
logging.basicConfig(format="%(message)s")
|
||||||
|
logger = logging.getLogger("test-driver")
|
||||||
|
logger.setLevel(logging.INFO)
|
||||||
|
|
||||||
def eprint(*args: object, **kwargs: Any) -> None:
|
|
||||||
print(*args, file=sys.stderr, **kwargs)
|
class MachineLogAdapter(logging.LoggerAdapter):
|
||||||
|
def process(self, msg: str, kwargs: Any) -> Tuple[str, Any]:
|
||||||
|
return f"{self.extra['machine']}: {msg}", kwargs
|
||||||
|
|
||||||
|
|
||||||
def make_command(args: list) -> str:
|
def make_command(args: list) -> str:
|
||||||
@ -102,8 +103,7 @@ def make_command(args: list) -> str:
|
|||||||
|
|
||||||
|
|
||||||
def create_vlan(vlan_nr: str) -> Tuple[str, str, "subprocess.Popen[bytes]", Any]:
|
def create_vlan(vlan_nr: str) -> Tuple[str, str, "subprocess.Popen[bytes]", Any]:
|
||||||
global log
|
logger.info(f"starting VDE switch for network {vlan_nr}")
|
||||||
log.log("starting VDE switch for network {}".format(vlan_nr))
|
|
||||||
vde_socket = tempfile.mkdtemp(
|
vde_socket = tempfile.mkdtemp(
|
||||||
prefix="nixos-test-vde-", suffix="-vde{}.ctl".format(vlan_nr)
|
prefix="nixos-test-vde-", suffix="-vde{}.ctl".format(vlan_nr)
|
||||||
)
|
)
|
||||||
@ -142,70 +142,6 @@ def retry(fn: Callable) -> None:
|
|||||||
raise Exception("action timed out")
|
raise Exception("action timed out")
|
||||||
|
|
||||||
|
|
||||||
class Logger:
|
|
||||||
def __init__(self) -> None:
|
|
||||||
self.logfile = os.environ.get("LOGFILE", "/dev/null")
|
|
||||||
self.logfile_handle = codecs.open(self.logfile, "wb")
|
|
||||||
self.xml = XMLGenerator(self.logfile_handle, encoding="utf-8")
|
|
||||||
self.queue: "Queue[Dict[str, str]]" = Queue()
|
|
||||||
|
|
||||||
self.xml.startDocument()
|
|
||||||
self.xml.startElement("logfile", attrs={})
|
|
||||||
|
|
||||||
def close(self) -> None:
|
|
||||||
self.xml.endElement("logfile")
|
|
||||||
self.xml.endDocument()
|
|
||||||
self.logfile_handle.close()
|
|
||||||
|
|
||||||
def sanitise(self, message: str) -> str:
|
|
||||||
return "".join(ch for ch in message if unicodedata.category(ch)[0] != "C")
|
|
||||||
|
|
||||||
def maybe_prefix(self, message: str, attributes: Dict[str, str]) -> str:
|
|
||||||
if "machine" in attributes:
|
|
||||||
return "{}: {}".format(attributes["machine"], message)
|
|
||||||
return message
|
|
||||||
|
|
||||||
def log_line(self, message: str, attributes: Dict[str, str]) -> None:
|
|
||||||
self.xml.startElement("line", attributes)
|
|
||||||
self.xml.characters(message)
|
|
||||||
self.xml.endElement("line")
|
|
||||||
|
|
||||||
def log(self, message: str, attributes: Dict[str, str] = {}) -> None:
|
|
||||||
eprint(self.maybe_prefix(message, attributes))
|
|
||||||
self.drain_log_queue()
|
|
||||||
self.log_line(message, attributes)
|
|
||||||
|
|
||||||
def enqueue(self, message: Dict[str, str]) -> None:
|
|
||||||
self.queue.put(message)
|
|
||||||
|
|
||||||
def drain_log_queue(self) -> None:
|
|
||||||
try:
|
|
||||||
while True:
|
|
||||||
item = self.queue.get_nowait()
|
|
||||||
attributes = {"machine": item["machine"], "type": "serial"}
|
|
||||||
self.log_line(self.sanitise(item["msg"]), attributes)
|
|
||||||
except Empty:
|
|
||||||
pass
|
|
||||||
|
|
||||||
@contextmanager
|
|
||||||
def nested(self, message: str, attributes: Dict[str, str] = {}) -> Iterator[None]:
|
|
||||||
eprint(self.maybe_prefix(message, attributes))
|
|
||||||
|
|
||||||
self.xml.startElement("nest", attrs={})
|
|
||||||
self.xml.startElement("head", attributes)
|
|
||||||
self.xml.characters(message)
|
|
||||||
self.xml.endElement("head")
|
|
||||||
|
|
||||||
tic = time.time()
|
|
||||||
self.drain_log_queue()
|
|
||||||
yield
|
|
||||||
self.drain_log_queue()
|
|
||||||
toc = time.time()
|
|
||||||
self.log("({:.2f} seconds)".format(toc - tic))
|
|
||||||
|
|
||||||
self.xml.endElement("nest")
|
|
||||||
|
|
||||||
|
|
||||||
class Machine:
|
class Machine:
|
||||||
def __init__(self, args: Dict[str, Any]) -> None:
|
def __init__(self, args: Dict[str, Any]) -> None:
|
||||||
if "name" in args:
|
if "name" in args:
|
||||||
@ -235,8 +171,8 @@ class Machine:
|
|||||||
self.pid: Optional[int] = None
|
self.pid: Optional[int] = None
|
||||||
self.socket = None
|
self.socket = None
|
||||||
self.monitor: Optional[socket.socket] = None
|
self.monitor: Optional[socket.socket] = None
|
||||||
self.logger: Logger = args["log"]
|
|
||||||
self.allow_reboot = args.get("allowReboot", False)
|
self.allow_reboot = args.get("allowReboot", False)
|
||||||
|
self.logger = MachineLogAdapter(logger, extra=dict(machine=self.name))
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def create_startcommand(args: Dict[str, str]) -> str:
|
def create_startcommand(args: Dict[str, str]) -> str:
|
||||||
@ -292,14 +228,6 @@ class Machine:
|
|||||||
def is_up(self) -> bool:
|
def is_up(self) -> bool:
|
||||||
return self.booted and self.connected
|
return self.booted and self.connected
|
||||||
|
|
||||||
def log(self, msg: str) -> None:
|
|
||||||
self.logger.log(msg, {"machine": self.name})
|
|
||||||
|
|
||||||
def nested(self, msg: str, attrs: Dict[str, str] = {}) -> _GeneratorContextManager:
|
|
||||||
my_attrs = {"machine": self.name}
|
|
||||||
my_attrs.update(attrs)
|
|
||||||
return self.logger.nested(msg, my_attrs)
|
|
||||||
|
|
||||||
def wait_for_monitor_prompt(self) -> str:
|
def wait_for_monitor_prompt(self) -> str:
|
||||||
assert self.monitor is not None
|
assert self.monitor is not None
|
||||||
answer = ""
|
answer = ""
|
||||||
@ -314,7 +242,7 @@ class Machine:
|
|||||||
|
|
||||||
def send_monitor_command(self, command: str) -> str:
|
def send_monitor_command(self, command: str) -> str:
|
||||||
message = ("{}\n".format(command)).encode()
|
message = ("{}\n".format(command)).encode()
|
||||||
self.log("sending monitor command: {}".format(command))
|
self.logger.info(f"sending monitor command: {command}")
|
||||||
assert self.monitor is not None
|
assert self.monitor is not None
|
||||||
self.monitor.send(message)
|
self.monitor.send(message)
|
||||||
return self.wait_for_monitor_prompt()
|
return self.wait_for_monitor_prompt()
|
||||||
@ -381,16 +309,19 @@ class Machine:
|
|||||||
return self.execute("systemctl {}".format(q))
|
return self.execute("systemctl {}".format(q))
|
||||||
|
|
||||||
def require_unit_state(self, unit: str, require_state: str = "active") -> None:
|
def require_unit_state(self, unit: str, require_state: str = "active") -> None:
|
||||||
with self.nested(
|
self.logger.info(
|
||||||
"checking if unit ‘{}’ has reached state '{}'".format(unit, require_state)
|
f"checking if unit ‘{unit}’ has reached state '{require_state}'"
|
||||||
):
|
)
|
||||||
info = self.get_unit_info(unit)
|
info = self.get_unit_info(unit)
|
||||||
state = info["ActiveState"]
|
state = info["ActiveState"]
|
||||||
if state != require_state:
|
if state != require_state:
|
||||||
raise Exception(
|
raise Exception(
|
||||||
"Expected unit ‘{}’ to to be in state ".format(unit)
|
"Expected unit ‘{}’ to to be in state ".format(unit)
|
||||||
+ "'{}' but it is in state ‘{}’".format(require_state, state)
|
+ "'{}' but it is in state ‘{}’".format(require_state, state)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def log(self, message: str) -> None:
|
||||||
|
self.logger.info(message)
|
||||||
|
|
||||||
def execute(self, command: str) -> Tuple[int, str]:
|
def execute(self, command: str) -> Tuple[int, str]:
|
||||||
self.connect()
|
self.connect()
|
||||||
@ -414,27 +345,25 @@ class Machine:
|
|||||||
"""Execute each command and check that it succeeds."""
|
"""Execute each command and check that it succeeds."""
|
||||||
output = ""
|
output = ""
|
||||||
for command in commands:
|
for command in commands:
|
||||||
with self.nested("must succeed: {}".format(command)):
|
self.logger.info(f"must succeed: {command}")
|
||||||
(status, out) = self.execute(command)
|
(status, out) = self.execute(command)
|
||||||
if status != 0:
|
if status != 0:
|
||||||
self.log("output: {}".format(out))
|
self.logger.info(f"output: {out}")
|
||||||
raise Exception(
|
raise Exception(
|
||||||
"command `{}` failed (exit code {})".format(command, status)
|
"command `{}` failed (exit code {})".format(command, status)
|
||||||
)
|
)
|
||||||
output += out
|
output += out
|
||||||
return output
|
return output
|
||||||
|
|
||||||
def fail(self, *commands: str) -> str:
|
def fail(self, *commands: str) -> str:
|
||||||
"""Execute each command and check that it fails."""
|
"""Execute each command and check that it fails."""
|
||||||
output = ""
|
output = ""
|
||||||
for command in commands:
|
for command in commands:
|
||||||
with self.nested("must fail: {}".format(command)):
|
self.logger.info(f"must fail: {command}")
|
||||||
(status, out) = self.execute(command)
|
(status, out) = self.execute(command)
|
||||||
if status == 0:
|
if status == 0:
|
||||||
raise Exception(
|
raise Exception("command `{}` unexpectedly succeeded".format(command))
|
||||||
"command `{}` unexpectedly succeeded".format(command)
|
output += out
|
||||||
)
|
|
||||||
output += out
|
|
||||||
return output
|
return output
|
||||||
|
|
||||||
def wait_until_succeeds(self, command: str) -> str:
|
def wait_until_succeeds(self, command: str) -> str:
|
||||||
@ -448,9 +377,9 @@ class Machine:
|
|||||||
status, output = self.execute(command)
|
status, output = self.execute(command)
|
||||||
return status == 0
|
return status == 0
|
||||||
|
|
||||||
with self.nested("waiting for success: {}".format(command)):
|
self.logger.info(f"waiting for success: {command}")
|
||||||
retry(check_success)
|
retry(check_success)
|
||||||
return output
|
return output
|
||||||
|
|
||||||
def wait_until_fails(self, command: str) -> str:
|
def wait_until_fails(self, command: str) -> str:
|
||||||
"""Wait until a command returns failure.
|
"""Wait until a command returns failure.
|
||||||
@ -463,21 +392,21 @@ class Machine:
|
|||||||
status, output = self.execute(command)
|
status, output = self.execute(command)
|
||||||
return status != 0
|
return status != 0
|
||||||
|
|
||||||
with self.nested("waiting for failure: {}".format(command)):
|
self.logger.info(f"waiting for failure: {command}")
|
||||||
retry(check_failure)
|
retry(check_failure)
|
||||||
return output
|
return output
|
||||||
|
|
||||||
def wait_for_shutdown(self) -> None:
|
def wait_for_shutdown(self) -> None:
|
||||||
if not self.booted:
|
if not self.booted:
|
||||||
return
|
return
|
||||||
|
|
||||||
with self.nested("waiting for the VM to power off"):
|
self.logger.info("waiting for the VM to power off")
|
||||||
sys.stdout.flush()
|
sys.stdout.flush()
|
||||||
self.process.wait()
|
self.process.wait()
|
||||||
|
|
||||||
self.pid = None
|
self.pid = None
|
||||||
self.booted = False
|
self.booted = False
|
||||||
self.connected = False
|
self.connected = False
|
||||||
|
|
||||||
def get_tty_text(self, tty: str) -> str:
|
def get_tty_text(self, tty: str) -> str:
|
||||||
status, output = self.execute(
|
status, output = self.execute(
|
||||||
@ -495,19 +424,19 @@ class Machine:
|
|||||||
def tty_matches(last: bool) -> bool:
|
def tty_matches(last: bool) -> bool:
|
||||||
text = self.get_tty_text(tty)
|
text = self.get_tty_text(tty)
|
||||||
if last:
|
if last:
|
||||||
self.log(
|
self.logger.info(
|
||||||
f"Last chance to match /{regexp}/ on TTY{tty}, "
|
f"Last chance to match /{regexp}/ on TTY{tty}, "
|
||||||
f"which currently contains: {text}"
|
f"which currently contains: {text}"
|
||||||
)
|
)
|
||||||
return len(matcher.findall(text)) > 0
|
return len(matcher.findall(text)) > 0
|
||||||
|
|
||||||
with self.nested("waiting for {} to appear on tty {}".format(regexp, tty)):
|
self.logger.info(f"waiting for {regexp} to appear on tty {tty}")
|
||||||
retry(tty_matches)
|
retry(tty_matches)
|
||||||
|
|
||||||
def send_chars(self, chars: List[str]) -> None:
|
def send_chars(self, chars: List[str]) -> None:
|
||||||
with self.nested("sending keys ‘{}‘".format(chars)):
|
self.logger.info(f"sending keys ‘{chars}‘")
|
||||||
for char in chars:
|
for char in chars:
|
||||||
self.send_key(char)
|
self.send_key(char)
|
||||||
|
|
||||||
def wait_for_file(self, filename: str) -> None:
|
def wait_for_file(self, filename: str) -> None:
|
||||||
"""Waits until the file exists in machine's file system."""
|
"""Waits until the file exists in machine's file system."""
|
||||||
@ -516,16 +445,16 @@ class Machine:
|
|||||||
status, _ = self.execute("test -e {}".format(filename))
|
status, _ = self.execute("test -e {}".format(filename))
|
||||||
return status == 0
|
return status == 0
|
||||||
|
|
||||||
with self.nested("waiting for file ‘{}‘".format(filename)):
|
self.logger.info(f"waiting for file ‘{filename}‘")
|
||||||
retry(check_file)
|
retry(check_file)
|
||||||
|
|
||||||
def wait_for_open_port(self, port: int) -> None:
|
def wait_for_open_port(self, port: int) -> None:
|
||||||
def port_is_open(_: Any) -> bool:
|
def port_is_open(_: Any) -> bool:
|
||||||
status, _ = self.execute("nc -z localhost {}".format(port))
|
status, _ = self.execute("nc -z localhost {}".format(port))
|
||||||
return status == 0
|
return status == 0
|
||||||
|
|
||||||
with self.nested("waiting for TCP port {}".format(port)):
|
self.logger.info(f"waiting for TCP port {port}")
|
||||||
retry(port_is_open)
|
retry(port_is_open)
|
||||||
|
|
||||||
def wait_for_closed_port(self, port: int) -> None:
|
def wait_for_closed_port(self, port: int) -> None:
|
||||||
def port_is_closed(_: Any) -> bool:
|
def port_is_closed(_: Any) -> bool:
|
||||||
@ -547,17 +476,17 @@ class Machine:
|
|||||||
if self.connected:
|
if self.connected:
|
||||||
return
|
return
|
||||||
|
|
||||||
with self.nested("waiting for the VM to finish booting"):
|
self.logger.info("waiting for the VM to finish booting")
|
||||||
self.start()
|
self.start()
|
||||||
|
|
||||||
tic = time.time()
|
tic = time.time()
|
||||||
self.shell.recv(1024)
|
self.shell.recv(1024)
|
||||||
# TODO: Timeout
|
# TODO: Timeout
|
||||||
toc = time.time()
|
toc = time.time()
|
||||||
|
|
||||||
self.log("connected to guest root shell")
|
self.logger.info("connected to guest root shell")
|
||||||
self.log("(connecting took {:.2f} seconds)".format(toc - tic))
|
self.logger.info(f"(connecting took {toc - tic:.2f} seconds)")
|
||||||
self.connected = True
|
self.connected = True
|
||||||
|
|
||||||
def screenshot(self, filename: str) -> None:
|
def screenshot(self, filename: str) -> None:
|
||||||
out_dir = os.environ.get("out", os.getcwd())
|
out_dir = os.environ.get("out", os.getcwd())
|
||||||
@ -566,15 +495,12 @@ class Machine:
|
|||||||
filename = os.path.join(out_dir, "{}.png".format(filename))
|
filename = os.path.join(out_dir, "{}.png".format(filename))
|
||||||
tmp = "{}.ppm".format(filename)
|
tmp = "{}.ppm".format(filename)
|
||||||
|
|
||||||
with self.nested(
|
self.logger.info(f"making screenshot {filename}")
|
||||||
"making screenshot {}".format(filename),
|
self.send_monitor_command("screendump {}".format(tmp))
|
||||||
{"image": os.path.basename(filename)},
|
ret = subprocess.run("pnmtopng {} > {}".format(tmp, filename), shell=True)
|
||||||
):
|
os.unlink(tmp)
|
||||||
self.send_monitor_command("screendump {}".format(tmp))
|
if ret.returncode != 0:
|
||||||
ret = subprocess.run("pnmtopng {} > {}".format(tmp, filename), shell=True)
|
raise Exception("Cannot convert screenshot")
|
||||||
os.unlink(tmp)
|
|
||||||
if ret.returncode != 0:
|
|
||||||
raise Exception("Cannot convert screenshot")
|
|
||||||
|
|
||||||
def copy_from_host_via_shell(self, source: str, target: str) -> None:
|
def copy_from_host_via_shell(self, source: str, target: str) -> None:
|
||||||
"""Copy a file from the host into the guest by piping it over the
|
"""Copy a file from the host into the guest by piping it over the
|
||||||
@ -650,20 +576,18 @@ class Machine:
|
|||||||
|
|
||||||
tess_args = "-c debug_file=/dev/null --psm 11 --oem 2"
|
tess_args = "-c debug_file=/dev/null --psm 11 --oem 2"
|
||||||
|
|
||||||
with self.nested("performing optical character recognition"):
|
self.logger.info("performing optical character recognition")
|
||||||
with tempfile.NamedTemporaryFile() as tmpin:
|
with tempfile.NamedTemporaryFile() as tmpin:
|
||||||
self.send_monitor_command("screendump {}".format(tmpin.name))
|
self.send_monitor_command("screendump {}".format(tmpin.name))
|
||||||
|
|
||||||
cmd = "convert {} {} tiff:- | tesseract - - {}".format(
|
cmd = "convert {} {} tiff:- | tesseract - - {}".format(
|
||||||
magick_args, tmpin.name, tess_args
|
magick_args, tmpin.name, tess_args
|
||||||
)
|
)
|
||||||
ret = subprocess.run(cmd, shell=True, capture_output=True)
|
ret = subprocess.run(cmd, shell=True, capture_output=True)
|
||||||
if ret.returncode != 0:
|
if ret.returncode != 0:
|
||||||
raise Exception(
|
raise Exception("OCR failed with exit code {}".format(ret.returncode))
|
||||||
"OCR failed with exit code {}".format(ret.returncode)
|
|
||||||
)
|
|
||||||
|
|
||||||
return ret.stdout.decode("utf-8")
|
return ret.stdout.decode("utf-8")
|
||||||
|
|
||||||
def wait_for_text(self, regex: str) -> None:
|
def wait_for_text(self, regex: str) -> None:
|
||||||
def screen_matches(last: bool) -> bool:
|
def screen_matches(last: bool) -> bool:
|
||||||
@ -671,15 +595,15 @@ class Machine:
|
|||||||
matches = re.search(regex, text) is not None
|
matches = re.search(regex, text) is not None
|
||||||
|
|
||||||
if last and not matches:
|
if last and not matches:
|
||||||
self.log("Last OCR attempt failed. Text was: {}".format(text))
|
self.logger.info(f"Last OCR attempt failed. Text was: {text}")
|
||||||
|
|
||||||
return matches
|
return matches
|
||||||
|
|
||||||
with self.nested("waiting for {} to appear on screen".format(regex)):
|
self.logger.info(f"waiting for {regex} to appear on screen")
|
||||||
retry(screen_matches)
|
retry(screen_matches)
|
||||||
|
|
||||||
def wait_for_console_text(self, regex: str) -> None:
|
def wait_for_console_text(self, regex: str) -> None:
|
||||||
self.log("waiting for {} to appear on console".format(regex))
|
self.logger.info(f"waiting for {regex} to appear on console")
|
||||||
# Buffer the console output, this is needed
|
# Buffer the console output, this is needed
|
||||||
# to match multiline regexes.
|
# to match multiline regexes.
|
||||||
console = io.StringIO()
|
console = io.StringIO()
|
||||||
@ -702,7 +626,7 @@ class Machine:
|
|||||||
if self.booted:
|
if self.booted:
|
||||||
return
|
return
|
||||||
|
|
||||||
self.log("starting vm")
|
self.logger.info("starting vm")
|
||||||
|
|
||||||
def create_socket(path: str) -> socket.socket:
|
def create_socket(path: str) -> socket.socket:
|
||||||
if os.path.exists(path):
|
if os.path.exists(path):
|
||||||
@ -759,7 +683,7 @@ class Machine:
|
|||||||
|
|
||||||
# Store last serial console lines for use
|
# Store last serial console lines for use
|
||||||
# of wait_for_console_text
|
# of wait_for_console_text
|
||||||
self.last_lines: Queue = Queue()
|
self.last_lines: queue.Queue = queue.Queue()
|
||||||
|
|
||||||
def process_serial_output() -> None:
|
def process_serial_output() -> None:
|
||||||
assert self.process.stdout is not None
|
assert self.process.stdout is not None
|
||||||
@ -767,8 +691,7 @@ class Machine:
|
|||||||
# Ignore undecodable bytes that may occur in boot menus
|
# Ignore undecodable bytes that may occur in boot menus
|
||||||
line = _line.decode(errors="ignore").replace("\r", "").rstrip()
|
line = _line.decode(errors="ignore").replace("\r", "").rstrip()
|
||||||
self.last_lines.put(line)
|
self.last_lines.put(line)
|
||||||
eprint("{} # {}".format(self.name, line))
|
self.logger.info(line)
|
||||||
self.logger.enqueue({"msg": line, "machine": self.name})
|
|
||||||
|
|
||||||
_thread.start_new_thread(process_serial_output, ())
|
_thread.start_new_thread(process_serial_output, ())
|
||||||
|
|
||||||
@ -777,10 +700,10 @@ class Machine:
|
|||||||
self.pid = self.process.pid
|
self.pid = self.process.pid
|
||||||
self.booted = True
|
self.booted = True
|
||||||
|
|
||||||
self.log("QEMU running (pid {})".format(self.pid))
|
self.logger.info(f"QEMU running (pid {self.pid})")
|
||||||
|
|
||||||
def cleanup_statedir(self) -> None:
|
def cleanup_statedir(self) -> None:
|
||||||
self.log("delete the VM state directory")
|
self.logger.info("delete the VM state directory")
|
||||||
if os.path.isfile(self.state_dir):
|
if os.path.isfile(self.state_dir):
|
||||||
shutil.rmtree(self.state_dir)
|
shutil.rmtree(self.state_dir)
|
||||||
|
|
||||||
@ -795,7 +718,7 @@ class Machine:
|
|||||||
if not self.booted:
|
if not self.booted:
|
||||||
return
|
return
|
||||||
|
|
||||||
self.log("forced crash")
|
self.logger.info("forced crash")
|
||||||
self.send_monitor_command("quit")
|
self.send_monitor_command("quit")
|
||||||
self.wait_for_shutdown()
|
self.wait_for_shutdown()
|
||||||
|
|
||||||
@ -815,8 +738,8 @@ class Machine:
|
|||||||
status, _ = self.execute("[ -e /tmp/.X11-unix/X0 ]")
|
status, _ = self.execute("[ -e /tmp/.X11-unix/X0 ]")
|
||||||
return status == 0
|
return status == 0
|
||||||
|
|
||||||
with self.nested("waiting for the X11 server"):
|
self.logger.info("waiting for the X11 server")
|
||||||
retry(check_x)
|
retry(check_x)
|
||||||
|
|
||||||
def get_window_names(self) -> List[str]:
|
def get_window_names(self) -> List[str]:
|
||||||
return self.succeed(
|
return self.succeed(
|
||||||
@ -829,15 +752,14 @@ class Machine:
|
|||||||
def window_is_visible(last_try: bool) -> bool:
|
def window_is_visible(last_try: bool) -> bool:
|
||||||
names = self.get_window_names()
|
names = self.get_window_names()
|
||||||
if last_try:
|
if last_try:
|
||||||
self.log(
|
self.logger.info(
|
||||||
"Last chance to match {} on the window list,".format(regexp)
|
f"Last chance to match {regexp} on the window list, "
|
||||||
+ " which currently contains: "
|
+ f"which currently contains: {', '.join(names)}"
|
||||||
+ ", ".join(names)
|
|
||||||
)
|
)
|
||||||
return any(pattern.search(name) for name in names)
|
return any(pattern.search(name) for name in names)
|
||||||
|
|
||||||
with self.nested("Waiting for a window to appear"):
|
self.logger.info("Waiting for a window to appear")
|
||||||
retry(window_is_visible)
|
retry(window_is_visible)
|
||||||
|
|
||||||
def sleep(self, secs: int) -> None:
|
def sleep(self, secs: int) -> None:
|
||||||
time.sleep(secs)
|
time.sleep(secs)
|
||||||
@ -865,23 +787,22 @@ class Machine:
|
|||||||
|
|
||||||
def create_machine(args: Dict[str, Any]) -> Machine:
|
def create_machine(args: Dict[str, Any]) -> Machine:
|
||||||
global log
|
global log
|
||||||
args["log"] = log
|
|
||||||
args["redirectSerial"] = os.environ.get("USE_SERIAL", "0") == "1"
|
args["redirectSerial"] = os.environ.get("USE_SERIAL", "0") == "1"
|
||||||
return Machine(args)
|
return Machine(args)
|
||||||
|
|
||||||
|
|
||||||
def start_all() -> None:
|
def start_all() -> None:
|
||||||
global machines
|
global machines
|
||||||
with log.nested("starting all VMs"):
|
logger.info("starting all VMs")
|
||||||
for machine in machines:
|
for machine in machines:
|
||||||
machine.start()
|
machine.start()
|
||||||
|
|
||||||
|
|
||||||
def join_all() -> None:
|
def join_all() -> None:
|
||||||
global machines
|
global machines
|
||||||
with log.nested("waiting for all VMs to finish"):
|
logger.info("waiting for all VMs to finish")
|
||||||
for machine in machines:
|
for machine in machines:
|
||||||
machine.wait_for_shutdown()
|
machine.wait_for_shutdown()
|
||||||
|
|
||||||
|
|
||||||
def test_script() -> None:
|
def test_script() -> None:
|
||||||
@ -892,13 +813,12 @@ def run_tests() -> None:
|
|||||||
global machines
|
global machines
|
||||||
tests = os.environ.get("tests", None)
|
tests = os.environ.get("tests", None)
|
||||||
if tests is not None:
|
if tests is not None:
|
||||||
with log.nested("running the VM test script"):
|
logger.info("running the VM test script")
|
||||||
try:
|
try:
|
||||||
exec(tests, globals())
|
exec(tests, globals())
|
||||||
except Exception as e:
|
except Exception:
|
||||||
eprint("error: ")
|
logging.exception("error:")
|
||||||
traceback.print_exc()
|
sys.exit(1)
|
||||||
sys.exit(1)
|
|
||||||
else:
|
else:
|
||||||
ptpython.repl.embed(locals(), globals())
|
ptpython.repl.embed(locals(), globals())
|
||||||
|
|
||||||
@ -911,18 +831,19 @@ def run_tests() -> None:
|
|||||||
|
|
||||||
@contextmanager
|
@contextmanager
|
||||||
def subtest(name: str) -> Iterator[None]:
|
def subtest(name: str) -> Iterator[None]:
|
||||||
with log.nested(name):
|
logger.info(name)
|
||||||
try:
|
try:
|
||||||
yield
|
yield
|
||||||
return True
|
return True
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
log.log(f'Test "{name}" failed with error: "{e}"')
|
logger.info(f'Test "{name}" failed with error: "{e}"')
|
||||||
raise e
|
raise e
|
||||||
|
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
def main() -> None:
|
||||||
|
global machines
|
||||||
arg_parser = argparse.ArgumentParser()
|
arg_parser = argparse.ArgumentParser()
|
||||||
arg_parser.add_argument(
|
arg_parser.add_argument(
|
||||||
"-K",
|
"-K",
|
||||||
@ -932,8 +853,6 @@ if __name__ == "__main__":
|
|||||||
)
|
)
|
||||||
(cli_args, vm_scripts) = arg_parser.parse_known_args()
|
(cli_args, vm_scripts) = arg_parser.parse_known_args()
|
||||||
|
|
||||||
log = Logger()
|
|
||||||
|
|
||||||
vlan_nrs = list(dict.fromkeys(os.environ.get("VLANS", "").split()))
|
vlan_nrs = list(dict.fromkeys(os.environ.get("VLANS", "").split()))
|
||||||
vde_sockets = [create_vlan(v) for v in vlan_nrs]
|
vde_sockets = [create_vlan(v) for v in vlan_nrs]
|
||||||
for nr, vde_socket, _, _ in vde_sockets:
|
for nr, vde_socket, _, _ in vde_sockets:
|
||||||
@ -944,23 +863,27 @@ if __name__ == "__main__":
|
|||||||
if not cli_args.keep_vm_state:
|
if not cli_args.keep_vm_state:
|
||||||
machine.cleanup_statedir()
|
machine.cleanup_statedir()
|
||||||
machine_eval = [
|
machine_eval = [
|
||||||
"{0} = machines[{1}]".format(m.name, idx) for idx, m in enumerate(machines)
|
"global {0}; {0} = machines[{1}]".format(m.name, idx)
|
||||||
|
for idx, m in enumerate(machines)
|
||||||
]
|
]
|
||||||
exec("\n".join(machine_eval))
|
exec("\n".join(machine_eval))
|
||||||
|
|
||||||
@atexit.register
|
@atexit.register
|
||||||
def clean_up() -> None:
|
def clean_up() -> None:
|
||||||
with log.nested("cleaning up"):
|
logger.info("cleaning up")
|
||||||
for machine in machines:
|
for machine in machines:
|
||||||
if machine.pid is None:
|
if machine.pid is None:
|
||||||
continue
|
continue
|
||||||
log.log("killing {} (pid {})".format(machine.name, machine.pid))
|
logger.info(f"killing {machine.name} (pid {machine.pid})")
|
||||||
machine.process.kill()
|
machine.process.kill()
|
||||||
for _, _, process, _ in vde_sockets:
|
for _, _, process, _ in vde_sockets:
|
||||||
process.terminate()
|
process.terminate()
|
||||||
log.close()
|
|
||||||
|
|
||||||
tic = time.time()
|
tic = time.time()
|
||||||
run_tests()
|
run_tests()
|
||||||
toc = time.time()
|
toc = time.time()
|
||||||
print("test script finished in {:.2f}s".format(toc - tic))
|
print("test script finished in {:.2f}s".format(toc - tic))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
|
@ -62,7 +62,7 @@ rec {
|
|||||||
''
|
''
|
||||||
mkdir -p $out
|
mkdir -p $out
|
||||||
|
|
||||||
LOGFILE=/dev/null tests='exec(os.environ["testScript"])' ${driver}/bin/nixos-test-driver
|
tests='exec(os.environ["testScript"])' ${driver}/bin/nixos-test-driver
|
||||||
|
|
||||||
for i in */xchg/coverage-data; do
|
for i in */xchg/coverage-data; do
|
||||||
mkdir -p $out/coverage-data
|
mkdir -p $out/coverage-data
|
||||||
|
Loading…
Reference in New Issue
Block a user