diff --git a/test/Polly.Core.Tests/Helpers/MockTimeProvider.cs b/test/Polly.Core.Tests/Helpers/MockTimeProvider.cs index 69549b4f16a..8e027185cef 100644 --- a/test/Polly.Core.Tests/Helpers/MockTimeProvider.cs +++ b/test/Polly.Core.Tests/Helpers/MockTimeProvider.cs @@ -4,49 +4,17 @@ namespace Polly.Core.Tests.Helpers; internal class MockTimeProvider : Mock { - private DateTimeOffset? _time; - public MockTimeProvider() : base(MockBehavior.Strict) { } - public MockTimeProvider SetupUtcNow(DateTimeOffset? time = null) - { - _time = time ?? DateTimeOffset.UtcNow; - Setup(x => x.GetUtcNow()).Returns(() => _time.Value); - return this; - } - - public MockTimeProvider AdvanceTime(TimeSpan time) - { - if (_time == null) - { - SetupUtcNow(DateTimeOffset.UtcNow); - } - - _time = _time!.Value.Add(time); - return this; - } - - public MockTimeProvider SetupTimestampFrequency() - { - Setup(x => x.TimestampFrequency).Returns(Stopwatch.Frequency); - return this; - } - public MockTimeProvider SetupAnyDelay(CancellationToken cancellationToken = default) { Setup(x => x.Delay(It.IsAny(), cancellationToken)).Returns(Task.CompletedTask); return this; } - public MockTimeProvider SetupGetTimestamp() - { - Setup(x => x.GetTimestamp()).Returns(0); - return this; - } - public MockTimeProvider SetupDelay(TimeSpan delay, CancellationToken cancellationToken = default) { Setup(x => x.Delay(delay, cancellationToken)).Returns(Task.CompletedTask);