rules/linker_config_test.bzl

"""
Copyright (C) 2022 The Android Open Source Project

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.
"""

load("@bazel_skylib//lib:unittest.bzl", "analysistest", "asserts")
load("@bazel_skylib//lib:paths.bzl", "paths")
load("//build/bazel/rules:linker_config.bzl", "linker_config")
load("//build/bazel/rules:prebuilt_file.bzl", "PrebuiltFileInfo")

SRC = "foo.json"
OUT_EXP = "foo.pb"

def _test_linker_config_actions_impl(ctx):
    env = analysistest.begin(ctx)
    actions = analysistest.target_actions(env)
    asserts.equals(env, 1, len(actions), "expected  1 action got {}".format(actions))

    in_file = actions[0].inputs.to_list()[0]
    out_files = actions[0].outputs.to_list()
    asserts.equals(env, 1, len(out_files), "expected 1 out file  got {}".format(out_files))

    asserts.equals(
        env,
        SRC,
        in_file.basename,
        "expected source file {} got {}".format(SRC, in_file.basename),
    )
    asserts.equals(
        env,
        OUT_EXP,
        out_files[0].basename,
        "expected out file {} got {}".format(OUT_EXP, out_files[0].basename),
    )

    # gets build target we are testing for
    target_under_test = analysistest.target_under_test(env)
    prebuilt_file_info = target_under_test[PrebuiltFileInfo]
    asserts.equals(
        env,
        "linker.config.pb",
        prebuilt_file_info.filename,
        "expected PrebuiltFileInfo filename to be {} but  got {}".format("linkerconfig.pb", prebuilt_file_info.filename),
    )
    asserts.equals(
        env,
        "etc",
        prebuilt_file_info.dir,
        "expected PrebuiltFileInfo dir to be {} but  got {}".format("etc", prebuilt_file_info.dir),
    )
    asserts.equals(
        env,
        out_files[0],
        prebuilt_file_info.src,
        "expected PrebuiltFileInfo src to be {} but got {}".format(out_files[0], prebuilt_file_info.src),
    )

    return analysistest.end(env)

linker_config_actions_test = analysistest.make(_test_linker_config_actions_impl)

def _test_linker_config_actions():
    name = "linker_config_actions"
    test_name = name + "_test"

    linker_config(
        name = name,
        src = SRC,
        tags = ["manual"],
    )

    linker_config_actions_test(
        name = test_name,
        target_under_test = name,
    )
    return test_name

def _test_linker_config_commands_impl(ctx):
    env = analysistest.begin(ctx)
    actions = analysistest.target_actions(env)

    in_files = actions[0].inputs.to_list()
    asserts.true(env, len(in_files) > 0, "expected at least 1 input file got {}".format(in_files))

    args = actions[0].argv
    asserts.equals(env, 6, len(args), "expected 4 args got {}".format(args))
    asserts.equals(env, "proto", args[1])
    asserts.equals(env, "-s", args[2])
    asserts.equals(env, "-o", args[4])

    return analysistest.end(env)

linker_config_commands_test = analysistest.make(_test_linker_config_commands_impl)

def _test_linker_config_commands():
    name = "linker_config_commands"
    test_name = name + "_test"
    linker_config(
        name = name,
        src = SRC,
        tags = ["manual"],
    )

    linker_config_commands_test(
        name = test_name,
        target_under_test = name,
    )

    return test_name

def linker_config_test_suite(name):
    native.test_suite(
        name = name,
        tests = [
            _test_linker_config_actions(),
            _test_linker_config_commands(),
        ],
    )