FIx hedging tests on .NET 8

App-vNext · Aug 10, 2023 · 75ea78e · 75ea78e
1 parent f0b3417
commit 75ea78e
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/test/Polly.Core.Tests/Hedging/HedgingTimeProvider.cs b/test/Polly.Core.Tests/Hedging/HedgingTimeProvider.cs
@@ -1,3 +1,5 @@
+using System;
+
 namespace Polly.Core.Tests.Hedging;
 
 internal class HedgingTimeProvider : TimeProvider
@@ -26,7 +28,13 @@ public void Advance(TimeSpan diff)
 
     public override ITimer CreateTimer(TimerCallback callback, object? state, TimeSpan dueTime, TimeSpan period)
     {
-        var entry = new TimerEntry(dueTime, new TaskCompletionSource<bool>(), _utcNow.Add(dueTime), () => callback(state));
+        var timeStamp = dueTime switch
+        {
+            _ when dueTime == global::System.Threading.Timeout.InfiniteTimeSpan => DateTimeOffset.MaxValue,
+            _ => _utcNow.Add(dueTime)
+        };
+
+        var entry = new TimerEntry(dueTime, new TaskCompletionSource<bool>(), timeStamp, () => callback(state));
         TimerEntries.Enqueue(entry);
 
         Advance(AutoAdvance);