mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-10-31 07:52:37 +01:00
398 lines
12 KiB
C#
398 lines
12 KiB
C#
using System.Collections.Generic;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.DecisionEngine.Specifications;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Qualities;
|
|
using NzbDrone.Core.Tv;
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
namespace NzbDrone.Core.Test.DecisionEngineTests
|
|
{
|
|
[TestFixture]
|
|
|
|
public class AcceptableSizeSpecificationFixture : CoreTest<AcceptableSizeSpecification>
|
|
{
|
|
private RemoteEpisode parseResultMulti;
|
|
private RemoteEpisode parseResultSingle;
|
|
private Series series30minutes;
|
|
private Series series60minutes;
|
|
private QualitySize qualityType;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
parseResultMulti = new RemoteEpisode
|
|
{
|
|
Report = new ReportInfo(),
|
|
ParsedEpisodeInfo = new ParsedEpisodeInfo { Quality = new QualityModel(Quality.SDTV, true) },
|
|
Episodes = new List<Episode> { new Episode(), new Episode() }
|
|
};
|
|
|
|
parseResultSingle = new RemoteEpisode
|
|
{
|
|
Report = new ReportInfo(),
|
|
ParsedEpisodeInfo = new ParsedEpisodeInfo { Quality = new QualityModel(Quality.SDTV, true) },
|
|
Episodes = new List<Episode> { new Episode() }
|
|
|
|
};
|
|
|
|
series30minutes = Builder<Series>.CreateNew()
|
|
.With(c => c.Runtime = 30)
|
|
.Build();
|
|
|
|
series60minutes = Builder<Series>.CreateNew()
|
|
.With(c => c.Runtime = 60)
|
|
.Build();
|
|
|
|
qualityType = Builder<QualitySize>.CreateNew()
|
|
.With(q => q.MinSize = 0)
|
|
.With(q => q.MaxSize = 10)
|
|
.With(q => q.QualityId = 1)
|
|
.Build();
|
|
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_single_episode_not_first_or_last_30_minute()
|
|
{
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Report.Size = 184572800;
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_single_episode_not_first_or_last_60_minute()
|
|
{
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Report.Size = 368572800;
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_single_episode_not_first_or_last_30_minute()
|
|
{
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Report.Size = 1.Gigabytes();
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_single_episode_not_first_or_last_60_minute()
|
|
{
|
|
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Report.Size = 1.Gigabytes();
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_multi_episode_not_first_or_last_30_minute()
|
|
{
|
|
|
|
|
|
parseResultMulti.Series = series30minutes;
|
|
parseResultMulti.Report.Size = 184572800;
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultMulti);
|
|
|
|
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_multi_episode_not_first_or_last_60_minute()
|
|
{
|
|
|
|
|
|
parseResultMulti.Series = series60minutes;
|
|
parseResultMulti.Report.Size = 368572800;
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultMulti);
|
|
|
|
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_multi_episode_not_first_or_last_30_minute()
|
|
{
|
|
|
|
|
|
parseResultMulti.Series = series30minutes;
|
|
parseResultMulti.Report.Size = 1.Gigabytes();
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultMulti);
|
|
|
|
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_multi_episode_not_first_or_last_60_minute()
|
|
{
|
|
|
|
|
|
parseResultMulti.Series = series60minutes;
|
|
parseResultMulti.Report.Size = 10.Gigabytes();
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultMulti);
|
|
|
|
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_single_episode_first_30_minute()
|
|
{
|
|
|
|
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Report.Size = 184572800;
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_single_episode_first_60_minute()
|
|
{
|
|
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Report.Size = 368572800;
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_single_episode_first_30_minute()
|
|
{
|
|
|
|
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Report.Size = 1.Gigabytes();
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_single_episode_first_60_minute()
|
|
{
|
|
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Report.Size = 10.Gigabytes();
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_unlimited_30_minute()
|
|
{
|
|
|
|
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Report.Size = 18457280000;
|
|
qualityType.MaxSize = 0;
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_unlimited_60_minute()
|
|
{
|
|
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Report.Size = 36857280000;
|
|
qualityType.MaxSize = 0;
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_should_treat_daily_series_as_single_episode()
|
|
{
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Series.SeriesType = SeriesTypes.Daily;
|
|
|
|
parseResultSingle.Report.Size = 300.Megabytes();
|
|
|
|
qualityType.MaxSize = (int)600.Megabytes();
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<IEpisodeService>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
|
|
bool result = Subject.IsSatisfiedBy(parseResultSingle);
|
|
|
|
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_true_if_RAWHD()
|
|
{
|
|
var parseResult = new RemoteEpisode
|
|
{
|
|
ParsedEpisodeInfo = new ParsedEpisodeInfo { Quality = new QualityModel(Quality.RAWHD, false) },
|
|
};
|
|
|
|
Subject.IsSatisfiedBy(parseResult).Should().BeTrue();
|
|
}
|
|
|
|
|
|
[Test]
|
|
public void should_always_return_false_if_unknow()
|
|
{
|
|
var parseResult = new RemoteEpisode
|
|
{
|
|
ParsedEpisodeInfo = new ParsedEpisodeInfo { Quality = new QualityModel(Quality.Unknown, false) },
|
|
};
|
|
|
|
Subject.IsSatisfiedBy(parseResult).Should().BeFalse();
|
|
|
|
|
|
Mocker.GetMock<IQualitySizeService>().Verify(c=>c.Get(It.IsAny<int>()),Times.Never());
|
|
}
|
|
}
|
|
} |