Fix tests waiting for WS event

Author: sierpinskid

Assets/Plugins/StreamChat/Tests/StatefulClient/ChannelsQueryFiltersTests.cs

@@ -108,13 +108,15 @@ private async Task When_query_channel_with_created_by_id_filter_expect_valid_res
             var channel1 = await CreateUniqueTempChannelAsync();
             var channel2 = await CreateUniqueTempChannelAsync();
             var channel3 = await CreateUniqueTempChannelAsync();
+            var allChannels = new[] { channel1, channel2, channel3 };
 
             var filters = new IFieldFilterRule[]
             {
                 ChannelFilter.CreatedById.EqualsTo(Client.LocalUserData.User),
             };
 
-            var channels = (await Client.QueryChannelsAsync(filters, _sortByCreatedAtAscending)).ToArray();
+            var channels = (await TryAsync(() => Client.QueryChannelsAsync(filters, _sortByCreatedAtAscending),
+                channels => allChannels.All(channels.Contains))).ToArray();
             Assert.Contains(channel1, channels);
             Assert.Contains(channel2, channels);
             Assert.Contains(channel3, channels);
@@ -215,14 +217,14 @@ private async Task When_query_channel_by_created_at_filter_expect_valid_results_
             var channel1 = await CreateUniqueTempChannelAsync();
             var channel2 = await CreateUniqueTempChannelAsync();
             var channel3 = await CreateUniqueTempChannelAsync();
-            var allChannels = new [] { channel1, channel2, channel3 };  
+            var allChannels = new[] { channel1, channel2, channel3 };
 
             var filters = new IFieldFilterRule[]
             {
                 ChannelFilter.CreatedAt.GreaterThanOrEquals(DateTime.UtcNow.AddMinutes(-5)),
             };
 
-		  var channels = await TryAsync(() => Client.QueryChannelsAsync(filters, _sortByCreatedAtAscending),
+            var channels = await TryAsync(() => Client.QueryChannelsAsync(filters, _sortByCreatedAtAscending),
                 channels => allChannels.All(channels.Contains));
             Assert.IsTrue(allChannels.All(channels.Contains));
         }

Assets/Plugins/StreamChat/Tests/StatefulClient/ClientStateSyncTests.cs

@@ -56,7 +56,8 @@ private async Task When_client_reconnects_expect_receiving_missed_messages_Async
             await GetConnectedOtherClientAsync();
 
             // Wait for sync request to complete
-            await WaitWhileTrueAsync(() => otherClientChannel.Messages.All(m => m.Id != message.Id));
+            await WaitWhileTrueAsync(()
+                => otherClientChannel.Messages.All(m => m.Id != message.Id || m.Id != message2.Id));
 
             // Messages should now be present on the second client with no duplicates
             Assert.IsNotNull(otherClientChannel.Messages.Single(m => m.Id == message.Id));