Merge pull request #267 from Togusa09/master

Added methods to handle loading data from embedded files

Author: tathamoddie

TestHelpers.Tests/MockFileSystemTests.cs

@@ -1,6 +1,8 @@
 using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
+using System.Reflection;
+using System.Text;
 using NUnit.Framework;
 
 namespace System.IO.Abstractions.TestingHelpers.Tests
@@ -204,5 +206,32 @@ public void MockFileSystem_AddFile_ShouldMatchCapitalization_PartialMatch_Furthe
             Assert.Contains(@"C:\test\subtest\new\SUBDirectory\", fileSystem.AllDirectories.ToList());
             Assert.Contains(@"C:\LOUD\SUBLOUD\new\SUBDirectory\", fileSystem.AllDirectories.ToList());
         }
+
+        [Test]
+        public void MockFileSystem_AddFileFromEmbeddedResource_ShouldAddTheFile()
+        {
+            // Arrange
+            var fileSystem = new MockFileSystem();
+
+            // Act
+            fileSystem.AddFileFromEmbeddedResource(@"C:\TestFile.txt", Assembly.GetExecutingAssembly(), "System.IO.Abstractions.TestingHelpers.Tests.TestFiles.TestFile.txt");
+            var result = fileSystem.GetFile(@"C:\TestFile.txt");
+
+            // Assert
+            Assert.AreEqual(new UTF8Encoding().GetBytes("This is a test file."), result.Contents);
+        }
+
+        [Test]
+        public void MockFileSystem_AddFilesFromEmbeddedResource_ShouldAddAllTheFiles()
+        {
+            // Arrange
+            var fileSystem = new MockFileSystem();
+
+            //Act
+            fileSystem.AddFilesFromEmbeddedNamespace(@"C:\", Assembly.GetExecutingAssembly(), "System.IO.Abstractions.TestingHelpers.Tests.TestFiles");
+
+            Assert.Contains(@"C:\TestFile.txt", fileSystem.AllFiles.ToList());
+            Assert.Contains(@"C:\SecondTestFile.txt", fileSystem.AllFiles.ToList());
+        }
     }
 }

TestHelpers.Tests/TestFiles/SecondTestFile.txt

@@ -0,0 +1 @@
+This is a the second test file.

TestHelpers.Tests/TestFiles/TestFile.txt

@@ -0,0 +1 @@
+This is a test file.

TestHelpers.Tests/TestHelpers.Tests.csproj

@@ -23,6 +23,16 @@
       <SignAssembly>True</SignAssembly>
   </PropertyGroup>
 
+  <ItemGroup>
+    <None Remove="TestFiles\SecondTestFile.txt" />
+    <None Remove="TestFiles\TestFile.txt" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <EmbeddedResource Include="TestFiles\SecondTestFile.txt" />
+    <EmbeddedResource Include="TestFiles\TestFile.txt" />
+  </ItemGroup>
+
   <ItemGroup>
     <ProjectReference Include="../System.IO.Abstractions/System.IO.Abstractions.csproj" />
     <ProjectReference Include="../TestingHelpers/TestingHelpers.csproj" />

TestingHelpers/IMockFileDataAccessor.cs

@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using System.Reflection;
 
 namespace System.IO.Abstractions.TestingHelpers
 {
@@ -17,6 +18,9 @@ public interface IMockFileDataAccessor
         void AddFile(string path, MockFileData mockFile);
         void AddDirectory(string path);
 
+        void AddFileFromEmbeddedResource(string path, Assembly resourceAssembly, string embeddedResourcePath);
+        void AddFilesFromEmbeddedNamespace(string path, Assembly resourceAssembly, string embeddedRresourcePath);
+
         /// <summary>
         /// Removes the file.
         /// </summary>

TestingHelpers/MockFileSystem.cs

@@ -1,6 +1,7 @@
 using System.Collections.Generic;
 using System.Globalization;
 using System.Linq;
+using System.Reflection;
 
 namespace System.IO.Abstractions.TestingHelpers
 {
@@ -190,7 +191,40 @@ public void AddDirectory(string path)
             }
         }
 
-        public void RemoveFile(string path)
+      public void AddFileFromEmbeddedResource(string path, Assembly resourceAssembly, string embeddedResourcePath)
+      {
+            using (var embeddedResourceStream = resourceAssembly.GetManifestResourceStream(embeddedResourcePath))
+            {
+                if (embeddedResourceStream == null)
+                {
+                    throw new Exception("Resource not found in assembly");
+                }
+
+                using (var streamReader = new BinaryReader(embeddedResourceStream))
+                {
+                    var fileData = streamReader.ReadBytes((int)embeddedResourceStream.Length);
+                    AddFile(path, new MockFileData(fileData));
+                }
+            }
+      }
+
+      public void AddFilesFromEmbeddedNamespace(string path, Assembly resourceAssembly, string embeddedRresourcePath)
+      {
+          var matchingResources = resourceAssembly.GetManifestResourceNames().Where(f => f.StartsWith(embeddedRresourcePath));
+          foreach (var resource in matchingResources)
+          {
+            using (var embeddedResourceStream = resourceAssembly.GetManifestResourceStream(resource))
+            using (var streamReader = new BinaryReader(embeddedResourceStream))
+            {
+                var fileName = resource.Substring(embeddedRresourcePath.Length + 1);
+                var fileData = streamReader.ReadBytes((int)embeddedResourceStream.Length);
+                var filePath = Path.Combine(path, fileName);
+                AddFile(filePath, new MockFileData(fileData));
+            }
+          }
+      }
+
+      public void RemoveFile(string path)
         {
             path = FixPath(path);