You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
391 lines
13 KiB
391 lines
13 KiB
// ReSharper disable RedundantUsingDirective
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.Model;
|
|
using NzbDrone.Core.Providers;
|
|
using NzbDrone.Core.Repository;
|
|
using NzbDrone.Core.Repository.Quality;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Test.Common.AutoMoq;
|
|
|
|
namespace NzbDrone.Core.Test.ProviderTests
|
|
{
|
|
[TestFixture]
|
|
// ReSharper disable InconsistentNaming
|
|
public class InventoryProvider_IsAcceptableSizeTest : CoreTest
|
|
{
|
|
private EpisodeParseResult parseResultMulti;
|
|
private EpisodeParseResult parseResultSingle;
|
|
private Series series30minutes;
|
|
private Series series60minutes;
|
|
private QualityType qualityType;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
parseResultMulti = new EpisodeParseResult
|
|
{
|
|
CleanTitle = "Title",
|
|
Language = LanguageType.English,
|
|
Quality = new Quality(QualityTypes.SDTV, true),
|
|
EpisodeNumbers = new List<int> { 3, 4 },
|
|
SeasonNumber = 12,
|
|
AirDate = DateTime.Now.AddDays(-12).Date
|
|
};
|
|
|
|
parseResultSingle = new EpisodeParseResult
|
|
{
|
|
CleanTitle = "Title",
|
|
Language = LanguageType.English,
|
|
Quality = new Quality(QualityTypes.SDTV, true),
|
|
EpisodeNumbers = new List<int> { 3 },
|
|
SeasonNumber = 12,
|
|
AirDate = DateTime.Now.AddDays(-12).Date
|
|
};
|
|
|
|
series30minutes = Builder<Series>.CreateNew()
|
|
.With(c => c.Monitored = true)
|
|
.With(d => d.CleanTitle = parseResultMulti.CleanTitle)
|
|
.With(c => c.Runtime = 30)
|
|
.Build();
|
|
|
|
series60minutes = Builder<Series>.CreateNew()
|
|
.With(c => c.Monitored = true)
|
|
.With(d => d.CleanTitle = parseResultMulti.CleanTitle)
|
|
.With(c => c.Runtime = 60)
|
|
.Build();
|
|
|
|
qualityType = Builder<QualityType>.CreateNew()
|
|
.With(q => q.MinSize = 0)
|
|
.With(q => q.MaxSize = 10)
|
|
.With(q => q.QualityTypeId = 1)
|
|
.Build();
|
|
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_single_episode_not_first_or_last_30_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Size = 184572800;
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_single_episode_not_first_or_last_60_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Size = 368572800;
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_single_episode_not_first_or_last_30_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Size = 1.Gigabytes();
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_single_episode_not_first_or_last_60_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Size = 1.Gigabytes();
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_multi_episode_not_first_or_last_30_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultMulti.Series = series30minutes;
|
|
parseResultMulti.Size = 184572800;
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultMulti);
|
|
|
|
//Assert
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_multi_episode_not_first_or_last_60_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultMulti.Series = series60minutes;
|
|
parseResultMulti.Size = 368572800;
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultMulti);
|
|
|
|
//Assert
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_multi_episode_not_first_or_last_30_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultMulti.Series = series30minutes;
|
|
parseResultMulti.Size = 1.Gigabytes();
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultMulti);
|
|
|
|
//Assert
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_multi_episode_not_first_or_last_60_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultMulti.Series = series60minutes;
|
|
parseResultMulti.Size = 10.Gigabytes();
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(false);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultMulti);
|
|
|
|
//Assert
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_single_episode_first_30_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Size = 184572800;
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_single_episode_first_60_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Size = 368572800;
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_single_episode_first_30_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Size = 1.Gigabytes();
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_false_single_episode_first_60_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Size = 10.Gigabytes();
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_unlimited_30_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series30minutes;
|
|
parseResultSingle.Size = 18457280000;
|
|
qualityType.MaxSize = 0;
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_true_unlimited_60_minute()
|
|
{
|
|
WithStrictMocker();
|
|
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Size = 36857280000;
|
|
qualityType.MaxSize = 0;
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void IsAcceptableSize_should_treat_daily_series_as_single_episode()
|
|
{
|
|
parseResultSingle.Series = series60minutes;
|
|
parseResultSingle.Size = 300.Megabytes();
|
|
parseResultSingle.AirDate = DateTime.Today;
|
|
parseResultSingle.EpisodeNumbers = null;
|
|
|
|
qualityType.MaxSize = (int)600.Megabytes();
|
|
|
|
Mocker.GetMock<QualityTypeProvider>().Setup(s => s.Get(1)).Returns(qualityType);
|
|
|
|
Mocker.GetMock<EpisodeProvider>().Setup(
|
|
s => s.IsFirstOrLastEpisodeOfSeason(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(true);
|
|
|
|
//Act
|
|
bool result = Mocker.Resolve<InventoryProvider>().IsAcceptableSize(parseResultSingle);
|
|
|
|
//Assert
|
|
result.Should().BeTrue();
|
|
}
|
|
}
|
|
} |