builders/java/acceptance/args.bzl

"""Module for initializing aruments by java version"""

load(":runtime.bzl", "flex_runtimes", "gae_runtimes", "gcf_runtimes")

# java8 is gen1 runtime so it's not using buildpacks
gae_runtime_versions = {n: n.replace("java", "") for n in gae_runtimes if n != "java8"}
gcf_runtime_versions = {n: n.replace("java", "") for n in gcf_runtimes}
flex_runtime_versions = {n: n.replace("java", "") for n in flex_runtimes}

# The union of all versions across all products.
gcp_runtime_versions = dict(dict(flex_runtime_versions, **gae_runtime_versions), **gcf_runtime_versions)
java_gcp_runtime_versions = [key for key in gcp_runtime_versions.keys()]

def javaargs(runImageTag = "", stack = ""):
    """Create a new key-value map of arguments for java test

    Returns:
        A key-value map of the arguments
    """
    args = {}
    for _n, version in gae_runtime_versions.items():
        args[version] = newArgs(version, runImageTag, stack)

    return args

def newArgs(version, runImageTag, stack):
    return {
        "-run-image-override": runImage(version, runImageTag, stack),
    }

def runImage(version, runImageTag, stack):
    if stack != "":
        if runImageTag != "":
            return "us-docker.pkg.dev/gae-runtimes-private/" + stack + "/runtimes/java" + version + ":" + runImageTag
        else:
            return "gcr.io/gae-runtimes/buildpacks/java" + version + "/run"

    if runImageTag != "":
        return "us-docker.pkg.dev/gae-runtimes-private/gcp/buildpacks/java" + version + "/run:" + runImageTag
    else:
        return "gcr.io/gae-runtimes/buildpacks/java" + version + "/run"