pantsbuild / pants / 19015773527

from __future__ import annotations

import json

from dataclasses import dataclass

from pants.backend.go.go_sources import load_go_binary

from pants.backend.go.go_sources.load_go_binary import LoadedGoBinaryRequest, setup_go_binary

from pants.engine.addresses import Address

from pants.engine.engine_aware import EngineAwareParameter

from pants.engine.fs import Digest, MergeDigests

from pants.engine.internals.native_engine import EMPTY_DIGEST

from pants.engine.intrinsics import execute_process, merge_digests

from pants.engine.process import Process

from pants.engine.rules import collect_rules, implicitly, rule

from pants.util.logging import LogLevel

from pants.util.ordered_set import FrozenOrderedSet

@dataclass(frozen=True)

class GenerateTestMainRequest(EngineAwareParameter):

    digest: Digest

    test_paths: FrozenOrderedSet[str]

    xtest_paths: FrozenOrderedSet[str]

    import_path: str

    register_cover: bool

    address: Address

    def debug_hint(self) -> str:

        return self.address.spec

@dataclass(frozen=True)

class GeneratedTestMain:

    digest: Digest

    has_tests: bool

    has_xtests: bool

    failed_exit_code_and_stderr: tuple[int, str] | None

    TEST_MAIN_FILE = "testmain.go"

    TEST_PKG = "_test"

    XTEST_PKG = "_xtest"

@rule

async def generate_testmain(request: GenerateTestMainRequest) -> GeneratedTestMain:

    generator_binary_name = "./generator"

    analyzer = await setup_go_binary(

    input_digest = await merge_digests(MergeDigests([request.digest, analyzer.digest]))

    test_paths = tuple(f"{GeneratedTestMain.TEST_PKG}:{path}" for path in request.test_paths)

    xtest_paths = tuple(f"{GeneratedTestMain.XTEST_PKG}:{path}" for path in request.xtest_paths)

    env = {}

    if request.register_cover:

        env["GENERATE_COVER"] = "1"

    result = await execute_process(

    if result.exit_code != 0:

        return GeneratedTestMain(

    metadata = json.loads(result.stdout.decode("utf-8"))

    return GeneratedTestMain(

def rules():

    return (*collect_rules(), *load_go_binary.rules())