fixed mock file system exception when using root path as a parameter

TestableIO · Jul 12, 2024 · 377be90 · 377be90
1 parent c708b42
commit 377be90
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/src/TestableIO.System.IO.Abstractions.TestingHelpers/MockFileSystem.cs b/src/TestableIO.System.IO.Abstractions.TestingHelpers/MockFileSystem.cs
@@ -243,6 +243,12 @@ public void AddFile(string path, MockFileData mockFile)
             }
 
             var directoryPath = Path.GetDirectoryName(fixedPath);
+            if (directoryPath == null)
+            {
+                AddDrive(fixedPath, new MockDriveData());
+                SetEntry(fixedPath, mockFile);
+                return;
+            }
 
             if (!DirectoryExistsWithoutFixingPath(directoryPath))
             {

diff --git a/tests/TestableIO.System.IO.Abstractions.TestingHelpers.Tests/MockFileSystemTests.cs b/tests/TestableIO.System.IO.Abstractions.TestingHelpers.Tests/MockFileSystemTests.cs
@@ -435,6 +435,26 @@ public void MockFileSystem_Constructor_ThrowsForNonRootedCurrentDirectory()
             );
             Assert.That(ae.ParamName, Is.EqualTo("currentDirectory"));
         }
+
+        [Test]
+        public void MockFileSystem_Constructor_ShouldSupportDifferentRootDrives()
+        {
+            var fileSystem = new MockFileSystem(new Dictionary<string, MockFileData>
+            {
+                [@"c:\"] = new MockDirectoryData(),
+                [@"z:\"] = new MockDirectoryData(),
+                [@"d:\"] = new MockDirectoryData(),
+            });
+
+            var cExists = fileSystem.Directory.Exists(@"c:\");
+            var zExists = fileSystem.Directory.Exists(@"z:\");
+            var dExists = fileSystem.Directory.Exists(@"d:\");
+
+            Assert.That(fileSystem, Is.Not.Null);
+            Assert.That(cExists, Is.True);
+            Assert.That(zExists, Is.True);
+            Assert.That(dExists, Is.True);
+        }
 
         [Test]
         public void MockFileSystem_DefaultState_DefaultTempDirectoryExists()