This change adds the standard 'env' attribute to the android_local_test rule.

The value of the attribute is a string dictionary. The values in the dictionary have 'make variables' expanded.

PiperOrigin-RevId: 582043784
Change-Id: I254cccafe642954aa6fb367a576708b065c0b52f

Author: A Googler

rules/android_local_test/attrs.bzl

@@ -172,6 +172,9 @@ def make_attrs(additional_aspects = [], native_libs_transition = None):
                       "added to the manifest if it does not already contain a " +
                       "superset listing.",
             ),
+            env = attr.string_dict(
+                doc = "A dictionary of environment variables set for the execution of the test. Will be subject to make variable and $(location) expansion.",
+            ),
             robolectric_properties_file = attr.string(
                 doc = "The classpath to robolectric-deps.properties file.",
                 default = "${JAVA_RUNFILES}/robolectric/bazel/robolectric-deps.properties",

rules/android_local_test/impl.bzl

@@ -310,10 +310,12 @@ def _process_stub(ctx, deploy_jar_ctx, jvm_ctx, stub_preprocess_ctx, **_unused_s
         jvm_ctx.coverage_start_class,
         merged_instr,
     )
+
     return ProviderInfo(
         name = "stub_ctx",
         value = struct(
             stub = stub,
+            providers = [testing.TestEnvironment(utils.expand_make_vars(ctx, ctx.attr.env))],
         ),
         runfiles = ctx.runfiles(
             files = runfiles,

test/rules/android_local_test/java/com/starlark_resources/BUILD

@@ -32,6 +32,24 @@ android_local_test(
     ],
 )
 
+android_local_test(
+    name = "environment",
+    srcs = ["SampleTestWithEnvSetting.java"],
+    data = ["env_location_data.txt"],
+    env = {
+        "android_local_test_test_env": "peekaboo",
+        "android_local_test_loc_env": "$(location :env_location_data.txt)",
+    },
+    manifest = "AndroidManifest.xml",
+    test_class = "com.starlark_resources.SampleTestWithEnvSetting",
+    deps = [
+        "//java/com/google/thirdparty/robolectric",
+        "//third_party/java/hamcrest:android",
+        "//third_party/java/robolectric",
+        "@rules_android_maven//:junit_junit",
+    ],
+)
+
 rule_test(
     name = "no_deps_with_resources_rule_test",
     target_under_test = ":no_deps_with_resources",

test/rules/android_local_test/java/com/starlark_resources/SampleTestWithEnvSetting.java

@@ -0,0 +1,33 @@
+/**
+ * Copyright 2021 The Bazel Authors. All rights reserved.
+ *
+ * <p>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
+ *
+ * <p>http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * <p>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.starlark_resources;
+
+import static org.hamcrest.CoreMatchers.containsString;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.Assert.assertEquals;
+
+import androidx.test.ext.junit.runners.AndroidJUnit4;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+@RunWith(AndroidJUnit4.class)
+public class SampleTestWithEnvSetting {
+
+  @Test
+  public void test() throws Exception {
+    assertEquals("peekaboo", System.getenv("android_local_test_test_env"));
+    assertThat(
+        System.getenv("android_local_test_loc_env"), containsString("env_location_data.txt"));
+  }
+}

test/rules/android_local_test/java/com/starlark_resources/env_location_data.txt

@@ -0,0 +1 @@
+This file is used to validate environment setting with $(location).