public DropSubchannelPicker(DropLoadBalancer loadBalancer)
{
_loadBalancer = loadBalancer;
}
public async Task PickAsync_HedgingWithDrop_ThrowsError()
{
// Arrange
string?authority = null;
var testMessageHandler = TestHttpMessageHandler.Create(async request =>
{
authority = request.RequestUri !.Authority;
var reply = new HelloReply {
Message = "Hello world"
};
var streamContent = await ClientTestHelpers.CreateResponseContent(reply).DefaultTimeout();
return(ResponseUtils.CreateResponse(HttpStatusCode.OK, streamContent));
});
var services = new ServiceCollection();
services.AddNUnitLogger();
services.AddSingleton <TestResolver>();
services.AddSingleton <ResolverFactory, TestResolverFactory>();
DropLoadBalancer?loadBalancer = null;
services.AddSingleton <LoadBalancerFactory>(new DropLoadBalancerFactory(c =>
{
loadBalancer = new DropLoadBalancer(c);
return(loadBalancer);
}));
services.AddSingleton <ISubchannelTransportFactory>(new TestSubchannelTransportFactory());
var invoker = HttpClientCallInvokerFactory.Create(testMessageHandler, "test:///localhost", configure: o =>
{
o.Credentials = ChannelCredentials.Insecure;
o.ServiceProvider = services.BuildServiceProvider();
o.ServiceConfig = new ServiceConfig
{
MethodConfigs =
{
new MethodConfig
{
Names = { MethodName.Default },
HedgingPolicy = new HedgingPolicy
{
MaxAttempts = 5,
HedgingDelay = TimeSpan.FromMinutes(10),
NonFatalStatusCodes ={ StatusCode.DataLoss }
}
}
},
LoadBalancingConfigs = { new LoadBalancingConfig("drop") }
};
});
// Act
var call = invoker.AsyncUnaryCall <HelloRequest, HelloReply>(ClientTestHelpers.ServiceMethod, string.Empty, new CallOptions().WithWaitForReady(), new HelloRequest());
// Assert
var ex = await ExceptionAssert.ThrowsAsync <RpcException>(() => call.ResponseAsync).DefaultTimeout();
Assert.AreEqual(StatusCode.DataLoss, ex.StatusCode);
Assert.AreEqual(1, loadBalancer !.PickCount);
}
