aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/test/java/com/google/devtools/build/skyframe
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/java/com/google/devtools/build/skyframe')
-rw-r--r--src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java67
-rw-r--r--src/test/java/com/google/devtools/build/skyframe/ParallelEvaluatorTest.java11
2 files changed, 29 insertions, 49 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 df97dc6b06..f3cf5a83d0 100644
--- a/src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java
+++ b/src/test/java/com/google/devtools/build/skyframe/MemoizingEvaluatorTest.java
@@ -15,9 +15,7 @@ package com.google.devtools.build.skyframe;
import static com.google.common.truth.Truth.assertThat;
import static com.google.common.truth.Truth.assertWithMessage;
-import static com.google.devtools.build.lib.testutil.MoreAsserts.assertContainsEvent;
-import static com.google.devtools.build.lib.testutil.MoreAsserts.assertEventCount;
-import static com.google.devtools.build.lib.testutil.MoreAsserts.assertNoEvents;
+import static com.google.devtools.build.lib.testutil.EventIterableSubjectFactory.assertThatEvents;
import static com.google.devtools.build.skyframe.ErrorInfoSubjectFactory.assertThatErrorInfo;
import static com.google.devtools.build.skyframe.EvaluationResultSubjectFactory.assertThatEvaluationResult;
import static com.google.devtools.build.skyframe.GraphTester.CONCATENATE;
@@ -150,16 +148,14 @@ public class MemoizingEvaluatorTest {
tester.set("x", new StringValue("y")).setWarning("fizzlepop");
StringValue value = (StringValue) tester.evalAndGet("x");
assertThat(value.getValue()).isEqualTo("y");
- assertContainsEvent(eventCollector, "fizzlepop");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("fizzlepop");
initializeReporter();
tester.invalidate();
value = (StringValue) tester.evalAndGet("x");
assertThat(value.getValue()).isEqualTo("y");
if (eventsStored()) {
- assertContainsEvent(eventCollector, "fizzlepop");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("fizzlepop");
}
}
@@ -443,9 +439,7 @@ public class MemoizingEvaluatorTest {
initializeReporter();
tester.evalAndGet("top");
if (i == 0 || eventsStored()) {
- assertContainsEvent(eventCollector, "warn-d1");
- assertContainsEvent(eventCollector, "warn-d2");
- assertEventCount(2, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("warn-d1", "warn-d2");
}
}
}
@@ -467,8 +461,7 @@ public class MemoizingEvaluatorTest {
.isEqualTo(topKey.toString());
assertThat(result.getError(topKey).getException()).isInstanceOf(SomeErrorException.class);
if (i == 0 || eventsStored()) {
- assertContainsEvent(eventCollector, "warn-dep");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("warn-dep");
}
}
}
@@ -489,8 +482,7 @@ public class MemoizingEvaluatorTest {
.isEqualTo(topKey.toString());
assertThat(result.getError(topKey).getException()).isInstanceOf(SomeErrorException.class);
if (i == 0 || eventsStored()) {
- assertContainsEvent(eventCollector, "warning msg");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("warning msg");
}
}
}
@@ -511,8 +503,7 @@ public class MemoizingEvaluatorTest {
.isEqualTo(topKey.toString());
assertThat(result.getError(topKey).getException()).isInstanceOf(SomeErrorException.class);
if (i == 0 || eventsStored()) {
- assertContainsEvent(eventCollector, "warning msg");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("warning msg");
}
}
}
@@ -527,8 +518,7 @@ public class MemoizingEvaluatorTest {
initializeReporter();
tester.eval(/*keepGoing=*/false, "t1", "t2");
if (i == 0 || eventsStored()) {
- assertContainsEvent(eventCollector, "look both ways before crossing");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("look both ways before crossing");
}
}
}
@@ -543,16 +533,14 @@ public class MemoizingEvaluatorTest {
initializeReporter();
tester.evalAndGetError("error-value");
if (i == 0 || eventsStored()) {
- assertContainsEvent(eventCollector, "don't chew with your mouth open");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("don't chew with your mouth open");
}
}
initializeReporter();
tester.evalAndGet("warning-value");
if (eventsStored()) {
- assertContainsEvent(eventCollector, "don't chew with your mouth open");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("don't chew with your mouth open");
}
}
@@ -566,17 +554,14 @@ public class MemoizingEvaluatorTest {
StringValue value = (StringValue) tester.evalAndGet("x");
assertThat(value.getValue()).isEqualTo("y");
- assertContainsEvent(eventCollector, "fizzlepop");
- assertContainsEvent(eventCollector, "just letting you know");
- assertEventCount(2, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("fizzlepop", "just letting you know");
if (eventsStored()) {
// On the rebuild, we only replay warning messages.
initializeReporter();
value = (StringValue) tester.evalAndGet("x");
assertThat(value.getValue()).isEqualTo("y");
- assertContainsEvent(eventCollector, "fizzlepop");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("fizzlepop");
}
}
@@ -622,8 +607,7 @@ public class MemoizingEvaluatorTest {
"just letting you know");
tester.evalAndGetError("error-value");
- assertContainsEvent(eventCollector, "just letting you know");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("just letting you know");
// Change the progress message.
tester.getOrCreate("error-value").setHasTransientError(true).setProgress(
@@ -633,15 +617,14 @@ public class MemoizingEvaluatorTest {
for (int i = 0; i < 2; i++) {
initializeReporter();
tester.evalAndGetError("error-value");
- assertNoEvents(eventCollector);
+ assertThatEvents(eventCollector).isEmpty();
}
// When invalidating errors, we should show the new progress message.
initializeReporter();
tester.invalidateTransientErrors();
tester.evalAndGetError("error-value");
- assertContainsEvent(eventCollector, "letting you know more");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("letting you know more");
}
@Test
@@ -1929,14 +1912,12 @@ public class MemoizingEvaluatorTest {
tester.set("x", new StringValue("y")).setWarning("fizzlepop");
StringValue value = (StringValue) tester.evalAndGet("x");
assertThat(value.getValue()).isEqualTo("y");
- assertContainsEvent(eventCollector, "fizzlepop");
- assertEventCount(1, eventCollector);
+ assertThatEvents(eventCollector).containsExactly("fizzlepop");
tester.invalidate();
value = (StringValue) tester.evalAndGet("x");
assertThat(value.getValue()).isEqualTo("y");
// No new events emitted.
- assertEventCount(1, eventCollector);
}
/**
@@ -2056,7 +2037,7 @@ public class MemoizingEvaluatorTest {
.that(numTopInvocations.get())
.isEqualTo(3);
}
- assertContainsEvent(eventCollector, warningText);
+ assertThatEvents(eventCollector).containsExactly(warningText);
assertThat(topSignaled.getCount()).isEqualTo(0);
assertThat(topRestartedBuild.getCount()).isEqualTo(0);
}
@@ -3511,10 +3492,11 @@ public class MemoizingEvaluatorTest {
assertStringValue(val, tester.evalAndGet( /*keepGoing=*/false, parent));
assertThat(parentEvaluated.get()).isEqualTo(1);
if (hasEvent) {
- assertContainsEvent(eventCollector, "shmoop");
+ assertThatEvents(eventCollector).containsExactly("shmoop");
} else {
- assertEventCount(0, eventCollector);
+ assertThatEvents(eventCollector).isEmpty();
}
+ eventCollector.clear();
tester.resetPlayedEvents();
tester.getOrCreate(child, /*markAsModified=*/true);
@@ -3522,9 +3504,9 @@ public class MemoizingEvaluatorTest {
assertStringValue(val, tester.evalAndGet( /*keepGoing=*/false, parent));
assertThat(parentEvaluated.get()).isEqualTo(2);
if (hasEvent) {
- assertContainsEvent(eventCollector, "shmoop");
+ assertThatEvents(eventCollector).containsExactly("shmoop");
} else {
- assertEventCount(0, eventCollector);
+ assertThatEvents(eventCollector).isEmpty();
}
}
@@ -3548,12 +3530,13 @@ public class MemoizingEvaluatorTest {
parentEvaluated, null, null, false, parentVal, ImmutableList.of(child)));
assertThat(tester.evalAndGet( /*keepGoing=*/false, parent)).isEqualTo(parentVal);
assertThat(parentEvaluated.getCount()).isEqualTo(1);
- assertContainsEvent(eventCollector, "bloop");
+ assertThatEvents(eventCollector).containsExactly("bloop");
+ eventCollector.clear();
tester.resetPlayedEvents();
tester.getOrCreate(child, /*markAsModified=*/ true);
tester.invalidate();
assertThat(tester.evalAndGet( /*keepGoing=*/false, parent)).isEqualTo(parentVal);
- assertContainsEvent(eventCollector, "bloop");
+ assertThatEvents(eventCollector).containsExactly("bloop");
assertThat(parentEvaluated.getCount()).isEqualTo(1);
}
diff --git a/src/test/java/com/google/devtools/build/skyframe/ParallelEvaluatorTest.java b/src/test/java/com/google/devtools/build/skyframe/ParallelEvaluatorTest.java
index db73be54fe..bc79489dc1 100644
--- a/src/test/java/com/google/devtools/build/skyframe/ParallelEvaluatorTest.java
+++ b/src/test/java/com/google/devtools/build/skyframe/ParallelEvaluatorTest.java
@@ -16,7 +16,7 @@ package com.google.devtools.build.skyframe;
import static com.google.common.collect.Iterables.getOnlyElement;
import static com.google.common.truth.Truth.assertThat;
import static com.google.common.truth.Truth.assertWithMessage;
-import static com.google.devtools.build.lib.testutil.MoreAsserts.assertContainsEvent;
+import static com.google.devtools.build.lib.testutil.EventIterableSubjectFactory.assertThatEvents;
import static com.google.devtools.build.skyframe.EvaluationResultSubjectFactory.assertThatEvaluationResult;
import static com.google.devtools.build.skyframe.GraphTester.CONCATENATE;
import static org.junit.Assert.fail;
@@ -412,8 +412,7 @@ public class ParallelEvaluatorTest {
set("a", "a").setWarning("warning on 'a'");
StringValue value = (StringValue) eval(false, GraphTester.toSkyKey("a"));
assertThat(value.getValue()).isEqualTo("a");
- assertContainsEvent(storedEventHandler.getEvents(), "warning on 'a'");
- assertThat(storedEventHandler.getEvents()).hasSize(1);
+ assertThatEvents(storedEventHandler.getEvents()).containsExactly("warning on 'a'");
}
/** Regression test: events from already-done value not replayed. */
@@ -498,15 +497,13 @@ public class ParallelEvaluatorTest {
evaluator.eval(ImmutableList.of(a));
assertThat(evaluated.get()).isTrue();
assertThat(storedEventHandler.getEvents()).hasSize(2);
- assertContainsEvent(storedEventHandler.getEvents(), "boop");
- assertContainsEvent(storedEventHandler.getEvents(), "beep");
+ assertThatEvents(storedEventHandler.getEvents()).containsExactly("boop", "beep");
storedEventHandler.clear();
evaluator = makeEvaluator(graph, tester.getSkyFunctionMap(), /*keepGoing=*/ false);
evaluated.set(false);
evaluator.eval(ImmutableList.of(a));
assertThat(evaluated.get()).isFalse();
- assertThat(storedEventHandler.getEvents()).hasSize(1);
- assertContainsEvent(storedEventHandler.getEvents(), "boop");
+ assertThatEvents(storedEventHandler.getEvents()).containsExactly("boop");
}
@Test