recyclarr/tests/Recyclarr.Cli.Tests/Pipelines/QualityProfile/QualityItemOrganizerTest.cs

using Recyclarr.Cli.Pipelines.QualityProfile;
using Recyclarr.Config.Models;
using Recyclarr.ServarrApi.QualityProfile;

namespace Recyclarr.Cli.Tests.Pipelines.QualityProfile;

[TestFixture]
public class QualityItemOrganizerTest
{
    private readonly QualityProfileConfig _config = new()
    {
        Qualities =
        [
            NewQp.QualityConfig("one"),
            NewQp.QualityConfig("three"),
            NewQp.QualityConfig("six", false),
            NewQp.QualityConfig("seven"),
            NewQp.QualityConfig("nonexistent1"),
            NewQp.GroupConfig("group3", "eight"),
            NewQp.GroupConfig("group4", false, "nine", "ten"),
            NewQp.GroupConfig("group5", "eleven"),
        ],
    };

    private readonly QualityProfileDto _dto = new()
    {
        Items =
        [
            NewQp.QualityDto(1, "one", true),
            NewQp.QualityDto(2, "two", true),
            NewQp.QualityDto(3, "three", true),
            NewQp.QualityDto(9, "nine", true),
            NewQp.GroupDto(50, "group5", true, NewQp.QualityDto(11, "eleven", true)),
            NewQp.QualityDto(10, "ten", true),
            NewQp.QualityDto(4, "four", true),
            NewQp.GroupDto(
                1001,
                "group1",
                true,
                NewQp.QualityDto(5, "five", true),
                NewQp.QualityDto(6, "six", true)
            ),
            NewQp.GroupDto(1002, "group2", true, NewQp.QualityDto(7, "seven", true)),
            NewQp.QualityDto(8, "eight", true),
        ],
    };

    [Test]
    public void Update_qualities_top_sort()
    {
        var sut = new QualityItemOrganizer();
        var result = sut.OrganizeItems(
            _dto,
            _config with
            {
                QualitySort = QualitySortAlgorithm.Top,
            }
        );

        result
            .Should()
            .BeEquivalentTo(
                new UpdatedQualities
                {
                    InvalidQualityNames = ["nonexistent1"],
                    NumWantedItems = 7,
                    Items =
                    [
                        // ------ IN CONFIG ------
                        NewQp.QualityDto(1, "one", true),
                        NewQp.QualityDto(3, "three", true),
                        NewQp.QualityDto(6, "six", false),
                        NewQp.QualityDto(7, "seven", true),
                        NewQp.GroupDto(1001, "group3", true, NewQp.QualityDto(8, "eight", true)),
                        NewQp.GroupDto(
                            1002,
                            "group4",
                            false,
                            NewQp.QualityDto(9, "nine", false),
                            NewQp.QualityDto(10, "ten", false)
                        ),
                        NewQp.GroupDto(50, "group5", true, NewQp.QualityDto(11, "eleven", true)),
                        // ------ NOT IN CONFIG ------
                        NewQp.QualityDto(2, "two", false),
                        NewQp.QualityDto(4, "four", false),
                        NewQp.QualityDto(5, "five", false),
                    ],
                }
            );
    }

    [Test]
    public void Update_qualities_bottom_sort()
    {
        var sut = new QualityItemOrganizer();
        var result = sut.OrganizeItems(
            _dto,
            _config with
            {
                QualitySort = QualitySortAlgorithm.Bottom,
            }
        );

        result
            .Should()
            .BeEquivalentTo(
                new UpdatedQualities
                {
                    InvalidQualityNames = ["nonexistent1"],
                    NumWantedItems = 7,
                    Items =
                    [
                        // ------ NOT IN CONFIG ------
                        NewQp.QualityDto(2, "two", false),
                        NewQp.QualityDto(4, "four", false),
                        NewQp.QualityDto(5, "five", false),
                        // ------ IN CONFIG ------
                        NewQp.QualityDto(1, "one", true),
                        NewQp.QualityDto(3, "three", true),
                        NewQp.QualityDto(6, "six", false),
                        NewQp.QualityDto(7, "seven", true),
                        NewQp.GroupDto(1001, "group3", true, NewQp.QualityDto(8, "eight", true)),
                        NewQp.GroupDto(
                            1002,
                            "group4",
                            false,
                            NewQp.QualityDto(9, "nine", false),
                            NewQp.QualityDto(10, "ten", false)
                        ),
                        NewQp.GroupDto(50, "group5", true, NewQp.QualityDto(11, "eleven", true)),
                    ],
                }
            );
    }

    [Test]
    public void Remove_empty_group()
    {
        var config = new QualityProfileConfig { Qualities = [NewQp.QualityConfig("one")] };

        var dto = new QualityProfileDto
        {
            Items = [NewQp.GroupDto(1001, "group1", true, NewQp.QualityDto(1, "one", true))],
        };

        var sut = new QualityItemOrganizer();
        var result = sut.OrganizeItems(dto, config);

        result.Items.Should().BeEquivalentTo([NewQp.QualityDto(1, "one", true)]);
    }

    [Test]
    public void Flatten_group_with_one_remaining_item()
    {
        var config = new QualityProfileConfig { Qualities = [NewQp.QualityConfig("one")] };

        var dto = new QualityProfileDto
        {
            Items =
            [
                NewQp.GroupDto(
                    1001,
                    "group1",
                    true,
                    NewQp.QualityDto(1, "one", true),
                    NewQp.QualityDto(2, "two", true)
                ),
            ],
        };

        var sut = new QualityItemOrganizer();
        var result = sut.OrganizeItems(dto, config);

        result
            .Items.Should()
            .BeEquivalentTo([NewQp.QualityDto(1, "one", true), NewQp.QualityDto(2, "two", false)]);
    }

    [Test]
    public void Do_not_flatten_group_with_two_remaining_items()
    {
        var config = new QualityProfileConfig { Qualities = [NewQp.QualityConfig("one")] };

        var dto = new QualityProfileDto
        {
            Items =
            [
                NewQp.GroupDto(
                    1001,
                    "group1",
                    true,
                    NewQp.QualityDto(1, "one", true),
                    NewQp.QualityDto(2, "two", true),
                    NewQp.QualityDto(3, "three", true)
                ),
            ],
        };

        var sut = new QualityItemOrganizer();
        var result = sut.OrganizeItems(dto, config);

        result
            .Items.Should()
            .BeEquivalentTo(
                [
                    NewQp.QualityDto(1, "one", true),
                    NewQp.GroupDto(
                        1001,
                        "group1",
                        false,
                        NewQp.QualityDto(2, "two", false),
                        NewQp.QualityDto(3, "three", false)
                    ),
                ]
            );
    }

    [Test]
    public void Remove_quality_from_existing_group()
    {
        var config = new QualityProfileConfig
        {
            Qualities = [NewQp.GroupConfig("group1", "one", "two", "three")],
        };

        var dto = new QualityProfileDto
        {
            Items =
            [
                NewQp.GroupDto(
                    1001,
                    "group1",
                    true,
                    NewQp.QualityDto(1, "one", true),
                    NewQp.QualityDto(2, "two", true),
                    NewQp.QualityDto(3, "three", true),
                    NewQp.QualityDto(4, "four", true)
                ),
            ],
        };

        var sut = new QualityItemOrganizer();
        var result = sut.OrganizeItems(dto, config);

        result
            .Items.Should()
            .BeEquivalentTo(
                [
                    NewQp.GroupDto(
                        1001,
                        "group1",
                        true,
                        NewQp.QualityDto(1, "one", true),
                        NewQp.QualityDto(2, "two", true),
                        NewQp.QualityDto(3, "three", true)
                    ),
                    NewQp.QualityDto(4, "four", false),
                ]
            );
    }
}