public void TestGetCompanies() { var mockSet = GenerateMock(); var mockContext = new Mock <CompanyDBEntities>(); mockContext.Setup(m => m.Companies).Returns(mockSet.Object); var companyContext = new CompanyDBContext(mockContext.Object); var comp = companyContext.GetAllCompanies().ToList(); Assert.AreEqual(3, comp.Count); Assert.AreEqual(1, comp[0].CompanyId); Assert.AreEqual(2, comp[1].CompanyId); Assert.AreEqual(3, comp[2].CompanyId); }
public void TestIfCompanyExistOnDelete() { var mockSet = new Mock <DbSet <Company> >(); var data = new List <Company> { new Company { CompanyId = 1 }, new Company { CompanyId = 2 }, new Company { CompanyId = 3 }, }.AsQueryable(); mockSet.As <IQueryable <Company> >().Setup(m => m.Provider).Returns(data.Provider); mockSet.As <IQueryable <Company> >().Setup(m => m.Expression).Returns(data.Expression); mockSet.As <IQueryable <Company> >().Setup(m => m.ElementType).Returns(data.ElementType); mockSet.As <IQueryable <Company> >().Setup(m => m.GetEnumerator()).Returns(data.GetEnumerator()); var mockContext = new Mock <CompanyDBEntities>(); mockContext.Setup(m => m.Companies).Returns(mockSet.Object); var commpanyContext = new CompanyDBContext(mockContext.Object); var comp = commpanyContext.GetAllCompanies().ToList(); Assert.AreEqual(3, comp.Count); Assert.AreEqual(1, comp[0].CompanyId); Assert.AreEqual(2, comp[1].CompanyId); Assert.AreEqual(3, comp[2].CompanyId); }