mirror of
https://github.com/Radarr/Radarr.git
synced 2024-08-18 16:29:41 +02:00
parent
386b33b624
commit
f0d10fe1cd
@ -0,0 +1,211 @@
|
|||||||
|
using System.Collections.Generic;
|
||||||
|
using FluentAssertions;
|
||||||
|
using NUnit.Framework;
|
||||||
|
using NzbDrone.Core.CustomFormats;
|
||||||
|
using NzbDrone.Core.DecisionEngine.Specifications;
|
||||||
|
using NzbDrone.Core.Profiles;
|
||||||
|
using NzbDrone.Core.Qualities;
|
||||||
|
using NzbDrone.Core.Test.Framework;
|
||||||
|
|
||||||
|
namespace NzbDrone.Core.Test.DecisionEngineTests
|
||||||
|
{
|
||||||
|
[TestFixture]
|
||||||
|
public class UpgradeAllowedSpecificationFixture : CoreTest<UpgradableSpecification>
|
||||||
|
{
|
||||||
|
private CustomFormat _customFormatOne;
|
||||||
|
private CustomFormat _customFormatTwo;
|
||||||
|
private Profile _qualityProfile;
|
||||||
|
|
||||||
|
[SetUp]
|
||||||
|
public void Setup()
|
||||||
|
{
|
||||||
|
_customFormatOne = new CustomFormat
|
||||||
|
{
|
||||||
|
Id = 1,
|
||||||
|
Name = "One"
|
||||||
|
};
|
||||||
|
_customFormatTwo = new CustomFormat
|
||||||
|
{
|
||||||
|
Id = 2,
|
||||||
|
Name = "Two"
|
||||||
|
};
|
||||||
|
|
||||||
|
_qualityProfile = new Profile
|
||||||
|
{
|
||||||
|
Cutoff = Quality.Bluray1080p.Id,
|
||||||
|
Items = Qualities.QualityFixture.GetDefaultQualities(),
|
||||||
|
UpgradeAllowed = false,
|
||||||
|
CutoffFormatScore = 100,
|
||||||
|
FormatItems = new List<ProfileFormatItem>
|
||||||
|
{
|
||||||
|
new ProfileFormatItem
|
||||||
|
{
|
||||||
|
Id = 1,
|
||||||
|
Format = _customFormatOne,
|
||||||
|
Score = 50
|
||||||
|
},
|
||||||
|
new ProfileFormatItem
|
||||||
|
{
|
||||||
|
Id = 1,
|
||||||
|
Format = _customFormatTwo,
|
||||||
|
Score = 100
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_false_when_quality_is_better_custom_formats_are_the_same_and_upgrading_is_not_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = false;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat>(),
|
||||||
|
new QualityModel(Quality.Bluray1080p),
|
||||||
|
new List<CustomFormat>())
|
||||||
|
.Should().BeFalse();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_false_when_quality_is_same_and_custom_format_is_upgrade_and_upgrading_is_not_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = false;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatOne },
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatTwo })
|
||||||
|
.Should().BeFalse();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_true_for_custom_format_upgrade_when_upgrading_is_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = true;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatOne },
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatTwo })
|
||||||
|
.Should().BeTrue();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_true_for_same_custom_format_score_when_upgrading_is_not_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = false;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatOne },
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatOne })
|
||||||
|
.Should().BeTrue();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_true_for_lower_custom_format_score_when_upgrading_is_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = true;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatTwo },
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatOne })
|
||||||
|
.Should().BeTrue();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_true_for_lower_language_when_upgrading_is_not_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = false;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatTwo },
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat> { _customFormatOne })
|
||||||
|
.Should().BeTrue();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_true_for_quality_upgrade_when_upgrading_is_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = true;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat>(),
|
||||||
|
new QualityModel(Quality.Bluray1080p),
|
||||||
|
new List<CustomFormat>())
|
||||||
|
.Should().BeTrue();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_true_for_same_quality_when_upgrading_is_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = true;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat>(),
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat>())
|
||||||
|
.Should().BeTrue();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_true_for_same_quality_when_upgrading_is_not_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = false;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat>(),
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat>())
|
||||||
|
.Should().BeTrue();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_true_for_lower_quality_when_upgrading_is_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = true;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat>(),
|
||||||
|
new QualityModel(Quality.SDTV),
|
||||||
|
new List<CustomFormat>())
|
||||||
|
.Should().BeTrue();
|
||||||
|
}
|
||||||
|
|
||||||
|
[Test]
|
||||||
|
public void should_return_true_for_lower_quality_when_upgrading_is_not_allowed()
|
||||||
|
{
|
||||||
|
_qualityProfile.UpgradeAllowed = false;
|
||||||
|
|
||||||
|
Subject.IsUpgradeAllowed(
|
||||||
|
_qualityProfile,
|
||||||
|
new QualityModel(Quality.DVD),
|
||||||
|
new List<CustomFormat>(),
|
||||||
|
new QualityModel(Quality.SDTV),
|
||||||
|
new List<CustomFormat>())
|
||||||
|
.Should().BeTrue();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -144,7 +144,13 @@ public bool IsUpgradeAllowed(Profile qualityProfile, QualityModel currentQuality
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
return false;
|
if ((isQualityUpgrade || isCustomFormatUpgrade) && !qualityProfile.UpgradeAllowed)
|
||||||
|
{
|
||||||
|
_logger.Debug("Quality profile does not allow upgrades, skipping");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user