1 files changed, 118 insertions, 0 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/skyframe/packages/AbstractPackageLoaderTest.java b/src/test/java/com/google/devtools/build/lib/skyframe/packages/AbstractPackageLoaderTest.java
new file mode 100644
index 0000000000..40deb5089d
--- /dev/null
+++ b/src/test/java/com/google/devtools/build/lib/skyframe/packages/AbstractPackageLoaderTest.java
@@ -0,0 +1,118 @@
+// Copyright 2017 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.skyframe.packages;
+
+import static com.google.common.truth.Truth.assertThat;
+import static com.google.devtools.build.lib.testutil.MoreAsserts.assertContainsEvent;
+import static com.google.devtools.build.lib.testutil.MoreAsserts.assertNoEvents;
+import static org.junit.Assert.fail;
+
+import com.google.common.base.Joiner;
+import com.google.common.eventbus.EventBus;
+import com.google.devtools.build.lib.cmdline.PackageIdentifier;
+import com.google.devtools.build.lib.events.Reporter;
+import com.google.devtools.build.lib.events.StoredEventHandler;
+import com.google.devtools.build.lib.packages.NoSuchPackageException;
+import com.google.devtools.build.lib.packages.Package;
+import com.google.devtools.build.lib.vfs.FileSystem;
+import com.google.devtools.build.lib.vfs.FileSystemUtils;
+import com.google.devtools.build.lib.vfs.Path;
+import com.google.devtools.build.lib.vfs.PathFragment;
+import com.google.devtools.build.lib.vfs.inmemoryfs.InMemoryFileSystem;
+import org.junit.Before;
+import org.junit.Test;
+
+/** Abstract base class of a unit test for a {@link AbstractPackageLoader} implementation. */
+public abstract class AbstractPackageLoaderTest {
+  private Path pkgRoot;
+  protected StoredEventHandler handler;
+  protected PackageLoader pkgLoader;
+
+  @Before
+  public final void init() throws Exception {
+    FileSystem fs = new InMemoryFileSystem();
+    pkgRoot = fs.getRootDirectory().getChild("pkgRoot");
+    FileSystemUtils.createDirectoryAndParents(pkgRoot);
+    Reporter reporter = new Reporter(new EventBus());
+    handler = new StoredEventHandler();
+    reporter.addHandler(handler);
+    pkgLoader = makeFreshBuilder(pkgRoot).setReporter(reporter).build();
+  }
+
+  protected abstract AbstractPackageLoader.Builder makeFreshBuilder(Path pkgRoot);
+
+  @Test
+  public void simpleNoPackage() throws Exception {
+    PackageIdentifier pkgId = PackageIdentifier.createInMainRepo(PathFragment.create("nope"));
+    try {
+      pkgLoader.loadPackage(pkgId);
+      fail();
+    } catch (NoSuchPackageException expected) {
+      assertThat(expected)
+          .hasMessageThat()
+          .isEqualTo("no such package 'nope': BUILD file not found on package path");
+    }
+    assertNoEvents(handler.getEvents());
+  }
+
+  @Test
+  public void simpleBadPackage() throws Exception {
+    file("bad/BUILD", "invalidBUILDsyntax");
+    PackageIdentifier pkgId = PackageIdentifier.createInMainRepo(PathFragment.create("bad"));
+    Package badPkg = pkgLoader.loadPackage(pkgId);
+    assertThat(badPkg.containsErrors()).isTrue();
+    assertContainsEvent(badPkg.getEvents(), "invalidBUILDsyntax");
+    assertContainsEvent(handler.getEvents(), "invalidBUILDsyntax");
+  }
+
+  @Test
+  public void simpleGoodPackage() throws Exception {
+    file("good/BUILD", "sh_library(name = 'good')");
+    PackageIdentifier pkgId = PackageIdentifier.createInMainRepo(PathFragment.create("good"));
+    Package goodPkg = pkgLoader.loadPackage(pkgId);
+    assertThat(goodPkg.containsErrors()).isFalse();
+    assertThat(
+        goodPkg.getTarget("good").getAssociatedRule().getRuleClass()).isEqualTo("sh_library");
+    assertNoEvents(goodPkg.getEvents());
+    assertNoEvents(handler.getEvents());
+  }
+
+  @Test
+  public void simpleGoodPackage_Skylark() throws Exception {
+    file("good/good.bzl",
+        "def f(x):",
+        "  native.sh_library(name = x)");
+    file("good/BUILD",
+        "load('//good:good.bzl', 'f')",
+        "f('good')");
+    PackageIdentifier pkgId = PackageIdentifier.createInMainRepo(PathFragment.create("good"));
+    Package goodPkg = pkgLoader.loadPackage(pkgId);
+    assertThat(goodPkg.containsErrors()).isFalse();
+    assertThat(
+        goodPkg.getTarget("good").getAssociatedRule().getRuleClass()).isEqualTo("sh_library");
+    assertNoEvents(goodPkg.getEvents());
+    assertNoEvents(handler.getEvents());
+  }
+
+  protected Path path(String rootRelativePath) {
+    return pkgRoot.getRelative(PathFragment.create(rootRelativePath));
+  }
+
+  protected Path file(String fileName, String... contents) throws Exception {
+    Path path = path(fileName);
+    FileSystemUtils.createDirectoryAndParents(path.getParentDirectory());
+    FileSystemUtils.writeContentAsLatin1(path, Joiner.on("\n").join(contents));
+    return path;
+  }
+}