recyclarr/tests/Recyclarr.Tests/Common/Extensions/FileSystemExtensionsTest.cs

using System.IO.Abstractions;
using System.Text.RegularExpressions;
using Recyclarr.Common;
using Recyclarr.Common.Extensions;
using Recyclarr.TestLibrary;

namespace Recyclarr.Tests.Common.Extensions;

[TestFixture]
public class FileSystemExtensionsTest
{
    private static IEnumerable<string> ReRootFiles(
        MockFileSystem fs,
        IEnumerable<string> files,
        string oldRoot,
        string newRoot
    )
    {
        return files.Select(x =>
        {
            var strippedPath = Regex.Replace(x, $"^{oldRoot}", newRoot);
            return fs.Path.GetFullPath(strippedPath);
        });
    }

    private static MockFileSystem NewMockFileSystem(IEnumerable<string> files, string cwd)
    {
        return NewMockFileSystem(files, Array.Empty<string>(), cwd);
    }

    private static MockFileSystem NewMockFileSystem(
        IEnumerable<string> files,
        IEnumerable<string> dirs,
        string cwd
    )
    {
        var dirData = dirs.Select(x => (x, (MockFileData)new MockDirectoryData()));
        var fileData = files.Select(x => (x, new MockFileData("")));

        return new MockFileSystem(
            fileData.Concat(dirData).ToDictionary(x => x.Item1, y => y.Item2),
            FileUtils.NormalizePath(cwd)
        );
    }

    [Test]
    public void Merge_directories_works()
    {
        var files = FileUtils.NormalizePaths(
            [
                @"path1\1\file1.txt",
                @"path1\1\file2.txt",
                @"path1\1\2\3\4\file3.txt",
                @"path1\file4.txt",
            ]
        );

        var dirs = FileUtils.NormalizePaths(
            [@"path1\empty1", @"path1\empty2", @"path1\1\2\empty3", @"path1\1\2\3\4\empty4"]
        );

        var fs = NewMockFileSystem(files, dirs, @"C:\root\path");

        fs.MergeDirectory(fs.DirectoryInfo.New("path1"), fs.DirectoryInfo.New("path2"));

        fs.AllDirectories.Select(MockUnixSupport.Path)
            .Should()
            .NotContain(x => x.Contains("path1") || x.Contains("empty"));

        fs.AllFiles.Should().BeEquivalentTo(ReRootFiles(fs, files, "path1", "path2"));
    }

    [Test]
    public void Fail_if_file_already_exists()
    {
        var files = FileUtils.NormalizePaths(
            [@"path1\1\file1.txt", @"path1\1\file2.txt", @"path2\1\file1.txt"]
        );

        var fs = NewMockFileSystem(files, @"C:\root\path");

        var act = () =>
            fs.MergeDirectory(fs.DirectoryInfo.New("path1"), fs.DirectoryInfo.New("path2"));

        act.Should().Throw<IOException>();
    }

    [Test]
    public void Fail_if_directory_exists_where_file_goes()
    {
        var files = FileUtils.NormalizePaths([@"path1\1\file1"]);

        var dirs = FileUtils.NormalizePaths([@"path2\1\file1"]);

        var fs = NewMockFileSystem(files, dirs, @"C:\root\path");

        var act = () =>
            fs.MergeDirectory(fs.DirectoryInfo.New("path1"), fs.DirectoryInfo.New("path2"));

        act.Should().Throw<IOException>();
    }

    [Test]
    public void Return_null_when_no_yaml_files_exist()
    {
        var fs = new MockFileSystem();
        var result = fs.CurrentDirectory().YamlFile("test");
        result.Should().BeNull();
    }

    [TestCase("test.yml")]
    [TestCase("test.yaml")]
    public void Return_non_null_when_single_yaml_file_exists(string yamlFilename)
    {
        var fs = new MockFileSystem();
        fs.AddEmptyFile(yamlFilename);

        var result = fs.CurrentDirectory().YamlFile("test");
        result.Should().NotBeNull();
        result!.Name.Should().Be(yamlFilename);
    }

    [Test]
    public void Throw_when_both_files_exist()
    {
        var fs = new MockFileSystem();
        fs.AddEmptyFile("test.yml");
        fs.AddEmptyFile("test.yaml");

        var act = () => fs.CurrentDirectory().YamlFile("test");
        act.Should().Throw<ConflictingYamlFilesException>();
    }
}