2020-01-22 21:47:33 +00:00
|
|
|
using System;
|
2019-06-11 02:59:39 +00:00
|
|
|
using System.Collections.Generic;
|
2020-01-22 21:47:33 +00:00
|
|
|
using FizzWare.NBuilder;
|
2018-08-05 14:28:05 +00:00
|
|
|
using FluentAssertions;
|
2020-01-22 21:47:33 +00:00
|
|
|
using Moq;
|
2013-03-07 00:19:49 +00:00
|
|
|
using NUnit.Framework;
|
2020-01-22 21:47:33 +00:00
|
|
|
using NzbDrone.Common.Serializer;
|
2020-01-02 01:55:35 +00:00
|
|
|
using NzbDrone.Core.CustomFormats;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.DecisionEngine.Specifications;
|
2020-01-22 21:47:33 +00:00
|
|
|
using NzbDrone.Core.MediaFiles;
|
|
|
|
using NzbDrone.Core.Movies;
|
2020-02-18 21:03:05 +00:00
|
|
|
using NzbDrone.Core.Parser;
|
2020-01-22 21:47:33 +00:00
|
|
|
using NzbDrone.Core.Parser.Model;
|
2014-06-08 08:22:55 +00:00
|
|
|
using NzbDrone.Core.Profiles;
|
2013-03-07 00:19:49 +00:00
|
|
|
using NzbDrone.Core.Qualities;
|
2020-01-02 01:55:35 +00:00
|
|
|
using NzbDrone.Core.Test.CustomFormats;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.Test.Framework;
|
2013-03-07 00:19:49 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.DecisionEngineTests
|
|
|
|
{
|
|
|
|
[TestFixture]
|
2020-01-22 21:47:33 +00:00
|
|
|
public class CutoffSpecificationFixture : CoreTest<CutoffSpecification>
|
2013-03-07 00:19:49 +00:00
|
|
|
{
|
2020-01-02 01:55:35 +00:00
|
|
|
private CustomFormat _customFormat;
|
2020-01-22 21:47:33 +00:00
|
|
|
private RemoteMovie _remoteMovie;
|
2018-08-05 14:28:05 +00:00
|
|
|
|
|
|
|
[SetUp]
|
|
|
|
public void Setup()
|
|
|
|
{
|
2020-01-22 21:47:33 +00:00
|
|
|
Mocker.SetConstant<IUpgradableSpecification>(Mocker.Resolve<UpgradableSpecification>());
|
|
|
|
|
|
|
|
_remoteMovie = new RemoteMovie()
|
|
|
|
{
|
|
|
|
Movie = Builder<Movie>.CreateNew().Build(),
|
|
|
|
ParsedMovieInfo = Builder<ParsedMovieInfo>.CreateNew().With(x => x.Quality = null).Build()
|
|
|
|
};
|
|
|
|
|
|
|
|
GivenOldCustomFormats(new List<CustomFormat>());
|
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenProfile(Profile profile)
|
|
|
|
{
|
|
|
|
CustomFormatsFixture.GivenCustomFormats(CustomFormat.None);
|
|
|
|
profile.FormatItems = CustomFormatsFixture.GetSampleFormatItems("None");
|
|
|
|
profile.FormatCutoff = CustomFormat.None.Id;
|
|
|
|
_remoteMovie.Movie.Profile = profile;
|
|
|
|
|
|
|
|
Console.WriteLine(profile.ToJson());
|
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenFileQuality(QualityModel quality)
|
|
|
|
{
|
|
|
|
_remoteMovie.Movie.MovieFile = Builder<MovieFile>.CreateNew().With(x => x.Quality = quality).Build();
|
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenNewQuality(QualityModel quality)
|
|
|
|
{
|
|
|
|
_remoteMovie.ParsedMovieInfo.Quality = quality;
|
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenOldCustomFormats(List<CustomFormat> formats)
|
|
|
|
{
|
|
|
|
Mocker.GetMock<ICustomFormatCalculationService>()
|
|
|
|
.Setup(x => x.ParseCustomFormat(It.IsAny<MovieFile>()))
|
|
|
|
.Returns(formats);
|
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenNewCustomFormats(List<CustomFormat> formats)
|
|
|
|
{
|
|
|
|
_remoteMovie.CustomFormats = formats;
|
2018-08-05 14:28:05 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenCustomFormatHigher()
|
|
|
|
{
|
2020-02-18 21:03:05 +00:00
|
|
|
_customFormat = new CustomFormat("My Format", new ResolutionSpecification { Value = (int)Resolution.R1080p }) { Id = 1 };
|
2018-08-05 14:28:05 +00:00
|
|
|
|
2020-01-02 01:55:35 +00:00
|
|
|
CustomFormatsFixture.GivenCustomFormats(_customFormat, CustomFormat.None);
|
2018-08-05 14:28:05 +00:00
|
|
|
}
|
|
|
|
|
2013-03-07 00:19:49 +00:00
|
|
|
[Test]
|
2013-09-11 05:45:08 +00:00
|
|
|
public void should_return_true_if_current_episode_is_less_than_cutoff()
|
2013-03-07 00:19:49 +00:00
|
|
|
{
|
2020-01-22 21:47:33 +00:00
|
|
|
GivenProfile(new Profile { Cutoff = Quality.Bluray1080p.Id, Items = Qualities.QualityFixture.GetDefaultQualities() });
|
|
|
|
GivenFileQuality(new QualityModel(Quality.DVD, new Revision(version: 2)));
|
|
|
|
Subject.IsSatisfiedBy(_remoteMovie, null).Accepted.Should().BeTrue();
|
2013-03-07 00:19:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
2013-09-11 05:45:08 +00:00
|
|
|
public void should_return_false_if_current_episode_is_equal_to_cutoff()
|
2013-03-07 00:19:49 +00:00
|
|
|
{
|
2020-01-22 21:47:33 +00:00
|
|
|
GivenProfile(new Profile { Cutoff = Quality.HDTV720p.Id, Items = Qualities.QualityFixture.GetDefaultQualities() });
|
|
|
|
GivenFileQuality(new QualityModel(Quality.HDTV720p, new Revision(version: 2)));
|
|
|
|
Subject.IsSatisfiedBy(_remoteMovie, null).Accepted.Should().BeFalse();
|
2013-03-07 00:19:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
2013-09-11 05:45:08 +00:00
|
|
|
public void should_return_false_if_current_episode_is_greater_than_cutoff()
|
2013-03-07 00:19:49 +00:00
|
|
|
{
|
2020-01-22 21:47:33 +00:00
|
|
|
GivenProfile(new Profile { Cutoff = Quality.HDTV720p.Id, Items = Qualities.QualityFixture.GetDefaultQualities() });
|
|
|
|
GivenFileQuality(new QualityModel(Quality.Bluray1080p, new Revision(version: 2)));
|
|
|
|
Subject.IsSatisfiedBy(_remoteMovie, null).Accepted.Should().BeFalse();
|
2013-03-07 00:19:49 +00:00
|
|
|
}
|
2013-09-11 05:45:08 +00:00
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_return_true_when_new_episode_is_proper_but_existing_is_not()
|
|
|
|
{
|
2020-01-22 21:47:33 +00:00
|
|
|
GivenProfile(new Profile { Cutoff = Quality.HDTV720p.Id, Items = Qualities.QualityFixture.GetDefaultQualities() });
|
|
|
|
GivenFileQuality(new QualityModel(Quality.HDTV720p, new Revision(version: 1)));
|
|
|
|
GivenNewQuality(new QualityModel(Quality.HDTV720p, new Revision(version: 2)));
|
|
|
|
Subject.IsSatisfiedBy(_remoteMovie, null).Accepted.Should().BeTrue();
|
2013-09-11 05:45:08 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_return_false_if_cutoff_is_met_and_quality_is_higher()
|
|
|
|
{
|
2020-01-22 21:47:33 +00:00
|
|
|
GivenProfile(new Profile { Cutoff = Quality.HDTV720p.Id, Items = Qualities.QualityFixture.GetDefaultQualities() });
|
|
|
|
GivenFileQuality(new QualityModel(Quality.HDTV720p, new Revision(version: 2)));
|
|
|
|
GivenNewQuality(new QualityModel(Quality.Bluray1080p, new Revision(version: 2)));
|
|
|
|
Subject.IsSatisfiedBy(_remoteMovie, null).Accepted.Should().BeFalse();
|
2013-09-11 05:45:08 +00:00
|
|
|
}
|
2018-08-05 14:28:05 +00:00
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_return_false_if_custom_formats_is_met_and_quality_and_format_higher()
|
|
|
|
{
|
2020-01-22 21:47:33 +00:00
|
|
|
GivenProfile(new Profile
|
|
|
|
{
|
|
|
|
Cutoff = Quality.HDTV720p.Id,
|
|
|
|
Items = Qualities.QualityFixture.GetDefaultQualities(),
|
|
|
|
FormatCutoff = CustomFormat.None.Id,
|
|
|
|
FormatItems = CustomFormatsFixture.GetSampleFormatItems("None", "My Format")
|
|
|
|
});
|
|
|
|
|
|
|
|
GivenFileQuality(new QualityModel(Quality.HDTV720p));
|
|
|
|
GivenNewQuality(new QualityModel(Quality.Bluray1080p));
|
|
|
|
|
2018-08-05 14:28:05 +00:00
|
|
|
GivenCustomFormatHigher();
|
2020-01-22 21:47:33 +00:00
|
|
|
|
|
|
|
GivenOldCustomFormats(new List<CustomFormat> { CustomFormat.None });
|
|
|
|
GivenNewCustomFormats(new List<CustomFormat> { _customFormat });
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_remoteMovie, null).Accepted.Should().BeFalse();
|
2018-08-05 14:28:05 +00:00
|
|
|
}
|
2019-08-31 02:50:03 +00:00
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_return_true_if_cutoffs_are_met_but_is_a_revision_upgrade()
|
|
|
|
{
|
2020-01-22 21:47:33 +00:00
|
|
|
GivenProfile(new Profile
|
2019-08-31 02:50:03 +00:00
|
|
|
{
|
|
|
|
Cutoff = Quality.HDTV1080p.Id,
|
|
|
|
Items = Qualities.QualityFixture.GetDefaultQualities(),
|
2020-01-22 21:47:33 +00:00
|
|
|
});
|
|
|
|
|
|
|
|
GivenFileQuality(new QualityModel(Quality.WEBDL1080p, new Revision(version: 1)));
|
|
|
|
GivenNewQuality(new QualityModel(Quality.WEBDL1080p, new Revision(version: 2)));
|
2019-08-31 02:50:03 +00:00
|
|
|
|
2020-01-22 21:47:33 +00:00
|
|
|
Subject.IsSatisfiedBy(_remoteMovie, null).Accepted.Should().BeTrue();
|
2019-08-31 02:50:03 +00:00
|
|
|
}
|
2013-03-07 00:19:49 +00:00
|
|
|
}
|
2014-09-01 02:28:42 +00:00
|
|
|
}
|