Introduce TelemetryResilienceStrategy (App-vNext#1140)

Author: martintmk

eng/Library.targets

@@ -40,7 +40,7 @@
     </ItemGroup>
   </Target>
 
-  <Target Name="AddInternalsVisibleToTest" BeforeTargets="BeforeCompile">
+  <Target Name="AddInternalsVisibleToTest" BeforeTargets="BeforeCompile" Condition="'@(InternalsVisibleToTest)' != ''">
     <ItemGroup>
       <InternalsVisibleTo Include="%(InternalsVisibleToTest.Identity)$(PollyPublicKeySuffix)" />
     </ItemGroup>

src/Polly.Core.Tests/Helpers/TestArguments.cs

@@ -13,9 +13,11 @@
 
   <ItemGroup>
     <ProjectReference Include="..\Polly.Core\Polly.Core.csproj" />
+    <ProjectReference Include="..\Polly.TestUtils\Polly.TestUtils.csproj" />
   </ItemGroup>
 
   <ItemGroup>
+    <Using Include="Polly.TestUtils" />
     <Using Include="Polly.Core.Tests.Helpers" />
   </ItemGroup>
 </Project>

src/Polly.Core.Tests/Helpers/TestUtils.cs

@@ -1,3 +1,5 @@
+using Polly.Strategy;
+
 namespace Polly.Core.Tests;
 
 public class ResilienceContextTests
@@ -19,7 +21,7 @@ public void Get_EnsureDefaults()
     [Fact]
     public async Task Get_EnsurePooled()
     {
-        await TestUtils.AssertWithTimeoutAsync(() =>
+        await TestUtilities.AssertWithTimeoutAsync(() =>
         {
             var context = ResilienceContext.Get();
 
@@ -35,17 +37,29 @@ public void Return_Null_Throws()
         Assert.Throws<ArgumentNullException>(() => ResilienceContext.Return(null!));
     }
 
+    [Fact]
+    public void AddResilienceEvent_Ok()
+    {
+        var context = ResilienceContext.Get();
+
+        context.AddResilienceEvent(new ReportedResilienceEvent("Dummy"));
+
+        context.ResilienceEvents.Should().HaveCount(1);
+        context.ResilienceEvents.Should().Contain(new ReportedResilienceEvent("Dummy"));
+    }
+
     [Fact]
     public async Task Return_EnsureDefaults()
     {
-        await TestUtils.AssertWithTimeoutAsync(() =>
+        await TestUtilities.AssertWithTimeoutAsync(() =>
         {
             using var cts = new CancellationTokenSource();
             var context = ResilienceContext.Get();
             context.CancellationToken = cts.Token;
             context.Initialize<bool>(true);
             context.CancellationToken.Should().Be(cts.Token);
             context.Properties.Set(new ResiliencePropertyKey<int>("abc"), 10);
+            context.AddResilienceEvent(new ReportedResilienceEvent("dummy"));
             ResilienceContext.Return(context);
 
             AssertDefaults(context);
@@ -91,5 +105,6 @@ private static void AssertDefaults(ResilienceContext context)
         context.IsSynchronous.Should().BeFalse();
         context.CancellationToken.Should().Be(CancellationToken.None);
         context.Properties.Should().BeEmpty();
+        context.ResilienceEvents.Should().BeEmpty();
     }
 }

src/Polly.Core.Tests/Polly.Core.Tests.csproj

@@ -292,6 +292,29 @@ public void BuildStrategy_EnsureCorrectContext()
         verified2.Should().BeTrue();
     }
 
+    [Fact]
+    public void Build_OnCreatingStrategy_EnsureRespected()
+    {
+        // arrange
+        var strategy = new TestResilienceStrategy();
+        var builder = new ResilienceStrategyBuilder
+        {
+            OnCreatingStrategy = strategies =>
+            {
+                strategies.Should().ContainSingle(s => s == strategy);
+                strategies.Insert(0, new TestResilienceStrategy());
+            }
+        };
+
+        builder.AddStrategy(strategy);
+
+        // act
+        var finalStrategy = builder.Build();
+
+        // assert
+        finalStrategy.Should().BeOfType<ResilienceStrategyPipeline>();
+    }
+
     private class Strategy : ResilienceStrategy
     {
         public Action? Before { get; set; }

src/Polly.Core.Tests/ResilienceContextTests.cs

@@ -11,7 +11,7 @@ public class RetryResilienceStrategyTests
     private readonly ResilienceStrategyTelemetry _telemetry;
     private readonly Mock<DiagnosticSource> _diagnosticSource = new();
 
-    public RetryResilienceStrategyTests() => _telemetry = TestUtils.CreateResilienceTelemetry(_diagnosticSource.Object);
+    public RetryResilienceStrategyTests() => _telemetry = TestUtilities.CreateResilienceTelemetry(_diagnosticSource.Object);
 
     [Fact]
     public void ShouldRetryEmpty_Skipped()

src/Polly.Core.Tests/ResilienceStrategyBuilderTests.cs

@@ -0,0 +1,22 @@
+using Polly.Strategy;
+
+namespace Polly.Core.Tests.Strategy;
+
+public class ReportedResilienceEventTests
+{
+    [Fact]
+    public void Ctor_Ok()
+    {
+        var ev = new ReportedResilienceEvent("A");
+
+        ev.ToString().Should().Be("A");
+    }
+
+    [Fact]
+    public void Equality_Ok()
+    {
+        (new ReportedResilienceEvent("A") == new ReportedResilienceEvent("A")).Should().BeTrue();
+        (new ReportedResilienceEvent("A") != new ReportedResilienceEvent("A")).Should().BeFalse();
+        (new ReportedResilienceEvent("A") == new ReportedResilienceEvent("B")).Should().BeFalse();
+    }
+}

src/Polly.Core.Tests/Retry/RetryResilienceStrategyTests.cs

@@ -1,5 +1,4 @@
 using Moq;
-using Polly.Core.Tests.Helpers;
 using Polly.Strategy;
 using Polly.Timeout;
 
@@ -16,7 +15,7 @@ public class TimeoutResilienceStrategyTests : IDisposable
 
     public TimeoutResilienceStrategyTests()
     {
-        _telemetry = TestUtils.CreateResilienceTelemetry(_diagnosticSource.Object);
+        _telemetry = TestUtilities.CreateResilienceTelemetry(_diagnosticSource.Object);
         _timeProvider = new FakeTimeProvider();
         _options = new TimeoutStrategyOptions();
         _cancellationSource = new CancellationTokenSource();

src/Polly.Core.Tests/Strategy/ReportedResilienceEventTests.cs

@@ -13,7 +13,7 @@ public void TimestampFrequency_Ok()
     [Fact]
     public async Task CancelAfter_Ok()
     {
-        await TestUtils.AssertWithTimeoutAsync(async () =>
+        await TestUtilities.AssertWithTimeoutAsync(async () =>
         {
             using var cts = new CancellationTokenSource();
             TimeProvider.System.CancelAfter(cts, TimeSpan.FromMilliseconds(10));
@@ -28,7 +28,7 @@ public async Task Delay_Ok()
     {
         using var cts = new CancellationTokenSource();
 
-        await TestUtils.AssertWithTimeoutAsync(() =>
+        await TestUtilities.AssertWithTimeoutAsync(() =>
         {
             TimeProvider.System.Delay(TimeSpan.FromMilliseconds(10)).IsCompleted.Should().BeFalse();
         });
@@ -48,16 +48,20 @@ public async Task GetElapsedTime_Ok()
         var delay = TimeSpan.FromMilliseconds(10);
         var delayWithTolerance = TimeSpan.FromMilliseconds(30);
 
-        await TestUtils.AssertWithTimeoutAsync(async () =>
+        await TestUtilities.AssertWithTimeoutAsync(async () =>
         {
             var stamp1 = TimeProvider.System.GetTimestamp();
             await Task.Delay(10);
             var stamp2 = TimeProvider.System.GetTimestamp();
 
             var elapsed = TimeProvider.System.GetElapsedTime(stamp1, stamp2);
+            var elapsed2 = TimeProvider.System.GetElapsedTime(stamp1);
 
             elapsed.Should().BeGreaterThanOrEqualTo(delay);
             elapsed.Should().BeLessThan(delayWithTolerance);
+
+            elapsed2.Should().BeGreaterThanOrEqualTo(delay);
+            elapsed2.Should().BeLessThan(delayWithTolerance);
         });
     }
 
@@ -81,7 +85,7 @@ public void GetElapsedTime_Mocked_Ok()
     [Fact]
     public async Task UtcNow_Ok()
     {
-        await TestUtils.AssertWithTimeoutAsync(() =>
+        await TestUtilities.AssertWithTimeoutAsync(() =>
         {
             var now = TimeProvider.System.UtcNow;
             (DateTimeOffset.UtcNow - now).Should().BeLessThanOrEqualTo(TimeSpan.FromMilliseconds(10));