mirror of
https://github.com/Sonarr/Sonarr
synced 2024-12-30 19:56:54 +00:00
872 lines
No EOL
35 KiB
C#
872 lines
No EOL
35 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
|
||
using FizzWare.NBuilder;
|
||
using FluentAssertions;
|
||
using NUnit.Framework;
|
||
using NzbDrone.Core.Providers;
|
||
using NzbDrone.Core.Providers.Core;
|
||
using NzbDrone.Core.Repository;
|
||
using NzbDrone.Core.Repository.Quality;
|
||
using NzbDrone.Core.Test.Framework;
|
||
|
||
// ReSharper disable InconsistentNaming
|
||
namespace NzbDrone.Core.Test.ProviderTests
|
||
{
|
||
[TestFixture]
|
||
public class SeriesProviderTest : CoreTest
|
||
{
|
||
private IList<QualityProfile> _qualityProfiles;
|
||
|
||
[SetUp]
|
||
public void Setup()
|
||
{
|
||
_qualityProfiles = Builder<QualityProfile>
|
||
.CreateListOfSize(2)
|
||
.All()
|
||
.With(p => p.Cutoff = QualityTypes.DVD)
|
||
.With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD })
|
||
.Build();
|
||
}
|
||
|
||
[TestCase(true)]
|
||
[TestCase(false)]
|
||
public void Add_new_series(bool useSeasonFolder)
|
||
{
|
||
WithRealDb();
|
||
|
||
Mocker.GetMock<ConfigProvider>()
|
||
.Setup(c => c.UseSeasonFolder).Returns(useSeasonFolder);
|
||
|
||
var fakeProfiles = Builder<QualityProfile>
|
||
.CreateListOfSize(2)
|
||
.All()
|
||
.With(p => p.Cutoff = QualityTypes.DVD)
|
||
.With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD })
|
||
.Build();
|
||
|
||
Db.InsertMany(fakeProfiles);
|
||
|
||
const string path = "C:\\Test\\";
|
||
const string title = "Test Title";
|
||
const int tvDbId = 1234;
|
||
const int qualityProfileId = 2;
|
||
|
||
//Act
|
||
var seriesProvider = Mocker.Resolve<SeriesProvider>();
|
||
seriesProvider.AddSeries(title, path, tvDbId, qualityProfileId, null);
|
||
|
||
//Assert
|
||
var series = seriesProvider.GetAllSeries();
|
||
series.Should().HaveCount(1);
|
||
Assert.AreEqual(path, series.First().Path);
|
||
Assert.AreEqual(tvDbId, series.First().SeriesId);
|
||
Assert.AreEqual(qualityProfileId, series.First().QualityProfileId);
|
||
Assert.AreEqual(title, series.First().Title);
|
||
series.First().SeasonFolder.Should().Be(useSeasonFolder);
|
||
}
|
||
|
||
|
||
[TestCase(0)]
|
||
[TestCase(-1)]
|
||
public void add_series_should_fail_if_series_is_less_than_zero(int seriesId)
|
||
{
|
||
WithRealDb();
|
||
Assert.Throws<ArgumentOutOfRangeException>(() => Mocker.Resolve<SeriesProvider>().AddSeries("Title", "C:\\Test", seriesId, 1, null));
|
||
}
|
||
|
||
[Test]
|
||
public void find_series_empty_repo()
|
||
{
|
||
WithRealDb();
|
||
|
||
//Act
|
||
var seriesProvider = Mocker.Resolve<SeriesProvider>();
|
||
var series = seriesProvider.FindSeries("My Title");
|
||
|
||
//Assert
|
||
Assert.IsNull(series);
|
||
}
|
||
|
||
[Test]
|
||
[ExpectedException(typeof(InvalidOperationException), ExpectedMessage = "Sequence contains no elements")]
|
||
public void Get_series_invalid_series_id_should_return_null()
|
||
{
|
||
WithRealDb();
|
||
|
||
//Act
|
||
var seriesProvider = Mocker.Resolve<SeriesProvider>();
|
||
var series = seriesProvider.GetSeries(2);
|
||
|
||
|
||
//Assert
|
||
Assert.IsNull(series);
|
||
}
|
||
|
||
[Test]
|
||
public void Get_series_by_id()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeSeries = Builder<Series>.CreateNew()
|
||
.With(c => c.QualityProfileId = 1)
|
||
.With(c => c.EpisodeCount = 0)
|
||
.With(c => c.EpisodeFileCount = 0)
|
||
.With(c => c.SeasonCount = 0)
|
||
.Build();
|
||
|
||
var fakeQuality = Builder<QualityProfile>
|
||
.CreateNew()
|
||
.With(p => p.Cutoff = QualityTypes.DVD)
|
||
.With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD })
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetSeries(1);
|
||
|
||
//Assert
|
||
series.ShouldHave().AllPropertiesBut(s => s.QualityProfile, s => s.SeriesId, s => s.NextAiring).EqualTo(fakeSeries);
|
||
series.QualityProfile.Should().NotBeNull();
|
||
series.QualityProfile.ShouldHave().Properties(q => q.Name, q => q.SonicAllowed, q => q.Cutoff, q => q.SonicAllowed).EqualTo(fakeQuality);
|
||
|
||
}
|
||
|
||
[Test]
|
||
public void Find_series_by_cleanName_mapped()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeSeries = Builder<Series>.CreateNew()
|
||
.With(c => c.QualityProfileId = 1)
|
||
.With(c => c.CleanTitle = "laworder")
|
||
.Build();
|
||
|
||
var fakeQuality = Builder<QualityProfile>
|
||
.CreateNew()
|
||
.With(p => p.Cutoff = QualityTypes.DVD)
|
||
.With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD })
|
||
.Build();
|
||
|
||
var id = Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
Mocker.GetMock<SceneMappingProvider>().Setup(s => s.GetSeriesId("laworder")).Returns(1);
|
||
|
||
var series = Mocker.Resolve<SeriesProvider>().FindSeries("laworder");
|
||
|
||
//Assert
|
||
series.ShouldHave().AllPropertiesBut(s => s.QualityProfile, s => s.SeriesId);
|
||
series.QualityProfile.Should().NotBeNull();
|
||
series.QualityProfile.ShouldHave().Properties(q => q.Name, q => q.SonicAllowed, q => q.Cutoff, q => q.AllowedString);
|
||
}
|
||
|
||
[Test]
|
||
public void find_series_empty_match()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeSEries = Builder<Series>.CreateNew()
|
||
.Build();
|
||
|
||
Db.Insert(fakeSEries);
|
||
|
||
//Act
|
||
var seriesProvider = Mocker.Resolve<SeriesProvider>();
|
||
|
||
//Assert
|
||
seriesProvider.FindSeries("WrongTitle").Should().BeNull();
|
||
}
|
||
|
||
[TestCase("The Test", "Test")]
|
||
[TestCase("Through the Wormhole", "Through.the.Wormhole")]
|
||
public void find_series_match(string title, string searchTitle)
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeSeries = Builder<Series>.CreateNew()
|
||
.With(c => c.Title = title)
|
||
.With(c => c.CleanTitle = Parser.NormalizeTitle(title))
|
||
.Build();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew()
|
||
.With(c => c.QualityProfileId = fakeSeries.QualityProfileId)
|
||
.With(p => p.Cutoff = QualityTypes.DVD)
|
||
.With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD })
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
var series = Mocker.Resolve<SeriesProvider>().FindSeries(searchTitle);
|
||
|
||
//Assert
|
||
series.Should().NotBeNull();
|
||
series.SeriesId.Should().Be(series.SeriesId);
|
||
series.Title.Should().BeEquivalentTo(series.Title);
|
||
}
|
||
|
||
[Test]
|
||
public void is_monitored()
|
||
{
|
||
WithRealDb();
|
||
|
||
Db.Insert(Builder<Series>.CreateNew()
|
||
.With(c => c.Monitored = true)
|
||
.With(c => c.SeriesId = 12)
|
||
.Build());
|
||
|
||
Db.Insert(Builder<Series>.CreateNew()
|
||
.With(c => c.Monitored = false)
|
||
.With(c => c.SeriesId = 11)
|
||
.Build());
|
||
|
||
Db.InsertMany(Builder<QualityProfile>.CreateListOfSize(3)
|
||
.All()
|
||
.With(p => p.Cutoff = QualityTypes.DVD)
|
||
.With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD })
|
||
.Build());
|
||
|
||
//Act, Assert
|
||
var provider = Mocker.Resolve<SeriesProvider>();
|
||
provider.IsMonitored(12).Should().BeTrue();
|
||
Assert.IsFalse(provider.IsMonitored(11));
|
||
Assert.IsFalse(provider.IsMonitored(1));
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_With_Count()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
|
||
.All().With(e => e.SeriesId = fakeSeries.SeriesId)
|
||
.With(e => e.Ignored = false)
|
||
.With(e => e.AirDate = DateTime.Today)
|
||
.TheFirst(5)
|
||
.With(e => e.EpisodeFileId = 0)
|
||
.TheLast(2)
|
||
.With(e => e.AirDate = DateTime.Today.AddDays(1))
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
Assert.AreEqual(8, series[0].EpisodeCount);
|
||
Assert.AreEqual(3, series[0].EpisodeFileCount);
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_With_Count_AllIgnored()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(10).All().With(e => e.SeriesId = fakeSeries.SeriesId).With(e => e.Ignored = true).Random(5).With(e => e.EpisodeFileId = 0).Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
Assert.AreEqual(0, series[0].EpisodeCount);
|
||
Assert.AreEqual(0, series[0].EpisodeFileCount);
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_With_Count_AllDownloaded()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
|
||
.All()
|
||
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
||
.With(e => e.Ignored = false)
|
||
.With(e => e.AirDate = DateTime.Today.AddDays(-1))
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
Assert.AreEqual(10, series[0].EpisodeCount);
|
||
Assert.AreEqual(10, series[0].EpisodeFileCount);
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_With_Count_Half_Ignored()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
|
||
.All()
|
||
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
||
.With(e => e.AirDate = DateTime.Today.AddDays(-1))
|
||
.TheFirst(5)
|
||
.With(e => e.Ignored = false)
|
||
.TheLast(5)
|
||
.With(e => e.Ignored = true)
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
Assert.AreEqual(5, series[0].EpisodeCount);
|
||
Assert.AreEqual(5, series[0].EpisodeFileCount);
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_should_not_return_series_that_do_not_have_info_synced_yet()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(5)
|
||
.All()
|
||
.With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
|
||
.TheLast(2)
|
||
.With(e => e.LastInfoSync = null)
|
||
.Build();
|
||
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
|
||
.All()
|
||
.With(e => e.SeriesId = fakeSeries.First().SeriesId)
|
||
.With(e => e.AirDate = DateTime.Today.AddDays(-1))
|
||
.TheFirst(5)
|
||
.With(e => e.Ignored = false)
|
||
.TheLast(5)
|
||
.With(e => e.Ignored = true)
|
||
.Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(3);
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Single_Series()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew()
|
||
.With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
|
||
.With(e => e.SeriesId = 1)
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetSeries(1);
|
||
|
||
//Assert
|
||
series.QualityProfile.Should().NotBeNull();
|
||
series.QualityProfileId.Should().Be(fakeQuality.QualityProfileId);
|
||
}
|
||
|
||
[Test]
|
||
public void SeriesPathExists_exact_match()
|
||
{
|
||
WithRealDb();
|
||
|
||
var path = @"C:\Test\TV\30 Rock";
|
||
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(10)
|
||
.All()
|
||
.With(c => c.QualityProfileId = 1)
|
||
.TheFirst(1)
|
||
.With(c => c.Path = path)
|
||
.Build();
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
//Mocker.GetMock<IDatabase>().Setup(s => s.Fetch<Series, QualityProfile>(It.IsAny<string>())).Returns(
|
||
//fakeSeries.ToList());
|
||
|
||
var result = Mocker.Resolve<SeriesProvider>().SeriesPathExists(path);
|
||
|
||
//Assert
|
||
result.Should().BeTrue();
|
||
}
|
||
|
||
[Test]
|
||
public void SeriesPathExists_match()
|
||
{
|
||
WithRealDb();
|
||
|
||
var path = @"C:\Test\TV\30 Rock";
|
||
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(10)
|
||
.All()
|
||
.With(c => c.QualityProfileId = 1)
|
||
.TheFirst(1)
|
||
.With(c => c.Path = path)
|
||
.Build();
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
//Mocker.GetMock<IDatabase>().Setup(s => s.Fetch<Series, QualityProfile>(It.IsAny<string>())).Returns(
|
||
//fakeSeries.ToList());
|
||
|
||
var result = Mocker.Resolve<SeriesProvider>().SeriesPathExists(path.ToUpper());
|
||
|
||
//Assert
|
||
result.Should().BeTrue();
|
||
}
|
||
|
||
[Test]
|
||
public void SeriesPathExists_match_alt()
|
||
{
|
||
WithRealDb();
|
||
|
||
var path = @"C:\Test\TV\The Simpsons";
|
||
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(10)
|
||
.All()
|
||
.With(c => c.QualityProfileId = 1)
|
||
.TheFirst(1)
|
||
.With(c => c.Path = path)
|
||
.Build();
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
//Mocker.GetMock<IDatabase>().Setup(s => s.Fetch<Series, QualityProfile>(It.IsAny<string>())).Returns(
|
||
//fakeSeries.ToList());
|
||
|
||
var result = Mocker.Resolve<SeriesProvider>().SeriesPathExists(@"c:\Test\Tv\the sIMpsons");
|
||
|
||
//Assert
|
||
result.Should().BeTrue();
|
||
}
|
||
|
||
[Test]
|
||
public void SeriesPathExists_match_false()
|
||
{
|
||
WithRealDb();
|
||
|
||
var path = @"C:\Test\TV\30 Rock";
|
||
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(10)
|
||
.All()
|
||
.With(c => c.QualityProfileId = 1)
|
||
.TheFirst(1)
|
||
.With(c => c.Path = path)
|
||
.Build();
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
//Mocker.GetMock<IDatabase>().Setup(s => s.Fetch<Series, QualityProfile>(It.IsAny<string>())).Returns(
|
||
//fakeSeries.ToList());
|
||
|
||
var result = Mocker.Resolve<SeriesProvider>().SeriesPathExists(@"C:\Test\TV\Not A match");
|
||
|
||
//Assert
|
||
result.Should().BeFalse();
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_NextAiring_Today()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
|
||
.All()
|
||
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
||
.With(e => e.Ignored = false)
|
||
.With(e => e.AirDate = DateTime.Today.AddDays(1))
|
||
.TheFirst(1)
|
||
.With(e => e.AirDate = DateTime.Today)
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
series[0].NextAiring.Should().Be(DateTime.Today);
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_NextAiring_Tomorrow_Last_Aired_Yesterday()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
|
||
.All()
|
||
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
||
.With(e => e.Ignored = false)
|
||
.With(e => e.AirDate = DateTime.Today.AddDays(1))
|
||
.TheFirst(1)
|
||
.With(e => e.AirDate = DateTime.Today.AddDays(-1))
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
series[0].NextAiring.Should().Be(DateTime.Today.AddDays(1));
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_NextAiring_Unknown()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
|
||
.All()
|
||
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
||
.With(e => e.AirDate = null)
|
||
.With(e => e.Ignored = false)
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
series[0].NextAiring.Should().NotHaveValue();
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_NextAiring_1_month()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
|
||
.All()
|
||
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
||
.With(e => e.Ignored = false)
|
||
.With(e => e.AirDate = DateTime.Today.AddMonths(1))
|
||
.TheFirst(1)
|
||
.With(e => e.AirDate = DateTime.Today.AddDays(-1))
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
series[0].NextAiring.Should().Be(DateTime.Today.AddMonths(1));
|
||
}
|
||
|
||
[Test]
|
||
public void Get_Series_NextAiring_skip_ignored()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
|
||
.All()
|
||
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
||
.With(e => e.AirDate = DateTime.Today.AddMonths(1))
|
||
.With(e => e.Ignored = false)
|
||
.TheFirst(1)
|
||
.With(e => e.AirDate = DateTime.Today.AddDays(1))
|
||
.With(e => e.Ignored = true)
|
||
.Build();
|
||
|
||
Db.Insert(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
Db.InsertMany(fakeEpisodes);
|
||
|
||
//Act
|
||
Mocker.Resolve<QualityProvider>();
|
||
var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
series[0].NextAiring.Should().Be(DateTime.Today.AddMonths(1));
|
||
}
|
||
|
||
[Test]
|
||
public void SearchForSeries_should_return_results_that_start_with_query()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(10)
|
||
.All()
|
||
.With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
|
||
.Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("Titl");
|
||
|
||
//Assert
|
||
series.Should().HaveCount(10);
|
||
}
|
||
|
||
[Test]
|
||
public void SearchForSeries_should_return_results_that_contain_the_query()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(10)
|
||
.All()
|
||
.With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
|
||
.Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("itl");
|
||
|
||
//Assert
|
||
series.Should().HaveCount(10);
|
||
}
|
||
|
||
[Test]
|
||
public void SearchForSeries_should_return_results_that_end_with_the_query()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(10)
|
||
.All()
|
||
.With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
|
||
.Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("2");
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
}
|
||
|
||
[Test]
|
||
public void SearchForSeries_should_not_return_results_that_do_not_contain_the_query()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(10)
|
||
.All()
|
||
.With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
|
||
.Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("NotATitle");
|
||
|
||
//Assert
|
||
series.Should().HaveCount(0);
|
||
}
|
||
|
||
[Test]
|
||
public void SearchForSeries_should_return_results_when_query_has_special_characters()
|
||
{
|
||
WithRealDb();
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(10)
|
||
.All()
|
||
.With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
|
||
.TheLast(1)
|
||
.With(s => s.Title = "It's Always Sunny")
|
||
.Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
//Act
|
||
var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("it's");
|
||
|
||
//Assert
|
||
series.Should().HaveCount(1);
|
||
}
|
||
|
||
[Test]
|
||
public void UpdateFromMassEdit_should_only_update_certain_values()
|
||
{
|
||
WithRealDb();
|
||
var newQualityProfileId = 10;
|
||
var newMonitored = false;
|
||
var newSeasonFolder = false;
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(1)
|
||
.All()
|
||
.With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
|
||
.With(e => e.Monitored = true)
|
||
.With(e => e.SeasonFolder = true)
|
||
.With(s => s.Title = "It's Always Sunny")
|
||
.Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
fakeSeries[0].QualityProfileId = newQualityProfileId;
|
||
fakeSeries[0].Monitored = newMonitored;
|
||
fakeSeries[0].SeasonFolder = newSeasonFolder;
|
||
|
||
//Act
|
||
Mocker.Resolve<SeriesProvider>().UpdateFromSeriesEditor(fakeSeries);
|
||
|
||
//Assert
|
||
var result = Db.Fetch<Series>();
|
||
result.Count.Should().Be(1);
|
||
result.First().QualityProfileId.Should().Be(newQualityProfileId);
|
||
result.First().Monitored.Should().Be(newMonitored);
|
||
result.First().SeasonFolder.Should().Be(newSeasonFolder);
|
||
}
|
||
|
||
[Test]
|
||
public void UpdateFromMassEdit_should_only_update_changed_values()
|
||
{
|
||
WithRealDb();
|
||
var newQualityProfileId = 10;
|
||
var newMonitored = false;
|
||
var newSeasonFolder = false;
|
||
var monitored = true;
|
||
var seasonFolder = true;
|
||
|
||
var fakeQuality = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(2)
|
||
.All()
|
||
.With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
|
||
.With(e => e.Monitored = monitored)
|
||
.With(e => e.SeasonFolder = seasonFolder)
|
||
.With(s => s.Title = "It's Always Sunny")
|
||
.Build();
|
||
|
||
Db.InsertMany(fakeSeries);
|
||
Db.Insert(fakeQuality);
|
||
|
||
fakeSeries[0].QualityProfileId = newQualityProfileId;
|
||
fakeSeries[0].Monitored = newMonitored;
|
||
fakeSeries[0].SeasonFolder = newSeasonFolder;
|
||
|
||
//Act
|
||
Mocker.Resolve<SeriesProvider>().UpdateFromSeriesEditor(fakeSeries);
|
||
|
||
//Assert
|
||
var result = Db.Fetch<Series>();
|
||
result.Count.Should().Be(2);
|
||
result.First().QualityProfileId.Should().Be(newQualityProfileId);
|
||
result.First().Monitored.Should().Be(newMonitored);
|
||
result.First().SeasonFolder.Should().Be(newSeasonFolder);
|
||
result.Last().QualityProfileId.Should().Be(fakeQuality.QualityProfileId);
|
||
result.Last().Monitored.Should().Be(monitored);
|
||
result.Last().SeasonFolder.Should().Be(seasonFolder);
|
||
}
|
||
|
||
[Test]
|
||
public void delete_series_should_delete_all_rows_related_to_the_series()
|
||
{
|
||
var fakeSeries = Builder<Series>.CreateListOfSize(3).Build();
|
||
var fakeEpisodes = Builder<Episode>.CreateListOfSize(30)
|
||
.TheFirst(10).With(c => c.SeriesId = fakeSeries[0].SeriesId)
|
||
.TheNext(10).With(c => c.SeriesId = fakeSeries[1].SeriesId)
|
||
.TheNext(10).With(c => c.SeriesId = fakeSeries[2].SeriesId)
|
||
.Build();
|
||
|
||
|
||
|
||
}
|
||
}
|
||
} |