public async Task FailSubtaskInProgressAsync_Should_MarkSubtaskInProgressAsError()
{
var dbContextOptions = DbContextOptionsFactory.CreateOptions("Mark_subtask_as_error");
using (var dbContext = new TestDbContext(dbContextOptions))
{
CreateMockData(dbContext);
await dbContext.SaveChangesAsync();
}
var subtaskInProgressCleanupServiceMock = new Mock <ISubtasksInProgressCleanupService>();
subtaskInProgressCleanupServiceMock.Setup(service => service.RemoveSubtasksInProgress(It.IsAny <int>()));
using (var dbContext = new TestDbContext(dbContextOptions))
{
var computationFailService =
new ComputationFailService(dbContext, subtaskInProgressCleanupServiceMock.Object);
await computationFailService.FailSubtaskInProgressAsync(1, new[] { "Some error" });
}
using (var dbContext = new TestDbContext(dbContextOptions))
{
var foundSubtaskInProgress = await dbContext.SubtasksInProgress
.Include(subtaskInProgress => subtaskInProgress.Subtask)
.FirstAsync(subtaskInProgress => subtaskInProgress.Id == 1);
Assert.AreEqual(SubtaskInProgressStatus.Error, foundSubtaskInProgress.Status);
Assert.AreEqual(SubtaskStatus.Executing, foundSubtaskInProgress.Subtask.Status);
}
}
FailSubtaskInProgressAsync_Should_MarkDistributedTaskAsError_WhenMultipleSubtasksInProgressFailed()
{
var dbContextOptions = DbContextOptionsFactory.CreateOptions("Mark_distributed task_as_error");
using (var dbContext = new TestDbContext(dbContextOptions))
{
CreateMockData(dbContext);
dbContext.SubtasksInProgress.Add(new SubtaskInProgress()
{
Id = 2,
SubtaskId = 1,
Status = SubtaskInProgressStatus.Executing
});
await dbContext.SaveChangesAsync();
}
var subtaskInProgressCleanupServiceMock = new Mock <ISubtasksInProgressCleanupService>();
subtaskInProgressCleanupServiceMock.Setup(service => service.RemoveSubtasksInProgress(It.IsAny <int>()));
using (var dbContext = new TestDbContext(dbContextOptions))
{
var computationFailService =
new ComputationFailService(dbContext, subtaskInProgressCleanupServiceMock.Object);
await computationFailService.FailSubtaskInProgressAsync(2, new[] { "Some error" });
}
subtaskInProgressCleanupServiceMock.Verify(service => service.RemoveSubtasksInProgress(1), Times.Once);
using (var dbContext = new TestDbContext(dbContextOptions))
{
var foundSubtaskInProgress = await dbContext.SubtasksInProgress
.Include(subtaskInProgress => subtaskInProgress.Subtask)
.ThenInclude(subtask => subtask.DistributedTask)
.FirstAsync(subtaskInProgress => subtaskInProgress.Id == 1);
Assert.AreEqual(SubtaskInProgressStatus.Error, foundSubtaskInProgress.Status);
Assert.AreEqual(SubtaskStatus.Error, foundSubtaskInProgress.Subtask.Status);
Assert.AreEqual(DistributedTaskStatus.Error, foundSubtaskInProgress.Subtask.DistributedTask.Status);
}
}
