Merge pull request #115 from JonCanning/master

Support for DirectoryInfo.EnumerateFiles()
TestableIO · Feb 23, 2015 · 30828b3 · 30828b3
2 parents 0417af3 + 574cddc
commit 30828b3
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/TestHelpers.Tests/MockDirectoryInfoTests.cs b/TestHelpers.Tests/MockDirectoryInfoTests.cs
@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using System.Linq;
 using NUnit.Framework;
 
 namespace System.IO.Abstractions.TestingHelpers.Tests
@@ -121,5 +122,22 @@ public void MockDirectoryInfo_GetParent_ShouldReturnDirectoriesAndNamesWithSearc
             // Assert
             Assert.AreEqual(XFS.Path(@"c:\a\b"), result.FullName);
         }
+
+        [Test]
+        public void MockDirectoryInfo_EnumerateFiles_ShouldReturnAllFiles()
+        {
+          // Arrange
+          var fileSystem = new MockFileSystem(new Dictionary<string, MockFileData>
+            {
+                { XFS.Path(@"c:\temp\folder\a.txt"), "" },
+                { XFS.Path(@"c:\temp\folder\b.txt"), "" }
+            });
+
+          // Act
+          var directoryInfo = new MockDirectoryInfo(fileSystem, XFS.Path(@"c:\temp\folder"));
+
+          // Assert
+          Assert.AreEqual(new[]{"a.txt", "b.txt"}, directoryInfo.EnumerateFiles().ToList().Select(x => x.Name).ToArray());
+        }
     }
 }
diff --git a/TestingHelpers/MockDirectoryInfo.cs b/TestingHelpers/MockDirectoryInfo.cs
@@ -161,7 +161,7 @@ public override IEnumerable<DirectoryInfoBase> EnumerateDirectories(string searc
 
         public override IEnumerable<FileInfoBase> EnumerateFiles()
         {
-            throw new NotImplementedException(Properties.Resources.NOT_IMPLEMENTED_EXCEPTION);
+            return mockFileDataAccessor.AllFiles.Select(x => new MockFileInfo(mockFileDataAccessor, x));
         }
 
         public override IEnumerable<FileInfoBase> EnumerateFiles(string searchPattern)