Move to using "well-known" SkyFunctionNames in tests exercising graph -- MOS_MIGRATED_REVID=105539869
diff --git a/src/test/java/com/google/devtools/build/skyframe/GraphConcurrencyTest.java b/src/test/java/com/google/devtools/build/skyframe/GraphConcurrencyTest.java index 62d5fad..3eb94a3 100644 --- a/src/test/java/com/google/devtools/build/skyframe/GraphConcurrencyTest.java +++ b/src/test/java/com/google/devtools/build/skyframe/GraphConcurrencyTest.java
@@ -47,8 +47,7 @@ /** Base class for concurrency sanity tests on {@link EvaluableGraph} implementations. */ public abstract class GraphConcurrencyTest { - private static final SkyFunctionName SKY_FUNCTION_NAME = - SkyFunctionName.create("GraphConcurrencyTestKey"); + private static final SkyFunctionName SKY_FUNCTION_NAME = SkyFunctionName.FOR_TESTING; private ProcessableGraph graph; private TestRunnableWrapper wrapper;
diff --git a/src/test/java/com/google/devtools/build/skyframe/GraphTester.java b/src/test/java/com/google/devtools/build/skyframe/GraphTester.java index 04d76db..f7c2da4 100644 --- a/src/test/java/com/google/devtools/build/skyframe/GraphTester.java +++ b/src/test/java/com/google/devtools/build/skyframe/GraphTester.java
@@ -45,7 +45,7 @@ */ public class GraphTester { - public static final SkyFunctionName NODE_TYPE = SkyFunctionName.create("Type"); + public static final SkyFunctionName NODE_TYPE = SkyFunctionName.FOR_TESTING; private final Map<SkyKey, TestFunction> values = new HashMap<>(); private final Set<SkyKey> modifiedValues = new LinkedHashSet<>();
diff --git a/src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java b/src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java index 550fea0..a265c76 100644 --- a/src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java +++ b/src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java
@@ -654,7 +654,7 @@ EvaluationResult<StringValue> result = tester.eval(/*keep_going=*/false, values); for (int i = 0; i < values.length; i++) { - SkyValue actual = result.get(new SkyKey(GraphTester.NODE_TYPE, values[i])); + SkyValue actual = result.get(toSkyKey(values[i])); assertEquals(new StringValue("leaf"), actual); } @@ -663,7 +663,7 @@ tester.invalidate(); result = tester.eval(/*keep_going=*/false, values); for (int i = 0; i < values.length; i++) { - SkyValue actual = result.get(new SkyKey(GraphTester.NODE_TYPE, values[i])); + SkyValue actual = result.get(toSkyKey(values[i])); assertEquals("Run " + j + ", value " + i, new StringValue("other" + j), actual); } } @@ -679,7 +679,7 @@ tester.set(values[i], new StringValue(values[i])); expected.append(values[i]); } - SkyKey rootKey = new SkyKey(GraphTester.NODE_TYPE, "root"); + SkyKey rootKey = toSkyKey("root"); TestFunction value = tester.getOrCreate(rootKey) .setComputedValue(CONCATENATE); for (int i = 0; i < values.length; i++) { @@ -928,7 +928,7 @@ .that(errorInfo.getCycleInfo()) .containsExactly( new CycleInfo(ImmutableList.of(top), ImmutableList.of(cycleKey1, cycleKey2))); - assertThat(errorInfo.getException()).hasMessage("Type:errorKey"); + assertThat(errorInfo.getException()).hasMessage(NODE_TYPE.getName() + ":errorKey"); assertThat(errorInfo.getRootCauseOfException()).isEqualTo(errorKey); } @@ -1895,7 +1895,7 @@ tester.set(values[i], new StringValue(valueName)); expected.append(valueName); } - SkyKey topKey = new SkyKey(GraphTester.NODE_TYPE, "top"); + SkyKey topKey = toSkyKey("top"); TestFunction value = tester.getOrCreate(topKey) .setComputedValue(CONCATENATE); for (int i = 0; i < values.length; i++) { @@ -3145,7 +3145,9 @@ tester.evalAndGet("value"); Assert.fail("injection over value with deps should have failed"); } catch (IllegalStateException e) { - assertEquals("existing entry for Type:value has deps: [Type:other]", e.getMessage()); + assertThat(e).hasMessage( + "existing entry for " + NODE_TYPE.getName() + ":value has deps: " + + "[" + NODE_TYPE.getName() + ":other]"); } } @@ -3162,7 +3164,9 @@ tester.evalAndGet("value"); Assert.fail("injection over value with deps should have failed"); } catch (IllegalStateException e) { - assertEquals("existing entry for Type:value has deps: [Type:other]", e.getMessage()); + assertThat(e).hasMessage( + "existing entry for " + NODE_TYPE.getName() + ":value has deps: " + + "[" + NODE_TYPE.getName() + ":other]"); } } @@ -3648,7 +3652,7 @@ public SkyValue evalAndGet(boolean keepGoing, String key) throws InterruptedException { - return evalAndGet(keepGoing, new SkyKey(NODE_TYPE, key)); + return evalAndGet(keepGoing, toSkyKey(key)); } public SkyValue evalAndGet(String key) throws InterruptedException { @@ -3671,7 +3675,7 @@ } public ErrorInfo evalAndGetError(String key) throws InterruptedException { - return evalAndGetError(new SkyKey(NODE_TYPE, key)); + return evalAndGetError(toSkyKey(key)); } @Nullable @@ -3681,7 +3685,7 @@ @Nullable public SkyValue getExistingValue(String key) { - return getExistingValue(new SkyKey(NODE_TYPE, key)); + return getExistingValue(toSkyKey(key)); } } }