Source code
Revision control
Copy as Markdown
Other Tools
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
import logging
import shutil
import subprocess
from mach.decorators import Command, CommandArgument
@Command(
"shell-test",
category="testing",
description="Run shell unit tests using shunit2.",
)
@CommandArgument("tests", nargs="+", metavar="PATH", help="Test files to run.")
def shell_test(command_context, tests):
if not _is_shunit2_on_path():
command_context.log(
logging.ERROR,
"shell-test",
{},
"shunit2 not found. Install it with: "
"apt install shunit2 (Debian/Ubuntu), "
"dnf install shunit2 (Fedora), "
"zypper install shunit2 (openSUSE), "
"or brew install shunit2 (macOS).",
)
return 1
failed = False
for path in tests:
result = subprocess.run([_read_shebang(path), path], check=False)
if result.returncode != 0:
failed = True
return 1 if failed else 0
def _read_shebang(path):
with open(path) as f:
first_line = f.readline().strip()
if first_line.startswith("#!"):
return first_line[2:].split()[0]
return "sh"
def _is_shunit2_on_path():
return shutil.which("shunit2") is not None