blob: 06e68ba701afde2972ddbfb9ae73a7bdc833bf5d [file] [log] [blame]
// 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.syntax;
import static com.google.common.truth.Truth.assertThat;
import static com.google.devtools.build.lib.testutil.MoreAsserts.assertThrows;
import com.google.devtools.build.lib.skylarkinterface.SkylarkPrinter;
import com.google.devtools.build.lib.skylarkinterface.SkylarkValue;
import java.util.List;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
/** Tests for {@link Runtime}. */
@RunWith(JUnit4.class)
public final class RuntimeTest {
private static final BuiltinFunction DUMMY_FUNC = new BuiltinFunction("dummyFunc") {
// This would normally be done by @SkylarkSignature annotation and configure(), but a simple
// stub suffices.
@Override
public Class<?> getObjectType() {
return DummyType.class;
}
};
private static class DummyType implements SkylarkValue {
@Override
public void repr(SkylarkPrinter printer) {
printer.append("DummyType");
}
}
@Test
public void checkRegistry_GetBuiltins() {
Object dummy = new Object();
Runtime.BuiltinRegistry reg = new Runtime.BuiltinRegistry();
reg.registerBuiltin(DummyType.class, "dummy", dummy);
assertThat(reg.getBuiltins()).contains(dummy);
}
@Test
public void checkRegistry_GetFunction() {
Runtime.BuiltinRegistry reg = new Runtime.BuiltinRegistry();
reg.registerFunction(DummyType.class, DUMMY_FUNC);
assertThat(reg.getFunction(DummyType.class, "dummyFunc")).isEqualTo(DUMMY_FUNC);
}
@Test
public void checkRegistry_GetFunctionNames() {
Runtime.BuiltinRegistry reg = new Runtime.BuiltinRegistry();
reg.registerFunction(DummyType.class, DUMMY_FUNC);
assertThat(reg.getFunctionNames(DummyType.class)).contains("dummyFunc");
}
/** Ensures that we still register all builtins, even when some are equal to one another. */
@Test
public void checkRegistry_EqualBuiltinsDontClash() {
// Create two distinct objects that compare equal under Object#equals. Use toCharArray() to
// not worry about whether the JVM does string interning.
String equalValue1 = "abc";
String equalValue2 = new String(equalValue1.toCharArray());
Runtime.BuiltinRegistry reg = new Runtime.BuiltinRegistry();
reg.registerBuiltin(DummyType.class, "eq1", equalValue1);
reg.registerBuiltin(DummyType.class, "eq2", equalValue2);
List<Object> values = reg.getBuiltins();
assertThat(values).hasSize(2);
assertThat(values.get(0)).isSameInstanceAs(equalValue1);
assertThat(values.get(1)).isSameInstanceAs(equalValue2);
}
@Test
public void checkRegistry_WriteAfterFreezeFails_Builtin() {
Runtime.BuiltinRegistry reg = new Runtime.BuiltinRegistry();
reg.freeze();
IllegalStateException expected = assertThrows(
IllegalStateException.class,
() -> reg.registerBuiltin(DummyType.class, "dummy", "foo"));
assertThat(expected).hasMessageThat()
.matches("Attempted to register builtin '(.*)DummyType.dummy' after registry has already "
+ "been frozen");
}
@Test
public void checkRegistry_WriteAfterFreezeFails_Function() {
Runtime.BuiltinRegistry reg = new Runtime.BuiltinRegistry();
reg.freeze();
IllegalStateException expected = assertThrows(
IllegalStateException.class,
() -> reg.registerFunction(DummyType.class, DUMMY_FUNC));
assertThat(expected).hasMessageThat()
.matches("Attempted to register function 'dummyFunc' in namespace '(.*)DummyType' after "
+ "registry has already been frozen");
}
}