Skip to content
Snippets Groups Projects
Select Git revision
  • c75c57c5af620491d0043047533fa0e2f078b09f
  • master default protected
  • arm-aarch-platform
  • arm-platform
  • vjpai-patch-3
  • vjpai-patch-1
  • v1.27.x
  • jtattermusch-patch-2
  • jtattermusch-patch-1
  • update-java-worker-example-in-performance-docs
  • revert-21805-revert-21797-reintroduce_21527
  • revert-21804-tls-credentials-1
  • zhen_cleanup_namecheck
  • revert-21806-revert-21767-revert-21725-revert-21680-cq_ordering
  • vjpai-patch-2
  • revert-21766-tls-credentials-1
  • revert-21640-change_local_tcp_security_level
  • revert-21680-cq_ordering
  • revert-21527-unify_boringssl_deps2
  • revert-20803-grpclb_stabilization
  • fix-kokoro-rvm-key
  • v1.27.0
  • v1.27.0-pre2
  • v1.27.0-pre1
  • v1.26.0
  • v1.26.0-pre1
  • v1.25.0
  • v1.25.0-pre1
  • v1.24.3
  • v1.24.2
  • v1.24.1
  • v1.23.1
  • v1.24.0
  • v1.24.0-pre2
  • v1.24.0-pre1
  • v1.22.1
  • v1.23.0
  • v1.23.0-pre1
  • v1.22.0
  • v1.22.0-pre1
  • v1.21.4
41 results

ClientServerTest.cs

Blame
  • ClientServerTest.cs 10.96 KiB
    #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.Diagnostics;
    using System.Linq;
    using System.Threading;
    using System.Threading.Tasks;
    using Grpc.Core;
    using Grpc.Core.Internal;
    using Grpc.Core.Utils;
    using NUnit.Framework;
    
    namespace Grpc.Core.Tests
    {
        public class ClientServerTest
        {
            const string Host = "127.0.0.1";
    
            MockServiceHelper helper;
            Server server;
            Channel channel;
    
            [SetUp]
            public void Init()
            {
                helper = new MockServiceHelper(Host);
                server = helper.GetServer();
                server.Start();
                channel = helper.GetChannel();
            }
    
            [TearDown]
            public void Cleanup()
            {
                channel.Dispose();
                server.ShutdownAsync().Wait();
            }
    
            [TestFixtureTearDown]
            public void CleanupClass()
            {
                GrpcEnvironment.Shutdown();
            }
    
            [Test]
            public async Task UnaryCall()
            {
                helper.UnaryHandler = new UnaryServerMethod<string, string>(async (request, context) =>
                {
                    return request;
                });
    
                Assert.AreEqual("ABC", Calls.BlockingUnaryCall(helper.CreateUnaryCall(), "ABC"));
    
                Assert.AreEqual("ABC", await Calls.AsyncUnaryCall(helper.CreateUnaryCall(), "ABC"));
            }
    
            [Test]
            public void UnaryCall_ServerHandlerThrows()
            {
                helper.UnaryHandler = new UnaryServerMethod<string, string>((request, context) =>
                {
                    throw new Exception("This was thrown on purpose by a test");
                });
                    
                var ex = Assert.Throws<RpcException>(() => Calls.BlockingUnaryCall(helper.CreateUnaryCall(), "abc"));
                Assert.AreEqual(StatusCode.Unknown, ex.Status.StatusCode); 
    
                var ex2 = Assert.Throws<RpcException>(async () => await Calls.AsyncUnaryCall(helper.CreateUnaryCall(), "abc"));
                Assert.AreEqual(StatusCode.Unknown, ex2.Status.StatusCode);
            }
    
            [Test]
            public void UnaryCall_ServerHandlerThrowsRpcException()
            {
                helper.UnaryHandler = new UnaryServerMethod<string, string>((request, context) =>
                {
                    throw new RpcException(new Status(StatusCode.Unauthenticated, ""));
                });
    
                var ex = Assert.Throws<RpcException>(() => Calls.BlockingUnaryCall(helper.CreateUnaryCall(), "abc"));
                Assert.AreEqual(StatusCode.Unauthenticated, ex.Status.StatusCode);
    
                var ex2 = Assert.Throws<RpcException>(async () => await Calls.AsyncUnaryCall(helper.CreateUnaryCall(), "abc"));
                Assert.AreEqual(StatusCode.Unauthenticated, ex2.Status.StatusCode);
            }
    
            [Test]
            public void UnaryCall_ServerHandlerSetsStatus()
            {
                helper.UnaryHandler = new UnaryServerMethod<string, string>(async (request, context) =>
                {
                    context.Status = new Status(StatusCode.Unauthenticated, "");
                    return "";
                });
    
                var ex = Assert.Throws<RpcException>(() => Calls.BlockingUnaryCall(helper.CreateUnaryCall(), "abc"));
                Assert.AreEqual(StatusCode.Unauthenticated, ex.Status.StatusCode);
    
                var ex2 = Assert.Throws<RpcException>(async () => await Calls.AsyncUnaryCall(helper.CreateUnaryCall(), "abc"));
                Assert.AreEqual(StatusCode.Unauthenticated, ex2.Status.StatusCode);
            }
    
            [Test]
            public async Task ClientStreamingCall()
            {
                helper.ClientStreamingHandler = new ClientStreamingServerMethod<string, string>(async (requestStream, context) =>
                {
                    string result = "";
                    await requestStream.ForEach(async (request) =>
                    {
                        result += request;
                    });
                    await Task.Delay(100);
                    return result;
                });
    
                var call = Calls.AsyncClientStreamingCall(helper.CreateClientStreamingCall());
                await call.RequestStream.WriteAll(new string[] { "A", "B", "C" });
                Assert.AreEqual("ABC", await call.ResponseAsync);
            }
    
            [Test]
            public async Task ClientStreamingCall_CancelAfterBegin()
            {
                var barrier = new TaskCompletionSource<object>();
    
                helper.ClientStreamingHandler = new ClientStreamingServerMethod<string, string>(async (requestStream, context) =>
                {
                    barrier.SetResult(null);
                    await requestStream.ToList();
                    return "";
                });
    
                var cts = new CancellationTokenSource();
                var call = Calls.AsyncClientStreamingCall(helper.CreateClientStreamingCall(new CallOptions(cancellationToken: cts.Token)));
    
                await barrier.Task;  // make sure the handler has started.
                cts.Cancel();
    
                var ex = Assert.Throws<RpcException>(async () => await call.ResponseAsync);
                Assert.AreEqual(StatusCode.Cancelled, ex.Status.StatusCode);
            }
    
            [Test]
            public async Task AsyncUnaryCall_EchoMetadata()
            {
                helper.UnaryHandler = new UnaryServerMethod<string, string>(async (request, context) =>
                {
                    foreach (Metadata.Entry metadataEntry in context.RequestHeaders)
                    {
                        if (metadataEntry.Key != "user-agent")
                        {
                            context.ResponseTrailers.Add(metadataEntry);
                        }
                    }
                    return "";
                });
    
                var headers = new Metadata
                {
                    new Metadata.Entry("ascii-header", "abcdefg"),
                    new Metadata.Entry("binary-header-bin", new byte[] { 1, 2, 3, 0, 0xff }),
                };
                var call = Calls.AsyncUnaryCall(helper.CreateUnaryCall(new CallOptions(headers: headers)), "ABC");
                await call;
    
                Assert.AreEqual(StatusCode.OK, call.GetStatus().StatusCode);
    
                var trailers = call.GetTrailers();
                Assert.AreEqual(2, trailers.Count);
                Assert.AreEqual(headers[0].Key, trailers[0].Key);
                Assert.AreEqual(headers[0].Value, trailers[0].Value);
    
                Assert.AreEqual(headers[1].Key, trailers[1].Key);
                CollectionAssert.AreEqual(headers[1].ValueBytes, trailers[1].ValueBytes);
            }
    
            [Test]
            public void UnaryCall_DisposedChannel()
            {
                channel.Dispose();
                Assert.Throws(typeof(ObjectDisposedException), () => Calls.BlockingUnaryCall(helper.CreateUnaryCall(), "ABC"));
            }
    
            [Test]
            public void UnaryCallPerformance()
            {
                helper.UnaryHandler = new UnaryServerMethod<string, string>(async (request, context) =>
                {
                    return request;
                });
    
                var callDetails = helper.CreateUnaryCall();
                BenchmarkUtil.RunBenchmark(100, 100,
                                           () => { Calls.BlockingUnaryCall(callDetails, "ABC"); });
            }
                
            [Test]
            public void UnknownMethodHandler()
            {
                var nonexistentMethod = new Method<string, string>(
                    MethodType.Unary,
                    MockServiceHelper.ServiceName,
                    "NonExistentMethod",
                    Marshallers.StringMarshaller,
                    Marshallers.StringMarshaller);
    
                var callDetails = new CallInvocationDetails<string, string>(channel, nonexistentMethod, new CallOptions());
    
                var ex = Assert.Throws<RpcException>(() => Calls.BlockingUnaryCall(callDetails, "abc"));
                Assert.AreEqual(StatusCode.Unimplemented, ex.Status.StatusCode);
            }
    
            [Test]
            public void UserAgentStringPresent()
            {
                helper.UnaryHandler = new UnaryServerMethod<string, string>(async (request, context) =>
                {
                    return context.RequestHeaders.Where(entry => entry.Key == "user-agent").Single().Value;
                });
    
                string userAgent = Calls.BlockingUnaryCall(helper.CreateUnaryCall(), "abc");
                Assert.IsTrue(userAgent.StartsWith("grpc-csharp/"));
            }
    
            [Test]
            public void PeerInfoPresent()
            {
                helper.UnaryHandler = new UnaryServerMethod<string, string>(async (request, context) =>
                {
                    return context.Peer;
                });
    
                string peer = Calls.BlockingUnaryCall(helper.CreateUnaryCall(), "abc");
                Assert.IsTrue(peer.Contains(Host));
            }
    
            [Test]
            public async Task Channel_WaitForStateChangedAsync()
            {
                helper.UnaryHandler = new UnaryServerMethod<string, string>(async (request, context) =>
                {
                    return request;
                });
    
                Assert.Throws(typeof(TaskCanceledException), 
                    async () => await channel.WaitForStateChangedAsync(channel.State, DateTime.UtcNow.AddMilliseconds(10)));
    
                var stateChangedTask = channel.WaitForStateChangedAsync(channel.State);
    
                await Calls.AsyncUnaryCall(helper.CreateUnaryCall(), "abc");
    
                await stateChangedTask;
                Assert.AreEqual(ChannelState.Ready, channel.State);
            }
    
            [Test]
            public async Task Channel_ConnectAsync()
            {
                await channel.ConnectAsync();
                Assert.AreEqual(ChannelState.Ready, channel.State);
    
                await channel.ConnectAsync(DateTime.UtcNow.AddMilliseconds(1000));
                Assert.AreEqual(ChannelState.Ready, channel.State);
            }
        }
    }