Add functional tests for using HttpContext (#106)

Author: JamesNK

test/FunctionalTests/HttpContextTests.cs

@@ -0,0 +1,86 @@
+#region Copyright notice and license
+
+// Copyright 2019 The gRPC Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#endregion
+
+using System.IO;
+using System.IO.Pipelines;
+using System.Linq;
+using System.Net;
+using System.Net.Http;
+using System.Threading.Tasks;
+using Greet;
+using Grpc.AspNetCore.FunctionalTests.Infrastructure;
+using Grpc.AspNetCore.Server.Internal;
+using Grpc.Core;
+using NUnit.Framework;
+
+namespace Grpc.AspNetCore.FunctionalTests
+{
+    [TestFixture]
+    public class HttpContextTests : FunctionalTestBase
+    {
+        [Test]
+        public async Task HttpContextAccessor_ReturnContextInTrailer()
+        {
+            // Arrange
+            var requestMessage = new HelloRequest
+            {
+                Name = "World"
+            };
+
+            var requestStream = new MemoryStream();
+            MessageHelpers.WriteMessage(requestStream, requestMessage);
+
+            var httpRequest = new HttpRequestMessage(HttpMethod.Post, "Greet.Greeter/SayHelloWithHttpContextAccessor?query=extra");
+            httpRequest.Content = new StreamContent(requestStream);
+
+            // Act
+            var response = await Fixture.Client.SendAsync(httpRequest).DefaultTimeout();
+
+            // Assert
+            Assert.AreEqual(HttpStatusCode.OK, response.StatusCode);
+
+            Assert.AreEqual(StatusCode.OK.ToTrailerString(), Fixture.TrailersContainer.Trailers[GrpcProtocolConstants.StatusTrailer].ToString());
+            Assert.AreEqual("/Greet.Greeter/SayHelloWithHttpContextAccessor?query=extra", Fixture.TrailersContainer.Trailers["Test-HttpContext-PathAndQueryString"].ToString());
+        }
+
+        [Test]
+        public async Task HttpContextExtensionMethod_ReturnContextInTrailer()
+        {
+            // Arrange
+            var requestMessage = new HelloRequest
+            {
+                Name = "World"
+            };
+
+            var requestStream = new MemoryStream();
+            MessageHelpers.WriteMessage(requestStream, requestMessage);
+
+            var httpRequest = new HttpRequestMessage(HttpMethod.Post, "Greet.Greeter/SayHelloWithHttpContextExtensionMethod?query=extra");
+            httpRequest.Content = new StreamContent(requestStream);
+
+            // Act
+            var response = await Fixture.Client.SendAsync(httpRequest).DefaultTimeout();
+
+            // Assert
+            Assert.AreEqual(HttpStatusCode.OK, response.StatusCode);
+
+            Assert.AreEqual(StatusCode.OK.ToTrailerString(), Fixture.TrailersContainer.Trailers[GrpcProtocolConstants.StatusTrailer].ToString());
+            Assert.AreEqual("/Greet.Greeter/SayHelloWithHttpContextExtensionMethod?query=extra", Fixture.TrailersContainer.Trailers["Test-HttpContext-PathAndQueryString"].ToString());
+        }
+    }
+}

testassets/FunctionalTestsWebsite/Greeter.cs

@@ -21,15 +21,18 @@
 using System.Threading.Tasks;
 using Greet;
 using Grpc.Core;
+using Microsoft.AspNetCore.Http;
 using Microsoft.Extensions.Logging;
 
 class GreeterService : Greeter.GreeterBase
 {
     private readonly ILogger _logger;
+    private readonly IHttpContextAccessor _httpContextAccessor;
 
-    public GreeterService(ILoggerFactory loggerFactory)
+    public GreeterService(ILoggerFactory loggerFactory, IHttpContextAccessor httpContextAccessor)
     {
         _logger = loggerFactory.CreateLogger<GreeterService>();
+        _httpContextAccessor = httpContextAccessor;
     }
 
     //Server side handler of the SayHello RPC
@@ -39,6 +42,22 @@ public override Task<HelloReply> SayHello(HelloRequest request, ServerCallContex
         return Task.FromResult(new HelloReply { Message = "Hello " + request.Name });
     }
 
+    public override Task<HelloReply> SayHelloWithHttpContextAccessor(HelloRequest request, ServerCallContext context)
+    {
+        var httpContext = _httpContextAccessor.HttpContext;
+        context.ResponseTrailers.Add("Test-HttpContext-PathAndQueryString", httpContext.Request.Path + httpContext.Request.QueryString);
+
+        return Task.FromResult(new HelloReply { Message = "Hello " + request.Name });
+    }
+
+    public override Task<HelloReply> SayHelloWithHttpContextExtensionMethod(HelloRequest request, ServerCallContext context)
+    {
+        var httpContext = context.GetHttpContext();
+        context.ResponseTrailers.Add("Test-HttpContext-PathAndQueryString", httpContext.Request.Path + httpContext.Request.QueryString);
+
+        return Task.FromResult(new HelloReply { Message = "Hello " + request.Name });
+    }
+
     public override Task<HelloReply> SayHelloSendLargeReply(HelloRequest request, ServerCallContext context)
     {
         _logger.LogInformation($"Sending hello to {request.Name}");

testassets/FunctionalTestsWebsite/Startup.cs

@@ -38,6 +38,8 @@ public void ConfigureServices(IServiceCollection services)
                     options.SendMaxMessageSize = 64 * 1024;
                     options.ReceiveMaxMessageSize = 64 * 1024;
                 });
+            services.AddHttpContextAccessor();
+
             services.AddScoped<IncrementingCounter>();
 
             // When the site is run from the test project a signaler will already be registered

testassets/Proto/greet.proto

@@ -20,6 +20,8 @@ package Greet;
 service Greeter {
   // Sends a greeting
   rpc SayHello (HelloRequest) returns (HelloReply) {}
+  rpc SayHelloWithHttpContextAccessor (HelloRequest) returns (HelloReply) {}
+  rpc SayHelloWithHttpContextExtensionMethod (HelloRequest) returns (HelloReply) {}
   rpc SayHelloSendLargeReply (HelloRequest) returns (HelloReply) {}
   rpc SayHelloSendHeadersTwice (HelloRequest) returns (HelloReply) {}
   rpc SayHellos (HelloRequest) returns (stream HelloReply) {}