// Copyright 2016 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.

package com.google.devtools.build.lib.rules.cpp;

import static com.google.common.truth.Truth.assertThat;

import com.google.devtools.build.lib.actions.util.ActionsTestUtil;
import com.google.devtools.build.lib.analysis.ConfiguredTarget;
import com.google.devtools.build.lib.analysis.util.BuildViewTestCase;
import com.google.devtools.build.lib.packages.util.Crosstool.CcToolchainConfig;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

/** Tests the output groups of cc_library. */
@RunWith(JUnit4.class)
public class CppOutputGroupsTest extends BuildViewTestCase {

  @Test
  public void testStaticLibraryOnlyOutputGroups() throws Exception {
    getAnalysisMock()
        .ccSupport()
        .setupCcToolchainConfig(
            mockToolsConfig,
            CcToolchainConfig.builder().withFeatures(CppRuleClasses.SUPPORTS_DYNAMIC_LINKER));
    scratch.file("src.cc");
    scratch.file(
        "a/BUILD",
        """
        cc_library(
            name = "lib",
            srcs = ["src.cc"],
            linkstatic = 1,
            alwayslink = 0,
        )

        filegroup(
            name = "group_archive",
            srcs = [":lib"],
            output_group = "archive",
        )

        filegroup(
            name = "group_dynamic",
            srcs = [":lib"],
            output_group = "dynamic_library",
        )
        """);

    ConfiguredTarget groupArchive = getConfiguredTarget("//a:group_archive");
    ConfiguredTarget groupDynamic = getConfiguredTarget("//a:group_dynamic");

    assertThat(ActionsTestUtil.prettyArtifactNames(getFilesToBuild(groupArchive)))
        .containsExactly("a/liblib.a");
    assertThat(ActionsTestUtil.prettyArtifactNames(getFilesToBuild(groupDynamic))).isEmpty();
  }

  @Test
  public void testSharedLibraryOnlyOutputGroups() throws Exception {
    getAnalysisMock()
        .ccSupport()
        .setupCcToolchainConfig(
            mockToolsConfig,
            CcToolchainConfig.builder().withFeatures(CppRuleClasses.SUPPORTS_DYNAMIC_LINKER));
    scratch.file("src.cc");
    scratch.file(
        "a/BUILD",
        """
        cc_library(
            name = "lib",
            srcs = ["src.cc"],
            linkstatic = 1,
            alwayslink = 1,
        )

        filegroup(
            name = "group_archive",
            srcs = [":lib"],
            output_group = "archive",
        )

        filegroup(
            name = "group_dynamic",
            srcs = [":lib"],
            output_group = "dynamic_library",
        )
        """);

    ConfiguredTarget groupArchive = getConfiguredTarget("//a:group_archive");
    ConfiguredTarget groupDynamic = getConfiguredTarget("//a:group_dynamic");

    assertThat(ActionsTestUtil.prettyArtifactNames(getFilesToBuild(groupArchive)))
        .containsExactly("a/liblib.lo");
    assertThat(ActionsTestUtil.prettyArtifactNames(getFilesToBuild(groupDynamic))).isEmpty();
  }

  @Test
  public void testStaticAndDynamicLibraryOutputGroups() throws Exception {
    getAnalysisMock()
        .ccSupport()
        .setupCcToolchainConfig(
            mockToolsConfig,
            CcToolchainConfig.builder().withFeatures(CppRuleClasses.SUPPORTS_DYNAMIC_LINKER));
    scratch.file("src.cc");
    scratch.file(
        "a/BUILD",
        """
        cc_library(
            name = "lib",
            srcs = ["src.cc"],
            linkstatic = 0,
            alwayslink = 0,
        )

        filegroup(
            name = "group_archive",
            srcs = [":lib"],
            output_group = "archive",
        )

        filegroup(
            name = "group_dynamic",
            srcs = [":lib"],
            output_group = "dynamic_library",
        )
        """);

    ConfiguredTarget groupArchive = getConfiguredTarget("//a:group_archive");
    ConfiguredTarget groupDynamic = getConfiguredTarget("//a:group_dynamic");

    assertThat(ActionsTestUtil.prettyArtifactNames(getFilesToBuild(groupArchive)))
        .containsExactly("a/liblib.a");
    // If supports_interface_shared_objects is true, .ifso could also be generated.
    // So we here use contains instead containsExactly.
    assertThat(ActionsTestUtil.prettyArtifactNames(getFilesToBuild(groupDynamic)))
        .contains("a/liblib.so");
  }

  @Test
  public void testSharedAndDynamicLibraryOutputGroups() throws Exception {
    getAnalysisMock()
        .ccSupport()
        .setupCcToolchainConfig(
            mockToolsConfig,
            CcToolchainConfig.builder().withFeatures(CppRuleClasses.SUPPORTS_DYNAMIC_LINKER));
    scratch.file("src.cc");
    scratch.file(
        "a/BUILD",
        """
        cc_library(
            name = "lib",
            srcs = ["src.cc"],
            linkstatic = 0,
            alwayslink = 1,
        )

        filegroup(
            name = "group_archive",
            srcs = [":lib"],
            output_group = "archive",
        )

        filegroup(
            name = "group_dynamic",
            srcs = [":lib"],
            output_group = "dynamic_library",
        )
        """);

    ConfiguredTarget groupArchive = getConfiguredTarget("//a:group_archive");
    ConfiguredTarget groupDynamic = getConfiguredTarget("//a:group_dynamic");

    assertThat(ActionsTestUtil.prettyArtifactNames(getFilesToBuild(groupArchive)))
        .containsExactly("a/liblib.lo");
    // If supports_interface_shared_objects is true, .ifso could also be generated.
    // So we here use contains instead containsExactly.
    assertThat(ActionsTestUtil.prettyArtifactNames(getFilesToBuild(groupDynamic)))
        .contains("a/liblib.so");
  }

  @Test
  public void testModuleOutputGroups() throws Exception {
    getAnalysisMock()
        .ccSupport()
        .setupCcToolchainConfig(
            mockToolsConfig,
            CcToolchainConfig.builder().withFeatures("header_modules_feature_configuration"));
    scratch.file("header.h");
    scratch.file(
        "a/BUILD",
        """
        cc_library(
            name = "lib",
            hdrs = ["src.h"],
            features = ["header_modules"],
        )

        filegroup(
            name = "group_modules",
            srcs = [":lib"],
            output_group = "module_files",
        )
        """);

    ConfiguredTarget groupArchive = getConfiguredTarget("//a:group_modules");

    assertThat(ActionsTestUtil.prettyArtifactNames(getFilesToBuild(groupArchive)))
        .containsExactly("a/_objs/lib/lib.pcm");
  }
}