public async Task <JsonResult> RemoveBudget(Guid id)
{
User currentUser = await this.authService.GetCurrentUser();
RemoveBudgetCommand command = new RemoveBudgetCommand(this.unitOfWork, id, currentUser);
await command.Run();
return(new JsonResult(new { success = true }));
}
public async Task Test_ExceptionThrown_WhenRemoveCommandCalled_ByUserThatDoesntOwnIt()
{
Mock <IUnitOfWork> unitOfWork = new Mock <IUnitOfWork>();
Mock <IRepository <Budget> > budgetRepo = new Mock <IRepository <Budget> >(); UserFactory userFactory = this.buildersAndFactories.GetService <UserFactory>();
Budget rootBudget = this.buildersAndFactories.BudgetBuilder.Build();
User unauthorizedUser = userFactory.NewUser();
IIncludableQuerySet <Budget> budgets = new InMemoryIncludableQuerySet <Budget>(new List <Budget>()
{
rootBudget
});
budgetRepo.Setup(r => r.GetAll()).Returns(budgets);
unitOfWork.Setup(u => u.SaveChangesAsync()).Returns(Task.CompletedTask);
unitOfWork.Setup(u => u.GetRepository <Budget>()).Returns(budgetRepo.Object);
RemoveBudgetCommand command = new RemoveBudgetCommand(unitOfWork.Object, rootBudget.Id, unauthorizedUser);
await Assert.ThrowsAsync <InvalidOperationException>(() => command.Run());
}
public async Task Test_BudgetNoLongerExists_WhenRemoveCommandCalled()
{
Mock <IUnitOfWork> unitOfWork = new Mock <IUnitOfWork>();
Mock <IRepository <Budget> > budgetRepo = new Mock <IRepository <Budget> >();
UserFactory userFactory = this.buildersAndFactories.GetService <UserFactory>();
Budget rootBudget = this.buildersAndFactories.BudgetBuilder.Build();
User user = userFactory.NewUser(rootBudget.Fund.UserId);
IIncludableQuerySet <Budget> budgets = new InMemoryIncludableQuerySet <Budget>(new List <Budget>()
{
rootBudget
});
budgetRepo.Setup(r => r.GetAll()).Returns(budgets);
unitOfWork.Setup(u => u.SaveChangesAsync()).Returns(Task.CompletedTask);
unitOfWork.Setup(u => u.GetRepository <Budget>()).Returns(budgetRepo.Object);
RemoveBudgetCommand command = new RemoveBudgetCommand(unitOfWork.Object, rootBudget.Id, user);
await command.Run();
budgetRepo.Verify(r => r.Remove(rootBudget), Times.Once);
}