diff options
author | 2016-11-01 16:59:46 +0000 | |
---|---|---|
committer | 2016-11-02 08:26:46 +0000 | |
commit | 6ea96f67a14f979d9edf95c844303afdd0488ed7 (patch) | |
tree | 82708340354be999a4146affbdb29e8dd3d535c6 /src/test/java/com/google/devtools/build/lib/rules/proto | |
parent | 2bd36c62696288d33504859eb780f4214b9d529b (diff) |
A version of ProtoCompileActionBuilder that uses proto_lang_toolchain() rules instead of a soup of methods.
--
MOS_MIGRATED_REVID=137835755
Diffstat (limited to 'src/test/java/com/google/devtools/build/lib/rules/proto')
-rw-r--r-- | src/test/java/com/google/devtools/build/lib/rules/proto/ProtoCompileActionBuilderTest.java | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/rules/proto/ProtoCompileActionBuilderTest.java b/src/test/java/com/google/devtools/build/lib/rules/proto/ProtoCompileActionBuilderTest.java new file mode 100644 index 0000000000..6d56d27a64 --- /dev/null +++ b/src/test/java/com/google/devtools/build/lib/rules/proto/ProtoCompileActionBuilderTest.java @@ -0,0 +1,159 @@ +// 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.rules.proto; + +import static com.google.common.truth.Truth.assertThat; +import static com.google.devtools.build.lib.collect.nestedset.Order.STABLE_ORDER; +import static com.google.devtools.build.lib.rules.proto.ProtoCompileActionBuilder.createCommandLineFromToolchains; +import static org.mockito.Mockito.mock; + +import com.google.common.base.Predicates; +import com.google.common.collect.ImmutableList; +import com.google.common.collect.ImmutableMap; +import com.google.devtools.build.lib.actions.Artifact; +import com.google.devtools.build.lib.actions.Root; +import com.google.devtools.build.lib.analysis.FilesToRunProvider; +import com.google.devtools.build.lib.analysis.TransitiveInfoCollection; +import com.google.devtools.build.lib.analysis.actions.CustomCommandLine; +import com.google.devtools.build.lib.collect.nestedset.NestedSetBuilder; +import com.google.devtools.build.lib.rules.proto.ProtoCompileActionBuilder.ToolchainInvocation; +import com.google.devtools.build.lib.util.LazyString; +import com.google.devtools.build.lib.vfs.PathFragment; +import com.google.devtools.build.lib.vfs.inmemoryfs.InMemoryFileSystem; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.junit.runners.JUnit4; + +@RunWith(JUnit4.class) +public class ProtoCompileActionBuilderTest { + + Root root = Root.asSourceRoot(new InMemoryFileSystem().getPath("/")); + + @Test + public void commandLine_basic() throws Exception { + FilesToRunProvider plugin = + new FilesToRunProvider( + ImmutableList.<Artifact>of(), + null /* runfilesSupport */, + artifact("protoc-gen-javalite.exe")); + + ProtoLangToolchainProvider toolchainNoPlugin = + ProtoLangToolchainProvider.create( + "--java_out=param1,param2:$(OUT)", + null /* pluginExecutable */, + mock(TransitiveInfoCollection.class) /* runtime */, + NestedSetBuilder.<Artifact>emptySet(STABLE_ORDER) /* blacklistedProtos */); + + ProtoLangToolchainProvider toolchainWithPlugin = + ProtoLangToolchainProvider.create( + "--$(PLUGIN_OUT)=param3,param4:$(OUT)", + plugin, + mock(TransitiveInfoCollection.class) /* runtime */, + NestedSetBuilder.<Artifact>emptySet(STABLE_ORDER) /* blacklistedProtos */); + + SupportData supportData = + SupportData.create( + Predicates.<TransitiveInfoCollection>alwaysFalse(), + ImmutableList.of(artifact("source_file.proto")), + NestedSetBuilder.create( + STABLE_ORDER, artifact("import1.proto"), artifact("import2.proto")), + null /* usedDirectDeps */, + true /* hasProtoSources */); + + CustomCommandLine cmdLine = + createCommandLineFromToolchains( + ImmutableMap.of( + "dontcare_because_no_plugin", + new ToolchainInvocation(toolchainNoPlugin, "foo.srcjar"), + "pluginName", new ToolchainInvocation(toolchainWithPlugin, "bar.srcjar")), + supportData, + true /* allowServices */, + ImmutableList.<String>of() /* protocOpts */); + + assertThat(cmdLine.arguments()) + .containsExactly( + "-Iimport1.proto=import1.proto", + "-Iimport2.proto=import2.proto", + "source_file.proto", + "--java_out=param1,param2:foo.srcjar", + "--plugin=protoc-gen-PLUGIN_pluginName=protoc-gen-javalite.exe", + "--PLUGIN_pluginName_out=param3,param4:bar.srcjar"); + } + + @Test + public void otherParameters() throws Exception { + SupportData supportData = + SupportData.create( + Predicates.<TransitiveInfoCollection>alwaysFalse(), + ImmutableList.<Artifact>of(), + NestedSetBuilder.<Artifact>emptySet(STABLE_ORDER), + null /* usedDirectDeps */, + true /* hasProtoSources */); + + CustomCommandLine cmdLine = + createCommandLineFromToolchains( + ImmutableMap.<String, ToolchainInvocation>of(), + supportData, + false /* allowServices */, + ImmutableList.of("--foo", "--bar") /* protocOpts */); + + assertThat(cmdLine.arguments()).containsAllOf("--disallow_services", "--foo", "--bar"); + } + + @Test + public void outReplacementAreLazilyEvaluated() throws Exception { + final boolean[] hasBeenCalled = new boolean[1]; + hasBeenCalled[0] = false; + + CharSequence outReplacement = + new LazyString() { + @Override + public String toString() { + hasBeenCalled[0] = true; + return "mu"; + } + }; + + ProtoLangToolchainProvider toolchain = + ProtoLangToolchainProvider.create( + "--java_out=param1,param2:$(OUT)", + null /* pluginExecutable */, + mock(TransitiveInfoCollection.class) /* runtime */, + NestedSetBuilder.<Artifact>emptySet(STABLE_ORDER) /* blacklistedProtos */); + + SupportData supportData = + SupportData.create( + Predicates.<TransitiveInfoCollection>alwaysFalse(), + ImmutableList.<Artifact>of(), + NestedSetBuilder.<Artifact>emptySet(STABLE_ORDER), + null /* usedDirectDeps */, + true /* hasProtoSources */); + + CustomCommandLine cmdLine = + createCommandLineFromToolchains( + ImmutableMap.of("pluginName", new ToolchainInvocation(toolchain, outReplacement)), + supportData, + true /* allowServices */, + ImmutableList.<String>of() /* protocOpts */); + + assertThat(hasBeenCalled[0]).isFalse(); + cmdLine.arguments(); + assertThat(hasBeenCalled[0]).isTrue(); + } + + private Artifact artifact(String path) { + return new Artifact(new PathFragment(path), root); + } +} |