From 235caec007ba1d3199a0cf2c0347bc9688cc742e Mon Sep 17 00:00:00 2001
From: aviv <aviv@ravendb.net>
Date: Mon, 29 Jul 2024 17:34:40 +0300
Subject: [PATCH] RavenDB-22611 : add debug info to flaky test

---
 .../PeriodicBackup/PeriodicBackupSlowTests.cs | 47 ++++++++++++++++++-
 1 file changed, 46 insertions(+), 1 deletion(-)

diff --git a/test/SlowTests/Server/Documents/PeriodicBackup/PeriodicBackupSlowTests.cs b/test/SlowTests/Server/Documents/PeriodicBackup/PeriodicBackupSlowTests.cs
index 0c4fa16b2c83..78e3daab3844 100644
--- a/test/SlowTests/Server/Documents/PeriodicBackup/PeriodicBackupSlowTests.cs
+++ b/test/SlowTests/Server/Documents/PeriodicBackup/PeriodicBackupSlowTests.cs
@@ -18,6 +18,7 @@
 using Raven.Client.Documents.Operations;
 using Raven.Client.Documents.Operations.Attachments;
 using Raven.Client.Documents.Operations.Backups;
+using Raven.Client.Documents.Operations.Backups.Sharding;
 using Raven.Client.Documents.Operations.CompareExchange;
 using Raven.Client.Documents.Operations.OngoingTasks;
 using Raven.Client.Documents.Operations.TimeSeries;
@@ -4023,7 +4024,24 @@ public async Task can_backup_and_restore_with_deleted_timeseries_ranges(Options
                     Assert.Null(ts);
                 }
 
-                await Backup.RunBackupForDatabaseModeAsync(Server, config, store, options.DatabaseMode, isFullBackup: false, backupTaskId);
+                // add debug info in order to investigate RavenDB-22611
+                // can revert this change when RavenDB-22611 is fixed
+
+                WaitHandle[] waitHandles;
+                if (options.DatabaseMode == RavenDatabaseMode.Sharded)
+                {
+                    waitHandles = await Sharding.Backup.WaitForBackupsToComplete(new[] { store });
+                    await Sharding.Backup.RunBackupAsync(store.Database, backupTaskId, isFullBackup : false);
+                }
+                else
+                {
+                    waitHandles = await Backup.WaitForBackupToComplete(store);
+                    await Backup.RunBackupAsync(Server, backupTaskId, store, isFullBackup : false);
+                }
+
+                Assert.True(WaitHandle.WaitAll(waitHandles, TimeSpan.FromMinutes(1)), AddDebugInfo(store, backupTaskId, options.DatabaseMode));
+
+                //await Backup.RunBackupForDatabaseModeAsync(Server, config, store, options.DatabaseMode, isFullBackup: false, backupTaskId);
             }
 
             using (var store = GetDocumentStore(options))
@@ -4366,5 +4384,32 @@ private static string AddDebugInfo(PeriodicBackupStatus fullBackupStatus, Period
                 .AppendLine(JsonConvert.SerializeObject(incrementalBackupStatus));
             return sb.ToString();
         }
+
+        private static string AddDebugInfo(IDocumentStore store, long backupTaskId, RavenDatabaseMode databaseMode)
+        {
+            var sb = new StringBuilder();
+            sb.AppendLine("backup failed to complete in 60 seconds");
+
+            if (databaseMode == RavenDatabaseMode.Sharded)
+            {
+                var shardedBackupStatus = store.Maintenance.Send(new GetShardedPeriodicBackupStatusOperation(backupTaskId));
+                sb.AppendLine("sharded backup statuses: ");
+                foreach (var kvp in shardedBackupStatus.Statuses)
+                {
+                    sb.AppendLine()
+                        .AppendLine($"shard {kvp.Key}:")
+                        .AppendLine(JsonConvert.SerializeObject(kvp.Value));
+                }
+            }
+            else
+            {
+                var backupStatus = store.Maintenance.Send(new GetPeriodicBackupStatusOperation(backupTaskId)).Status;
+                sb.AppendLine("backup status: ")
+                    .AppendLine(JsonConvert.SerializeObject(backupStatus));
+            }
+
+            return sb.ToString();
+
+        }
     }
 }