Fix tests

grpc · Feb 17, 2021 · e549c80 · e549c80
1 parent 31eaaf8
commit e549c80
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/Grpc.Net.Client.Tests/AsyncUnaryCallTests.cs b/test/Grpc.Net.Client.Tests/AsyncUnaryCallTests.cs
@@ -43,10 +43,12 @@ public async Task AsyncUnaryCall_Success_HttpRequestMessagePopulated()
         {
             // Arrange
             HttpRequestMessage? httpRequestMessage = null;
+            long? requestContentLength = null;
 
             var httpClient = ClientTestHelpers.CreateTestClient(async request =>
             {
                 httpRequestMessage = request;
+                requestContentLength = httpRequestMessage!.Content!.Headers!.ContentLength;
 
                 HelloReply reply = new HelloReply
                 {
@@ -72,7 +74,7 @@ public async Task AsyncUnaryCall_Success_HttpRequestMessagePopulated()
             Assert.AreEqual(new MediaTypeHeaderValue("application/grpc"), httpRequestMessage.Content?.Headers?.ContentType);
             Assert.AreEqual(GrpcProtocolConstants.TEHeaderValue, httpRequestMessage.Headers.TE.Single().Value);
             Assert.AreEqual("identity,gzip", httpRequestMessage.Headers.GetValues(GrpcProtocolConstants.MessageAcceptEncodingHeader).Single());
-            Assert.AreEqual(null, httpRequestMessage!.Content!.Headers!.ContentLength);
+            Assert.AreEqual(null, requestContentLength);
 
             var userAgent = httpRequestMessage.Headers.UserAgent.Single()!;
             Assert.AreEqual("grpc-dotnet", userAgent.Product?.Name);
@@ -89,10 +91,12 @@ public async Task AsyncUnaryCall_HasWinHttpHandler_ContentLengthOnHttpRequestMes
         {
             // Arrange
             HttpRequestMessage? httpRequestMessage = null;
+            long? requestContentLength = null;
 
             var handler = TestHttpMessageHandler.Create(async request =>
             {
                 httpRequestMessage = request;
+                requestContentLength = httpRequestMessage!.Content!.Headers!.ContentLength;
 
                 HelloReply reply = new HelloReply
                 {
@@ -114,7 +118,7 @@ public async Task AsyncUnaryCall_HasWinHttpHandler_ContentLengthOnHttpRequestMes
             Assert.AreEqual("Hello world", rs.Message);
 
             Assert.IsNotNull(httpRequestMessage);
-            Assert.AreEqual(18, httpRequestMessage!.Content!.Headers!.ContentLength);
+            Assert.AreEqual(18, requestContentLength);
         }
 
         [Test]