Fix Postable forwarding and replay

We were previously duplicate-posting Postable events posted to the
Skyframe environment.

PiperOrigin-RevId: 162323598

1 files changed, 9 insertions, 0 deletions

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 455a36ec44..75ebadf4c9 100644
--- a/src/test/java/com/google/devtools/build/skyframe/GraphTester.java
+++ b/src/test/java/com/google/devtools/build/skyframe/GraphTester.java
@@ -20,6 +20,7 @@ import com.google.common.collect.ImmutableMap;
 import com.google.common.collect.ImmutableSet;
 import com.google.common.collect.Iterables;
 import com.google.devtools.build.lib.events.Event;
+import com.google.devtools.build.lib.events.ExtendedEventHandler.Postable;
 import com.google.devtools.build.lib.util.Pair;
 import com.google.devtools.build.lib.util.Preconditions;
 import com.google.devtools.build.skyframe.SkyFunction.Environment;
@@ -103,6 +104,9 @@ public class GraphTester {
         if (builder.progress != null) {
           env.getListener().handle(Event.progress(builder.progress));
         }
+        if (builder.postable != null) {
+          env.getListener().post(builder.postable);
+        }
         Map<SkyKey, SkyValue> deps = new LinkedHashMap<>();
         boolean oneMissing = false;
         for (Pair<SkyKey, SkyValue> dep : builder.deps) {
@@ -175,6 +179,7 @@ public class GraphTester {
 
     private String warning;
     private String progress;
+    private Postable postable;
 
     private String tag;
 
@@ -260,6 +265,10 @@ public class GraphTester {
       return this;
     }
 
+    public TestFunction setPostable(Postable postable) {
+      this.postable = postable;
+      return this;
+    }
   }
 
   public static ImmutableList<SkyKey> toSkyKeys(String... names) {