mirror of https://github.com/grpc/grpc.git
* C# add RetryTest with service config to reproduce #23618 * fixup testspull/29305/head
parent
1973b4f9b1
commit
2223877d41
2 changed files with 120 additions and 0 deletions
@ -0,0 +1,119 @@ |
||||
#region Copyright notice and license |
||||
|
||||
// Copyright 2020 The gRPC Authors |
||||
// |
||||
// Licensed under the Apache License, Version 2.0 (the "License"); |
||||
// you may not use this file except in compliance with the License. |
||||
// You may obtain a copy of the License at |
||||
// |
||||
// http://www.apache.org/licenses/LICENSE-2.0 |
||||
// |
||||
// Unless required by applicable law or agreed to in writing, software |
||||
// distributed under the License is distributed on an "AS IS" BASIS, |
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
// See the License for the specific language governing permissions and |
||||
// limitations under the License. |
||||
|
||||
#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 |
||||
{ |
||||
/// <summary> |
||||
/// Tests for service-config based retries. |
||||
/// </summary> |
||||
public class RetryTest |
||||
{ |
||||
MockServiceHelper helper; |
||||
Server server; |
||||
Channel channel; |
||||
|
||||
[SetUp] |
||||
public void Init() |
||||
{ |
||||
var channelOptions = new ChannelOption[] |
||||
{ |
||||
new ChannelOption( |
||||
"grpc.service_config", |
||||
"{\"methodConfig\":[{\"name\":[{\"service\":\"" + MockServiceHelper.ServiceName + "\",\"method\":\"Unary\"}],\"retryPolicy\":{\"maxAttempts\":3,\"initialBackoff\":\"0.5s\",\"maxBackoff\":\"5s\",\"backoffMultiplier\":2.0,\"retryableStatusCodes\":[\"UNAVAILABLE\"]}}, {\"name\":[{\"service\":\"" + MockServiceHelper.ServiceName + "\",\"method\":\"ServerStreaming\"}],\"retryPolicy\":{\"maxAttempts\":3,\"initialBackoff\":\"0.5s\",\"maxBackoff\":\"5s\",\"backoffMultiplier\":2.0,\"retryableStatusCodes\":[\"UNAVAILABLE\"]}}]}") |
||||
}; |
||||
helper = new MockServiceHelper(channelOptions: channelOptions); |
||||
|
||||
server = helper.GetServer(); |
||||
server.Start(); |
||||
channel = helper.GetChannel(); |
||||
} |
||||
|
||||
[TearDown] |
||||
public void Cleanup() |
||||
{ |
||||
channel.ShutdownAsync().Wait(); |
||||
server.ShutdownAsync().Wait(); |
||||
} |
||||
|
||||
[Test] |
||||
public void ServiceConfigRetryPolicy_UnaryCall() |
||||
{ |
||||
var counter = new AtomicCounter(); |
||||
|
||||
helper.UnaryHandler = new UnaryServerMethod<string, string>((request, context) => |
||||
{ |
||||
var attempt = counter.Increment(); |
||||
if (attempt <= 2) |
||||
{ |
||||
throw new RpcException(new Status(StatusCode.Unavailable, $"Attempt {attempt} failed on purpose")); |
||||
} |
||||
return Task.FromResult("PASS"); |
||||
}); |
||||
|
||||
Assert.AreEqual("PASS", Calls.BlockingUnaryCall(helper.CreateUnaryCall(), "abc")); |
||||
} |
||||
|
||||
[Test] |
||||
public async Task ServiceConfigRetryPolicy_AsyncUnaryCall() |
||||
{ |
||||
var counter = new AtomicCounter(); |
||||
|
||||
helper.UnaryHandler = new UnaryServerMethod<string, string>((request, context) => |
||||
{ |
||||
var attempt = counter.Increment(); |
||||
if (attempt <= 2) |
||||
{ |
||||
throw new RpcException(new Status(StatusCode.Unavailable, $"Attempt {attempt} failed on purpose")); |
||||
} |
||||
return Task.FromResult("PASS"); |
||||
}); |
||||
|
||||
Assert.AreEqual("PASS", await Calls.AsyncUnaryCall(helper.CreateUnaryCall(), "abc")); |
||||
} |
||||
|
||||
[Test] |
||||
public async Task ServiceConfigRetryPolicy_ServerStreaming() |
||||
{ |
||||
var counter = new AtomicCounter(); |
||||
|
||||
helper.ServerStreamingHandler = new ServerStreamingServerMethod<string, string>(async (request, responseStream, context) => |
||||
{ |
||||
var attempt = counter.Increment(); |
||||
if (attempt <= 2) |
||||
{ |
||||
throw new RpcException(new Status(StatusCode.Unavailable, $"Attempt {attempt} failed on purpose")); |
||||
} |
||||
await responseStream.WriteAllAsync(request.Split(new []{' '})); |
||||
}); |
||||
|
||||
var call = Calls.AsyncServerStreamingCall(helper.CreateServerStreamingCall(), "A B C"); |
||||
CollectionAssert.AreEqual(new string[] { "A", "B", "C" }, await call.ResponseStream.ToListAsync()); |
||||
Assert.AreEqual(StatusCode.OK, call.GetStatus().StatusCode); |
||||
} |
||||
} |
||||
} |
Loading…
Reference in new issue