static async Task TransportRouting_TryOpenFailedWithSupportedExceptionFirstTimes_SuccessOnSecondTry(Func <Exception> exceptionFactory)
{
var amqpTransportSettings = Substitute.For <ITransportSettings>();
var mqttTransportSettings = Substitute.For <ITransportSettings>();
var innerHandler = Substitute.For <IDelegatingHandler>();
innerHandler.CloseAsync().Returns(TaskConstants.Completed);
int openCallCounter = 0;
innerHandler.OpenAsync(Arg.Is(false), Arg.Any <CancellationToken>()).Returns(async ci =>
{
openCallCounter++;
await Task.Yield();
if (openCallCounter == 1)
{
throw exceptionFactory();
}
});
var sut = new RoutingDelegatingHandler((cs, ts) => innerHandler, null, new[] { amqpTransportSettings, mqttTransportSettings });
var cancellationToken = new CancellationToken();
await sut.OpenAsync(Arg.Is(false), cancellationToken);
await innerHandler.Received(1).CloseAsync();
Assert.AreEqual(2, openCallCounter);
}
public async Task TransportRouting_FirstTrySucceed_Open()
{
var amqpTransportSettings = Substitute.For <ITransportSettings>();
var mqttTransportSettings = Substitute.For <ITransportSettings>();
var innerHandler = Substitute.For <IDelegatingHandler>();
innerHandler.OpenAsync(Arg.Is(false), Arg.Any <CancellationToken>()).Returns(TaskConstants.Completed);
var sut = new RoutingDelegatingHandler((cs, ts) => innerHandler, null, new [] { amqpTransportSettings, mqttTransportSettings });
var cancellationToken = new CancellationToken();
await sut.OpenAsync(false, cancellationToken);
await innerHandler.Received(1).OpenAsync(Arg.Is(false), Arg.Any <CancellationToken>());
}
public async Task TransportRouting_CancellationTokenCanceled_Open()
{
var transportSettings = Substitute.For <ITransportSettings>();
var innerHandler = Substitute.For <IDelegatingHandler>();
innerHandler.OpenAsync(Arg.Any <bool>(), Arg.Any <CancellationToken>()).Returns(TaskConstants.Completed);
var sut = new RoutingDelegatingHandler((cs, ts) => innerHandler, null, new[] { transportSettings });
var userDefinedTimeoutCancellationTokenSource = new CancellationTokenSource();
userDefinedTimeoutCancellationTokenSource.Cancel();
await sut.OpenAsync(false, userDefinedTimeoutCancellationTokenSource.Token);
await innerHandler.Received(0).OpenAsync(Arg.Any <bool>(), userDefinedTimeoutCancellationTokenSource.Token);
}
public async Task TransportRouting_TryOpenFailedWithSupportedExceptionTwoTimes_Fail()
{
var amqpTransportSettings = Substitute.For <ITransportSettings>();
var mqttTransportSettings = Substitute.For <ITransportSettings>();
var innerHandler = Substitute.For <IDelegatingHandler>();
innerHandler.CloseAsync().Returns(TaskConstants.Completed);
int openCallCounter = 0;
innerHandler.OpenAsync(Arg.Is(false), Arg.Any <CancellationToken>()).Returns(async ci =>
{
openCallCounter++;
await Task.Yield();
throw new TimeoutException();
});
var sut = new RoutingDelegatingHandler((cs, ts) => innerHandler, null, new [] { amqpTransportSettings, mqttTransportSettings });
var cancellationToken = new CancellationToken();
await sut.OpenAsync(Arg.Is(false), cancellationToken).ExpectedAsync <IotHubCommunicationException>();
await innerHandler.Received(2).CloseAsync();
Assert.AreEqual(2, openCallCounter);
}
public async Task TransportRouting_TryOpenFailedWithUnsupportedException_FailOnFirstTry()
{
var amqpTransportSettings = Substitute.For <ITransportSettings>();
var mqttTransportSettings = Substitute.For <ITransportSettings>();
var innerHandler = Substitute.For <IDelegatingHandler>();
innerHandler.CloseAsync().Returns(TaskConstants.Completed);
int openCallCounter = 0;
innerHandler.OpenAsync(Arg.Is(false)).Returns(async ci =>
{
openCallCounter++;
await Task.Yield();
throw new InvalidOperationException();
});
var sut = new RoutingDelegatingHandler((cs, ts) => innerHandler, null, new [] { amqpTransportSettings, mqttTransportSettings });
await sut.OpenAsync(Arg.Is(false)).ExpectedAsync <InvalidOperationException>();
await innerHandler.Received(1).CloseAsync();
Assert.AreEqual(1, openCallCounter);
}
