using FluentAssertions; using NUnit.Framework; using NzbDrone.Core.Configuration; using NzbDrone.Core.Profiles.Qualities; using NzbDrone.Core.Qualities; using NzbDrone.Core.DecisionEngine; using NzbDrone.Core.Test.Framework; using NzbDrone.Core.Languages; using NzbDrone.Core.Profiles.Languages; using NzbDrone.Core.Test.Languages; namespace NzbDrone.Core.Test.DecisionEngineTests { [TestFixture] public class QualityUpgradeSpecificationFixture : CoreTest { public static object[] IsUpgradeTestCases = { new object[] { Quality.MP3_192, 1, Quality.MP3_192, 2, Quality.MP3_192, true }, new object[] { Quality.MP3_320, 1, Quality.MP3_320, 2, Quality.MP3_320, true }, new object[] { Quality.MP3_192, 1, Quality.MP3_192, 1, Quality.MP3_192, false }, new object[] { Quality.MP3_320, 1, Quality.MP3_256, 2, Quality.MP3_320, false }, new object[] { Quality.MP3_320, 1, Quality.MP3_256, 2, Quality.MP3_320, false }, new object[] { Quality.MP3_320, 1, Quality.MP3_320, 1, Quality.MP3_320, false }, new object[] { Quality.MP3_512, 1, Quality.MP3_512, 1, Quality.MP3_512, false } }; public static object[] IsUpgradeTestCasesLanguages = { new object[] { Quality.MP3_192, 1, Language.English, Quality.MP3_192, 2, Language.English, Quality.MP3_192, Language.Spanish, true }, new object[] { Quality.MP3_192, 1, Language.English, Quality.MP3_192, 1, Language.Spanish, Quality.MP3_192, Language.Spanish, true }, new object[] { Quality.MP3_320, 1, Language.French, Quality.MP3_320, 2, Language.English, Quality.MP3_320, Language.Spanish, true }, new object[] { Quality.MP3_192, 1, Language.English, Quality.MP3_192, 1, Language.English, Quality.MP3_192, Language.English, false }, new object[] { Quality.MP3_320, 1, Language.English, Quality.MP3_320, 2, Language.Spanish, Quality.FLAC, Language.Spanish, false }, new object[] { Quality.MP3_320, 1, Language.Spanish, Quality.MP3_320, 2, Language.French, Quality.MP3_320, Language.Spanish, false } }; [SetUp] public void Setup() { } private void GivenAutoDownloadPropers(bool autoDownloadPropers) { Mocker.GetMock() .SetupGet(s => s.AutoDownloadPropers) .Returns(autoDownloadPropers); } [Test, TestCaseSource(nameof(IsUpgradeTestCases))] public void IsUpgradeTest(Quality current, int currentVersion, Quality newQuality, int newVersion, Quality cutoff, bool expected) { GivenAutoDownloadPropers(true); var profile = new Profile { Items = Qualities.QualityFixture.GetDefaultQualities() }; var langProfile = new LanguageProfile { Languages = LanguageFixture.GetDefaultLanguages(), Cutoff = Language.English }; Subject.IsUpgradable(profile, langProfile, new QualityModel(current, new Revision(version: currentVersion)), Language.English, new QualityModel(newQuality, new Revision(version: newVersion)), Language.English) .Should().Be(expected); } [Test, TestCaseSource("IsUpgradeTestCasesLanguages")] public void IsUpgradeTestLanguage(Quality current, int currentVersion, Language currentLanguage, Quality newQuality, int newVersion, Language newLanguage, Quality cutoff, Language languageCutoff, bool expected) { GivenAutoDownloadPropers(true); var profile = new Profile { Items = Qualities.QualityFixture.GetDefaultQualities(), Cutoff = cutoff.Id, }; var langProfile = new LanguageProfile { Languages = LanguageFixture.GetDefaultLanguages(), Cutoff = languageCutoff }; Subject.IsUpgradable(profile, langProfile, new QualityModel(current, new Revision(version: currentVersion)), currentLanguage, new QualityModel(newQuality, new Revision(version: newVersion)), newLanguage).Should().Be(expected); } [Test] public void should_return_false_if_proper_and_autoDownloadPropers_is_false() { GivenAutoDownloadPropers(false); var profile = new Profile { Items = Qualities.QualityFixture.GetDefaultQualities(), }; var langProfile = new LanguageProfile { Languages = LanguageFixture.GetDefaultLanguages(), Cutoff = Language.English }; Subject.IsUpgradable(profile, langProfile, new QualityModel(Quality.MP3_256, new Revision(version: 2)), Language.English, new QualityModel(Quality.MP3_256, new Revision(version: 1)), Language.English) .Should().BeFalse(); } } }