Allow cancellation token in endpoints

src/Acheve.TestHost/Routing/TestServerAction.cs

@@ -27,14 +27,20 @@ public void AddArgument(int order, Expression expression, bool activeBodyApiCont
             var isFromHeader = argument.GetCustomAttributes<FromHeaderAttribute>().Any();
             var isFromRoute = argument.GetCustomAttributes<FromRouteAttribute>().Any();
 
-            bool isPrimitive = argument.ParameterType.IsPrimitive || argument.ParameterType.Equals(typeof(string));
+            bool isPrimitive = argument.ParameterType.IsPrimitiveType();
             bool hasNoAttributes = !isFromBody && !isFromForm && !isFromHeader && !isFromRoute;
 
             if (activeBodyApiController && hasNoAttributes && !isPrimitive)
             {
                 isFromBody = true;
             }
 
+            var isCancellationToken = argument.ParameterType == typeof(System.Threading.CancellationToken);
+            if (isCancellationToken)
+            {
+                isFromBody = isFromForm = isFromHeader = false;
+            }
+
             if (!ArgumentValues.ContainsKey(order))
             {
                 if (IsNullable(argument.ParameterType))

tests/UnitTests/Acheve.TestHost/Routing/Builders/ValuesV5Controller.cs

@@ -1,4 +1,5 @@
 using Microsoft.AspNetCore.Mvc;
+using System.Threading;
 
 namespace UnitTests.Acheve.TestHost.Builders
 {
@@ -44,5 +45,27 @@ public IActionResult Patch1(int id, Pagination pagination1)
 
             return Ok();
         }
+
+        [HttpGet(nameof(GetWithCancellationToken))]
+        public ActionResult<string> GetWithCancellationToken([FromQuery] string value, CancellationToken cancellationToken)
+        {
+            if (cancellationToken.IsCancellationRequested)
+            {
+                return BadRequest();
+            }
+
+            return Ok(value);
+        }
+
+        [HttpPost(nameof(PostWithCancellationToken))]
+        public ActionResult<string> PostWithCancellationToken([FromBody] string value, CancellationToken cancellationToken)
+        {
+            if (cancellationToken.IsCancellationRequested)
+            {
+                return BadRequest();
+            }
+
+            return Ok(value);
+        }
     }
 }

tests/UnitTests/Acheve.TestHost/Routing/TestServerExtensionsTests.cs

@@ -7,6 +7,7 @@
 using System.Net;
 using System.Net.Http;
 using System.Text.Json;
+using System.Threading;
 using System.Threading.Tasks;
 using UnitTests.Acheve.TestHost.Builders;
 using UnitTests.Acheve.TestHost.Routing.Models;
@@ -1676,6 +1677,44 @@ public async Task Create_request_with_list_parameter()
             response.Should().BeEquivalentTo(param);
         }
 
+        [Fact]
+        public async Task Create_get_request_with_cancellation_token_parameter()
+        {
+            var server = new TestServerBuilder()
+                .UseDefaultStartup()
+                .Build();
+
+            var param = "one";
+            var source = new CancellationTokenSource();
+            var token = source.Token;
+
+            var request = server.CreateHttpApiRequest<ValuesV5Controller>(controller => controller.GetWithCancellationToken(param, token));
+            var responseMessage = await request.GetAsync();
+
+            responseMessage.EnsureSuccessStatusCode();
+            var response = await responseMessage.Content.ReadAsStringAsync();
+            response.Should().Be(param);
+        }
+
+        [Fact]
+        public async Task Create_post_request_with_cancellation_token_parameter()
+        {
+            var server = new TestServerBuilder()
+                .UseDefaultStartup()
+                .Build();
+
+            var param = "one";
+            var source = new CancellationTokenSource();
+            var token = source.Token;
+
+            var request = server.CreateHttpApiRequest<ValuesV5Controller>(controller => controller.PostWithCancellationToken(param, token));
+            var responseMessage = await request.PostAsync();
+
+            responseMessage.EnsureSuccessStatusCode();
+            var response = await responseMessage.Content.ReadAsStringAsync();
+            response.Should().Be(param);
+        }
+
         private class PrivateNonControllerClass
         {
             public int SomeAction()