public void Should_cancel_remaining_requests_when_receiving_accepted_result()
{
DropSynchronizationContext();
var tokens = new List <CancellationToken>();
sender
.When(s => s.SendToReplicaAsync(Arg.Any <Uri>(), Arg.Any <Request>(), Arg.Any <TimeSpan>(), Arg.Any <CancellationToken>()))
.Do(info => tokens.Add(info.Arg <CancellationToken>()));
strategy = new ParallelRequestStrategy(int.MaxValue);
var sendTask = strategy.SendAsync(request, sender, Budget.WithRemaining(5.Seconds()), replicas, replicas.Length, token);
CompleteRequest(replicas.Last(), ResponseVerdict.Accept);
sendTask.GetAwaiter().GetResult();
tokens.Should().HaveCount(replicas.Length);
foreach (var t in tokens)
{
t.IsCancellationRequested.Should().BeTrue();
}
}
public void Should_fire_initial_requests_to_all_replicas_if_parallelism_level_is_greater_than_replicas_count()
{
strategy = new ParallelRequestStrategy(int.MaxValue);
strategy.SendAsync(request, parameters, sender, Budget.WithRemaining(5.Seconds()), replicas, replicas.Length, token);
sender.ReceivedCalls().Should().HaveCount(replicas.Length);
foreach (var replica in replicas)
{
sender.Received(1).SendToReplicaAsync(replica, request, Arg.Any <TimeSpan?>(), Arg.Any <TimeSpan>(), Arg.Any <CancellationToken>());
}
}
public SimpleClusterClient(Uri[] addressList, TimeSpan?timeout = null, ILog log = null)
{
replicasCount = addressList.Length;
globalTimeout = timeout ?? TimeSpan.FromSeconds(15);
clusterClient = new ClusterClient(log, config =>
{
config.ClusterProvider = new ConstantTopologyClusterProvider(addressList);
config.SetupWebRequestTransport();
config.SetupWeighedReplicaOrdering(builder => { builder.AddAdaptiveHealthModifierWithLinearDecay(TuningPolicies.ByResponseVerdict, 10.Minutes()); });
});
forkingRequestStrategy = new ForkingRequestStrategy(new EqualDelaysProvider(addressList.Length), addressList.Length);
parallelRequestStrategy = new ParallelRequestStrategy(addressList.Length);
}
public void Should_ignore_connection_timeout()
{
strategy = new ParallelRequestStrategy(int.MaxValue);
parameters = parameters.WithConnectionTimeout(5.Seconds());
strategy.SendAsync(request, parameters, sender, Budget.WithRemaining(5.Seconds()), replicas, replicas.Length, token);
sender.ReceivedCalls().Should().HaveCount(replicas.Length);
foreach (var replica in replicas)
{
sender.Received(1).SendToReplicaAsync(replica, Arg.Any <Request>(), null, Arg.Any <TimeSpan>(), Arg.Any <CancellationToken>());
}
}
public void SetUp()
{
request = Request.Get("foo/bar");
replicas = Enumerable.Range(0, 10).Select(i => new Uri($"http://replica-{i}/")).ToArray();
resultSources = replicas.ToDictionary(r => r, _ => new TaskCompletionSource <ReplicaResult>());
sender = Substitute.For <IRequestSender>();
sender
.SendToReplicaAsync(Arg.Any <Uri>(), Arg.Any <Request>(), Arg.Any <TimeSpan>(), Arg.Any <CancellationToken>())
.Returns(info => resultSources[info.Arg <Uri>()].Task);
token = new CancellationTokenSource().Token;
strategy = new ParallelRequestStrategy(3);
}
