fix shouldbootandthenclose test

heroiclabs · Jul 14, 2021 · c10a3c5 · c10a3c5
1 parent 36bfafc
commit c10a3c5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/Nakama.Tests/Socket/WebSocketPartyTest.cs b/tests/Nakama.Tests/Socket/WebSocketPartyTest.cs
@@ -316,7 +316,6 @@ public async Task PresencesInitializedWithConcurrentJoins()
 
             while (partyObjCounter < numMembers)
             {
-                System.Console.WriteLine(partyObjCounter);
                 await Task.Delay(25);
             }
 
@@ -358,9 +357,11 @@ public async Task ShouldBootThenClose()
             var socket2PresenceTcs = new TaskCompletionSource<IUserPresence>();
 
             socket1.ReceivedPartyPresence += presences => {
-                if (!socket2PresenceTcs.Task.IsCompleted)
+
+                var session2Join = presences.Joins.FirstOrDefault(presence => presence.UserId == session2.UserId);
+                if (session2Join != null)
                 {
-                    socket2PresenceTcs.SetResult(presences.Joins.FirstOrDefault(presence => presence.UserId == session2.UserId));
+                    socket2PresenceTcs.SetResult(session2Join);
                 }
             };