Add tests

Author: jerviscui

src/plugins/ConsolePollyTest/ConsolePollyTest.csproj

@@ -8,10 +8,11 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="Microsoft.Extensions.Logging" Version="9.0.6" />
-    <PackageReference Include="Microsoft.Extensions.Logging.Console" Version="9.0.6" />
-    <PackageReference Include="Polly.Core" Version="8.6.1" />
-    <PackageReference Include="Polly.Extensions" Version="8.6.1" />
+    <PackageReference Include="Microsoft.Extensions.Logging" Version="9.0.7" />
+    <PackageReference Include="Microsoft.Extensions.Logging.Console" Version="9.0.7" />
+    <PackageReference Include="Polly" Version="8.6.2" />
+    <PackageReference Include="Polly.Extensions" Version="8.6.2" />
+    <PackageReference Include="Polly.RateLimiting" Version="8.6.2" />
   </ItemGroup>
 
 </Project>

src/plugins/ConsolePollyTest/PollyFallbackTest.cs

@@ -0,0 +1,66 @@
+using Polly;
+using Polly.Fallback;
+
+namespace ConsolePollyTest;
+
+public static class PollyFallbackTest
+{
+
+    #region Constants & Statics
+
+    public static async Task Fallback_TestAsync()
+    {
+        var options = TestBase.GetTelemetry();
+        var builder = new ResiliencePipelineBuilder<MyClass>().ConfigureTelemetry(options)
+            .AddFallback(
+                new FallbackStrategyOptions<MyClass>
+                {
+                    ShouldHandle = static (args) =>
+                            {
+                                // Handle all exceptions
+                                return ValueTask.FromResult(true);
+                            },
+                    FallbackAction = static (args) =>
+                            {
+                                // Return a new instance of MyClass as fallback
+                                return Outcome.FromResultAsValueTask(new MyClass("fallback"));
+                            },
+                    OnFallback = static (args) =>
+                            {
+                                Console.WriteLine($"Fallback executed for: {args.Outcome.Result?.PropertyName}");
+                                return ValueTask.CompletedTask;
+                            }
+                });
+
+        var pipeline = builder.Build();
+
+        using var client = TestBase.GetClient();
+
+        var result = await pipeline.ExecuteAsync(
+            async (token) =>
+            {
+                var r = await client.GetAsync(TestBase.GetUri("/weatherforecast/GetWeatherForecast"), token);
+
+                return new MyClass("return");
+            },
+            cancellationToken: default);
+
+        Console.WriteLine(result.PropertyName);
+    }
+
+    #endregion
+
+    public class MyClass
+    {
+        public MyClass(string propertyName)
+        {
+            PropertyName = propertyName;
+        }
+
+        #region Properties
+
+        public string PropertyName { get; private set; }
+
+        #endregion
+    }
+}

src/plugins/ConsolePollyTest/PollyTimeoutTest.cs

@@ -0,0 +1,98 @@
+using Polly;
+using Polly.Timeout;
+
+namespace ConsolePollyTest;
+
+public static class PollyTimeoutTest
+{
+
+    #region Constants & Statics
+
+    public static async Task Timeout_OuterCancel_TestAsync()
+    {
+        var options = TestBase.GetTelemetry();
+        var builder = new ResiliencePipelineBuilder().ConfigureTelemetry(options)
+            .AddTimeout(
+                new TimeoutStrategyOptions
+                {
+                    Timeout = TimeSpan.FromSeconds(1),
+                    OnTimeout = (args) =>
+                            {
+                                Console.WriteLine(args.Context);
+
+                                return ValueTask.CompletedTask;
+                            }
+                });
+
+        var pipeline = builder.Build();
+
+        using var client = TestBase.GetClient();
+
+        using var cts = new CancellationTokenSource();
+        var outerToken = cts.Token;
+
+        try
+        {
+            var result = await pipeline.ExecuteAsync(
+                async (token) =>
+                {
+                    //使用 outerToken 而不是 token，不会抛出 TimeoutRejectedException 异常
+                    await Task.Delay(2_000, outerToken);
+                    var r = await client.GetAsync(TestBase.GetUri("/weatherforecast/GetWeatherForecast"), outerToken);
+
+                    return r;
+                },
+                cancellationToken: outerToken);
+
+            Console.WriteLine(result);
+        }
+        catch (TimeoutRejectedException ex)
+        {
+            Console.WriteLine(ex);
+        }
+    }
+
+    public static async Task Timeout_TestAsync()
+    {
+        var options = TestBase.GetTelemetry();
+        var builder = new ResiliencePipelineBuilder().ConfigureTelemetry(options)
+            .AddTimeout(
+                new TimeoutStrategyOptions
+                {
+                    Timeout = TimeSpan.FromSeconds(1),
+                    OnTimeout = (args) =>
+                            {
+                                Console.WriteLine(args.Context);
+
+                                return ValueTask.CompletedTask;
+                            }
+                });
+
+        var pipeline = builder.Build();
+
+        using var client = TestBase.GetClient();
+
+        try
+        {
+            var result = await pipeline.ExecuteAsync(
+                async (token) =>
+                {
+                    await Task.Delay(2_000, token);
+
+                    var r = await client.GetAsync(TestBase.GetUri("/weatherforecast/GetWeatherForecast"), token);
+
+                    return r;
+                },
+                cancellationToken: default);
+
+            Console.WriteLine(result);
+        }
+        catch (TimeoutRejectedException ex)
+        {
+            Console.WriteLine(ex);
+        }
+    }
+
+    #endregion
+
+}

src/plugins/ConsolePollyTest/Program.cs

@@ -11,7 +11,12 @@ private static async Task Main(string[] args)
         //await PollyRetryTest.Retry_ExecThree_TestAsync();
         //await PollyRetryTest.Retry_UseOnePipeline_TestAsync();
 
-        await PollyCircuitTest.Circuit_3Per2sec_TestAsync();
+        //await PollyCircuitTest.Circuit_3Per2sec_TestAsync();
+
+        //await PollyFallbackTest.Fallback_TestAsync();
+
+        //await PollyTimeoutTest.Timeout_TestAsync();
+        await PollyTimeoutTest.Timeout_OuterCancel_TestAsync();
 
         _ = Console.ReadKey();
     }