RavenDB-17793 : fix failing tests

aviv86 · Feb 7, 2024 · f06099f · f06099f
1 parent 501f713
commit f06099f
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/src/Raven.Server/ServerWide/ShardingStore.cs b/src/Raven.Server/ServerWide/ShardingStore.cs
@@ -30,7 +30,10 @@ public ShardingStore([NotNull] ServerStore serverStore)
             _serverStore = serverStore ?? throw new ArgumentNullException(nameof(serverStore));
         }
 
-        public Task<(long Index, object Result)> StartBucketMigration(string database, int bucket, int toShard, string prefix = null, string raftId = null)
+        public Task<(long Index, object Result)> StartBucketMigration(string database, int bucket, int toShard, string raftId = null) =>
+            StartBucketMigration(database, bucket, toShard, prefix: null, raftId);
+
+        public Task<(long Index, object Result)> StartBucketMigration(string database, int bucket, int toShard, string prefix, string raftId)
         {
             var cmd = new StartBucketMigrationCommand(bucket, toShard, database, prefix, raftId ?? RaftIdGenerator.NewId());
             return _serverStore.SendToLeaderAsync(cmd);

diff --git a/test/FastTests/Client/RavenCommandTest.cs b/test/FastTests/Client/RavenCommandTest.cs
@@ -70,7 +70,8 @@ public void WhenCommandCanBeCheckedForFastestNode_ItCanRunInParallel()
                 "AddDatabaseShardCommand", "GetNextServerOperationIdCommand", "KillServerOperationCommand", "ModifyDatabaseTopologyCommand", "DelayBackupCommand",
                 "PutDatabaseClientConfigurationCommand", "PutDatabaseSettingsCommand", "PutDatabaseStudioConfigurationCommand", "GetTcpInfoForReplicationCommand",
                 "AddQueueSinkCommand", "UpdateQueueSinkCommand", "ConfigureDataArchivalCommand",
-                "AdoptOrphanedRevisionsCommand"
+                "AdoptOrphanedRevisionsCommand",
+                "AddPrefixedShardingSettingCommand", "DeletePrefixedShardingSettingCommand", "UpdatePrefixedShardingSettingCommand"
             }.OrderBy(t => t);
 
             var commandBaseType = typeof(RavenCommand<>);

diff --git a/test/SlowTests/Sharding/PrefixedSharding.cs b/test/SlowTests/Sharding/PrefixedSharding.cs
@@ -10,6 +10,7 @@
 using Raven.Client.Exceptions;
 using Raven.Client.ServerWide.Operations;
 using Raven.Client.ServerWide.Sharding;
+using Raven.Client.Util;
 using Raven.Server.Documents;
 using Raven.Server.Documents.Sharding;
 using Raven.Server.Rachis;
@@ -1558,7 +1559,7 @@ public async Task ShouldThrowOnAttemptToMovePrefixedBucketToShardNotInPrefixSett
 
         // shard #2 is not a part of Prefixed['users/'].Shards 
         await Assert.ThrowsAsync<RachisApplyException>(async ()=> 
-            await Server.ServerStore.Sharding.StartBucketMigration(store.Database, bucket, toShard : 2, prefix: "users/"));
+            await Server.ServerStore.Sharding.StartBucketMigration(store.Database, bucket, toShard : 2, prefix: "users/", RaftIdGenerator.NewId()));
     }
 
     [RavenFact(RavenTestCategory.Sharding)]

diff --git a/test/Tests.Infrastructure/RavenTestBase.ReshardingTestBase.cs b/test/Tests.Infrastructure/RavenTestBase.ReshardingTestBase.cs
@@ -6,6 +6,7 @@
 using Raven.Client.Documents;
 using Raven.Client.ServerWide.Operations;
 using Raven.Client.ServerWide.Sharding;
+using Raven.Client.Util;
 using Raven.Server;
 using Raven.Server.Utils;
 using Sparrow.Json;
@@ -65,7 +66,7 @@ public async Task<int> StartMovingShardForId(IDocumentStore store, string id, in
             {
                 try
                 {
-                    await server.ServerStore.Sharding.StartBucketMigration(store.Database, bucket, moveToShard, prefixedSetting?.Prefix);
+                    await server.ServerStore.Sharding.StartBucketMigration(store.Database, bucket, moveToShard, prefix: prefixedSetting?.Prefix, raftId: RaftIdGenerator.NewId());
                     break;
                 }
                 catch