Expose test "data" attribute to IDE.

This is needed to deploy android_tests to the device for testing. For
symmetry we expose it for all tests rules.

--
MOS_MIGRATED_REVID=129918247

2 files changed, 17 insertions, 2 deletions

diff --git a/src/test/java/com/google/devtools/build/lib/ideinfo/AndroidStudioInfoAspectTest.java b/src/test/java/com/google/devtools/build/lib/ideinfo/AndroidStudioInfoAspectTest.java
index 998ec257b5..b8df3e0fff 100644
--- a/src/test/java/com/google/devtools/build/lib/ideinfo/AndroidStudioInfoAspectTest.java
+++ b/src/test/java/com/google/devtools/build/lib/ideinfo/AndroidStudioInfoAspectTest.java
@@ -475,6 +475,7 @@ public class AndroidStudioInfoAspectTest extends AndroidStudioInfoAspectTestBase
         "   srcs = ['FooBarTest.java'],",
         "   size = 'large',",
         "   deps = [':foobar'],",
+        "   data = ['MyData'],",
         ")");
     Map<String, RuleIdeInfo> ruleIdeInfos = buildRuleIdeInfo(
         "//java/com/google/example:FooBarTest");
@@ -501,6 +502,13 @@ public class AndroidStudioInfoAspectTest extends AndroidStudioInfoAspectTestBase
         .isEqualTo("java/com/google/example/FooBarTest.jdeps");
 
     assertThat(testInfo.getTestInfo().getSize()).isEqualTo("large");
+    assertThat(testInfo.getTestInfo().getDataList()).containsExactly(
+        ArtifactLocation.newBuilder()
+            .setIsSource(true)
+            .setRootPath(testLegacyAswbPluginVersionCompatibility() ? "/workspace" : "")
+            .setRelativePath("java/com/google/example/MyData")
+        .build()
+    );
   }
 
   @Test
diff --git a/src/test/java/com/google/devtools/build/lib/ideinfo/intellij_info.bzl b/src/test/java/com/google/devtools/build/lib/ideinfo/intellij_info.bzl
index d3f27d2a33..ffaf80a28f 100644
--- a/src/test/java/com/google/devtools/build/lib/ideinfo/intellij_info.bzl
+++ b/src/test/java/com/google/devtools/build/lib/ideinfo/intellij_info.bzl
@@ -358,13 +358,20 @@ def build_android_rule_ide_info(target, ctx, legacy_resource_label):
   ide_resolve_files = set(jars_from_output(target.android.idl.output))
   return (android_rule_ide_info, ide_resolve_files)
 
+def _get_data(ctx):
+  """Returns a list of artifact locations from the data attribute."""
+  if not hasattr(ctx.rule.attr, "data"):
+    return []
+  return [artifact_location(f) for data in ctx.rule.attr.data for f in data.files]
+
 def build_test_info(target, ctx):
   """Build TestInfo"""
   if not is_test_rule(ctx):
     return None
   return struct_omit_none(
-           size = ctx.rule.attr.size,
-         )
+      size = ctx.rule.attr.size,
+      data = _get_data(ctx),
+  )
 
 def is_test_rule(ctx):
   kind_string = ctx.rule.kind