Using a Func<Task> may be a better option
This commit is contained in:
@@ -9,26 +9,45 @@ namespace WpfViewModelFirst.Tests
|
||||
[Fact]
|
||||
public void Part1WillSetPart1ViewModelToCurrentViewModel()
|
||||
{
|
||||
Part1ViewModel part1ViewModel = new(() => _ = true);
|
||||
Part1ViewModel part1ViewModel = new(() => Task.CompletedTask);
|
||||
Mock<IViewModelFactory> mockViewModelFactory = new Mock<IViewModelFactory>();
|
||||
mockViewModelFactory.Setup(vmf => vmf.GetPart1ViewModel(It.IsAny<Action>()))
|
||||
mockViewModelFactory.Setup(vmf => vmf.GetPart1ViewModel(It.IsAny<Func<Task>>()))
|
||||
.Returns(part1ViewModel);
|
||||
MainWindowViewModel main = new(mockViewModelFactory.Object);
|
||||
main.Part1(null);
|
||||
Assert.Equal(part1ViewModel, main.CurrentViewModel);
|
||||
}
|
||||
[Fact]
|
||||
public void Part1CallbackIsSuccessfull()
|
||||
public async Task Part1CallbackIsSuccessfull()
|
||||
{
|
||||
TaskCompletionSource<bool> taskCompletionSource = new TaskCompletionSource<bool>();
|
||||
Mock<IViewModelFactory> mockViewModelFactory = new Mock<IViewModelFactory>();
|
||||
mockViewModelFactory.Setup(vmf => vmf.GetPart1ViewModel(It.IsAny<Action>()))
|
||||
.Returns(() => new Part1ViewModel(() => Task.Delay(0)))
|
||||
.Callback<Action>(c => c.Invoke());
|
||||
mockViewModelFactory.Setup(vmf => vmf.GetPart1ViewModel(It.IsAny<Func<Task>>()))
|
||||
|
||||
.Returns(() => new Part1ViewModel(() =>
|
||||
{
|
||||
taskCompletionSource.SetResult(true);
|
||||
return Task.CompletedTask;
|
||||
}))
|
||||
.Callback<Func<Task>>(c =>
|
||||
{
|
||||
c.Invoke().ContinueWith(t =>
|
||||
{
|
||||
taskCompletionSource.SetResult(true);
|
||||
});
|
||||
})
|
||||
;
|
||||
mockViewModelFactory.Setup(vmf => vmf.GetPart2ViewModel())
|
||||
.Returns(() => new Part2ViewModel());
|
||||
MainWindowViewModel main = new(mockViewModelFactory.Object);
|
||||
main.Part1(null);
|
||||
await taskCompletionSource.Task.WaitAsync(CancellationToken.None);
|
||||
Assert.IsType<Part2ViewModel>(main.CurrentViewModel);
|
||||
}
|
||||
|
||||
Func<Task> Part1Func()
|
||||
{
|
||||
return () => Task.CompletedTask;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user