From bc3bc7948ec858cccfcd8a7f27c5271c86d3912e Mon Sep 17 00:00:00 2001
From: Daniel Tavares Agostinho <comptesdaniel@gmail.com>
Date: Sun, 30 Jun 2024 22:32:13 +0200
Subject: [PATCH] fix tests

---
 .../scala/ch/epfl/pop/decentralized/GossipManagerSuite.scala  | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/be2-scala/src/test/scala/ch/epfl/pop/decentralized/GossipManagerSuite.scala b/be2-scala/src/test/scala/ch/epfl/pop/decentralized/GossipManagerSuite.scala
index d2a0770e86..23ccbaa19b 100644
--- a/be2-scala/src/test/scala/ch/epfl/pop/decentralized/GossipManagerSuite.scala
+++ b/be2-scala/src/test/scala/ch/epfl/pop/decentralized/GossipManagerSuite.scala
@@ -309,6 +309,7 @@ class GossipManagerSuite extends TestKit(ActorSystem("GossipManagerSuiteActorSys
     // register server
     for (peer <- peers) {
       connectionMediatorRef ? ConnectionMediator.NewServerConnected(peer.ref, GreetServer(PublicKey(Base64Data("")), "", ""))
+      peer.receiveOne(duration)
     }
     checkPeersWritten(connectionMediatorRef)
 
@@ -317,7 +318,7 @@ class GossipManagerSuite extends TestKit(ActorSystem("GossipManagerSuiteActorSys
     Await.result(output, duration)
 
     // include rumor state
-    peers.map(_.receiveOne(duration)).count(_ != null) shouldBe 2
+    peers.map(_.receiveOne(duration)).count(_ != null) shouldBe 1
 
   }
 
@@ -340,6 +341,7 @@ class GossipManagerSuite extends TestKit(ActorSystem("GossipManagerSuiteActorSys
     var n = 0
     for (peer <- peers) {
       connectionMediatorRef ? ConnectionMediator.NewServerConnected(peer.ref, GreetServer(PublicKey(Base64Data.encode(s"$n")), "", ""))
+      peer.receiveOne(duration)
       n += 1
     }
     checkPeersWritten(connectionMediatorRef)