| // Copyright 2018 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.starlark; |
| |
| import static com.google.common.truth.Truth.assertThat; |
| import static org.junit.Assert.assertThrows; |
| |
| import com.google.common.collect.ImmutableList; |
| import com.google.common.collect.ImmutableMap; |
| import com.google.devtools.build.lib.events.Event; |
| import com.google.devtools.build.lib.events.ExtendedEventHandler; |
| import com.google.devtools.build.lib.events.ExtendedEventHandler.Postable; |
| import com.google.devtools.build.lib.pkgcache.TargetParsingCompleteEvent; |
| import com.google.devtools.build.lib.runtime.StarlarkOptionsParser; |
| import com.google.devtools.build.lib.starlark.util.StarlarkOptionsTestCase; |
| import com.google.devtools.build.lib.util.Pair; |
| import com.google.devtools.common.options.OptionsParsingException; |
| import com.google.devtools.common.options.OptionsParsingResult; |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.stream.Collectors; |
| import net.starlark.java.eval.StarlarkInt; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.JUnit4; |
| |
| /** Unit test for the {@code StarlarkOptionsParser}. */ |
| @RunWith(JUnit4.class) |
| public class StarlarkOptionsParsingTest extends StarlarkOptionsTestCase { |
| |
| private List<Postable> postedEvents; |
| |
| @Before |
| public void addPostableEventHandler() { |
| postedEvents = new ArrayList<>(); |
| reporter.addHandler( |
| new ExtendedEventHandler() { |
| @Override |
| public void post(Postable obj) { |
| postedEvents.add(obj); |
| } |
| |
| @Override |
| public void handle(Event event) {} |
| }); |
| } |
| |
| /** Returns only the posted events of the given class. */ |
| private List<Postable> eventsOfType(Class<? extends Postable> clazz) { |
| return postedEvents.stream() |
| .filter(event -> event.getClass().equals(clazz)) |
| .collect(Collectors.toList()); |
| } |
| |
| // test --flag=value |
| @Test |
| public void testFlagEqualsValueForm() throws Exception { |
| writeBasicIntFlag(); |
| |
| OptionsParsingResult result = parseStarlarkOptions("--//test:my_int_setting=666"); |
| |
| assertThat(result.getStarlarkOptions()).hasSize(1); |
| assertThat(result.getStarlarkOptions().get("//test:my_int_setting")) |
| .isEqualTo(StarlarkInt.of(666)); |
| assertThat(result.getResidue()).isEmpty(); |
| } |
| |
| // test --@workspace//flag=value |
| @Test |
| public void testFlagNameWithWorkspace() throws Exception { |
| writeBasicIntFlag(); |
| rewriteWorkspace("workspace(name = 'starlark_options_test')"); |
| |
| OptionsParsingResult result = |
| parseStarlarkOptions("--@starlark_options_test//test:my_int_setting=666"); |
| |
| assertThat(result.getStarlarkOptions()).hasSize(1); |
| assertThat(result.getStarlarkOptions().get("@starlark_options_test//test:my_int_setting")) |
| .isEqualTo(StarlarkInt.of(666)); |
| assertThat(result.getResidue()).isEmpty(); |
| } |
| |
| // test --fake_flag=value |
| @Test |
| public void testBadFlag_equalsForm() throws Exception { |
| scratch.file("test/BUILD"); |
| reporter.removeHandler(failFastHandler); |
| |
| OptionsParsingException e = |
| assertThrows( |
| OptionsParsingException.class, |
| () -> parseStarlarkOptions("--//fake_flag=blahblahblah")); |
| |
| assertThat(e).hasMessageThat().contains("Error loading option //fake_flag"); |
| assertThat(e.getInvalidArgument()).isEqualTo("//fake_flag"); |
| } |
| |
| // test --fake_flag |
| @Test |
| public void testBadFlag_boolForm() throws Exception { |
| scratch.file("test/BUILD"); |
| reporter.removeHandler(failFastHandler); |
| |
| OptionsParsingException e = |
| assertThrows(OptionsParsingException.class, () -> parseStarlarkOptions("--//fake_flag")); |
| |
| assertThat(e).hasMessageThat().contains("Error loading option //fake_flag"); |
| assertThat(e.getInvalidArgument()).isEqualTo("//fake_flag"); |
| } |
| |
| @Test |
| public void testBadFlag_keepGoing() throws Exception { |
| optionsParser.parse("--keep_going"); |
| scratch.file("test/BUILD"); |
| reporter.removeHandler(failFastHandler); |
| |
| OptionsParsingException e = |
| assertThrows(OptionsParsingException.class, () -> parseStarlarkOptions("--//fake_flag")); |
| |
| assertThat(e).hasMessageThat().contains("Error loading option //fake_flag"); |
| assertThat(e.getInvalidArgument()).isEqualTo("//fake_flag"); |
| } |
| |
| @Test |
| public void testSingleDash_notAllowed() throws Exception { |
| writeBasicIntFlag(); |
| |
| OptionsParsingResult result = parseStarlarkOptions("-//test:my_int_setting=666"); |
| |
| assertThat(result.getStarlarkOptions()).isEmpty(); |
| assertThat(result.getResidue()).containsExactly("-//test:my_int_setting=666"); |
| } |
| |
| // test --non_flag_setting=value |
| @Test |
| public void testNonFlagParsing() throws Exception { |
| scratch.file( |
| "test/build_setting.bzl", |
| "def _build_setting_impl(ctx):", |
| " return []", |
| "int_flag = rule(", |
| " implementation = _build_setting_impl,", |
| " build_setting = config.int(flag=False)", |
| ")"); |
| scratch.file( |
| "test/BUILD", |
| "load('//test:build_setting.bzl', 'int_flag')", |
| "int_flag(name = 'my_int_setting', build_setting_default = 42)"); |
| |
| OptionsParsingException e = |
| assertThrows( |
| OptionsParsingException.class, |
| () -> parseStarlarkOptions("--//test:my_int_setting=666")); |
| |
| assertThat(e).hasMessageThat().isEqualTo("Unrecognized option: //test:my_int_setting=666"); |
| } |
| |
| // test --bool_flag |
| @Test |
| public void testBooleanFlag() throws Exception { |
| writeBasicBoolFlag(); |
| |
| OptionsParsingResult result = parseStarlarkOptions("--//test:my_bool_setting=false"); |
| |
| assertThat(result.getStarlarkOptions()).hasSize(1); |
| assertThat(result.getStarlarkOptions().get("//test:my_bool_setting")).isEqualTo(false); |
| assertThat(result.getResidue()).isEmpty(); |
| } |
| |
| // test --nobool_flag |
| @Test |
| public void testNoPrefixedBooleanFlag() throws Exception { |
| writeBasicBoolFlag(); |
| |
| OptionsParsingResult result = parseStarlarkOptions("--no//test:my_bool_setting"); |
| |
| assertThat(result.getStarlarkOptions()).hasSize(1); |
| assertThat(result.getStarlarkOptions().get("//test:my_bool_setting")).isEqualTo(false); |
| assertThat(result.getResidue()).isEmpty(); |
| } |
| |
| // test --noint_flag |
| @Test |
| public void testNoPrefixedNonBooleanFlag() throws Exception { |
| writeBasicIntFlag(); |
| |
| OptionsParsingException e = |
| assertThrows( |
| OptionsParsingException.class, () -> parseStarlarkOptions("--no//test:my_int_setting")); |
| |
| assertThat(e) |
| .hasMessageThat() |
| .isEqualTo("Illegal use of 'no' prefix on non-boolean option: //test:my_int_setting"); |
| } |
| |
| // test --int_flag |
| @Test |
| public void testFlagWithoutValue() throws Exception { |
| writeBasicIntFlag(); |
| |
| OptionsParsingException e = |
| assertThrows( |
| OptionsParsingException.class, () -> parseStarlarkOptions("--//test:my_int_setting")); |
| |
| assertThat(e).hasMessageThat().isEqualTo("Expected value after --//test:my_int_setting"); |
| } |
| |
| // test --flag --flag |
| @Test |
| public void testRepeatFlagLastOneWins() throws Exception { |
| writeBasicIntFlag(); |
| |
| OptionsParsingResult result = |
| parseStarlarkOptions("--//test:my_int_setting=4 --//test:my_int_setting=7"); |
| |
| assertThat(result.getStarlarkOptions()).hasSize(1); |
| assertThat(result.getStarlarkOptions().get("//test:my_int_setting")) |
| .isEqualTo(StarlarkInt.of(7)); |
| assertThat(result.getResidue()).isEmpty(); |
| } |
| |
| // test --flagA=valueA --flagB=valueB |
| @Test |
| public void testMultipleFlags() throws Exception { |
| scratch.file( |
| "test/build_setting.bzl", |
| "def _build_setting_impl(ctx):", |
| " return []", |
| "int_flag = rule(", |
| " implementation = _build_setting_impl,", |
| " build_setting = config.int(flag=True)", |
| ")"); |
| scratch.file( |
| "test/BUILD", |
| "load('//test:build_setting.bzl', 'int_flag')", |
| "int_flag(name = 'my_int_setting', build_setting_default = 42)", |
| "int_flag(name = 'my_other_int_setting', build_setting_default = 77)"); |
| |
| OptionsParsingResult result = |
| parseStarlarkOptions("--//test:my_int_setting=0 --//test:my_other_int_setting=0"); |
| |
| assertThat(result.getResidue()).isEmpty(); |
| assertThat(result.getStarlarkOptions()).hasSize(2); |
| assertThat(result.getStarlarkOptions().get("//test:my_int_setting")) |
| .isEqualTo(StarlarkInt.of(0)); |
| assertThat(result.getStarlarkOptions().get("//test:my_other_int_setting")) |
| .isEqualTo(StarlarkInt.of(0)); |
| } |
| |
| // test --non_build_setting |
| @Test |
| public void testNonBuildSetting() throws Exception { |
| scratch.file( |
| "test/rules.bzl", |
| "def _impl(ctx):", |
| " return []", |
| "my_rule = rule(", |
| " implementation = _impl,", |
| ")"); |
| scratch.file("test/BUILD", "load('//test:rules.bzl', 'my_rule')", "my_rule(name = 'my_rule')"); |
| OptionsParsingException e = |
| assertThrows(OptionsParsingException.class, () -> parseStarlarkOptions("--//test:my_rule")); |
| assertThat(e).hasMessageThat().isEqualTo("Unrecognized option: //test:my_rule"); |
| } |
| |
| // test --non_rule_configured_target |
| @Test |
| public void testNonRuleConfiguredTarget() throws Exception { |
| scratch.file( |
| "test/BUILD", |
| "genrule(", |
| " name = 'my_gen',", |
| " srcs = ['x.in'],", |
| " outs = ['x.cc'],", |
| " cmd = '$(locations :tool) $< >$@',", |
| " tools = [':tool'],", |
| ")", |
| "cc_library(name = 'tool-dep')"); |
| OptionsParsingException e = |
| assertThrows(OptionsParsingException.class, () -> parseStarlarkOptions("--//test:x.in")); |
| assertThat(e).hasMessageThat().isEqualTo("Unrecognized option: //test:x.in"); |
| } |
| |
| // test --int_flag=non_int_value |
| @Test |
| public void testWrongValueType_int() throws Exception { |
| writeBasicIntFlag(); |
| |
| OptionsParsingException e = |
| assertThrows( |
| OptionsParsingException.class, |
| () -> parseStarlarkOptions("--//test:my_int_setting=woohoo")); |
| |
| assertThat(e) |
| .hasMessageThat() |
| .isEqualTo("While parsing option //test:my_int_setting=woohoo: 'woohoo' is not a int"); |
| } |
| |
| // test --bool_flag=non_bool_value |
| @Test |
| public void testWrongValueType_bool() throws Exception { |
| writeBasicBoolFlag(); |
| |
| OptionsParsingException e = |
| assertThrows( |
| OptionsParsingException.class, |
| () -> parseStarlarkOptions("--//test:my_bool_setting=woohoo")); |
| |
| assertThat(e) |
| .hasMessageThat() |
| .isEqualTo("While parsing option //test:my_bool_setting=woohoo: 'woohoo' is not a boolean"); |
| } |
| |
| // test --int-flag=same value as default |
| @Test |
| public void testDontStoreDefaultValue() throws Exception { |
| // build_setting_default = 42 |
| writeBasicIntFlag(); |
| |
| OptionsParsingResult result = parseStarlarkOptions("--//test:my_int_setting=42"); |
| |
| assertThat(result.getStarlarkOptions()).isEmpty(); |
| } |
| |
| @Test |
| public void testOptionsAreParsedWithBuildTestsOnly() throws Exception { |
| writeBasicIntFlag(); |
| optionsParser.parse("--build_tests_only"); |
| |
| OptionsParsingResult result = parseStarlarkOptions("--//test:my_int_setting=15"); |
| |
| assertThat(result.getStarlarkOptions().get("//test:my_int_setting")) |
| .isEqualTo(StarlarkInt.of(15)); |
| } |
| |
| @Test |
| public void testRemoveStarlarkOptionsWorks() throws Exception { |
| Pair<ImmutableList<String>, ImmutableList<String>> residueAndStarlarkOptions = |
| StarlarkOptionsParser.removeStarlarkOptions( |
| ImmutableList.of( |
| "--//local/starlark/option", |
| "--@some_repo//external/starlark/option", |
| "--@//main/repo/option", |
| "some-random-residue", |
| "--mangled//external/starlark/option")); |
| assertThat(residueAndStarlarkOptions.getFirst()) |
| .containsExactly( |
| "--//local/starlark/option", |
| "--@some_repo//external/starlark/option", |
| "--@//main/repo/option"); |
| assertThat(residueAndStarlarkOptions.getSecond()) |
| .containsExactly("some-random-residue", "--mangled//external/starlark/option"); |
| } |
| |
| /** |
| * When Starlark flags are only set as flags, they shouldn't produce {@link |
| * TargetParsingCompleteEvent}s. That's intended to communicate (to the build event protocol) |
| * which of the targets in {@code blaze build //foo:all //bar:all} were built. |
| */ |
| @Test |
| public void testExpectedBuildEventOutput_asFlag() throws Exception { |
| writeBasicIntFlag(); |
| scratch.file("blah/BUILD", "cc_library(name = 'mylib')"); |
| useConfiguration(ImmutableMap.of("//test:my_int_setting", "15")); |
| update( |
| ImmutableList.of("//blah:mylib"), |
| /*keepGoing=*/ false, |
| /*loadingPhaseThreads=*/ LOADING_PHASE_THREADS, |
| /*doAnalysis*/ true, |
| eventBus); |
| List<Postable> targetParsingCompleteEvents = eventsOfType(TargetParsingCompleteEvent.class); |
| assertThat(targetParsingCompleteEvents).hasSize(1); |
| assertThat( |
| ((TargetParsingCompleteEvent) targetParsingCompleteEvents.get(0)) |
| .getOriginalTargetPattern()) |
| .containsExactly("//blah:mylib"); |
| } |
| |
| /** |
| * But Starlark are also targets. When they're requested as normal build targets they should |
| * produce {@link TargetParsingCompleteEvent} just like any other target. |
| */ |
| @Test |
| public void testExpectedBuildEventOutput_asTarget() throws Exception { |
| writeBasicIntFlag(); |
| scratch.file("blah/BUILD", "cc_library(name = 'mylib')"); |
| useConfiguration(ImmutableMap.of("//test:my_int_setting", "15")); |
| update( |
| ImmutableList.of("//blah:mylib", "//test:my_int_setting"), |
| /*keepGoing=*/ false, |
| /*loadingPhaseThreads=*/ LOADING_PHASE_THREADS, |
| /*doAnalysis*/ true, |
| eventBus); |
| List<Postable> targetParsingCompleteEvents = eventsOfType(TargetParsingCompleteEvent.class); |
| assertThat(targetParsingCompleteEvents).hasSize(1); |
| assertThat( |
| ((TargetParsingCompleteEvent) targetParsingCompleteEvents.get(0)) |
| .getOriginalTargetPattern()) |
| .containsExactly("//blah:mylib", "//test:my_int_setting"); |
| } |
| } |