refactor TestBindingSource attribute

mariusz96 · Dec 15, 2023 · 09a9409 · 09a9409
1 parent 9091b64
commit 09a9409
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 20 deletions.
diff --git a/test/UriGeneration.IntegrationTests/Controllers/AttributeRoutingController.cs b/test/UriGeneration.IntegrationTests/Controllers/AttributeRoutingController.cs
@@ -236,7 +236,7 @@ public void VoidAction()
         }
 
         [HttpGet]
-        public string? Test25([TestBindingSourceMetadata("Header")] string apiKey)
+        public string? Test25([TestBindingSource("Header")] string apiKey)
         {
             return _uriGenerator.GetUriByExpression<AttributeRoutingController>(
                 HttpContext,

diff --git a/test/UriGeneration.IntegrationTests/TestBindingSourceAttribute.cs b/test/UriGeneration.IntegrationTests/TestBindingSourceAttribute.cs
@@ -0,0 +1,18 @@
+using Microsoft.AspNetCore.Mvc.ModelBinding;
+
+namespace UriGeneration.IntegrationTests
+{
+    public class TestBindingSourceAttribute : Attribute, IBindingSourceMetadata
+    {
+        public TestBindingSourceAttribute(string bindingSource)
+        {
+            BindingSource = bindingSource switch
+            {
+                "Header" => BindingSource.Header,
+                _ => null
+            };
+        }
+
+        public BindingSource? BindingSource { get; }
+    }
+}
diff --git a/test/UriGeneration.IntegrationTests/TestBindingSourceMetadataAttribute.cs b/test/UriGeneration.IntegrationTests/TestBindingSourceMetadataAttribute.cs