public void CanEditValueAndCommentOfABudget()
{
IRepository <Client> clientRepository = new LocalRepository <Client>();
IRepository <Budget> budgetRepository = new LocalRepository <Budget>();
var userManager = new Mock <IIdentityMembershipProvider>();
Budget budget1 = new Budget();
budget1.Id = 1;
budget1.Value = 1000;
budget1.Month = new DateTime(2014, 1, 1);
budget1.ClientId = 1;
budgetRepository.Add(budget1);
Assert.AreEqual(1, budgetRepository.All().Count());
var controller = new BudgetController(clientRepository, budgetRepository, userManager.Object);
var createBudgetPostMessage = new EditBudgetViewModel
{
id = 1,
clientId = 1,
month = "2014-01-01",
value = 3000,
comment = "New Comment"
};
controller.Edit(createBudgetPostMessage);
Assert.AreEqual(1, budgetRepository.All().Count());
Assert.AreEqual(3000, budgetRepository.All().First().Value);
Assert.AreEqual("New Comment", budgetRepository.All().First().Comment);
}
public void EditGetReturnsCorrectData()
{
IRepository <Client> clientRepository = new LocalRepository <Client>();
IRepository <Budget> budgetRepository = new LocalRepository <Budget>();
var userManager = new Mock <IIdentityMembershipProvider>();
Budget budget1 = new Budget();
budget1.Id = 1;
budget1.Value = 1000;
budget1.Month = new DateTime(2014, 1, 1);
budget1.ClientId = 1;
budget1.Comment = "Comment";
budget1.Client = new Client {
Name = "Client Name"
};
budgetRepository.Add(budget1);
Assert.AreEqual(1, budgetRepository.All().Count());
userManager.Setup(um => um.GetRoles(It.IsAny <string>())).Returns(new List <string> {
"mockRole"
});
var controller = new BudgetController(clientRepository, budgetRepository, userManager.Object);
var claim = new Claim("test", "AnyId");
ModelHelper.SetClaimAsCurrentUserForController(claim, controller);
var expectedEditBudgetInRepository = new CreateEditClientBudgetViewModel
{
BudgetId = 1,
ClientId = 1,
Month = "2014-01",
Value = 1000,
Comment = "Comment",
ClientName = "Client Name"
};
var result = controller.Edit(1) as ViewResult;
var resultModel = result.Model as CreateEditClientBudgetViewModel;
Assert.AreEqual(expectedEditBudgetInRepository.BudgetId, resultModel.BudgetId);
Assert.AreEqual(expectedEditBudgetInRepository.ClientId, resultModel.ClientId);
Assert.AreEqual(expectedEditBudgetInRepository.Month, resultModel.Month);
Assert.AreEqual(expectedEditBudgetInRepository.Value, resultModel.Value);
Assert.AreEqual(expectedEditBudgetInRepository.Comment, resultModel.Comment);
Assert.AreEqual(expectedEditBudgetInRepository.ClientName, resultModel.ClientName);
}
