public void ShouldWaitProvidedTime()
{
// given
var retries = 0;
var countdownEvent = new CountdownEvent(2);
var strategy = new TransientGrpcErrorRetryStrategy(retry => TimeSpan.FromSeconds(1));
// when
strategy.DoWithRetry(() =>
{
countdownEvent.Signal();
if (retries == 3)
{
return(Task.FromResult(retries));
}
retries++;
throw new RpcException(new Status(StatusCode.ResourceExhausted, "resourceExhausted"));
});
countdownEvent.Wait(TimeSpan.FromMilliseconds(10));
// then
Assert.AreEqual(countdownEvent.CurrentCount, 1);
Assert.AreEqual(retries, 1);
}
public async Task ShouldIncrementRetriesOnWaitTimeProvider()
{
// given
var retries = 0;
var values = new List <int>();
var strategy = new TransientGrpcErrorRetryStrategy(retry =>
{
values.Add(retry);
return(TimeSpan.Zero);
});
// when
var result = await strategy.DoWithRetry(() =>
{
if (retries == 3)
{
return(Task.FromResult(retries));
}
retries++;
throw new RpcException(new Status(StatusCode.ResourceExhausted, "resourceExhausted"));
});
// then
Assert.AreEqual(3, result);
CollectionAssert.AreEqual(new List <int> {
1, 2, 3
}, values);
}
public async Task ShouldRetryOnUnavailableException()
{
// given
int retries = 0;
var strategy = new TransientGrpcErrorRetryStrategy(retry => TimeSpan.Zero);
// when
var result = await strategy.DoWithRetry(() =>
{
if (retries == 3)
{
return(Task.FromResult(retries));
}
retries++;
throw new RpcException(new Status(StatusCode.Unavailable, "resourceExhausted"));
});
// then
Assert.AreEqual(3, result);
}
public void ShouldNotRetryOnOtherException()
{
// given
int retries = 0;
var strategy = new TransientGrpcErrorRetryStrategy(retry => TimeSpan.Zero);
// when
var resultTask = strategy.DoWithRetry(() =>
{
if (retries == 3)
{
return(Task.FromResult(retries));
}
retries++;
throw new Exception("exception");
});
// then
var aggregateException = Assert.Throws <AggregateException>(() => resultTask.Wait());
var exception = aggregateException.InnerExceptions[0];
Assert.AreEqual("exception", exception.Message);
}
public void ShouldNotRetryOnOtherRpcException()
{
// given
int retries = 0;
var strategy = new TransientGrpcErrorRetryStrategy(retry => TimeSpan.Zero);
// when
var resultTask = strategy.DoWithRetry(() =>
{
if (retries == 3)
{
return(Task.FromResult(retries));
}
retries++;
throw new RpcException(new Status(StatusCode.Unknown, "idk"));
});
// then
var aggregateException = Assert.Throws <AggregateException>(() => resultTask.Wait());
var rpcException = (RpcException)aggregateException.InnerExceptions[0];
Assert.AreEqual(StatusCode.Unknown, rpcException.Status.StatusCode);
}
