using System; 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 { [TestCase(true)] [TestCase(false)] public void Add_new_series(bool useSeasonFolder) { WithRealDb(); Mocker.GetMock() .Setup(c => c.UseSeasonFolder).Returns(useSeasonFolder); var fakeProfiles = Builder.CreateListOfSize(2).Build(); Db.InsertMany(fakeProfiles); const string path = "C:\\Test\\"; const int tvDbId = 1234; const int qualityProfileId = 2; //Act var seriesProvider = Mocker.Resolve(); seriesProvider.AddSeries(path, tvDbId, qualityProfileId); //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); 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(() => Mocker.Resolve().AddSeries("C:\\Test", seriesId, 1)); } [Test] public void find_series_empty_repo() { WithRealDb(); //Act var seriesProvider = Mocker.Resolve(); 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(); var series = seriesProvider.GetSeries(2); //Assert Assert.IsNull(series); } [Test] public void Get_series_by_id() { WithRealDb(); var fakeSeries = Builder.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.CreateNew().Build(); Db.Insert(fakeSeries); Db.Insert(fakeQuality); //Act Mocker.Resolve(); var series = Mocker.Resolve().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.CreateNew() .With(c => c.QualityProfileId = 1) .With(c => c.CleanTitle = "laworder") .Build(); var fakeQuality = Builder.CreateNew().Build(); var id = Db.Insert(fakeSeries); Db.Insert(fakeQuality); //Act Mocker.Resolve(); Mocker.GetMock().Setup(s => s.GetSeriesId("laworder")).Returns(1); var series = Mocker.Resolve().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.CreateNew() .Build(); Db.Insert(fakeSEries); //Act var seriesProvider = Mocker.Resolve(); //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.CreateNew() .With(c => c.Title = title) .With(c => c.CleanTitle = Parser.NormalizeTitle(title)) .Build(); var fakeQuality = Builder.CreateNew() .With(c => c.QualityProfileId = fakeSeries.QualityProfileId) .Build(); Db.Insert(fakeSeries); Db.Insert(fakeQuality); //Act var series = Mocker.Resolve().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.CreateNew() .With(c => c.Monitored = true) .With(c => c.SeriesId = 12) .Build()); Db.Insert(Builder.CreateNew() .With(c => c.Monitored = false) .With(c => c.SeriesId = 11) .Build()); Db.InsertMany(Builder.CreateListOfSize(3).Build()); //Act, Assert var provider = Mocker.Resolve(); 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.CreateNew().Build(); var fakeSeries = Builder.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build(); var fakeEpisodes = Builder.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(); var series = Mocker.Resolve().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.CreateNew().Build(); var fakeSeries = Builder.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build(); var fakeEpisodes = Builder.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(); var series = Mocker.Resolve().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.CreateNew().Build(); var fakeSeries = Builder.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build(); var fakeEpisodes = Builder.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(); var series = Mocker.Resolve().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.CreateNew().Build(); var fakeSeries = Builder.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build(); var fakeEpisodes = Builder.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(); var series = Mocker.Resolve().GetAllSeriesWithEpisodeCount(); //Assert series.Should().HaveCount(1); Assert.AreEqual(5, series[0].EpisodeCount); Assert.AreEqual(5, series[0].EpisodeFileCount); } [Test] public void Get_Single_Series() { WithRealDb(); var fakeQuality = Builder.CreateNew().Build(); var fakeSeries = Builder.CreateNew() .With(e => e.QualityProfileId = fakeQuality.QualityProfileId) .With(e => e.SeriesId = 1) .Build(); Db.Insert(fakeSeries); Db.Insert(fakeQuality); //Act Mocker.Resolve(); var series = Mocker.Resolve().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.CreateListOfSize(10) .All() .With(c => c.QualityProfileId = 1) .TheFirst(1) .With(c => c.Path = path) .Build(); var fakeQuality = Builder.CreateNew().Build(); Db.InsertMany(fakeSeries); Db.Insert(fakeQuality); //Act Mocker.Resolve(); //Mocker.GetMock().Setup(s => s.Fetch(It.IsAny())).Returns( //fakeSeries.ToList()); var result = Mocker.Resolve().SeriesPathExists(path); //Assert result.Should().BeTrue(); } [Test] public void SeriesPathExists_match() { WithRealDb(); var path = @"C:\Test\TV\30 Rock"; var fakeSeries = Builder.CreateListOfSize(10) .All() .With(c => c.QualityProfileId = 1) .TheFirst(1) .With(c => c.Path = path) .Build(); var fakeQuality = Builder.CreateNew().Build(); Db.InsertMany(fakeSeries); Db.Insert(fakeQuality); //Act Mocker.Resolve(); //Mocker.GetMock().Setup(s => s.Fetch(It.IsAny())).Returns( //fakeSeries.ToList()); var result = Mocker.Resolve().SeriesPathExists(path.ToUpper()); //Assert result.Should().BeTrue(); } [Test] public void SeriesPathExists_match_alt() { WithRealDb(); var path = @"C:\Test\TV\The Simpsons"; var fakeSeries = Builder.CreateListOfSize(10) .All() .With(c => c.QualityProfileId = 1) .TheFirst(1) .With(c => c.Path = path) .Build(); var fakeQuality = Builder.CreateNew().Build(); Db.InsertMany(fakeSeries); Db.Insert(fakeQuality); //Act Mocker.Resolve(); //Mocker.GetMock().Setup(s => s.Fetch(It.IsAny())).Returns( //fakeSeries.ToList()); var result = Mocker.Resolve().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.CreateListOfSize(10) .All() .With(c => c.QualityProfileId = 1) .TheFirst(1) .With(c => c.Path = path) .Build(); var fakeQuality = Builder.CreateNew().Build(); Db.InsertMany(fakeSeries); Db.Insert(fakeQuality); //Act Mocker.Resolve(); //Mocker.GetMock().Setup(s => s.Fetch(It.IsAny())).Returns( //fakeSeries.ToList()); var result = Mocker.Resolve().SeriesPathExists(@"C:\Test\TV\Not A match"); //Assert result.Should().BeFalse(); } [Test] public void Get_Series_NextAiring_Today() { WithRealDb(); var fakeQuality = Builder.CreateNew().Build(); var fakeSeries = Builder.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build(); var fakeEpisodes = Builder.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(); var series = Mocker.Resolve().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.CreateNew().Build(); var fakeSeries = Builder.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build(); var fakeEpisodes = Builder.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(); var series = Mocker.Resolve().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.CreateNew().Build(); var fakeSeries = Builder.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build(); var fakeEpisodes = Builder.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(); var series = Mocker.Resolve().GetAllSeriesWithEpisodeCount(); //Assert series.Should().HaveCount(1); series[0].NextAiring.Should().NotHaveValue(); } [Test] public void Get_Series_NextAiring_1_month() { WithRealDb(); var fakeQuality = Builder.CreateNew().Build(); var fakeSeries = Builder.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build(); var fakeEpisodes = Builder.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(); var series = Mocker.Resolve().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.CreateNew().Build(); var fakeSeries = Builder.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build(); var fakeEpisodes = Builder.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(); var series = Mocker.Resolve().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.CreateNew().Build(); var fakeSeries = Builder.CreateListOfSize(10) .All() .With(e => e.QualityProfileId = fakeQuality.QualityProfileId) .Build(); Db.InsertMany(fakeSeries); Db.Insert(fakeQuality); //Act var series = Mocker.Resolve().SearchForSeries("Titl"); //Assert series.Should().HaveCount(10); } [Test] public void SearchForSeries_should_return_results_that_contain_the_query() { WithRealDb(); var fakeQuality = Builder.CreateNew().Build(); var fakeSeries = Builder.CreateListOfSize(10) .All() .With(e => e.QualityProfileId = fakeQuality.QualityProfileId) .Build(); Db.InsertMany(fakeSeries); Db.Insert(fakeQuality); //Act var series = Mocker.Resolve().SearchForSeries("itl"); //Assert series.Should().HaveCount(10); } [Test] public void SearchForSeries_should_return_results_that_end_with_the_query() { WithRealDb(); var fakeQuality = Builder.CreateNew().Build(); var fakeSeries = Builder.CreateListOfSize(10) .All() .With(e => e.QualityProfileId = fakeQuality.QualityProfileId) .Build(); Db.InsertMany(fakeSeries); Db.Insert(fakeQuality); //Act var series = Mocker.Resolve().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.CreateNew().Build(); var fakeSeries = Builder.CreateListOfSize(10) .All() .With(e => e.QualityProfileId = fakeQuality.QualityProfileId) .Build(); Db.InsertMany(fakeSeries); Db.Insert(fakeQuality); //Act var series = Mocker.Resolve().SearchForSeries("NotATitle"); //Assert series.Should().HaveCount(0); } [Test] public void SearchForSeries_should_return_results_when_query_has_special_characters() { WithRealDb(); var fakeQuality = Builder.CreateNew().Build(); var fakeSeries = Builder.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().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.CreateNew().Build(); var fakeSeries = Builder.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().UpdateFromMassEdit(fakeSeries); //Assert var result = Db.Fetch(); 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.CreateNew().Build(); var fakeSeries = Builder.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().UpdateFromMassEdit(fakeSeries); //Assert var result = Db.Fetch(); 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); } } }