mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 21:00:13 +01:00
Richard Smith
e090a0ef65
A check for `args.verbose` was missing for one line of autoupdate's verbose output. Add a wrapper function to print verbose output to fix this and to make it easier to get verbose output right in future.
125 lines
3.7 KiB
Python
Executable File
125 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env -S uv run --script
|
|
|
|
# /// script
|
|
# requires-python = ">=3.12"
|
|
# ///
|
|
|
|
"""Autoupdates testdata in toolchain."""
|
|
|
|
__copyright__ = """
|
|
Part of the Carbon Language project, under the Apache License v2.0 with LLVM
|
|
Exceptions. See /LICENSE for license information.
|
|
SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
"""
|
|
|
|
import argparse
|
|
import re
|
|
import shlex
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def main() -> None:
|
|
# Parse arguments.
|
|
parser = argparse.ArgumentParser(__doc__)
|
|
parser.add_argument("--non-fatal-checks", action="store_true")
|
|
parser.add_argument(
|
|
"--print_slowest_tests", default=0, help="Forwarded to file_test"
|
|
)
|
|
parser.add_argument("--threads", help="Forwarded to file_test")
|
|
parser.add_argument(
|
|
"--verbose", "-v", action="store_true", help="Produce verbose output"
|
|
)
|
|
parser.add_argument("files", nargs="*")
|
|
args = parser.parse_args()
|
|
|
|
printv = print if args.verbose else lambda _: None
|
|
|
|
bazel = str(Path(__file__).parents[1] / "scripts" / "run_bazel.py")
|
|
configs = []
|
|
# Use the most recently used build mode, or `fastbuild` if missing
|
|
# `bazel-bin`.
|
|
build_mode = "fastbuild"
|
|
workspace = subprocess.check_output(
|
|
[
|
|
bazel,
|
|
"info",
|
|
"workspace",
|
|
"--ui_event_filters=stdout",
|
|
],
|
|
encoding="utf-8",
|
|
).strip()
|
|
bazel_bin_path = Path(workspace).joinpath("bazel-bin")
|
|
if bazel_bin_path.exists():
|
|
link = str(bazel_bin_path.readlink())
|
|
m = re.search(r"-(\w+)/bin$", link)
|
|
if m:
|
|
build_mode = m[1]
|
|
printv(f"Detected --compilation_mode: {build_mode}")
|
|
else:
|
|
exit(f"Build mode not found in `bazel-bin` symlink: {link}")
|
|
else:
|
|
printv(
|
|
"Detected --compilation_mode: none (no `./bazel-bin`), "
|
|
+ f"falling back to {build_mode}"
|
|
)
|
|
|
|
if args.non_fatal_checks:
|
|
if build_mode == "optimize":
|
|
exit(
|
|
"`--non-fatal-checks` is incompatible with inferred "
|
|
"`-c optimize` build mode"
|
|
)
|
|
configs.append("--config=non-fatal-checks")
|
|
|
|
argv = [
|
|
bazel,
|
|
"run",
|
|
"-c",
|
|
build_mode,
|
|
*configs,
|
|
"--experimental_convenience_symlinks=ignore",
|
|
"--ui_event_filters=-info,-stdout,-stderr,-finish",
|
|
"//toolchain/testing:file_test",
|
|
"--",
|
|
"--autoupdate",
|
|
"--print_slowest_tests",
|
|
str(args.print_slowest_tests),
|
|
]
|
|
if args.threads:
|
|
argv += ["--threads", args.threads]
|
|
# Support specifying tests to update, such as:
|
|
# ./autoupdate_testdata.py lex/**/*
|
|
if args.files:
|
|
repo_root = Path(__file__).parents[1]
|
|
file_tests = []
|
|
# Filter down to just test files.
|
|
for f in args.files:
|
|
if f.endswith(".carbon"):
|
|
path = str(Path(f).resolve().relative_to(repo_root))
|
|
if path.count("/testdata/"):
|
|
file_tests.append(path)
|
|
if not file_tests:
|
|
sys.exit(
|
|
"Args do not seem to be test files; for example, "
|
|
f"{args.files[0]}"
|
|
)
|
|
argv.append("--file_tests=" + ",".join(file_tests))
|
|
printv(shlex.join(argv))
|
|
# Provide an empty stdin so that the driver tests that read from stdin
|
|
# don't block waiting for input. This matches the behavior of `bazel test`.
|
|
result = subprocess.run(argv)
|
|
if result.returncode != 0:
|
|
sys.exit(
|
|
f"Command `{shlex.join(argv)}` failed with exit code "
|
|
f"`{result.returncode}`"
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
sys.exit(1)
|