From 8a032df0d3f220442451a59b96f672e32eff95b6 Mon Sep 17 00:00:00 2001
From: NetherGranite <nethergranite@gmail.com>
Date: Sat, 20 Apr 2024 16:22:10 -0600
Subject: [PATCH] Add a test to ensure UriReplacementHandler respects request
 options

---
 .../Middleware/UriReplacementHandlerTests.cs  | 35 +++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/Microsoft.Kiota.Http.HttpClientLibrary.Tests/Middleware/UriReplacementHandlerTests.cs b/Microsoft.Kiota.Http.HttpClientLibrary.Tests/Middleware/UriReplacementHandlerTests.cs
index 02df8ce..293edf4 100644
--- a/Microsoft.Kiota.Http.HttpClientLibrary.Tests/Middleware/UriReplacementHandlerTests.cs
+++ b/Microsoft.Kiota.Http.HttpClientLibrary.Tests/Middleware/UriReplacementHandlerTests.cs
@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.Net.Http;
 using System.Threading.Tasks;
+using Microsoft.Kiota.Abstractions;
 using Microsoft.Kiota.Http.HttpClientLibrary.Middleware;
 using Microsoft.Kiota.Http.HttpClientLibrary.Middleware.Options;
 using Moq;
@@ -65,4 +66,38 @@ public async Task Calls_Uri_ReplacementAsync()
 
         mockReplacement.Verify(static x=> x.Replace(It.IsAny<Uri>()), Times.Once());
     }
+
+    [Fact]
+    public async Task Calls_Uri_Replacement_From_Request_OptionsAsync()
+    {
+        var mockReplacement = new Mock<IUriReplacementHandlerOption>();
+        mockReplacement.Setup(static x => x.IsEnabled()).Returns(true);
+        mockReplacement.Setup(static x => x.Replace(It.IsAny<Uri>())).Returns(new Uri("http://changed"));
+
+        var handler = new UriReplacementHandler<IUriReplacementHandlerOption>()
+        {
+            InnerHandler = new FakeSuccessHandler()
+        };
+        var msg = new HttpRequestMessage(HttpMethod.Get, "http://localhost");
+        SetRequestOption(msg, mockReplacement.Object);
+        var client = new HttpClient(handler);
+        await client.SendAsync(msg);
+
+        mockReplacement.Verify(static x=> x.Replace(It.IsAny<Uri>()), Times.Once());
+    }
+
+    /// <summary>
+    /// Sets a <see cref="IRequestOption"/> in <see cref="HttpRequestMessage"/>.
+    /// </summary>
+    /// <typeparam name="T"></typeparam>
+    /// <param name="httpRequestMessage">The <see cref="HttpRequestMessage"/> representation of the request.</param>
+    /// <param name="option">The request option.</param>
+    private static void SetRequestOption<T>(HttpRequestMessage httpRequestMessage, T option) where T : IRequestOption
+    {
+#if NET5_0_OR_GREATER
+        httpRequestMessage.Options.Set(new HttpRequestOptionsKey<T>(typeof(T).FullName!), option);
+#else
+        httpRequestMessage.Properties.Add(typeof(T).FullName!, option);
+#endif
+    }
 }