Add a test

Author: thii

src/test/java/com/google/devtools/build/lib/rules/apple/AppleFragmentTest.java

@@ -0,0 +1,109 @@
+// Copyright 2022 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.apple;
+
+import static com.google.common.truth.Truth.assertThat;
+
+import com.google.common.collect.ImmutableMap;
+import com.google.devtools.build.lib.analysis.ConfiguredTarget;
+import com.google.devtools.build.lib.analysis.config.BuildConfigurationValue;
+import com.google.devtools.build.lib.analysis.util.BuildViewTestCase;
+import com.google.devtools.build.lib.cmdline.Label;
+import com.google.devtools.build.lib.packages.Provider;
+import com.google.devtools.build.lib.packages.StarlarkProvider;
+import com.google.devtools.build.lib.packages.StructImpl;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+/** Tests for the action properties on rule configured targets of Apple related rules. */
+@RunWith(JUnit4.class)
+public class AppleFragmentTest extends BuildViewTestCase  {
+
+  @Before
+  public void setup() throws Exception {
+    scratch.file(
+        "rules.bzl",
+        "MyInfo = provider()",
+        "def _my_binary_impl(ctx):",
+        "    out = ctx.actions.declare_file(ctx.label.name)",
+        "    ctx.actions.write(out, '')",
+        "    return [",
+        "        DefaultInfo(executable = out),",
+        "        MyInfo(",
+        "            single_arch_cpu = ctx.fragments.apple.single_arch_cpu,",
+        "            single_arch_platform = ctx.fragments.apple.single_arch_platform,",
+        "        ),",
+        "    ]",
+        "my_binary = rule(",
+        "    fragments = ['apple'],",
+        "    implementation = _my_binary_impl,",
+        ")",
+        "def _my_rule_impl(ctx):",
+        "    out = ctx.actions.declare_file(ctx.label.name)",
+        "    ctx.actions.run(",
+        "        executable = ctx.executable._tool,",
+        "        arguments = [out.path],",
+        "        outputs = [out],",
+        "    )",
+        "    return ctx.attr._tool[MyInfo]",
+        "my_rule = rule(",
+        "    _my_rule_impl,",
+        "    attrs = {",
+        "        '_tool': attr.label(",
+        "            cfg = 'exec',",
+        "            executable = True,",
+        "            default = ('//:bin'),",
+        "        ),",
+        "    },",
+        ")");
+    scratch.file(
+        "BUILD",
+        "load(':rules.bzl', 'my_binary', 'my_rule')",
+        "my_binary(name = 'bin')",
+        "my_rule(name = 'a')",
+        "platform(",
+        "    name = 'macos_arm64',",
+        "    constraint_values = [",
+        "        '@platforms//cpu:aarch64',",
+        "        '@platforms//os:osx',",
+        "    ],",
+        ")");
+    scratch.file(
+        "/workspace/platform_mappings",
+        "platforms:",
+        "  //:macos_arm64",
+        "    --cpu=darwin_arm64",
+        "flags:",
+        "  --cpu=darwin_arm64",
+        "  --apple_platform_type=macos",
+        "    //:macos_arm64");
+    invalidatePackages(false);
+  }
+
+  @Test
+  public void appleFragmentSingleArchCpu() throws Exception {
+    // Test that ctx.fragments.apple.single_arch_cpu returns the execution
+    // platform's cpu in a tool's rule context.
+    useConfiguration("--extra_execution_platforms=//:macos_arm64");
+    ConfiguredTarget configuredTarget = getConfiguredTarget("//:a");
+    Provider.Key key =
+        new StarlarkProvider.Key(
+            Label.parseAbsolute("//:rules.bzl", ImmutableMap.of()), "MyInfo");
+    StructImpl myInfo = (StructImpl) configuredTarget.get(key);
+    assertThat((String) myInfo.getValue("single_arch_cpu")).isEqualTo("arm64");
+  }
+}