using System; using System.Collections.Generic; using System.IO; using System.IO.Abstractions.TestingHelpers; using System.Linq; using System.Text; using System.Threading.Tasks; using API.Services; using Microsoft.Extensions.Logging; using NSubstitute; using Xunit; namespace API.Tests.Services { public class DirectoryServiceTests { private readonly ILogger _logger = Substitute.For>(); #region TraverseTreeParallelForEach [Fact] public void TraverseTreeParallelForEach_JustArchives_ShouldBe28() { var testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 28; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}file_{29}.jpg", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = new List(); var fileCount = ds.TraverseTreeParallelForEach(testDirectory, s => files.Add(s), API.Parser.Parser.ArchiveFileExtensions, _logger); Assert.Equal(28, fileCount); Assert.Equal(28, files.Count); } [Fact] public void TraverseTreeParallelForEach_LongDirectory_ShouldBe1() { var fileSystem = new MockFileSystem(); // Create a super long path var testDirectory = "/manga/"; for (var i = 0; i < 200; i++) { testDirectory = fileSystem.FileSystem.Path.Join(testDirectory, "supercalifragilisticexpialidocious"); } fileSystem.AddFile(fileSystem.FileSystem.Path.Join(testDirectory, "file_29.jpg"), new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = new List(); try { var fileCount = ds.TraverseTreeParallelForEach("/manga/", s => files.Add(s), API.Parser.Parser.ImageFileExtensions, _logger); Assert.Equal(1, fileCount); } catch (Exception ex) { Assert.False(true); } Assert.Equal(1, files.Count); } [Fact] public void TraverseTreeParallelForEach_DontCountExcludedDirectories_ShouldBe28() { var testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 28; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{Path.Join(testDirectory, "@eaDir")}file_{29}.jpg", new MockFileData("")); fileSystem.AddFile($"{Path.Join(testDirectory, ".DS_Store")}file_{30}.jpg", new MockFileData("")); fileSystem.AddFile($"{Path.Join(testDirectory, ".qpkg")}file_{30}.jpg", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = new List(); var fileCount = ds.TraverseTreeParallelForEach(testDirectory, s => files.Add(s), API.Parser.Parser.ArchiveFileExtensions, _logger); Assert.Equal(28, fileCount); Assert.Equal(28, files.Count); } #endregion #region GetFilesWithCertainExtensions [Fact] public void GetFilesWithCertainExtensions_ShouldBe10() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}file_{29}.jpg", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFilesWithExtension(testDirectory, API.Parser.Parser.ArchiveFileExtensions); Assert.Equal(10, files.Length); Assert.All(files, s => fileSystem.Path.GetExtension(s).Equals(".zip")); } [Fact] public void GetFilesWithCertainExtensions_OnlyArchives() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}file_{29}.rar", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFilesWithExtension(testDirectory, ".zip|.rar"); Assert.Equal(11, files.Length); } #endregion #region GetFiles [Fact] public void GetFiles_ArchiveOnly_ShouldBe10() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}file_{29}.jpg", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFiles(testDirectory, API.Parser.Parser.ArchiveFileExtensions).ToList(); Assert.Equal(10, files.Count()); Assert.All(files, s => fileSystem.Path.GetExtension(s).Equals(".zip")); } [Fact] public void GetFiles_All_ShouldBe11() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}file_{29}.jpg", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFiles(testDirectory).ToList(); Assert.Equal(11, files.Count()); } [Fact] public void GetFiles_All_MixedPathSeparators() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"/manga\\file_{29}.jpg", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFiles(testDirectory).ToList(); Assert.Equal(11, files.Count()); } [Fact] public void GetFiles_All_TopDirectoryOnly_ShouldBe10() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}/SubDir/file_{29}.jpg", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFiles(testDirectory).ToList(); Assert.Equal(10, files.Count()); } [Fact] public void GetFiles_WithSubDirectories_ShouldCountOnlyTopLevel() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}/SubDir/file_{29}.jpg", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFiles(testDirectory).ToList(); Assert.Equal(10, files.Count()); } [Fact] public void GetFiles_ShouldNotReturnFilesThatAreExcluded() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}/._file_{29}.jpg", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFiles(testDirectory).ToList(); Assert.Equal(10, files.Count()); } [Fact] public void GetFiles_WithCustomRegex_ShouldBe10() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}data-{i}.txt", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}joe.txt", new MockFileData("")); fileSystem.AddFile($"{testDirectory}0d.txt", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFiles(testDirectory, @".*d.*\.txt"); Assert.Equal(11, files.Count()); } [Fact] public void GetFiles_WithCustomRegexThatContainsFolder_ShouldBe10() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file/data-{i}.txt", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}joe.txt", new MockFileData("")); fileSystem.AddFile($"{testDirectory}0d.txt", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var files = ds.GetFiles(testDirectory, @".*d.*\.txt", SearchOption.AllDirectories); Assert.Equal(11, files.Count()); } #endregion #region GetTotalSize [Fact] public void GetTotalSize_ShouldBeGreaterThan0() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file/data-{i}.txt", new MockFileData("abc")); } fileSystem.AddFile($"{testDirectory}joe.txt", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var fileSize = ds.GetTotalSize(fileSystem.AllFiles); Assert.True(fileSize > 0); } #endregion #region CopyFileToDirectory [Fact] public void CopyFileToDirectory_ShouldCopyFileToNonExistentDirectory() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}file/data-0.txt", new MockFileData("abc")); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyFileToDirectory($"{testDirectory}file/data-0.txt", "/manga/output/"); Assert.True(fileSystem.FileExists("manga/output/data-0.txt")); Assert.True(fileSystem.FileExists("manga/file/data-0.txt")); } [Fact] public void CopyFileToDirectory_ShouldCopyFileToExistingDirectoryAndOverwrite() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}file/data-0.txt", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}output/data-0.txt", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyFileToDirectory($"{testDirectory}file/data-0.txt", "/manga/output/"); Assert.True(fileSystem.FileExists("/manga/output/data-0.txt")); Assert.True(fileSystem.FileExists("/manga/file/data-0.txt")); Assert.True(fileSystem.FileInfo.FromFileName("/manga/file/data-0.txt").Length == fileSystem.FileInfo.FromFileName("/manga/output/data-0.txt").Length); } #endregion #region CopyDirectoryToDirectory [Fact] public void CopyDirectoryToDirectory_ShouldThrowWhenSourceDestinationDoesntExist() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}file/data-0.txt", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}output/data-0.txt", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var ex = Assert.Throws(() => ds.CopyDirectoryToDirectory("/comics/", "/manga/output/")); Assert.Equal(ex.Message, "Source directory does not exist or could not be found: " + "/comics/"); } [Fact] public void CopyDirectoryToDirectory_ShouldCopyEmptyDirectory() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}file/data-0.txt", new MockFileData("abc")); fileSystem.AddDirectory($"{testDirectory}empty/"); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyDirectoryToDirectory($"{testDirectory}empty/", "/manga/output/"); Assert.Empty(fileSystem.DirectoryInfo.FromDirectoryName("/manga/output/").GetFiles()); } [Fact] public void CopyDirectoryToDirectory_ShouldCopyAllFileAndNestedDirectoriesOver() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}file/data-0.txt", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}data-1.txt", new MockFileData("abc")); fileSystem.AddDirectory($"{testDirectory}empty/"); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyDirectoryToDirectory($"{testDirectory}", "/manga/output/"); Assert.Equal(2, ds.GetFiles("/manga/output/", searchOption: SearchOption.AllDirectories).Count()); } #endregion #region IsDriveMounted [Fact] public void IsDriveMounted_DriveIsNotMounted() { const string testDirectory = "c:/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}data-0.txt", new MockFileData("abc")); var ds = new DirectoryService(Substitute.For>(), fileSystem); Assert.False(ds.IsDriveMounted("d:/manga/")); } [Fact] public void IsDriveMounted_DriveIsMounted() { const string testDirectory = "c:/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}data-0.txt", new MockFileData("abc")); var ds = new DirectoryService(Substitute.For>(), fileSystem); Assert.True(ds.IsDriveMounted("c:/manga/file")); } #endregion #region IsDirectoryEmpty [Fact] public void IsDirectoryEmpty_DirectoryIsEmpty() { const string testDirectory = "c:/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddDirectory(testDirectory); var ds = new DirectoryService(Substitute.For>(), fileSystem); Assert.True(ds.IsDirectoryEmpty("c:/manga/")); } [Fact] public void IsDirectoryEmpty_DirectoryIsNotEmpty() { const string testDirectory = "c:/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}data-0.txt", new MockFileData("abc")); var ds = new DirectoryService(Substitute.For>(), fileSystem); Assert.False(ds.IsDirectoryEmpty("c:/manga/")); } #endregion #region ExistOrCreate [Fact] public void ExistOrCreate_ShouldCreate() { var fileSystem = new MockFileSystem(); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.ExistOrCreate("c:/manga/output/"); Assert.True(ds.FileSystem.DirectoryInfo.FromDirectoryName("c:/manga/output/").Exists); } #endregion #region ClearAndDeleteDirectory [Fact] public void ClearAndDeleteDirectory_ShouldDeleteSelfAndAllFilesAndFolders() { const string testDirectory = "/manga/base/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file/data-{i}.txt", new MockFileData("abc")); } fileSystem.AddFile($"{testDirectory}data-a.txt", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}data-b.txt", new MockFileData("abc")); fileSystem.AddDirectory($"{testDirectory}empty/"); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.ClearAndDeleteDirectory($"{testDirectory}"); Assert.Empty(ds.GetFiles("/manga/", searchOption: SearchOption.AllDirectories)); Assert.Empty(ds.FileSystem.DirectoryInfo.FromDirectoryName("/manga/").GetDirectories()); Assert.True(ds.FileSystem.DirectoryInfo.FromDirectoryName("/manga/").Exists); Assert.False(ds.FileSystem.DirectoryInfo.FromDirectoryName("/manga/base").Exists); } #endregion #region ClearDirectory [Fact] public void ClearDirectory_ShouldDeleteAllFilesAndFolders_LeaveSelf() { const string testDirectory = "/manga/base/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file/data-{i}.txt", new MockFileData("abc")); } fileSystem.AddFile($"{testDirectory}data-a.txt", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}data-b.txt", new MockFileData("abc")); fileSystem.AddDirectory($"{testDirectory}file/empty/"); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.ClearDirectory($"{testDirectory}file/"); Assert.Empty(ds.FileSystem.DirectoryInfo.FromDirectoryName($"{testDirectory}file/").GetDirectories()); Assert.True(ds.FileSystem.DirectoryInfo.FromDirectoryName("/manga/").Exists); Assert.True(ds.FileSystem.DirectoryInfo.FromDirectoryName($"{testDirectory}file/").Exists); } [Fact] public void ClearDirectory_ShouldDeleteFoldersWithOneFileInside() { const string testDirectory = "/manga/base/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file/data-{i}.txt", new MockFileData("abc")); } var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.ClearDirectory($"{testDirectory}"); Assert.Empty(ds.FileSystem.DirectoryInfo.FromDirectoryName($"{testDirectory}").GetDirectories()); Assert.True(ds.FileSystem.DirectoryInfo.FromDirectoryName(testDirectory).Exists); Assert.False(ds.FileSystem.DirectoryInfo.FromDirectoryName($"{testDirectory}file/").Exists); } #endregion #region CopyFilesToDirectory [Fact] public void CopyFilesToDirectory_ShouldMoveAllFiles() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyFilesToDirectory(new []{$"{testDirectory}file_{0}.zip", $"{testDirectory}file_{1}.zip"}, "/manga/output/"); Assert.Equal(2, ds.GetFiles("/manga/output/").Count()); } [Fact] public void CopyFilesToDirectory_ShouldMoveAllFilesAndNotFailOnNonExistentFiles() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyFilesToDirectory(new []{$"{testDirectory}file_{0}.zip", $"{testDirectory}file_{200}.zip", $"{testDirectory}file_{1}.zip"}, "/manga/output/"); Assert.Equal(2, ds.GetFiles("/manga/output/").Count()); } [Fact] public void CopyFilesToDirectory_ShouldMoveAllFiles_InclFilesInNestedFolders() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } fileSystem.AddFile($"{testDirectory}nested/file_11.zip", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyFilesToDirectory(new []{$"{testDirectory}file_{0}.zip", $"{testDirectory}file_{1}.zip", $"{testDirectory}nested/file_11.zip"}, "/manga/output/"); Assert.Equal(3, ds.GetFiles("/manga/output/").Count()); } [Fact] public void CopyFilesToDirectory_ShouldMoveAllFiles_WithPrepend() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyFilesToDirectory(new []{$"{testDirectory}file_{0}.zip", $"{testDirectory}file_{1}.zip", $"{testDirectory}nested/file_11.zip"}, "/manga/output/", "mangarocks_"); Assert.Equal(2, ds.GetFiles("/manga/output/").Count()); Assert.All(ds.GetFiles("/manga/output/"), filepath => ds.FileSystem.Path.GetFileName(filepath).StartsWith("mangarocks_")); } [Fact] public void CopyFilesToDirectory_ShouldMoveOnlyFilesThatExist() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); for (var i = 0; i < 10; i++) { fileSystem.AddFile($"{testDirectory}file_{i}.zip", new MockFileData("")); } var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyFilesToDirectory(new []{$"{testDirectory}file_{0}.zip", $"{testDirectory}file_{1}.zip", $"{testDirectory}nested/file_11.zip"}, "/manga/output/"); Assert.Equal(2, ds.GetFiles("/manga/output/").Count()); } [Fact] public void CopyFilesToDirectory_ShouldAppendWhenTargetFileExists() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile(MockUnixSupport.Path($"{testDirectory}file.zip"), new MockFileData("")); fileSystem.AddFile(MockUnixSupport.Path($"/manga/output/file (1).zip"), new MockFileData("")); fileSystem.AddFile(MockUnixSupport.Path($"/manga/output/file (2).zip"), new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.CopyFilesToDirectory(new []{MockUnixSupport.Path($"{testDirectory}file.zip")}, "/manga/output/"); ds.CopyFilesToDirectory(new []{MockUnixSupport.Path($"{testDirectory}file.zip")}, "/manga/output/"); var outputFiles = ds.GetFiles("/manga/output/").Select(API.Parser.Parser.NormalizePath).ToList(); Assert.Equal(4, outputFiles.Count()); // we have 2 already there and 2 copies // For some reason, this has C:/ on directory even though everything is emulated (System.IO.Abstractions issue, not changing) // https://github.com/TestableIO/System.IO.Abstractions/issues/831 Assert.True(outputFiles.Contains(API.Parser.Parser.NormalizePath("/manga/output/file (3).zip")) || outputFiles.Contains(API.Parser.Parser.NormalizePath("C:/manga/output/file (3).zip"))); } #endregion #region ListDirectory [Fact] public void ListDirectory_EmptyForNonExistent() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}file_0.zip", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); Assert.Empty(ds.ListDirectory("/comics/")); } [Fact] public void ListDirectory_ListsAllDirectories() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddDirectory($"{testDirectory}dir1"); fileSystem.AddDirectory($"{testDirectory}dir2"); fileSystem.AddDirectory($"{testDirectory}dir3"); fileSystem.AddFile($"{testDirectory}file_0.zip", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); Assert.Equal(3, ds.ListDirectory(testDirectory).Count()); } [Fact] public void ListDirectory_ListsOnlyNonSystemAndHiddenOnly() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddDirectory($"{testDirectory}dir1"); var di = fileSystem.DirectoryInfo.FromDirectoryName($"{testDirectory}dir1"); di.Attributes |= FileAttributes.System; fileSystem.AddDirectory($"{testDirectory}dir2"); di = fileSystem.DirectoryInfo.FromDirectoryName($"{testDirectory}dir2"); di.Attributes |= FileAttributes.Hidden; fileSystem.AddDirectory($"{testDirectory}dir3"); fileSystem.AddFile($"{testDirectory}file_0.zip", new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); Assert.Equal(1, ds.ListDirectory(testDirectory).Count()); } #endregion #region ReadFileAsync [Fact] public async Task ReadFileAsync_ShouldGetBytes() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}file_1.zip", new MockFileData("Hello")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var bytes = await ds.ReadFileAsync($"{testDirectory}file_1.zip"); Assert.Equal(Encoding.UTF8.GetBytes("Hello"), bytes); } [Fact] public async Task ReadFileAsync_ShouldReadNothingFromNonExistent() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddFile($"{testDirectory}file_1.zip", new MockFileData("Hello")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var bytes = await ds.ReadFileAsync($"{testDirectory}file_32123.zip"); Assert.Empty(bytes); } #endregion #region FindHighestDirectoriesFromFiles [Theory] [InlineData(new [] {"C:/Manga/"}, new [] {"C:/Manga/Love Hina/Vol. 01.cbz"}, "C:/Manga/Love Hina")] [InlineData(new [] {"C:/Manga/Dir 1/", "c://Manga/Dir 2/"}, new [] {"C:/Manga/Dir 1/Love Hina/Vol. 01.cbz"}, "C:/Manga/Dir 1/Love Hina")] [InlineData(new [] {"C:/Manga/Dir 1/", "c://Manga/"}, new [] {"D:/Manga/Love Hina/Vol. 01.cbz", "D:/Manga/Vol. 01.cbz"}, "")] public void FindHighestDirectoriesFromFilesTest(string[] rootDirectories, string[] files, string expectedDirectory) { var fileSystem = new MockFileSystem(); foreach (var directory in rootDirectories) { fileSystem.AddDirectory(directory); } foreach (var f in files) { fileSystem.AddFile(f, new MockFileData("")); } var ds = new DirectoryService(Substitute.For>(), fileSystem); var actual = ds.FindHighestDirectoriesFromFiles(rootDirectories, files); var expected = new Dictionary(); if (!string.IsNullOrEmpty(expectedDirectory)) { expected = new Dictionary {{expectedDirectory, ""}}; } Assert.Equal(expected, actual); } #endregion #region GetFoldersTillRoot [Theory] [InlineData("C:/Manga/", "C:/Manga/Love Hina/Specials/Omake/", "Omake,Specials,Love Hina")] [InlineData("C:/Manga/", "C:/Manga/Love Hina/Specials/Omake", "Omake,Specials,Love Hina")] [InlineData("C:/Manga", "C:/Manga/Love Hina/Specials/Omake/", "Omake,Specials,Love Hina")] [InlineData("C:/Manga", @"C:\Manga\Love Hina\Specials\Omake\", "Omake,Specials,Love Hina")] [InlineData(@"/manga/", @"/manga/Love Hina/Specials/Omake/", "Omake,Specials,Love Hina")] [InlineData(@"/manga/", @"/manga/", "")] [InlineData(@"E:\test", @"E:\test\Sweet X Trouble\Sweet X Trouble - Chapter 001.cbz", "Sweet X Trouble")] [InlineData(@"C:\/mount/gdrive/Library/Test Library/Comics/", @"C:\/mount/gdrive/Library/Test Library/Comics\godzilla rivals vs hedorah\vol 1\", "vol 1,godzilla rivals vs hedorah")] [InlineData(@"/manga/", @"/manga/Btooom!/Vol.1 Chapter 2/1.cbz", "Vol.1 Chapter 2,Btooom!")] [InlineData(@"C:/", @"C://Btooom!/Vol.1 Chapter 2/1.cbz", "Vol.1 Chapter 2,Btooom!")] [InlineData(@"C:\\", @"C://Btooom!/Vol.1 Chapter 2/1.cbz", "Vol.1 Chapter 2,Btooom!")] [InlineData(@"C://mount/gdrive/Library/Test Library/Comics", @"C://mount/gdrive/Library/Test Library/Comics/Dragon Age/Test", "Test,Dragon Age")] [InlineData(@"M:\", @"M:\Toukyou Akazukin\Vol. 01 Ch. 005.cbz", @"Toukyou Akazukin")] public void GetFoldersTillRoot_Test(string rootPath, string fullpath, string expectedArray) { var fileSystem = new MockFileSystem(); fileSystem.AddDirectory(rootPath); fileSystem.AddFile(fullpath, new MockFileData("")); var ds = new DirectoryService(Substitute.For>(), fileSystem); var expected = expectedArray.Split(","); if (expectedArray.Equals(string.Empty)) { expected = Array.Empty(); } Assert.Equal(expected, ds.GetFoldersTillRoot(rootPath, fullpath)); } #endregion #region RemoveNonImages [Fact] public void RemoveNonImages() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddDirectory(testDirectory); fileSystem.AddFile($"{testDirectory}file/data-0.txt", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}data-1.jpg", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}data-2.png", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}data-3.webp", new MockFileData("abc")); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.RemoveNonImages($"{testDirectory}"); Assert.False(fileSystem.FileExists($"{testDirectory}file/data-0.txt")); Assert.Equal(3, ds.GetFiles($"{testDirectory}", searchOption:SearchOption.AllDirectories).Count()); } #endregion #region Flatten [Fact] public void Flatten_ShouldDoNothing() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddDirectory(testDirectory); fileSystem.AddFile($"{testDirectory}data-1.jpg", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}data-2.png", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}data-3.webp", new MockFileData("abc")); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.Flatten($"{testDirectory}"); Assert.True(fileSystem.FileExists($"{testDirectory}data-1.jpg")); Assert.True(fileSystem.FileExists($"{testDirectory}data-2.png")); Assert.True(fileSystem.FileExists($"{testDirectory}data-3.webp")); } [Fact] public void Flatten_ShouldFlatten() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddDirectory(testDirectory); fileSystem.AddFile($"{testDirectory}data-1.jpg", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}subdir/data-3.webp", new MockFileData("abc")); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.Flatten($"{testDirectory}"); Assert.Equal(2, ds.GetFiles(testDirectory).Count()); Assert.False(fileSystem.FileExists($"{testDirectory}subdir/data-3.webp")); Assert.True(fileSystem.Directory.Exists($"{testDirectory}subdir/")); } [Fact] public void Flatten_ShouldFlatten_WithoutMacosx() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); fileSystem.AddDirectory(testDirectory); fileSystem.AddFile($"{testDirectory}data-1.jpg", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}subdir/data-3.webp", new MockFileData("abc")); fileSystem.AddFile($"{testDirectory}__MACOSX/data-4.webp", new MockFileData("abc")); var ds = new DirectoryService(Substitute.For>(), fileSystem); ds.Flatten($"{testDirectory}"); Assert.Equal(2, ds.GetFiles(testDirectory).Count()); Assert.False(fileSystem.FileExists($"{testDirectory}data-4.webp")); } #endregion #region CheckWriteAccess [Fact] public async Task CheckWriteAccess_ShouldHaveAccess() { const string testDirectory = "/manga/"; var fileSystem = new MockFileSystem(); var ds = new DirectoryService(Substitute.For>(), fileSystem); var hasAccess = await ds.CheckWriteAccess(ds.FileSystem.Path.Join(testDirectory, "bookmarks")); Assert.True(hasAccess); Assert.False(ds.FileSystem.Directory.Exists(ds.FileSystem.Path.Join(testDirectory, "bookmarks"))); Assert.False(ds.FileSystem.File.Exists(ds.FileSystem.Path.Join(testDirectory, "bookmarks", "test.txt"))); } #endregion #region GetHumanReadableBytes [Theory] [InlineData(1200, "1.17 KB")] [InlineData(1, "1 B")] [InlineData(10000000, "9.54 MB")] [InlineData(10000000000, "9.31 GB")] public void GetHumanReadableBytesTest(long bytes, string expected) { Assert.Equal(expected, DirectoryService.GetHumanReadableBytes(bytes)); } #endregion } }