mirror of https://github.com/Radarr/Radarr
114 lines
4.0 KiB
C#
114 lines
4.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using FluentAssertions;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.Profiles;
|
|
using NzbDrone.Core.Qualities;
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
namespace NzbDrone.Core.Test.Qualities
|
|
{
|
|
[TestFixture]
|
|
public class QualityFixture : CoreTest
|
|
{
|
|
public static object[] FromIntCases =
|
|
{
|
|
new object[] { 0, Quality.Unknown },
|
|
new object[] { 1, Quality.SDTV },
|
|
new object[] { 2, Quality.DVD },
|
|
new object[] { 3, Quality.WEBDL1080p },
|
|
new object[] { 4, Quality.HDTV720p },
|
|
new object[] { 5, Quality.WEBDL720p },
|
|
new object[] { 6, Quality.Bluray720p },
|
|
new object[] { 7, Quality.Bluray1080p },
|
|
new object[] { 8, Quality.WEBDL480p },
|
|
new object[] { 9, Quality.HDTV1080p },
|
|
|
|
//new object[] {10, Quality.RAWHD},
|
|
new object[] { 16, Quality.HDTV2160p },
|
|
new object[] { 18, Quality.WEBDL2160p },
|
|
new object[] { 19, Quality.Bluray2160p },
|
|
};
|
|
|
|
public static object[] ToIntCases =
|
|
{
|
|
new object[] { Quality.Unknown, 0 },
|
|
new object[] { Quality.SDTV, 1 },
|
|
new object[] { Quality.DVD, 2 },
|
|
new object[] { Quality.WEBDL1080p, 3 },
|
|
new object[] { Quality.HDTV720p, 4 },
|
|
new object[] { Quality.WEBDL720p, 5 },
|
|
new object[] { Quality.Bluray720p, 6 },
|
|
new object[] { Quality.Bluray1080p, 7 },
|
|
new object[] { Quality.WEBDL480p, 8 },
|
|
new object[] { Quality.HDTV1080p, 9 },
|
|
|
|
//new object[] {Quality.RAWHD, 10},
|
|
new object[] { Quality.HDTV2160p, 16 },
|
|
new object[] { Quality.WEBDL2160p, 18 },
|
|
new object[] { Quality.Bluray2160p, 19 },
|
|
};
|
|
|
|
[Test]
|
|
[TestCaseSource("FromIntCases")]
|
|
public void should_be_able_to_convert_int_to_qualityTypes(int source, Quality expected)
|
|
{
|
|
var quality = (Quality)source;
|
|
quality.Should().Be(expected);
|
|
}
|
|
|
|
[Test]
|
|
[TestCaseSource("ToIntCases")]
|
|
public void should_be_able_to_convert_qualityTypes_to_int(Quality source, int expected)
|
|
{
|
|
var i = (int)source;
|
|
i.Should().Be(expected);
|
|
}
|
|
|
|
public static List<ProfileQualityItem> GetDefaultQualities(params Quality[] allowed)
|
|
{
|
|
var qualities = new List<Quality>
|
|
{
|
|
Quality.CAM,
|
|
Quality.TELECINE,
|
|
Quality.DVDSCR,
|
|
Quality.REGIONAL,
|
|
Quality.SDTV,
|
|
Quality.DVD,
|
|
Quality.DVDR,
|
|
Quality.HDTV720p,
|
|
Quality.HDTV1080p,
|
|
Quality.HDTV2160p,
|
|
Quality.WEBDL480p,
|
|
Quality.WEBDL720p,
|
|
Quality.WEBDL1080p,
|
|
Quality.WEBDL2160p,
|
|
Quality.Bluray480p,
|
|
Quality.Bluray576p,
|
|
Quality.Bluray720p,
|
|
Quality.Bluray1080p,
|
|
Quality.Bluray2160p,
|
|
Quality.BRDISK,
|
|
Quality.RAWHD
|
|
};
|
|
|
|
if (allowed.Length == 0)
|
|
{
|
|
allowed = qualities.ToArray();
|
|
}
|
|
|
|
var items = qualities
|
|
.Except(allowed)
|
|
.Concat(allowed)
|
|
.Select(v => new ProfileQualityItem
|
|
{
|
|
Quality = v,
|
|
Allowed = allowed.Contains(v)
|
|
}).ToList();
|
|
|
|
return items;
|
|
}
|
|
}
|
|
}
|