Add another test for repository all

cwetanow · Nov 16, 2017 · ac29112 · ac29112
1 parent bf14a21
commit ac29112
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/src/Logs.Data.Tests/EfGenericRepositoryTests/AllTests.cs b/src/Logs.Data.Tests/EfGenericRepositoryTests/AllTests.cs
@@ -46,5 +46,29 @@ public void TestAll_ShouldCallDbContextSet()
             // Assert
             mockedDbContext.Verify(db => db.DbSet<FakeGenericRepositoryType>(), Times.Once);
         }
+
+        [Test]
+        public void TestAll_ShouldReturnCorrectly()
+        {
+            // Arrange
+            var data = this.GetData();
+
+            var mockedSet = new Mock<IDbSet<FakeGenericRepositoryType>>();
+            mockedSet.Setup(m => m.Provider).Returns(data.Provider);
+            mockedSet.Setup(m => m.Expression).Returns(data.Expression);
+            mockedSet.Setup(m => m.ElementType).Returns(data.ElementType);
+            mockedSet.Setup(m => m.GetEnumerator()).Returns(data.GetEnumerator());
+
+            var mockedDbContext = new Mock<ILogsDbContext>();
+            mockedDbContext.Setup(x => x.DbSet<FakeGenericRepositoryType>()).Returns(mockedSet.Object);
+
+            var repository = new EntityFrameworkRepository<FakeGenericRepositoryType>(mockedDbContext.Object);
+
+            // Act
+            var result = repository.All;
+
+            // Assert
+            CollectionAssert.AreEqual(mockedSet.Object, result);
+        }
     }
 }