wait for message to be sent

wavesplatform · Nov 27, 2023 · b147166 · b147166
1 parent 6ee8d21
commit b147166
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/node-it/src/test/scala/com/wavesplatform/it/api/AsyncNetworkApi.scala b/node-it/src/test/scala/com/wavesplatform/it/api/AsyncNetworkApi.scala
@@ -18,8 +18,11 @@ object AsyncNetworkApi {
         s"it-client-to-${node.name}",
         nonce
       )
-      sender.connect(node.networkAddress).map { ch =>
-        if (ch.isActive) sender.send(ch, messages*).map(_ => sender.close()) else sender.close()
+      sender.connect(node.networkAddress).flatMap { ch =>
+        if (ch.isActive) sender.send(ch, messages*).map(_ => sender.close()) else {
+          sender.close()
+          Future.failed(new Exception("Channel is inactive"))
+        }
       }
     }
   }