recyclarr/tests/Recyclarr.Cli.Tests/NewQp.cs

using System.Diagnostics.CodeAnalysis;
using Recyclarr.Cli.Pipelines.QualityProfile;
using Recyclarr.Cli.Pipelines.QualityProfile.Models;
using Recyclarr.Config.Models;
using Recyclarr.ServarrApi.QualityProfile;

namespace Recyclarr.Cli.Tests;

public static class NewQp
{
    public static ProcessedQualityProfileData Processed(
        string profileName,
        params (string TrashId, int FormatId, int Score)[] scores)
    {
        return Processed(profileName, false, scores);
    }

    public static ProcessedQualityProfileData Processed(
        string profileName,
        bool resetUnmatchedScores,
        params (string TrashId, int FormatId, int Score)[] scores)
    {
        return Processed(profileName, resetUnmatchedScores,
            scores.Select(x => ("", x.TrashId, x.FormatId, x.Score)).ToArray());
    }

    public static ProcessedQualityProfileData Processed(
        string profileName,
        bool resetUnmatchedScores,
        params (string CfName, string TrashId, int FormatId, int Score)[] scores)
    {
        var profileConfig = new QualityProfileConfig
        {
            Name = profileName,
            ResetUnmatchedScores = new ResetUnmatchedScoresConfig
            {
                Enabled = resetUnmatchedScores
            }
        };

        return Processed(profileConfig, scores);
    }

    public static ProcessedQualityProfileData Processed(
        QualityProfileConfig profileConfig,
        params (string CfName, string TrashId, int FormatId, int Score)[] scores)
    {
        return new ProcessedQualityProfileData
        {
            Profile = profileConfig,
            CfScores = scores
                .Select(x => new ProcessedQualityProfileScore(x.TrashId, x.CfName, x.FormatId, x.Score))
                .ToList()
        };
    }

    public static UpdatedFormatScore UpdatedScore(
        string name,
        int oldScore,
        int newScore,
        FormatScoreUpdateReason reason)
    {
        return new UpdatedFormatScore(
            new ProfileFormatItemDto {Name = name, Score = oldScore},
            newScore,
            reason);
    }

    public static ProfileItemDto GroupDto(
        int itemId,
        string itemName,
        bool enabled,
        params ProfileItemDto[] nestedItems)
    {
        return new ProfileItemDto
        {
            Id = itemId,
            Name = itemName,
            Allowed = enabled,
            Items = nestedItems
        };
    }

    public static ProfileItemDto QualityDto(int itemId, string itemName, bool enabled)
    {
        return new ProfileItemDto
        {
            Allowed = enabled,
            Quality = new ProfileItemQualityDto
            {
                Id = itemId,
                Name = itemName
            }
        };
    }

    [SuppressMessage("ReSharper", "IntroduceOptionalParameters.Global", Justification =
        "This is for unit test purposes and we want to be explicit sometimes")]
    public static QualityProfileQualityConfig QualityConfig(string itemName)
    {
        return QualityConfig(itemName, true);
    }

    public static QualityProfileQualityConfig QualityConfig(string itemName, bool enabled)
    {
        return new QualityProfileQualityConfig
        {
            Enabled = enabled,
            Name = itemName
        };
    }

    public static QualityProfileQualityConfig GroupConfig(string itemName, params string[] nestedItems)
    {
        return GroupConfig(itemName, true, nestedItems);
    }

    public static QualityProfileQualityConfig GroupConfig(string itemName, bool enabled, params string[] nestedItems)
    {
        return QualityConfig(itemName, enabled) with {Qualities = nestedItems};
    }
}