check with completed count

Author: atakavci

tests/NRedisStack.Tests/Search/SearchTests.cs

@@ -8,9 +8,6 @@
 using System.Runtime.InteropServices;
 using NetTopologySuite.IO;
 using NetTopologySuite.Geometries;
-using System.Linq.Expressions;
-using System.Collections;
-
 
 namespace NRedisStack.Tests.Search;
 
@@ -3451,6 +3448,7 @@ public async void TestDocumentLoadWithDB_Issue352(string endpointId)
 
         Document droppedDocument = null;
         int numberOfAttempts = 0;
+        Int32 completed = 0;
 
         do
         {
@@ -3474,6 +3472,7 @@ public async void TestDocumentLoadWithDB_Issue352(string endpointId)
                     // doc would show up in search result with no fields 
                     if (docs.Count == 0)
                     {
+                        Interlocked.Increment(ref completed);
                         break;
                     }
                     // if we get a document with no fields then we know that the key 
@@ -3482,6 +3481,7 @@ public async void TestDocumentLoadWithDB_Issue352(string endpointId)
                     else if (docs[0].GetProperties().Count() == 0)
                     {
                         droppedDocument = docs[0];
+                        Interlocked.Increment(ref completed);
                         break;
                     }
                 }
@@ -3495,8 +3495,7 @@ public async void TestDocumentLoadWithDB_Issue352(string endpointId)
             }
             Task checkTask = Task.WhenAll(tasks);
             await Task.WhenAny(checkTask, Task.Delay(1500));
-            Assert.True(checkTask.IsCompleted);
-            Assert.Null(checkTask.Exception);
+            Assert.Equal(3, completed);
             cancelled = true;
         } while (droppedDocument == null && numberOfAttempts++ < 5);
         // we wont do an actual assert here since