Implementing MockFileInfo.Directory

Author: Dirk Rombauts

TestHelpers.Tests/MockFileInfoTests.cs

@@ -176,5 +176,17 @@ public void MockFileInfo_GetDirectoryName_ShouldReturnCompleteDirectoryPath()
 
             Assert.AreEqual(@"c:\temp\level1\level2", result);
         }
-   }
+
+        [Test]
+        public void MockFileInfo_GetDirectory_ShouldReturnDirectoryInfoWithCorrectPath()
+        {
+            // Arrange
+            var fileInfo = new MockFileInfo(new MockFileSystem(), @"c:\temp\level1\level2\file.txt");
+
+            // Act
+            var result = fileInfo.Directory;
+
+            Assert.AreEqual(@"c:\temp\level1\level2\", result.FullName);
+        }
+    }
 }

TestingHelpers/IMockFileDataAccessor.cs

@@ -14,5 +14,6 @@ public interface IMockFileDataAccessor {
         DirectoryBase Directory { get; }
         IFileInfoFactory FileInfo {get; }
         PathBase Path { get; }
+        IDirectoryInfoFactory DirectoryInfo { get; }
     }
 }

TestingHelpers/MockFileInfo.cs

@@ -215,7 +215,10 @@ public override void SetAccessControl(FileSecurity fileSecurity)
 
         public override DirectoryInfoBase Directory
         {
-            get { throw new NotImplementedException("This test helper hasn't been implemented yet. They are implemented on an as-needed basis. As it seems like you need it, now would be a great time to send us a pull request over at https://github.com/tathamoddie/System.IO.Abstractions. You know, because it's open source and all."); }
+            get
+            {
+                return mockFileSystem.DirectoryInfo.FromDirectoryName(this.DirectoryName);
+            }
         }
 
         public override string DirectoryName