feat: prevent serialization of CancellationToken? (#1917)

Co-authored-by: Chris Pulman <[email protected]>

Refit.Tests/RestService.cs

@@ -321,6 +321,18 @@ public interface IFragmentApi
     Task QueryAfterFragment();
 }
 
+public interface ICancellableApi
+{
+    [Get("/foo")]
+    Task GetWithCancellation(CancellationToken token = default);
+
+    [Get("/foo")]
+    Task<string> GetWithCancellationAndReturn(CancellationToken token = default);
+
+    [Get("/foo")]
+    Task GetWithNullableCancellation(CancellationToken? token);
+}
+
 public class HttpBinGet
 {
     public Dictionary<string, object> Args { get; set; }
@@ -2547,6 +2559,50 @@ public async Task ShouldStripQueryAfterFragment()
         mockHttp.VerifyNoOutstandingExpectation();
     }
 
+    [Fact]
+    public async Task TaskShouldCancelWhenRequested()
+    {
+        var ctSource = new CancellationTokenSource();
+        var token = ctSource.Token;
+
+        var fixture = RestService.For<ICancellableApi>("https://github.com");
+
+        ctSource.Cancel();
+        var task = fixture.GetWithCancellation(token);
+        await Assert.ThrowsAsync<TaskCanceledException>(async () => await task);
+    }
+
+    [Fact]
+    public async Task TaskResultShouldCancelWhenRequested()
+    {
+        var ctSource = new CancellationTokenSource();
+        var token = ctSource.Token;
+
+        var fixture = RestService.For<ICancellableApi>("https://github.com");
+
+        ctSource.Cancel();
+        var task = fixture.GetWithCancellationAndReturn(token);
+        await Assert.ThrowsAsync<TaskCanceledException>(async () => await task);
+    }
+
+
+    [Fact]
+    public async Task NullableCancellationTokenShouldBeIgnored()
+    {
+        var mockHttp = new MockHttpMessageHandler();
+        var settings = new RefitSettings { HttpMessageHandlerFactory = () => mockHttp, };
+
+        mockHttp
+            .Expect(HttpMethod.Get, "https://github.com/foo")
+            .Respond(HttpStatusCode.OK);
+
+        var fixture = RestService.For<ICancellableApi>("https://github.com", settings);
+
+        await fixture.GetWithNullableCancellation(null);
+
+        mockHttp.VerifyNoOutstandingExpectation();
+    }
+
     [Fact]
     public async Task TypeCollisionTest()
     {

Refit/RestMethodInfo.cs

@@ -89,7 +89,7 @@ public RestMethodInfoInternal(
             // Exclude cancellation token parameters from this list
             ParameterInfoArray = methodInfo
                 .GetParameters()
-                .Where(static p => p.ParameterType != typeof(CancellationToken))
+                .Where(static p => p.ParameterType != typeof(CancellationToken) && p.ParameterType != typeof(CancellationToken?))
                 .ToArray();
             (ParameterMap, FragmentPath) = BuildParameterMap(RelativePath, ParameterInfoArray);
             BodyParameterInfo = FindBodyParameter(ParameterInfoArray, IsMultipart, hma.Method);