First steps towards implementing AndroidStudioInfoAspect in Skylark.

--
MOS_MIGRATED_REVID=112042343

3 files changed, 171 insertions, 0 deletions

diff --git a/src/test/java/com/google/devtools/build/lib/BUILD b/src/test/java/com/google/devtools/build/lib/BUILD
index 46f09fe850..d2366813e9 100644
--- a/src/test/java/com/google/devtools/build/lib/BUILD
+++ b/src/test/java/com/google/devtools/build/lib/BUILD
@@ -618,6 +618,7 @@ java_test(
         "ideinfo/*.java",
     ]),
     args = ["com.google.devtools.build.lib.AllTests"],
+    resources = ["ideinfo/intellij_info.bzl"],
     tags = ["ideinfo"],
     deps = [
         ":analysis_testutil",
@@ -640,6 +641,7 @@ java_test(
         "//third_party:jsr305",
         "//third_party:junit4",
         "//third_party:mockito",
+        "//third_party:protobuf",
         "//third_party:truth",
     ],
 )
diff --git a/src/test/java/com/google/devtools/build/lib/ideinfo/IntelliJSkylarkAspectTest.java b/src/test/java/com/google/devtools/build/lib/ideinfo/IntelliJSkylarkAspectTest.java
new file mode 100644
index 0000000000..7efbb6351c
--- /dev/null
+++ b/src/test/java/com/google/devtools/build/lib/ideinfo/IntelliJSkylarkAspectTest.java
@@ -0,0 +1,103 @@
+// 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.ideinfo;
+
+import static com.google.common.truth.Truth.assertThat;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.eventbus.EventBus;
+import com.google.devtools.build.lib.actions.Action;
+import com.google.devtools.build.lib.actions.Artifact;
+import com.google.devtools.build.lib.analysis.BuildView.AnalysisResult;
+import com.google.devtools.build.lib.analysis.OutputGroupProvider;
+import com.google.devtools.build.lib.analysis.actions.FileWriteAction;
+import com.google.devtools.build.lib.analysis.util.BuildViewTestCase;
+import com.google.devtools.build.lib.collect.nestedset.NestedSet;
+import com.google.devtools.build.lib.ideinfo.androidstudio.AndroidStudioIdeInfo.RuleIdeInfo;
+import com.google.devtools.build.lib.ideinfo.androidstudio.AndroidStudioIdeInfo.RuleIdeInfo.Builder;
+import com.google.devtools.build.lib.ideinfo.androidstudio.AndroidStudioIdeInfo.RuleIdeInfo.Kind;
+import com.google.devtools.build.lib.skyframe.AspectValue;
+import com.google.protobuf.TextFormat;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+import java.io.BufferedReader;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
+import java.util.Collection;
+
+/**
+ * Tests for Skylark implementation of Android Studio info aspect
+ */
+@RunWith(JUnit4.class)
+public class IntelliJSkylarkAspectTest extends BuildViewTestCase {
+  @Before
+  public void setupBzl() throws Exception {
+    InputStream stream = IntelliJSkylarkAspectTest.class
+        .getResourceAsStream("intellij_info.bzl");
+    BufferedReader reader =
+        new BufferedReader(new InputStreamReader(stream, StandardCharsets.UTF_8));
+    String line;
+    ArrayList<String> contents = new ArrayList<>();
+    while ((line = reader.readLine()) != null) {
+      contents.add(line);
+    }
+
+    scratch.file("intellij_tools/BUILD", "# empty");
+    scratch.file("intellij_tools/intellij_info.bzl",
+        contents.toArray(new String[contents.size()]));
+  }
+
+  @Test
+  public void testSimple() throws Exception {
+    scratch.file(
+        "com/google/example/BUILD",
+        "java_library(",
+        "    name = 'simple',",
+        "    srcs = ['simple/Simple.java']",
+        ")");
+    AnalysisResult analysisResult = update(
+        ImmutableList.of("//com/google/example:simple"),
+        ImmutableList.of("intellij_tools/intellij_info.bzl%intellij_info_aspect"),
+        false,
+        LOADING_PHASE_THREADS,
+        true,
+        new EventBus()
+    );
+    Collection<AspectValue> aspects = analysisResult.getAspects();
+    assertThat(aspects).hasSize(1);
+    AspectValue aspectValue = aspects.iterator().next();
+    OutputGroupProvider provider = aspectValue.getConfiguredAspect()
+        .getProvider(OutputGroupProvider.class);
+    NestedSet<Artifact> outputGroup = provider.getOutputGroup("ide-info-text");
+    assertThat(outputGroup.toList()).hasSize(1);
+    for (Artifact artifact : outputGroup) {
+      Action generatingAction = getGeneratingAction(artifact);
+      assertThat(generatingAction).isInstanceOf(FileWriteAction.class);
+      String fileContents = ((FileWriteAction) generatingAction).getFileContents();
+      Builder builder = RuleIdeInfo.newBuilder();
+      TextFormat.getParser().merge(fileContents, builder);
+      RuleIdeInfo build = builder.build();
+      assertThat(build.getLabel()).isEqualTo("//com/google/example:simple");
+      assertThat(build.getKind()).isEqualTo(Kind.JAVA_LIBRARY);
+    }
+
+  }
+}
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
new file mode 100644
index 0000000000..251a0a2155
--- /dev/null
+++ b/src/test/java/com/google/devtools/build/lib/ideinfo/intellij_info.bzl
@@ -0,0 +1,66 @@
+# 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.
+
+_kind_to_kind_id = {
+  "android_binary"  : 0,
+  "android_library" : 1,
+  "android_test" :    2,
+  "android_roboelectric_test" : 3,
+  "java_library" : 4,
+  "java_test" : 5,
+  "java_import" : 6,
+  "java_binary" : 7,
+  "proto_library" : 8,
+  "android_sdk" : 9,
+  "java_plugin" : 10,
+}
+
+_unrecognized_rule = -1;
+
+def get_kind(target, ctx):
+  return _kind_to_kind_id.get(ctx.rule.kind, _unrecognized_rule)
+
+def _aspect_impl(target, ctx):
+  ide_info_text = set()
+  kind = get_kind(target, ctx)
+  if kind != _unrecognized_rule:
+      info = struct(
+          label = str(target.label),
+          kind = kind,
+          # build_file = ???
+      )
+      output = ctx.new_file(target.label.name + ".aswb-build.txt")
+      ctx.file_action(output, info.to_proto())
+      ide_info_text += set([output])
+  for dep in ctx.rule.attr.deps:
+    ide_info_text += dep.android_studio_info_files
+
+  return struct(
+      output_groups = {
+        "ide-info-text" : ide_info_text
+      },
+      android_studio_info_files = ide_info_text
+    )
+
+intellij_info_aspect = aspect(implementation = _aspect_impl,
+    attr_aspects = [
+          "deps",
+          "exports",
+          "_robolectric", # From android_robolectric_test
+          "_junit", # From android_robolectric_test
+          "binary_under_test", #  From android_test
+          "java_lib",# From proto_library
+          "_proto1_java_lib", # From proto_library
+    ]
+)
\ No newline at end of file