| // Copyright 2020 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.packages; |
| |
| import static com.google.common.truth.Truth.assertThat; |
| import static org.junit.Assert.assertThrows; |
| |
| import com.google.common.collect.ImmutableMap; |
| import com.google.common.collect.Iterables; |
| import com.google.devtools.build.lib.packages.StarlarkLibrary.SelectLibrary; |
| import java.util.List; |
| import net.starlark.java.eval.EvalException; |
| import net.starlark.java.eval.Module; |
| import net.starlark.java.eval.Mutability; |
| import net.starlark.java.eval.Starlark; |
| import net.starlark.java.eval.StarlarkInt; |
| import net.starlark.java.eval.StarlarkSemantics; |
| import net.starlark.java.eval.StarlarkThread; |
| import net.starlark.java.syntax.FileOptions; |
| import net.starlark.java.syntax.ParserInput; |
| import net.starlark.java.syntax.SyntaxError; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.JUnit4; |
| |
| /** Tests of @{code select} function and data type. */ |
| @RunWith(JUnit4.class) |
| public class SelectTest { |
| |
| private static Object eval(String expr) |
| throws SyntaxError.Exception, EvalException, InterruptedException { |
| ParserInput input = ParserInput.fromLines(expr); |
| ImmutableMap.Builder<String, Object> predeclared = ImmutableMap.builder(); |
| predeclared.putAll(StarlarkLibrary.COMMON); |
| Starlark.addMethods(predeclared, new SelectLibrary()); |
| Module module = Module.withPredeclared(StarlarkSemantics.DEFAULT, predeclared.buildOrThrow()); |
| try (Mutability mu = Mutability.create()) { |
| StarlarkThread thread = new StarlarkThread(mu, StarlarkSemantics.DEFAULT); |
| return Starlark.eval(input, FileOptions.DEFAULT, module, thread); |
| } |
| } |
| |
| private static void assertFails(String expr, String wantError) { |
| EvalException ex = assertThrows(EvalException.class, () -> eval(expr)); |
| assertThat(ex).hasMessageThat().contains(wantError); |
| } |
| |
| @Test |
| public void testSelect() throws Exception { |
| SelectorList result = (SelectorList) eval("select({'a': 1})"); |
| assertThat(((SelectorValue) Iterables.getOnlyElement(result.getElements())).getDictionary()) |
| .containsExactly("a", StarlarkInt.of(1)); |
| } |
| |
| @Test |
| public void testPlus() throws Exception { |
| SelectorList x = (SelectorList) eval("select({'foo': ['FOO'], 'bar': ['BAR']}) + []"); |
| List<Object> elements = x.getElements(); |
| assertThat(elements).hasSize(2); |
| assertThat(elements.get(0)).isInstanceOf(SelectorValue.class); |
| assertThat((Iterable<?>) elements.get(1)).isEmpty(); |
| } |
| |
| @Test |
| public void testPlusIncompatibleType() throws Exception { |
| |
| assertFails( |
| "select({'foo': ['FOO'], 'bar': ['BAR']}) + 1", |
| "Cannot combine incompatible types (select of list, int)"); |
| assertFails( |
| "select({'foo': ['FOO']}) + select({'bar': 2})", |
| "Cannot combine incompatible types (select of list, select of int)"); |
| |
| assertFails( |
| "select({'foo': ['FOO']}) + select({'bar': {'a': 'a'}})", |
| "Cannot combine incompatible types (select of list, select of dict)"); |
| assertFails( |
| "select({'bar': {'a': 'a'}}) + select({'foo': ['FOO']})", |
| "Cannot combine incompatible types (select of dict, select of list)"); |
| assertFails( |
| "['FOO'] + select({'bar': {'a': 'a'}})", |
| "Cannot combine incompatible types (list, select of dict)"); |
| assertFails( |
| "select({'bar': {'a': 'a'}}) + ['FOO']", |
| "Cannot combine incompatible types (select of dict, list)"); |
| assertFails( |
| "select({'foo': ['FOO']}) + {'a': 'a'}", "unsupported binary operation: select + dict"); |
| assertFails( |
| "{'a': 'a'} + select({'foo': ['FOO']})", "unsupported binary operation: dict + select"); |
| } |
| |
| @Test |
| public void testUnionIncompatibleType() throws Exception { |
| assertFails( |
| "select({'foo': ['FOO']}) | select({'bar': {'a': 'a'}})", |
| "Cannot combine incompatible types (select of list, select of dict)"); |
| assertFails( |
| "select({'bar': {'a': 'a'}}) | select({'foo': ['FOO']})", |
| "Cannot combine incompatible types (select of dict, select of list)"); |
| assertFails( |
| "['FOO'] | select({'bar': {'a': 'a'}})", "unsupported binary operation: list | select"); |
| assertFails( |
| "select({'bar': {'a': 'a'}}) | ['FOO']", "unsupported binary operation: select | list"); |
| assertFails( |
| "select({'foo': ['FOO']}) | {'a': 'a'}", |
| "Cannot combine incompatible types (select of list, dict)"); |
| assertFails( |
| "{'a': 'a'} | select({'foo': ['FOO']})", |
| "Cannot combine incompatible types (dict, select of list)"); |
| } |
| } |