Index: third_party/grpc/src/csharp/Grpc.IntegrationTesting/HeaderInterceptorTest.cs |
diff --git a/third_party/grpc/src/csharp/Grpc.IntegrationTesting/HeaderInterceptorTest.cs b/third_party/grpc/src/csharp/Grpc.IntegrationTesting/HeaderInterceptorTest.cs |
new file mode 100644 |
index 0000000000000000000000000000000000000000..1d758b75402087b73f91ec5eae4f1573f7168d1e |
--- /dev/null |
+++ b/third_party/grpc/src/csharp/Grpc.IntegrationTesting/HeaderInterceptorTest.cs |
@@ -0,0 +1,113 @@ |
+#region Copyright notice and license |
+ |
+// Copyright 2015, Google Inc. |
+// All rights reserved. |
+// |
+// Redistribution and use in source and binary forms, with or without |
+// modification, are permitted provided that the following conditions are |
+// met: |
+// |
+// * Redistributions of source code must retain the above copyright |
+// notice, this list of conditions and the following disclaimer. |
+// * Redistributions in binary form must reproduce the above |
+// copyright notice, this list of conditions and the following disclaimer |
+// in the documentation and/or other materials provided with the |
+// distribution. |
+// * Neither the name of Google Inc. nor the names of its |
+// contributors may be used to endorse or promote products derived from |
+// this software without specific prior written permission. |
+// |
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
+// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
+// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
+// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
+// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
+// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
+// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
+// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
+// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
+// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
+ |
+#endregion |
+ |
+using System; |
+using System.Collections.Generic; |
+using System.Linq; |
+using System.Threading; |
+using System.Threading.Tasks; |
+using Grpc.Core; |
+using Grpc.Core.Utils; |
+using Grpc.Testing; |
+using NUnit.Framework; |
+ |
+namespace Grpc.IntegrationTesting |
+{ |
+ public class HeaderInterceptorTest |
+ { |
+ const string Host = "localhost"; |
+ Server server; |
+ Channel channel; |
+ TestService.TestServiceClient client; |
+ |
+ [TestFixtureSetUp] |
+ public void Init() |
+ { |
+ server = new Server |
+ { |
+ Services = { TestService.BindService(new TestServiceImpl()) }, |
+ Ports = { { Host, ServerPort.PickUnused, ServerCredentials.Insecure } } |
+ }; |
+ server.Start(); |
+ |
+ channel = new Channel(Host, server.Ports.Single().BoundPort, ChannelCredentials.Insecure); |
+ client = TestService.NewClient(channel); |
+ } |
+ |
+ [TestFixtureTearDown] |
+ public void Cleanup() |
+ { |
+ channel.ShutdownAsync().Wait(); |
+ server.ShutdownAsync().Wait(); |
+ } |
+ |
+ [Test] |
+ public async Task HeaderInterceptor_CreateMetadata() |
+ { |
+ var key = "x-grpc-test-echo-initial"; |
+ client.HeaderInterceptor = new HeaderInterceptor((method, metadata) => |
+ { |
+ metadata.Add(key, "ABC"); |
+ }); |
+ |
+ var call = client.UnaryCallAsync(new SimpleRequest()); |
+ await call; |
+ |
+ var responseHeaders = await call.ResponseHeadersAsync; |
+ Assert.AreEqual("ABC", responseHeaders.First((entry) => entry.Key == key).Value); |
+ } |
+ |
+ [Test] |
+ public async Task HeaderInterceptor_AppendMetadata() |
+ { |
+ var initialKey = "x-grpc-test-echo-initial"; |
+ var trailingKey = "x-grpc-test-echo-trailing-bin"; |
+ |
+ client.HeaderInterceptor = new HeaderInterceptor((method, metadata) => |
+ { |
+ metadata.Add(initialKey, "ABC"); |
+ }); |
+ |
+ var headers = new Metadata |
+ { |
+ { trailingKey, new byte[] {0xaa} } |
+ }; |
+ var call = client.UnaryCallAsync(new SimpleRequest(), headers: headers); |
+ await call; |
+ |
+ var responseHeaders = await call.ResponseHeadersAsync; |
+ Assert.AreEqual("ABC", responseHeaders.First((entry) => entry.Key == initialKey).Value); |
+ CollectionAssert.AreEqual(new byte[] {0xaa}, call.GetTrailers().First((entry) => entry.Key == trailingKey).ValueBytes); |
+ } |
+ } |
+} |