test: improve reliability of mockwebserver websocket test

Signed-off-by: Marc Nuri <[email protected]>
fabric8io · Aug 29, 2024 · 3dabd69 · 3dabd69
1 parent 75dc347
commit 3dabd69
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...kwebserver/src/test/groovy/io/fabric8/mockwebserver/DefaultMockServerWebSocketTest.groovy b/...kwebserver/src/test/groovy/io/fabric8/mockwebserver/DefaultMockServerWebSocketTest.groovy
@@ -193,7 +193,12 @@ class DefaultMockServerWebSocketTest extends Specification {
 				.addSubProtocol("v4.channel.k8s.io"))
 		and: "A WebSocket listener"
 		wsReq.onComplete { ws ->
-			future.complete(ws.result().headers().get("sec-websocket-protocol"))
+			if (ws.result() != null && ws.result().headers() != null) {
+				future.complete(ws.result().headers().get("sec-websocket-protocol"))
+			}
+		}
+		if (wsReq.result() != null && wsReq.result().headers() != null) {
+			future.complete(wsReq.result().headers().get("sec-websocket-protocol"))
 		}
 		and: "An instance of AsyncConditions"
 		def async = new AsyncConditions(1)