1 #region Copyright notice and license 2 3 // Copyright 2015 gRPC authors. 4 // 5 // Licensed under the Apache License, Version 2.0 (the "License"); 6 // you may not use this file except in compliance with the License. 7 // You may obtain a copy of the License at 8 // 9 // http://www.apache.org/licenses/LICENSE-2.0 10 // 11 // Unless required by applicable law or agreed to in writing, software 12 // distributed under the License is distributed on an "AS IS" BASIS, 13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 // See the License for the specific language governing permissions and 15 // limitations under the License. 16 17 #endregion 18 19 using System; 20 using System.Diagnostics; 21 using System.Linq; 22 using System.Threading; 23 using System.Threading.Tasks; 24 using Grpc.Core; 25 using Grpc.Core.Internal; 26 using Grpc.Core.Utils; 27 using NUnit.Framework; 28 29 namespace Grpc.Core.Tests 30 { 31 public class ShutdownHookPendingCallTest 32 { 33 const string Host = "127.0.0.1"; 34 35 [Test] ProcessExitHookCanCleanupAbandonedCall()36 public void ProcessExitHookCanCleanupAbandonedCall() 37 { 38 var helper = new MockServiceHelper(Host); 39 var server = helper.GetServer(); 40 server.Start(); 41 var channel = helper.GetChannel(); 42 43 var readyToShutdown = new TaskCompletionSource<object>(); 44 helper.DuplexStreamingHandler = new DuplexStreamingServerMethod<string, string>(async (requestStream, responseStream, context) => 45 { 46 readyToShutdown.SetResult(null); 47 await requestStream.ToListAsync(); 48 }); 49 50 var call = Calls.AsyncDuplexStreamingCall(helper.CreateDuplexStreamingCall()); 51 readyToShutdown.Task.Wait(); // make sure handler is running 52 } 53 } 54 } 55