blob: 4c3b413152e3f6ae977c1fe4d3dbd85072062faa [file] [log] [blame]
# Copyright 2021 The Bazel Authors. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""cc_test Starlark implementation."""
load(":common/cc/cc_binary.bzl", "cc_binary_impl")
load(":common/cc/cc_binary_attrs.bzl", "cc_binary_attrs")
load(":common/cc/cc_helper.bzl", "cc_helper")
load(":common/cc/semantics.bzl", "semantics")
load(":common/paths.bzl", "paths")
cc_internal = _builtins.internal.cc_internal
config_common = _builtins.toplevel.config_common
platform_common = _builtins.toplevel.platform_common
testing = _builtins.toplevel.testing
_CC_TEST_TOOLCHAIN_TYPE = "@" + semantics.get_repo() + "//tools/cpp:test_runner_toolchain_type"
def _cc_test_impl(ctx):
binary_info, providers = cc_binary_impl(ctx, [])
test_env = {}
test_env.update(cc_helper.get_expanded_env(ctx, {}))
coverage_runfiles, coverage_env = semantics.get_coverage_env(ctx)
runfiles_list = [binary_info.runfiles]
if coverage_runfiles:
runfiles_list.append(coverage_runfiles)
runfiles = ctx.runfiles()
runfiles = runfiles.merge_all(runfiles_list)
test_env.update(coverage_env)
providers.append(testing.TestEnvironment(
environment = test_env,
inherited_environment = ctx.attr.env_inherit,
))
providers.append(DefaultInfo(
files = binary_info.files,
runfiles = runfiles,
executable = binary_info.executable,
))
if cc_helper.has_target_constraints(ctx, ctx.attr._apple_constraints):
# When built for Apple platforms, require the execution to be on a Mac.
providers.append(testing.ExecutionInfo({"requires-darwin": ""}))
return providers
def _impl(ctx):
cc_test_toolchain = ctx.exec_groups["test"].toolchains[_CC_TEST_TOOLCHAIN_TYPE]
if cc_test_toolchain:
cc_test_info = cc_test_toolchain.cc_test_info
else:
# This is the "legacy" cc_test flow
return _cc_test_impl(ctx)
binary_info, providers = cc_binary_impl(ctx, cc_test_info.linkopts)
processed_environment = cc_helper.get_expanded_env(ctx, {})
test_providers = cc_test_info.get_runner.func(
ctx,
binary_info,
processed_environment = processed_environment,
**cc_test_info.get_runner.args
)
providers.extend(test_providers)
return providers
_cc_test_attrs = dict(cc_binary_attrs)
# Update cc_test defaults:
_cc_test_attrs.update(
_is_test = attr.bool(default = True),
_apple_constraints = attr.label_list(
default = [
"@" + paths.join(semantics.get_platforms_root(), "os:ios"),
"@" + paths.join(semantics.get_platforms_root(), "os:macos"),
"@" + paths.join(semantics.get_platforms_root(), "os:tvos"),
"@" + paths.join(semantics.get_platforms_root(), "os:watchos"),
],
),
# Starlark tests don't get `env_inherit` by default.
env_inherit = attr.string_list(),
stamp = attr.int(values = [-1, 0, 1], default = 0),
linkstatic = attr.bool(default = False),
)
_cc_test_attrs.update(semantics.get_test_malloc_attr())
_cc_test_attrs.update(semantics.get_coverage_attrs())
cc_test = rule(
implementation = _impl,
attrs = _cc_test_attrs,
outputs = {
# TODO(b/198254254): Handle case for windows.
"stripped_binary": "%{name}.stripped",
"dwp_file": "%{name}.dwp",
},
fragments = ["cpp", "coverage"] + semantics.additional_fragments(),
exec_groups = {
"cpp_link": exec_group(toolchains = cc_helper.use_cpp_toolchain()),
# testing.ExecutionInfo defaults to an exec_group of "test".
"test": exec_group(toolchains = [config_common.toolchain_type(_CC_TEST_TOOLCHAIN_TYPE, mandatory = False)]),
},
toolchains = [] +
cc_helper.use_cpp_toolchain() +
semantics.get_runtimes_toolchain(),
test = True,
)