Fix integration test

ACINQ · Nov 10, 2021 · 92cf397 · 92cf397
1 parent 529cf32
commit 92cf397
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/eclair-core/src/test/scala/fr/acinq/eclair/integration/ChannelIntegrationSpec.scala b/eclair-core/src/test/scala/fr/acinq/eclair/integration/ChannelIntegrationSpec.scala
@@ -534,7 +534,7 @@ class StandardChannelIntegrationSpec extends ChannelIntegrationSpec {
         address_opt = Some(HostAndPort.fromParts(funder.nodeParams.publicAddresses.head.socketAddress.getHostString, funder.nodeParams.publicAddresses.head.socketAddress.getPort)),
         sender.ref.toTyped
       ))
-      sender.expectMsgAnyOf(30 seconds, PeerConnection.ConnectionResult.Connected, PeerConnection.ConnectionResult.AlreadyConnected)
+      sender.expectMsgType[PeerConnection.ConnectionResult.HasConnection](30 seconds)
 
       fundee.register ! Register.Forward(sender.ref, channelId, CMD_GETSTATE(ActorRef.noSender))
       val fundeeState = sender.expectMsgType[RES_GETSTATE[ChannelState]].state

diff --git a/eclair-core/src/test/scala/fr/acinq/eclair/integration/IntegrationSpec.scala b/eclair-core/src/test/scala/fr/acinq/eclair/integration/IntegrationSpec.scala
@@ -158,7 +158,7 @@ abstract class IntegrationSpec extends TestKitBaseClass with BitcoindService wit
       address_opt = Some(HostAndPort.fromParts(address.socketAddress.getHostString, address.socketAddress.getPort)),
       sender.ref.toTyped
     ))
-    sender.expectMsgAnyOf(10 seconds, PeerConnection.ConnectionResult.Connected, PeerConnection.ConnectionResult.AlreadyConnected)
+    sender.expectMsgType[PeerConnection.ConnectionResult.HasConnection](10 seconds)
     sender.send(node1.switchboard, Peer.OpenChannel(
       remoteNodeId = node2.nodeParams.nodeId,
       fundingSatoshis = fundingSatoshis,