"""Fake java toolchains for testing"""

load("//java/common:java_semantics.bzl", "semantics")

load("//java/toolchains:java_runtime.bzl", _java_runtime_rule = "java_runtime")

load("//java/toolchains:java_toolchain.bzl", "java_toolchain")

# buildifier: disable=function-docstring

def mock_java_toolchain(

*,

name,

singlejar = "singlejar",

javabuilder = "JavaBuilder_deploy.jar",

header_compiler = "turbine_canary_deploy.jar",

header_compiler_direct = "turbine_direct",

ijar = "ijar",

genclass = "genclass",

java_runtime = None,

oneversion = None,

oneversion_allowlist = None,

oneversion_allowlist_for_tests = None,

tags = None, # for util.helper_target

**kwargs):

if not java_runtime:

java_runtime = name + "_runtime"

_java_runtime_rule(name = java_runtime)

one_version_args = {

"oneversion": oneversion,

"oneversion_allowlist": oneversion_allowlist,

"oneversion_allowlist_for_tests": oneversion_allowlist_for_tests,

} if semantics.java_toolchain_supports_one_version else {}

java_toolchain(

name = name + "_java",

javabuilder = javabuilder,

singlejar = singlejar,

header_compiler = header_compiler,

header_compiler_direct = header_compiler_direct,

ijar = ijar,

java_runtime = java_runtime,

genclass = genclass,

tags = tags,

**(kwargs | one_version_args)

)

native.toolchain(

name = name,

toolchain = name + "_java",

toolchain_type = semantics.JAVA_TOOLCHAIN_TYPE,

tags = tags,

)

# buildifier: disable=function-docstring

def mock_java_runtime_toolchain(

*,

name,

srcs = [],

java_home = None,

java = None,

version = None,

**kwargs):

_java_runtime_rule(

name = name + "_runtime",

srcs = srcs,

java_home = java_home,

java = java,

version = version,

**kwargs

)

native.toolchain(

name = name,

toolchain = name + "_runtime",

toolchain_type = semantics.JAVA_RUNTIME_TOOLCHAIN_TYPE,

**kwargs

)