public async Task ShouldThrowTimeoutExceptionOnTimeout(Mock <IContextHelper> contextHelper, ISeedingHelper seedingHelper)
{
// Assert
contextHelper.Setup(c => c.IsMigrated(It.IsAny <CancellationToken>())).ReturnsAsync(false);
var sut = new MigrationWaitHelper(contextHelper.Object, seedingHelper, this.Logger);
// Act & Assert
await Assert.ThrowsAsync <TimeoutException>(
() => sut.WaitForMigrationsToComplete(TimeSpan.FromSeconds(5), DefaultPollTimeSpan, CancellationToken.None));
contextHelper.Verify(c => c.IsMigrated(It.IsAny <CancellationToken>()), Times.AtLeast(2));
}
public async Task ShouldWaitForMigrationToComplete(Mock <IContextHelper> contextHelper, ISeedingHelper seedingHelper)
{
// Assert
var completedAt = DateTimeOffset.Now.AddSeconds(5);
contextHelper.Setup(c => c.IsMigrated(It.IsAny <CancellationToken>())).Returns <CancellationToken>(cts => Task.FromResult(DateTimeOffset.Now > completedAt));
var sut = new MigrationWaitHelper(contextHelper.Object, seedingHelper, this.Logger);
// Act
await sut.WaitForMigrationsToComplete(DefaultWaitTimeSpan, DefaultPollTimeSpan, CancellationToken.None);
// Assert
contextHelper.Verify(c => c.IsMigrated(It.IsAny <CancellationToken>()), Times.AtLeast(3));
}
public async Task ShouldReturnOnMigrated(Mock <IContextHelper> contextHelper, Mock <ISeedingHelper> seedingHelper)
{
// Assert
contextHelper.Setup(s => s.IsMigrated(It.IsAny <CancellationToken>())).ReturnsAsync(true);
seedingHelper.Setup(s => s.IsSeeded(It.IsAny <CancellationToken>())).ReturnsAsync(true);
var sut = new MigrationWaitHelper(contextHelper.Object, seedingHelper.Object, this.Logger);
// Act
await sut.WaitForMigrationsToComplete(DefaultWaitTimeSpan, DefaultPollTimeSpan, CancellationToken.None);
// Assert
contextHelper.Verify(c => c.IsMigrated(It.IsAny <CancellationToken>()), Times.Once);
seedingHelper.Verify(c => c.IsSeeded(It.IsAny <CancellationToken>()), Times.Once);
}
public async Task ShouldThrowOperationCancelledExceptionOnExternalCancellation(Mock <IContextHelper> contextHelper, ISeedingHelper seedingHelper)
{
// Assert
contextHelper.Setup(c => c.IsMigrated(It.IsAny <CancellationToken>())).ReturnsAsync(false);
var sut = new MigrationWaitHelper(contextHelper.Object, seedingHelper, this.Logger);
using (var cts = new CancellationTokenSource(TimeSpan.FromSeconds(5)))
{
// Act & Assert
await Assert.ThrowsAnyAsync <OperationCanceledException>(
() => sut.WaitForMigrationsToComplete(DefaultWaitTimeSpan, DefaultPollTimeSpan, cts.Token));
contextHelper.Verify(c => c.IsMigrated(It.IsAny <CancellationToken>()), Times.AtLeast(2));
}
}
