recyclarr/tests/Recyclarr.IntegrationTests/IncludePostProcessorIntegrationTest.cs

using System.IO.Abstractions;
using Recyclarr.Config.Parsing;
using Recyclarr.Config.Parsing.PostProcessing;
using Recyclarr.Config.Parsing.PostProcessing.ConfigMerging;
using Recyclarr.TestLibrary;

namespace Recyclarr.IntegrationTests;

[TestFixture]
public class IncludePostProcessorIntegrationTest : IntegrationTestFixture
{
    [Test]
    public void No_change_when_no_includes()
    {
        var sut = Resolve<IncludePostProcessor>();

        var config = new RootConfigYaml
        {
            Radarr = new Dictionary<string, RadarrConfigYaml?>
            {
                ["service1"] = new() { ApiKey = "asdf", BaseUrl = "fdsa" },
            },
        };

        var result = sut.Process(config);

        result.Should().BeEquivalentTo(config);
    }

    [Test]
    public void Throw_when_unable_to_parse()
    {
        var sut = Resolve<IncludePostProcessor>();

        var configPath = Fs.CurrentDirectory().File("my-include.yml");

        Fs.AddFile(
            configPath,
            new MockFileData(
                """
                asdf: invalid
                """
            )
        );

        var config = new RootConfigYaml
        {
            Radarr = new Dictionary<string, RadarrConfigYaml?>
            {
                ["service1"] = new()
                {
                    Include = [new ConfigYamlInclude { Config = configPath.FullName }],
                },
            },
        };

        var act = () => sut.Process(config);

        act.Should()
            .Throw<YamlIncludeException>()
            .WithMessage("*parse include file*my-include.yml*");
    }

    [Test]
    public void Throw_when_unable_to_validate()
    {
        var sut = Resolve<IncludePostProcessor>();

        var configPath = Fs.CurrentDirectory().File("my-include.yml");

        Fs.AddFile(
            configPath,
            new MockFileData(
                """
                custom_formats:
                """
            )
        );

        var config = new RootConfigYaml
        {
            Radarr = new Dictionary<string, RadarrConfigYaml?>
            {
                ["service1"] = new()
                {
                    Include = [new ConfigYamlInclude { Config = configPath.FullName }],
                },
            },
        };

        var act = () => sut.Process(config);

        act.Should()
            .Throw<YamlIncludeException>()
            .WithMessage("*Validation*failed*my-include.yml*");
    }

    [Test]
    public void Merge_works()
    {
        var sut = Resolve<IncludePostProcessor>();

        var configPath1 = Fs.CurrentDirectory().File("my-include1.yml");
        Fs.AddFile(
            configPath1,
            new MockFileData(
                """
                custom_formats:
                  - trash_ids:
                      - 496f355514737f7d83bf7aa4d24f8169

                quality_definition:
                  type: anime
                  preferred_ratio: 0.75

                delete_old_custom_formats: false
                """
            )
        );

        var configPath2 = Fs.CurrentDirectory().File("sub_dir/my-include2.yml");
        Fs.AddFile(
            configPath2,
            new MockFileData(
                """
                custom_formats:
                  - trash_ids:
                      - 240770601cc226190c367ef59aba7463

                delete_old_custom_formats: true
                """
            )
        );

        var config = new RootConfigYaml
        {
            Radarr = new Dictionary<string, RadarrConfigYaml?>
            {
                ["service1"] = new()
                {
                    BaseUrl = "the_base_url",
                    ApiKey = "the_api_key",
                    CustomFormats =
                    [
                        new CustomFormatConfigYaml
                        {
                            TrashIds = ["2f22d89048b01681dde8afe203bf2e95"],
                        },
                    ],
                    QualityDefinition = new QualitySizeConfigYaml { Type = "series" },
                    Include =
                    [
                        new ConfigYamlInclude { Config = configPath1.FullName },
                        new ConfigYamlInclude { Config = configPath2.FullName },
                    ],
                },
            },
        };

        var result = sut.Process(config);

        result
            .Should()
            .BeEquivalentTo(
                new RootConfigYaml
                {
                    Radarr = new Dictionary<string, RadarrConfigYaml?>
                    {
                        ["service1"] = new()
                        {
                            BaseUrl = "the_base_url",
                            ApiKey = "the_api_key",
                            Include = null,
                            CustomFormats =
                            [
                                new CustomFormatConfigYaml
                                {
                                    TrashIds =
                                    [
                                        "496f355514737f7d83bf7aa4d24f8169",
                                        "240770601cc226190c367ef59aba7463",
                                    ],
                                },
                                new CustomFormatConfigYaml
                                {
                                    TrashIds = ["2f22d89048b01681dde8afe203bf2e95"],
                                },
                            ],
                            QualityDefinition = new QualitySizeConfigYaml
                            {
                                Type = "series",
                                PreferredRatio = 0.75m,
                            },
                            DeleteOldCustomFormats = true,
                        },
                    },
                }
            );
    }
}