Fix flaky test
diff --git a/hobbits/src/test/kotlin/org/apache/tuweni/hobbits/HobbitsTransportTest.kt b/hobbits/src/test/kotlin/org/apache/tuweni/hobbits/HobbitsTransportTest.kt
index c527a22..6088852 100644
--- a/hobbits/src/test/kotlin/org/apache/tuweni/hobbits/HobbitsTransportTest.kt
+++ b/hobbits/src/test/kotlin/org/apache/tuweni/hobbits/HobbitsTransportTest.kt
@@ -70,12 +70,15 @@
it.handler {
completion.complete(Bytes.wrapBuffer(it))
}
- }.listen(0, "localhost")
- val server = HobbitsTransport(vertx)
- server.start()
- val msg = Message(protocol = Protocol.RPC, headers = Bytes.EMPTY, body = Bytes.EMPTY)
- server.sendMessage(msg, Transport.TCP, "localhost", listening.actualPort())
- val result = completion.await()
- assertEquals(msg.toBytes(), result)
+ }.listen(0, "localhost") {
+ runBlocking {
+ val server = HobbitsTransport(vertx)
+ server.start()
+ val msg = Message(protocol = Protocol.RPC, headers = Bytes.EMPTY, body = Bytes.EMPTY)
+ server.sendMessage(msg, Transport.TCP, "localhost", listening.actualPort())
+ val result = completion.await()
+ assertEquals(msg.toBytes(), result)
+ }
+ }
}
}
