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

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

namespace Recyclarr.Cli.Tests.Pipelines.QualityProfile;

[TestFixture]
public class UpdatedQualityProfileTest
{
    [Test]
    public void Profile_name_uses_dto_first()
    {
        var profile = new UpdatedQualityProfile
        {
            ProfileDto = new QualityProfileDto {Name = "dto_name"},
            ProfileConfig = NewQp.Processed("config_name"),
            UpdateReason = QualityProfileUpdateReason.New
        };

        profile.ProfileName.Should().Be("dto_name");
    }

    [Test]
    public void Profile_name_uses_config_second()
    {
        var profile = new UpdatedQualityProfile
        {
            ProfileDto = new QualityProfileDto(),
            ProfileConfig = NewQp.Processed("config_name"),
            UpdateReason = QualityProfileUpdateReason.New
        };

        profile.ProfileName.Should().Be("config_name");
    }

    [Test]
    public void Dto_updated_from_config_with_qualities()
    {
        var profile = new UpdatedQualityProfile
        {
            ProfileDto = new QualityProfileDto
            {
                Id = 1,
                Name = "dto_name",
                MinFormatScore = 100,
                CutoffFormatScore = 200,
                UpgradeAllowed = false,
                Cutoff = 1
            },
            ProfileConfig = NewQp.Processed(new QualityProfileConfig
            {
                Name = "config_name",
                MinFormatScore = 110,
                UpgradeAllowed = true,
                UpgradeUntilScore = 220
            }),
            UpdatedQualities = new UpdatedQualities
            {
                NumWantedItems = 1,
                Items = new List<ProfileItemDto>
                {
                    NewQp.QualityDto(1, "Quality Item 1", true),
                    NewQp.QualityDto(2, "Quality Item 2", true),
                    NewQp.GroupDto(3, "Quality Item 3", true,
                        NewQp.QualityDto(4, "Quality Item 4", true))
                }
            },
            UpdateReason = QualityProfileUpdateReason.New
        };

        var result = profile.BuildUpdatedDto();

        result.Should().BeEquivalentTo(new QualityProfileDto
        {
            // For right now, names are used for lookups (since QPs have no cache yet). As such, two profiles with
            // different names will never be matched and so the names should normally be identical. However, for testing
            // purposes, I made them different to make sure it doesn't get overwritten.
            Name = "dto_name",
            Id = 1,
            MinFormatScore = 110,
            CutoffFormatScore = 220,
            UpgradeAllowed = true,
            Items = profile.UpdatedQualities.Items
        }, o => o.Excluding(x => x.Cutoff));
    }

    [Test]
    public void Dto_quality_items_updated_from_config_with_no_qualities()
    {
        var profile = new UpdatedQualityProfile
        {
            ProfileDto = new QualityProfileDto
            {
                Items = new List<ProfileItemDto>
                {
                    NewQp.QualityDto(8, "Quality Item 8", true),
                    NewQp.QualityDto(9, "Quality Item 9", true)
                }
            },
            ProfileConfig = NewQp.Processed(""),
            UpdatedQualities = new UpdatedQualities
            {
                NumWantedItems = 0,
                Items = new List<ProfileItemDto>
                {
                    NewQp.QualityDto(1, "Quality Item 1", true),
                    NewQp.QualityDto(2, "Quality Item 2", true),
                    NewQp.GroupDto(3, "Quality Item 3", true,
                        NewQp.QualityDto(4, "Quality Item 4", true))
                }
            },
            UpdateReason = QualityProfileUpdateReason.New
        };

        var result = profile.BuildUpdatedDto();

        result.Items.Should().BeEquivalentTo(profile.ProfileDto.Items);
    }

    [Test]
    public void Dto_name_is_updated_when_empty()
    {
        var profile = new UpdatedQualityProfile
        {
            ProfileDto = new QualityProfileDto {Name = ""},
            ProfileConfig = NewQp.Processed("config_name"),
            UpdatedQualities = new UpdatedQualities
            {
                Items =
                [
                    new ProfileItemDto()
                ]
            },
            UpdateReason = QualityProfileUpdateReason.New
        };

        var dto = profile.BuildUpdatedDto();

        dto.Name.Should().Be("config_name");
    }

    [Test]
    public void Cutoff_obtained_from_updated_qualities()
    {
        var profile = new UpdatedQualityProfile
        {
            ProfileDto = new QualityProfileDto
            {
                Items = new List<ProfileItemDto>
                {
                    NewQp.QualityDto(8, "Quality Item 8", true),
                    NewQp.QualityDto(9, "Quality Item 9", true)
                }
            },
            ProfileConfig = NewQp.Processed(new QualityProfileConfig
            {
                UpgradeUntilQuality = "Quality Item 2"
            }),
            UpdatedQualities = new UpdatedQualities
            {
                NumWantedItems = 1,
                Items = new List<ProfileItemDto>
                {
                    NewQp.QualityDto(1, "Quality Item 1", true),
                    NewQp.QualityDto(2, "Quality Item 2", true),
                    NewQp.GroupDto(3, "Quality Item 3", true,
                        NewQp.QualityDto(4, "Quality Item 4", true))
                }
            },
            UpdateReason = QualityProfileUpdateReason.New
        };

        var dto = profile.BuildUpdatedDto();

        dto.Cutoff.Should().Be(2);
    }

    [Test]
    public void Cutoff_obtained_from_original_qualities()
    {
        var profile = new UpdatedQualityProfile
        {
            ProfileDto = new QualityProfileDto
            {
                Items = new List<ProfileItemDto>
                {
                    NewQp.QualityDto(8, "Quality Item 8", true),
                    NewQp.QualityDto(9, "Quality Item 9", true)
                }
            },
            ProfileConfig = NewQp.Processed(new QualityProfileConfig
            {
                UpgradeUntilQuality = "Quality Item 9"
            }),
            UpdatedQualities = new UpdatedQualities
            {
                NumWantedItems = 0, // zero forces cutoff search to fall back to original DTO items
                Items = new List<ProfileItemDto>
                {
                    NewQp.QualityDto(1, "Quality Item 1", true),
                    NewQp.QualityDto(2, "Quality Item 2", true),
                    NewQp.GroupDto(3, "Quality Item 3", true,
                        NewQp.QualityDto(4, "Quality Item 4", true))
                }
            },
            UpdateReason = QualityProfileUpdateReason.New
        };

        var dto = profile.BuildUpdatedDto();

        dto.Cutoff.Should().Be(9);
    }

    [Test]
    public void Cutoff_fall_back_to_first()
    {
        var profile = new UpdatedQualityProfile
        {
            ProfileDto = new QualityProfileDto
            {
                Items = new List<ProfileItemDto>
                {
                    NewQp.QualityDto(8, "Quality Item 8", true),
                    NewQp.QualityDto(9, "Quality Item 9", true)
                }
            },
            ProfileConfig = NewQp.Processed(new QualityProfileConfig
            {
                // Do not specify an `UpgradeUntilQuality` here to simulate fallback
            }),
            UpdatedQualities = new UpdatedQualities
            {
                NumWantedItems = 1,
                Items = new List<ProfileItemDto>
                {
                    NewQp.QualityDto(1, "Quality Item 1", true),
                    NewQp.QualityDto(2, "Quality Item 2", true),
                    NewQp.GroupDto(3, "Quality Item 3", true,
                        NewQp.QualityDto(4, "Quality Item 4", true))
                }
            },
            UpdateReason = QualityProfileUpdateReason.New
        };

        var dto = profile.BuildUpdatedDto();

        dto.Cutoff.Should().Be(1);
    }
}