aboutsummaryrefslogtreecommitdiffhomepage
path: root/src/test/java/com/google/devtools/build/lib/skylarkdebug/server/DebugServerTransportTest.java
diff options
context:
space:
mode:
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.java18
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);