diff options
Diffstat (limited to 'src/test/java')
-rw-r--r-- | src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java | 69 |
1 files changed, 69 insertions, 0 deletions
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 3fc2e27ff7..a2fa72ce7f 100644 --- a/src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java +++ b/src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java @@ -26,6 +26,7 @@ import static org.junit.Assert.fail; import com.google.auto.value.AutoValue; import com.google.common.base.Preconditions; +import com.google.common.base.Predicate; import com.google.common.base.Predicates; import com.google.common.collect.ImmutableList; import com.google.common.collect.ImmutableMap; @@ -114,12 +115,14 @@ public class MemoizingEvaluatorTest { Differencer differencer, EvaluationProgressReceiver progressReceiver, GraphInconsistencyReceiver graphInconsistencyReceiver, + EventFilter eventFilter, boolean keepEdges) { return new InMemoryMemoizingEvaluator( functions, differencer, progressReceiver, graphInconsistencyReceiver, + eventFilter, emittedEventState, true); } @@ -4057,6 +4060,62 @@ public class MemoizingEvaluatorTest { assertThat(parentEvaluated.getCount()).isEqualTo(1); } + @Test + public void depEventPredicate() throws Exception { + if (!eventsStored()) { + return; + } + SkyKey parent = GraphTester.toSkyKey("parent"); + SkyKey excludedDep = GraphTester.toSkyKey("excludedDep"); + SkyKey includedDep = GraphTester.toSkyKey("includedDep"); + tester.setEventFilter( + new EventFilter() { + @Override + public boolean storeEventsAndPosts() { + return true; + } + + @Override + public boolean apply(Event input) { + return true; + } + + @Override + public Predicate<SkyKey> depEdgeFilterForEventsAndPosts(SkyKey primaryKey) { + if (primaryKey.equals(parent)) { + return includedDep::equals; + } else { + return Predicates.alwaysTrue(); + } + } + }); + tester.initialize(/*keepEdges=*/ true); + tester + .getOrCreate(parent) + .addDependency(excludedDep) + .addDependency(includedDep) + .setComputedValue(CONCATENATE); + tester + .getOrCreate(excludedDep) + .setWarning("excludedDep warning") + .setConstantValue(new StringValue("excludedDep")); + tester + .getOrCreate(includedDep) + .setWarning("includedDep warning") + .setConstantValue(new StringValue("includedDep")); + tester.eval(/*keepGoing=*/ false, includedDep, excludedDep); + assertThatEvents(eventCollector).containsExactly("excludedDep warning", "includedDep warning"); + eventCollector.clear(); + emittedEventState.clear(); + tester.eval(/*keepGoing=*/ true, parent); + assertThatEvents(eventCollector).containsExactly("includedDep warning"); + assertThat( + ValueWithMetadata.getEvents( + tester.driver.getEntryForTesting(parent).getValueMaybeWithMetadata())) + .containsExactly( + new TaggedEvents(null, ImmutableList.of(Event.warn("includedDep warning")))); + } + // Tests that we have a sane implementation of error transience. @Test public void errorTransienceBug() throws Exception { @@ -5055,6 +5114,7 @@ public class MemoizingEvaluatorTest { private TrackingProgressReceiver progressReceiver = new TrackingProgressReceiver(); private GraphInconsistencyReceiver graphInconsistencyReceiver = GraphInconsistencyReceiver.THROWING; + private EventFilter eventFilter = InMemoryMemoizingEvaluator.DEFAULT_STORED_EVENT_FILTER; public void initialize(boolean keepEdges) { this.differencer = getRecordingDifferencer(); @@ -5064,6 +5124,7 @@ public class MemoizingEvaluatorTest { differencer, progressReceiver, graphInconsistencyReceiver, + eventFilter, keepEdges); this.driver = getBuildDriver(evaluator); } @@ -5077,6 +5138,14 @@ public class MemoizingEvaluatorTest { } /** + * Sets the {@link #eventFilter}. {@link #initialize} must be called after this to have any + * effect. + */ + public void setEventFilter(EventFilter eventFilter) { + this.eventFilter = eventFilter; + } + + /** * Sets the {@link #graphInconsistencyReceiver}. {@link #initialize} must be called after this * to have any effect. */ |