Addressed styling issues.

microsoft · Feb 23, 2021 · a37e772 · a37e772
1 parent 0c977e2
commit a37e772
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 25 deletions.
diff --git a/test/ReverseProxy.FunctionalTests/Common/TestEnvironment.cs b/test/ReverseProxy.FunctionalTests/Common/TestEnvironment.cs
@@ -4,6 +4,7 @@
 using System;
 using System.Collections.Generic;
 using System.Net;
+using System.Text;
 using System.Threading;
 using System.Threading.Tasks;
 using Microsoft.AspNetCore.Builder;
@@ -14,7 +15,6 @@
 using Microsoft.Extensions.Hosting;
 using Microsoft.ReverseProxy.Abstractions;
 using Microsoft.ReverseProxy.Utilities.Tests;
-using System.Text;
 
 namespace Microsoft.ReverseProxy.Common
 {

diff --git a/test/ReverseProxy.FunctionalTests/HeaderEncodingTests.cs b/test/ReverseProxy.FunctionalTests/HeaderEncodingTests.cs
@@ -59,9 +59,12 @@ public async Task ProxyAsync_RequestWithEncodedHeaderValue(string headerValue, s
                             return Task.CompletedTask;
                         }
                     });
-                    proxyApp.UseMiddleware<CheckHeaderMiddleware>(HeaderNames.Referer, headerValue);
                     proxyApp.Use(async (context, next) =>
                     {
+                        Assert.True(context.Request.Headers.TryGetValue(HeaderNames.Referer, out var header));
+                        var value = Assert.Single(header);
+                        Assert.Equal(headerValue, value);
+
                         await next();
                         proxyError = context.Features.Get<IProxyErrorFeature>();
                     });
@@ -201,29 +204,6 @@ public async Task ProxyAsync_ResponseWithEncodedHeaderValue(string headerValue,
             }
         }
 
-        private class CheckHeaderMiddleware
-        {
-            private readonly RequestDelegate _next;
-            private readonly string _headerName;
-            private readonly string _headerValue;
-
-            public CheckHeaderMiddleware(RequestDelegate next, string headerName, string headerValue)
-            {
-                _next = next;
-                _headerName = headerName;
-                _headerValue = headerValue;
-            }
-
-            public async Task Invoke(HttpContext context)
-            {
-                Assert.True(context.Request.Headers.TryGetValue(_headerName, out var header));
-                var value = Assert.Single(header);
-                Assert.Equal(_headerValue, value);
-
-                await _next.Invoke(context);
-            }
-        }
-
         private class HeaderEncodingClientFactory : IProxyHttpClientFactory
         {
             private Encoding _encoding;