RavenDB-21724 : fix flaky test - wait for database deletion

aviv86 · Nov 21, 2023 · 226599d · 226599d
1 parent 013e95b
commit 226599d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/test/SlowTests/Sharding/ShardingTopologyTests.cs b/test/SlowTests/Sharding/ShardingTopologyTests.cs
@@ -190,7 +190,13 @@ public async Task AddAndDeleteShardFromDatabase_ShardHasNoBucketsMapping()
                     await Cluster.WaitForRaftIndexToBeAppliedInClusterAsync(deleteShardDatabaseRes.RaftCommandIndex);
                 }
 
-                var record = await store.Maintenance.Server.SendAsync(new GetDatabaseRecordOperation(store.Database));
+                DatabaseRecord record = null;
+                await WaitAndAssertForValueAsync(async () =>
+                {
+                    record = await store.Maintenance.Server.SendAsync(new GetDatabaseRecordOperation(store.Database));
+                    return record.DeletionInProgress.Count;
+                }, expectedVal: 0);
+
                 Assert.Equal(2, record.Sharding.Shards.Count);
 
                 //make sure the nodes that held the deleted shard no longer have any of this shard's db instances