diff options
Diffstat (limited to 'src/test/java/com/google/devtools/build/lib/skylarkdebug/server/DebugServerTransportTest.java')
-rw-r--r-- | src/test/java/com/google/devtools/build/lib/skylarkdebug/server/DebugServerTransportTest.java | 18 |
1 files changed, 16 insertions, 2 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/skylarkdebug/server/DebugServerTransportTest.java b/src/test/java/com/google/devtools/build/lib/skylarkdebug/server/DebugServerTransportTest.java index 89a0405cd4..2dcdf2543e 100644 --- a/src/test/java/com/google/devtools/build/lib/skylarkdebug/server/DebugServerTransportTest.java +++ b/src/test/java/com/google/devtools/build/lib/skylarkdebug/server/DebugServerTransportTest.java @@ -16,6 +16,7 @@ package com.google.devtools.build.lib.skylarkdebug.server; import static com.google.common.truth.Truth.assertThat; +import com.google.devtools.build.lib.events.EventHandler; import com.google.devtools.build.lib.events.EventKind; import com.google.devtools.build.lib.events.util.EventCollectionApparatus; import com.google.devtools.build.lib.skylarkdebugging.SkylarkDebuggingProtos.ContinueExecutionResponse; @@ -47,6 +48,17 @@ public class DebugServerTransportTest { private final EventCollectionApparatus events = new EventCollectionApparatus(EventKind.ALL_EVENTS); + /** + * The default fail-fast {@link EventHandler} fails on debug-level events, so we instead use a + * handler suppressing those events. + */ + private final EventHandler eventHandler = + event -> { + if (event.getKind() != EventKind.DEBUG) { + events.reporter().handle(event); + } + }; + /** A simple debug client for testing purposes. */ private static class MockDebugClient { @@ -87,7 +99,8 @@ public class DebugServerTransportTest { ServerSocket serverSocket = new ServerSocket(0, 1, InetAddress.getByName(null)); Future<DebugServerTransport> future = executor.submit( - () -> DebugServerTransport.createAndWaitForClient(events.reporter(), serverSocket)); + () -> + DebugServerTransport.createAndWaitForClient(eventHandler, serverSocket)); MockDebugClient client = new MockDebugClient(); client.connect(Duration.ofSeconds(10), serverSocket); @@ -109,7 +122,8 @@ public class DebugServerTransportTest { ServerSocket serverSocket = new ServerSocket(0, 1, InetAddress.getByName(null)); Future<DebugServerTransport> future = executor.submit( - () -> DebugServerTransport.createAndWaitForClient(events.reporter(), serverSocket)); + () -> + DebugServerTransport.createAndWaitForClient(eventHandler, serverSocket)); MockDebugClient client = new MockDebugClient(); client.connect(Duration.ofSeconds(10), serverSocket); |