crosvm/infra/recipes/health_check.py

# Copyright 2022 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from recipe_engine.post_process import Filter

PYTHON_VERSION_COMPATIBILITY = "PY3"

DEPS = [
    "crosvm",
    "recipe_engine/raw_io",
    "recipe_engine/buildbucket",
    "recipe_engine/context",
    "recipe_engine/properties",
    "recipe_engine/step",
]


def RunSteps(api):
    with api.crosvm.container_build_context():
        api.step(
            "Self-test dev-container",
            [
                "vpython3",
                api.crosvm.source_dir.join("tools/dev_container"),
                "--verbose",
                "--self-test",
            ],
        )
        result = api.step(
            "List checks to run",
            [
                "vpython3",
                api.crosvm.source_dir.join("tools/health-check"),
                "--list-checks",
            ],
            stdout=api.raw_io.output(),
        )
        check_list = result.stdout.strip().decode("utf-8").split("\n")
        for check in check_list:
            api.crosvm.step_in_container(
                "Checking %s" % check, ["./tools/health-check", "--all", check]
            )

        api.crosvm.step_in_container("Checking mdbook", ["mdbook", "build", "docs/book/"])
        api.crosvm.step_in_container(
            "Checking cargo docs",
            ["./tools/cargo-doc"],
        )


def GenTests(api):
    yield (
        api.test(
            "basic",
            api.buildbucket.ci_build(project="crosvm/crosvm"),
            # Provide a fake response to --list-checks
            api.step_data(
                "List checks to run",
                stdout=api.raw_io.output("a\nb"),
            ),
        )
        + api.post_process(Filter().include_re(r"Checking.*"))
    )