public void InitializeTestData()
{
expectedCompanies = CompanyMock.Companies();
mockCompanyRepository = new Mock <ICompanyRepository>();
companyService = new CompanyService(mockCompanyRepository.Object);
mockCompanyRepository.Setup(m => m.GetAll()).Returns(expectedCompanies);
mockCompanyRepository.Setup(m => m.AddCompany(It.IsAny <Company>())).Returns(
(Company target) =>
{
expectedCompanies.Add(target);
return(target.Id);
});
mockCompanyRepository.Setup(m => m.UpdateCompany(It.IsAny <Company>())).Returns(
(Company target) =>
{
Company companyToUpdate = expectedCompanies.Find(c => c.Id == target.Id);
expectedCompanies.Remove(companyToUpdate);
expectedCompanies.Add(target);
return(target);
});
mockCompanyRepository.Setup(m => m.GetCompanyById(It.IsAny <int>())).Returns(
(int companyId) =>
{
Company company = expectedCompanies.Find(c => c.Id == companyId);
return(company);
});
}
