mirror of https://github.com/grpc/grpc.git
parent
de0d8b534f
commit
66a653df5e
2 changed files with 114 additions and 0 deletions
@ -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); |
||||
} |
||||
} |
||||
} |
Loading…
Reference in new issue