diff options
-rw-r--r-- | src/test/java/com/google/devtools/build/lib/BUILD | 3 | ||||
-rw-r--r-- | src/test/java/com/google/devtools/build/lib/bazel/repository/skylark/SkylarkRepositoryContextTest.java | 99 |
2 files changed, 102 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 9a9f8caac7..ff5c22a5bc 100644 --- a/src/test/java/com/google/devtools/build/lib/BUILD +++ b/src/test/java/com/google/devtools/build/lib/BUILD @@ -1042,7 +1042,10 @@ java_test( "//src/main/java/com/google/devtools/build/lib:bazel-repository", "//src/main/java/com/google/devtools/build/lib:build-base", "//src/main/java/com/google/devtools/build/lib:collect", + "//src/main/java/com/google/devtools/build/lib:events", "//src/main/java/com/google/devtools/build/lib:inmemoryfs", + "//src/main/java/com/google/devtools/build/lib:packages-internal", + "//src/main/java/com/google/devtools/build/lib:syntax", "//src/main/java/com/google/devtools/build/lib:vfs", "//src/main/java/com/google/devtools/build/skyframe", "//third_party:guava", diff --git a/src/test/java/com/google/devtools/build/lib/bazel/repository/skylark/SkylarkRepositoryContextTest.java b/src/test/java/com/google/devtools/build/lib/bazel/repository/skylark/SkylarkRepositoryContextTest.java new file mode 100644 index 0000000000..2e782ef1cc --- /dev/null +++ b/src/test/java/com/google/devtools/build/lib/bazel/repository/skylark/SkylarkRepositoryContextTest.java @@ -0,0 +1,99 @@ +// Copyright 2014 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.bazel.repository.skylark; + +import static com.google.common.truth.Truth.assertThat; + +import com.google.common.collect.ImmutableList; +import com.google.common.collect.ImmutableMap; +import com.google.devtools.build.lib.events.Location; +import com.google.devtools.build.lib.packages.Attribute; +import com.google.devtools.build.lib.packages.Package; +import com.google.devtools.build.lib.packages.Rule; +import com.google.devtools.build.lib.packages.RuleClass; +import com.google.devtools.build.lib.packages.RuleClass.Builder.RuleClassType; +import com.google.devtools.build.lib.syntax.Argument.Passed; +import com.google.devtools.build.lib.syntax.BuiltinFunction; +import com.google.devtools.build.lib.syntax.FuncallExpression; +import com.google.devtools.build.lib.syntax.Identifier; +import com.google.devtools.build.lib.syntax.Type; +import com.google.devtools.build.lib.testutil.Scratch; +import com.google.devtools.build.lib.vfs.Path; + +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.junit.runners.JUnit4; + +import java.util.Map; + +/** + * Unit tests for complex function of SkylarkRepositoryContext. + */ +@RunWith(JUnit4.class) +public class SkylarkRepositoryContextTest { + + private Scratch scratch; + private Path outputDirectory; + private Path workspaceFile; + private SkylarkRepositoryContext context; + + @Before + public void setUp() throws Exception { + scratch = new Scratch("/"); + outputDirectory = scratch.dir("/outputDir"); + workspaceFile = scratch.file("/WORKSPACE"); + } + + protected static RuleClass buildRuleClass(Attribute... attributes) { + RuleClass.Builder ruleClassBuilder = + new RuleClass.Builder("test", RuleClassType.WORKSPACE, true); + for (Attribute attr : attributes) { + ruleClassBuilder.addOrOverrideAttribute(attr); + } + ruleClassBuilder.setWorkspaceOnly(); + ruleClassBuilder.setConfiguredTargetFunction(new BuiltinFunction("test") {}); + return ruleClassBuilder.build(); + } + + protected void setUpContextForRule(Map<String, Object> kwargs, Attribute... attributes) + throws Exception { + Package.LegacyBuilder packageBuilder = + Package.newExternalPackageBuilder(workspaceFile, "runfiles"); + FuncallExpression ast = + new FuncallExpression(new Identifier("test"), ImmutableList.<Passed>of()); + ast.setLocation(Location.BUILTIN); + Rule rule = + packageBuilder + .externalPackageData() + .createAndAddRepositoryRule( + packageBuilder, buildRuleClass(attributes), null, kwargs, ast); + context = new SkylarkRepositoryContext(rule, outputDirectory); + } + + protected void setUpContexForRule(String name) throws Exception { + setUpContextForRule(ImmutableMap.<String, Object>of("name", name)); + } + + @Test + public void testAttr() throws Exception { + setUpContextForRule( + ImmutableMap.<String, Object>of("name", "test", "foo", "bar"), + Attribute.attr("foo", Type.STRING).build()); + + assertThat(context.getAttr().getKeys()).contains("foo"); + assertThat(context.getAttr().getValue("foo")).isEqualTo("bar"); + } +} |