Hacky thing

milvus-io · Feb 28, 2024 · ddbc986 · ddbc986
1 parent 9877164
commit ddbc986
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Milvus.Client.Tests/DataTests.cs b/Milvus.Client.Tests/DataTests.cs
@@ -119,7 +119,7 @@ public async Task FlushAllAsync_and_wait()
         await Client.WaitForFlushAllAsync(timestamp);
 
         IEnumerable<PersistentSegmentInfo> segmentInfos = await Collection.GetPersistentSegmentInfosAsync();
-        Assert.True(segmentInfos.All(p => p.State == SegmentState.Sealed));
+        Assert.True(segmentInfos.All(p => p.State is SegmentState.Flushed or SegmentState.Sealed));
     }
 
     [Fact]