fix breaking tests

microsoft · Feb 15, 2022 · 0b48e00 · 0b48e00
1 parent 0b204c6
commit 0b48e00
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/cs/src/core/FasterLog/FasterLog.cs b/cs/src/core/FasterLog/FasterLog.cs
@@ -1826,8 +1826,15 @@ private void RestoreLatest(out Dictionary<string, long> iterators, out byte[] co
 
                 if (headAddress == 0)
                     headAddress = Constants.kFirstValidAddress;
-
-                allocator.RestoreHybridLog(info.BeginAddress, headAddress, info.UntilAddress, info.UntilAddress);
+
+                try
+                {
+                    allocator.RestoreHybridLog(info.BeginAddress, headAddress, info.UntilAddress, info.UntilAddress);
+                }
+                catch
+                {
+                    if (!tolerateDeviceFailure) throw;
+                }
             }
 
             iterators = CompleteRestoreFromCommit(info);
@@ -1890,7 +1897,14 @@ private void RestoreSpecificCommit(long requestedCommitNum, out Dictionary<strin
 
                 if (headAddress == 0)
                     headAddress = Constants.kFirstValidAddress;
-                allocator.RestoreHybridLog(info.BeginAddress, headAddress, info.UntilAddress, info.UntilAddress);
+                try
+                {
+                    allocator.RestoreHybridLog(info.BeginAddress, headAddress, info.UntilAddress, info.UntilAddress);
+                }
+                catch
+                {
+                    if (!tolerateDeviceFailure) throw;
+                }
             }
 
             iterators = CompleteRestoreFromCommit(info);