// ReSharper disable RedundantUsingDirective using System; using System.Collections.Generic; using System.Data.SqlServerCe; using System.Linq; using AutoMoq; using FizzWare.NBuilder; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Core.Model; using NzbDrone.Core.Providers; using NzbDrone.Core.Providers.Core; using NzbDrone.Core.Repository; using NzbDrone.Core.Repository.Quality; using NzbDrone.Core.Test.Framework; using PetaPoco; using TvdbLib.Data; namespace NzbDrone.Core.Test { [TestFixture] // ReSharper disable InconsistentNaming public class EpisodeProviderTest : TestBase { [Test] public void GetEpisodes_exists() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var fakeEpisodes = Builder.CreateListOfSize(5) .WhereAll().Have(e => e.SeriesId = 1).Have(e => e.EpisodeFileId = 0).Build(); db.Insert(fakeSeries); db.InsertMany(fakeEpisodes); //Act var episode = mocker.Resolve().GetEpisode(1); //Assert episode.ShouldHave().AllPropertiesBut(e => e.Series, e => e.EpisodeFile).EqualTo(fakeEpisodes.First()); episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries); } [Test] public void GetEpisodes_by_season_episode_exists() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew() .With(s => s.SeriesId = 1) .Build(); var fakeEpisodes = Builder.CreateNew() .With(e => e.SeriesId = 1) .With(e => e.EpisodeNumber = 1) .And(e => e.SeasonNumber = 2) .With(e => e.EpisodeFileId = 0).Build(); db.Insert(fakeSeries); db.Insert(fakeEpisodes); //Act var episode = mocker.Resolve().GetEpisode(fakeSeries.SeriesId, 2, 1); //Assert episode.ShouldHave().AllPropertiesBut(e => e.Series).EqualTo(fakeEpisodes); episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries); } [Test] public void GetEpisodes_by_season_episode_doesnt_exists() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); //Act var episode = mocker.Resolve().GetEpisode(1, 1, 1); //Assert episode.Should().BeNull(); } [Test] public void GetEpisode_with_EpisodeFile() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var fakeFile = Builder.CreateNew().With(f => f.EpisodeFileId).Build(); var fakeEpisodes = Builder.CreateListOfSize(5) .WhereAll().Have(e => e.SeriesId = 1).WhereTheFirst(1).Have(e => e.EpisodeFileId = 1).Have(e => e.EpisodeFile = fakeFile).Build(); db.Insert(fakeSeries); db.InsertMany(fakeEpisodes); db.Insert(fakeFile); //Act var episode = mocker.Resolve().GetEpisode(1); //Assert episode.ShouldHave().AllPropertiesBut(e => e.Series, e => e.EpisodeFile).EqualTo(fakeEpisodes.First()); episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries); episode.EpisodeFile.Should().NotBeNull(); } [Test] [ExpectedException(typeof(InvalidOperationException), ExpectedMessage = "Sequence contains no elements")] public void GetEpisodes_invalid_series() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); mocker.Resolve(); var fakeEpisodes = Builder.CreateListOfSize(5) .WhereAll().Have(e => e.SeriesId = 1).Build(); db.InsertMany(fakeEpisodes); //Act mocker.Resolve().GetEpisode(1); } [Test] public void AttachSeries_empty_list() { var mocker = new AutoMoqer(); //Act var result = mocker.Resolve().AttachSeries(new List()); //Assert result.Should().HaveCount(0); } [Test] public void AttachSeries_list_success() { var mocker = new AutoMoqer(); var fakeSeries = Builder.CreateNew().With(s => s.SeriesId = 12).Build(); var fakeEpisodes = Builder.CreateListOfSize(5) .WhereAll().Have(e => e.SeriesId = 12).Build(); mocker.GetMock() .Setup(c => c.GetSeries(12)) .Returns(fakeSeries); //Act fakeEpisodes.Should().OnlyContain(e => e.Series == null); var returnedSeries = mocker.Resolve().AttachSeries(fakeEpisodes); //Assert fakeEpisodes.Should().OnlyContain(e => e.Series == fakeSeries); returnedSeries.Should().BeEquivalentTo(fakeEpisodes); } [Test] public void AttachSeries_null_episode_should_return_null() { var mocker = new AutoMoqer(); Episode episode = null; //Act var result = mocker.Resolve().AttachSeries(episode); //Assert result.Should().BeNull(); } [Test] public void AttachSeries_single_success() { var mocker = new AutoMoqer(); var fakeSeries = Builder.CreateNew().With(s => s.SeriesId = 12).Build(); var fakeEpisodes = Builder.CreateNew().With(e => e.SeriesId = 12).Build(); mocker.GetMock() .Setup(c => c.GetSeries(12)) .Returns(fakeSeries); //Act var returnedEpisode = mocker.Resolve().AttachSeries(fakeEpisodes); //Assert fakeEpisodes.Series.Should().Be(fakeSeries); returnedEpisode.Should().Be(fakeEpisodes); } [Test] [ExpectedException(typeof(InvalidOperationException), ExpectedMessage = "Sequence contains no elements")] public void AttachSeries_single_invalid_series() { var mocker = new AutoMoqer(); mocker.SetConstant(MockLib.GetEmptyDatabase()); mocker.Resolve(); var fakeEpisodes = Builder.CreateNew().With(e => e.SeriesId = 12).Build(); //Act var returnedEpisode = mocker.Resolve().AttachSeries(fakeEpisodes); } [Test] public void GetEpisodesBySeason_success() { var fakeSeries = Builder.CreateNew() .With(s => s.SeriesId = 12) .Build(); var episodes = Builder.CreateListOfSize(10) .WhereAll().Have(c => c.SeriesId = 12) .WhereTheFirst(5).Have(c => c.SeasonNumber = 1) .AndTheRemaining().Have(c => c.SeasonNumber = 2).Build(); var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); db.Insert(fakeSeries); db.InsertMany(episodes); //Act var seasonEposodes = mocker.Resolve().GetEpisodesBySeason(12, 2); //Assert db.Fetch().Should().HaveCount(10); seasonEposodes.Should().HaveCount(5); } [Test] public void RefreshEpisodeInfo_emptyRepo() { //Arrange const int seriesId = 71663; const int episodeCount = 10; var fakeEpisodes = Builder.CreateNew().With( c => c.Episodes = new List(Builder.CreateListOfSize(episodeCount). WhereAll() .Have(l => l.Language = new TvdbLanguage(0, "eng", "a")) .Build()) ).With(c => c.Id = seriesId).Build(); var fakeSeries = Builder.CreateNew().With(c => c.SeriesId = seriesId).Build(); var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); db.Insert(fakeSeries); mocker.GetMock() .Setup(c => c.GetSeries(seriesId, true)) .Returns(fakeEpisodes); //Act mocker.Resolve().RefreshEpisodeInfo(fakeSeries); //Assert var actualCount = mocker.Resolve().GetEpisodeBySeries(seriesId).ToList().Count; mocker.GetMock().VerifyAll(); actualCount.Should().Be(episodeCount); mocker.VerifyAllMocks(); } [Test] public void RefreshEpisodeInfo_should_set_older_than_1900_to_null() { //Arrange const int seriesId = 71663; var fakeEpisodes = Builder.CreateNew().With( c => c.Episodes = new List(Builder.CreateListOfSize(10). WhereAll() .Have(l => l.Language = new TvdbLanguage(0, "eng", "a")) .WhereTheFirst(7).Have(e => e.FirstAired = new DateTime(1800, 1, 1)) .AndTheRemaining().Have(e => e.FirstAired = DateTime.Now) .Build()) ).With(c => c.Id = seriesId).Build(); var fakeSeries = Builder.CreateNew().With(c => c.SeriesId = seriesId).Build(); var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); db.Insert(fakeSeries); mocker.GetMock() .Setup(c => c.GetSeries(seriesId, true)) .Returns(fakeEpisodes); //Act mocker.Resolve().RefreshEpisodeInfo(fakeSeries); //Assert var storedEpisodes = mocker.Resolve().GetEpisodeBySeries(seriesId).ToList(); storedEpisodes.Should().HaveCount(10); storedEpisodes.Where(e => e.AirDate == null).Should().HaveCount(7); storedEpisodes.Where(e => e.AirDate != null).Should().HaveCount(3); mocker.VerifyAllMocks(); } [Test] public void RefreshEpisodeInfo_ignore_episode_zero() { //Arrange const int seriesId = 71663; const int episodeCount = 10; var fakeEpisodes = Builder.CreateNew().With( c => c.Episodes = new List(Builder.CreateListOfSize(episodeCount). WhereAll() .Have(l => l.Language = new TvdbLanguage(0, "eng", "a")) .WhereTheFirst(1) .Has(e => e.EpisodeNumber = 0) .Has(e => e.SeasonNumber = 15) .Build()) ).With(c => c.Id = seriesId).Build(); var fakeSeries = Builder.CreateNew().With(c => c.SeriesId = seriesId).Build(); var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); db.Insert(fakeSeries); mocker.GetMock() .Setup(c => c.GetSeries(seriesId, true)) .Returns(fakeEpisodes); //Act mocker.Resolve().RefreshEpisodeInfo(fakeSeries); //Assert var result = mocker.Resolve().GetEpisodeBySeries(seriesId).ToList(); mocker.GetMock().VerifyAll(); result.Should().HaveCount(episodeCount); mocker.VerifyAllMocks(); result.Where(e => e.EpisodeNumber == 0 && e.SeasonNumber == 15).Single().Ignored.Should().BeTrue(); } [Test] public void new_episodes_only_calls_Insert() { const int seriesId = 71663; var tvdbSeries = Builder.CreateNew() .With(c => c.Episodes = new List(Builder.CreateListOfSize(5).Build()) ).With(c => c.Id = seriesId).Build(); var fakeSeries = Builder.CreateNew().With(c => c.SeriesId = seriesId).Build(); var currentEpisodes = new List(); var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); mocker.GetMock(MockBehavior.Strict) .Setup(c => c.GetSeries(seriesId, true)) .Returns(tvdbSeries); mocker.GetMock() .Setup(d => d.Fetch(It.IsAny(), It.IsAny())) .Returns(currentEpisodes); //Act mocker.Resolve().RefreshEpisodeInfo(fakeSeries); //Assert mocker.GetMock().Verify(c => c.InsertMany(It.Is>(l => l.Count() == 5)), Times.Once()); mocker.GetMock().Verify(c => c.Update(It.IsAny>()), Times.Never()); mocker.VerifyAllMocks(); } [Test] public void existing_episodes_only_calls_Update() { const int seriesId = 71663; var tvdbSeries = Builder.CreateNew() .With(c => c.Episodes = new List(Builder.CreateListOfSize(5).Build()) ).With(c => c.Id = seriesId).Build(); var fakeSeries = Builder.CreateNew().With(c => c.SeriesId = seriesId).Build(); var currentEpisodes = new List(); foreach (var tvDbEpisode in tvdbSeries.Episodes) { currentEpisodes.Add(new Episode { TvDbEpisodeId = tvDbEpisode.Id, Series = fakeSeries }); } var mocker = new AutoMoqer(); mocker.GetMock(MockBehavior.Strict) .Setup(c => c.GetSeries(seriesId, true)) .Returns(tvdbSeries); mocker.GetMock() .Setup(d => d.Fetch(It.IsAny(), It.IsAny())) .Returns(currentEpisodes); //Act mocker.Resolve().RefreshEpisodeInfo(fakeSeries); //Assert mocker.GetMock().Verify(c => c.InsertMany(It.Is>(l => l.Count() == 0)), Times.Once()); mocker.GetMock().Verify(c => c.UpdateMany(It.Is>(l => l.Count() == 5)), Times.Once()); mocker.VerifyAllMocks(); } [Test] public void should_try_to_get_existing_episode_using_tvdbid_first() { const int seriesId = 71663; var fakeTvDbResult = Builder.CreateNew() .With(c => c.Id = seriesId) .With(c => c.Episodes = new List( Builder.CreateListOfSize(1) .WhereAll().Have(g => g.Id = 99) .Build()) ) .Build(); var fakeSeries = Builder.CreateNew().With(c => c.SeriesId = seriesId).Build(); var fakeEpisodeList = new List { new Episode { TvDbEpisodeId = 99, SeasonNumber = 10, EpisodeNumber = 10, Series = fakeSeries } }; var mocker = new AutoMoqer(); mocker.GetMock() .Setup(d => d.Fetch(It.IsAny(), It.IsAny())) .Returns(fakeEpisodeList); mocker.GetMock() .Setup(c => c.GetSeries(seriesId, true)) .Returns(fakeTvDbResult); //Act mocker.Resolve().RefreshEpisodeInfo(fakeSeries); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.UpdateMany(fakeEpisodeList), Times.Once()); } [Test] public void should_try_to_get_existing_episode_using_tvdbid_first_then_season_episode() { const int seriesId = 71663; var tvdbSeries = Builder.CreateNew() .With(c => c.Id = seriesId) .With(c => c.Episodes = new List{ Builder.CreateNew() .With(g => g.Id = 99) .With(g => g.SeasonNumber = 4) .With(g => g.EpisodeNumber = 15) .With(g=>g.SeriesId = seriesId) .Build() }) .Build(); var localEpisode = Builder.CreateNew() .With(c => c.SeriesId = seriesId) .With(c => c.SeasonNumber = 4) .With(c => c.EpisodeNumber = 15) .Build(); var fakeSeries = Builder.CreateNew().With(c => c.SeriesId = seriesId).Build(); var mocker = new AutoMoqer(); mocker.GetMock(MockBehavior.Strict) .Setup(c => c.GetSeries(seriesId, true)) .Returns(tvdbSeries); mocker.GetMock() .Setup(d => d.Fetch(It.IsAny(), It.IsAny())) .Returns(new List { localEpisode }); //Act mocker.Resolve().RefreshEpisodeInfo(fakeSeries); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.UpdateMany(new List { localEpisode }), Times.Once()); } [Test] public void existing_episodes_keep_their_episodeId_file_id() { const int seriesId = 71663; var tvdbSeries = Builder.CreateNew() .With(c => c.Episodes = new List(Builder.CreateListOfSize(5).Build()) ).With(c => c.Id = seriesId).Build(); var fakeSeries = Builder.CreateNew().With(c => c.SeriesId = seriesId).Build(); var currentEpisodes = new List(); foreach (var tvDbEpisode in tvdbSeries.Episodes) { currentEpisodes.Add(new Episode { TvDbEpisodeId = tvDbEpisode.Id, EpisodeId = 99, EpisodeFileId = 69, Ignored = true, Series = fakeSeries }); } var mocker = new AutoMoqer(); mocker.GetMock(MockBehavior.Strict) .Setup(c => c.GetSeries(seriesId, true)) .Returns(tvdbSeries); var updatedEpisodes = new List(); mocker.GetMock() .Setup(d => d.Fetch(It.IsAny(), It.IsAny())) .Returns(currentEpisodes); mocker.GetMock() .Setup(c => c.UpdateMany(It.IsAny>())) .Callback>(ep => updatedEpisodes = ep.ToList()); //Act mocker.Resolve().RefreshEpisodeInfo(fakeSeries); //Assert updatedEpisodes.Should().HaveSameCount(tvdbSeries.Episodes); updatedEpisodes.Should().OnlyContain(c => c.EpisodeId == 99); updatedEpisodes.Should().OnlyContain(c => c.EpisodeFileId == 69); updatedEpisodes.Should().OnlyContain(c => c.Ignored == true); } [Test] public void IsSeasonIgnored_should_return_true_if_all_episodes_ignored() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.Ignored = true) .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 2) .Build(); episodes.ToList().ForEach(c => db.Insert(c)); //Act var result = mocker.Resolve().IsIgnored(10, 2); //Assert result.Should().BeTrue(); } [Test] public void IsSeasonIgnored_should_return_false_if_none_of_episodes_are_ignored() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.Ignored = false) .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 2) .Build(); episodes.ToList().ForEach(c => db.Insert(c)); //Act var result = mocker.Resolve().IsIgnored(10, 2); //Assert result.Should().BeFalse(); } [Test] public void IsSeasonIgnored_should_return_false_if_some_of_episodes_are_ignored() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 2) .Have(c => c.Ignored = true) .Build(); episodes[2].Ignored = false; episodes.ToList().ForEach(c => db.Insert(c)); //Act var result = mocker.Resolve().IsIgnored(10, 2); //Assert result.Should().BeFalse(); } [Test] public void IsSeasonIgnored_should_return_false_if_zero_episodes_in_db_for_season() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 3) .Have(c => c.Ignored = true) .Build(); episodes.ToList().ForEach(c => db.Insert(c)); //Act var result = mocker.Resolve().IsIgnored(10, 2); //Assert result.Should().BeFalse(); } [Test] public void IsSeasonIgnored_should_return_true_if_zero_episodes_in_db_for_season_and_previous_is_ignored() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 3) .Have(c => c.Ignored = true) .Build(); episodes.ToList().ForEach(c => db.Insert(c)); //Act var result = mocker.Resolve().IsIgnored(10, 4); //Assert result.Should().BeTrue(); } [Test] public void IsSeasonIgnored_should_return_false_if_zero_episodes_in_db_for_season_and_previous_is_not_ignored() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 3) .Have(c => c.Ignored = false) .Build(); episodes.ToList().ForEach(c => db.Insert(c)); //Act var result = mocker.Resolve().IsIgnored(10, 4); //Assert result.Should().BeFalse(); } [Test] public void IsSeasonIgnored_should_return_false_if_zero_episodes_in_db_for_season_one() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.SetConstant(db); //Act var result = mocker.Resolve().IsIgnored(10, 1); //Assert result.Should().BeFalse(); } [Test] public void IsSeasonIgnored_should_return_true_if_zero_episodes_in_db_for_season_zero() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.SetConstant(db); //Act var result = mocker.Resolve().IsIgnored(10, 0); //Assert result.Should().BeTrue(); } [Test] public void IsSeasonIgnored_should_return_false_if_season_zero_is_not_ignored() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 0) .Have(c => c.Ignored = false) .Build(); episodes.ToList().ForEach(c => db.Insert(c)); //Act var result = mocker.Resolve().IsIgnored(10, 0); //Assert result.Should().BeFalse(); } [Test] [Explicit] public void Add_daily_show_episodes() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); mocker.Resolve(); mocker.GetMock() .Setup(e => e.DefaultQualityProfile).Returns(1); db.Insert(Builder.CreateNew().Build()); const int tvDbSeriesId = 71256; //act var seriesProvider = mocker.Resolve(); seriesProvider.AddSeries("c:\\test\\", tvDbSeriesId, 1); var episodeProvider = mocker.Resolve(); episodeProvider.RefreshEpisodeInfo(seriesProvider.GetSeries(tvDbSeriesId)); //assert var episodes = episodeProvider.GetEpisodeBySeries(tvDbSeriesId); episodes.Should().NotBeEmpty(); } [Test] public void GetEpisode_by_Season_Episode_none_existing() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); //Act var episode = mocker.Resolve().GetEpisode(1, 1, 1); //Assert episode.Should().BeNull(); } [Test] public void GetEpisode_by_Season_Episode_with_EpisodeFile() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var fakeFile = Builder.CreateNew().With(f => f.EpisodeFileId).Build(); var fakeEpisodes = Builder.CreateListOfSize(5) .WhereAll().Have(e => e.SeriesId = 1).WhereTheFirst(1).Have(e => e.EpisodeFileId = 1).Have(e => e.EpisodeFile = fakeFile).Build(); db.Insert(fakeSeries); db.InsertMany(fakeEpisodes); db.Insert(fakeFile); //Act var episode = mocker.Resolve().GetEpisode(1, 1, 1); //Assert episode.ShouldHave().AllPropertiesBut(e => e.Series, e => e.EpisodeFile).EqualTo(fakeEpisodes.First()); episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries); episode.EpisodeFile.Should().NotBeNull(); } [Test] public void GetEpisode_by_Season_Episode_without_EpisodeFile() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var fakeEpisodes = Builder.CreateListOfSize(5) .WhereAll().Have(e => e.SeriesId = 1).WhereTheFirst(1).Have(e => e.EpisodeFileId = 0).Build(); db.Insert(fakeSeries); db.InsertMany(fakeEpisodes); //Act var episode = mocker.Resolve().GetEpisode(1, 1, 1); //Assert episode.ShouldHave().AllPropertiesBut(e => e.Series).EqualTo(fakeEpisodes.First()); episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries); episode.EpisodeFile.Should().BeNull(); } [Test] public void GetEpisode_by_AirDate_with_EpisodeFile() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var fakeFile = Builder.CreateNew().With(f => f.EpisodeFileId).Build(); var fakeEpisodes = Builder.CreateListOfSize(5) .WhereAll().Have(e => e.SeriesId = 1).WhereTheFirst(1).Have(e => e.EpisodeFileId = 1).Have(e => e.EpisodeFile = fakeFile).Build(); db.Insert(fakeSeries); db.InsertMany(fakeEpisodes); db.Insert(fakeFile); //Act var episode = mocker.Resolve().GetEpisode(1, fakeEpisodes[0].AirDate.Value); //Assert episode.ShouldHave().AllPropertiesBut(e => e.Series, e => e.EpisodeFile).EqualTo(fakeEpisodes.First()); episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries); episode.EpisodeFile.Should().NotBeNull(); } [Test] public void GetEpisode_by_AirDate_without_EpisodeFile() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var fakeEpisodes = Builder.CreateListOfSize(5) .WhereAll().Have(e => e.SeriesId = 1).WhereTheFirst(1).Have(e => e.EpisodeFileId = 0).Build(); db.InsertMany(fakeEpisodes); db.Insert(fakeSeries); //Act var episode = mocker.Resolve().GetEpisode(1, fakeEpisodes[0].AirDate.Value); //Assert episode.ShouldHave().AllPropertiesBut(e => e.Series).EqualTo(fakeEpisodes.First()); episode.Series.ShouldHave().AllPropertiesBut(s => s.EpisodeCount, s => s.EpisodeFileCount, s => s.SeasonCount, s => s.NextAiring).EqualTo(fakeSeries); episode.EpisodeFile.Should().BeNull(); } [Test] public void MarkEpisodeAsFetched() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeEpisodes = Builder.CreateListOfSize(5) .WhereAll().Have(e => e.GrabDate = null) .Build(); db.InsertMany(fakeEpisodes); //Act mocker.Resolve().MarkEpisodeAsFetched(2); var episodes = db.Fetch(); //Assert episodes.Where(e => e.EpisodeId == 2).Single().GrabDate.Should().BeWithin(TimeSpan.FromSeconds(5)).Before( DateTime.Now); episodes.Where(e => e.GrabDate == null).Should().HaveCount(4); } [Test] public void AddEpisode_episode_is_ignored_when_full_season_is_ignored() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.Ignored = true) .Build().ToList(); episodes.ForEach(c => db.Insert(c)); var newEpisode = Builder.CreateNew() .With(e => e.SeriesId = 10) .With(e => e.SeasonNumber = 1) .With(e => e.EpisodeNumber = 8) .With(e => e.SeasonNumber = 1) .With(e => e.Ignored = false) .Build(); //Act mocker.Resolve().AddEpisode(newEpisode); //Assert var episodesInDb = db.Fetch(@"SELECT * FROM Episodes"); episodesInDb.Should().HaveCount(5); episodesInDb.Should().OnlyContain(e => e.Ignored); mocker.VerifyAllMocks(); } [Test] public void AddEpisode_episode_is_not_ignored_when_full_season_is_not_ignored() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.Ignored = false) .Build().ToList(); episodes.ForEach(c => db.Insert(c)); var newEpisode = Builder.CreateNew() .With(e => e.SeriesId = 10) .With(e => e.SeasonNumber = 1) .With(e => e.EpisodeNumber = 8) .With(e => e.SeasonNumber = 1) .With(e => e.Ignored = false) .Build(); //Act mocker.Resolve().AddEpisode(newEpisode); //Assert var episodesInDb = db.Fetch(@"SELECT * FROM Episodes"); episodesInDb.Should().HaveCount(5); episodesInDb.Should().OnlyContain(e => e.Ignored == false); mocker.VerifyAllMocks(); } [Test] public void AddEpisode_episode_is_not_ignored_when_not_full_season_is_not_ignored() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .WhereTheFirst(2) .Have(c => c.Ignored = false) .AndTheRemaining() .Have(c => c.Ignored = true) .Build().ToList(); episodes.ForEach(c => db.Insert(c)); var newEpisode = Builder.CreateNew() .With(e => e.SeriesId = 10) .With(e => e.SeasonNumber = 1) .With(e => e.EpisodeNumber = 8) .With(e => e.SeasonNumber = 1) .With(e => e.Ignored = false) .Build(); //Act mocker.Resolve().AddEpisode(newEpisode); //Assert var episodesInDb = db.Fetch(@"SELECT * FROM Episodes"); episodesInDb.Should().HaveCount(5); episodesInDb.Where(e => e.EpisodeNumber == 8 && !e.Ignored).Should().HaveCount(1); mocker.VerifyAllMocks(); } [Test] public void IgnoreEpisode_Ignore() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.Ignored = false) .Build().ToList(); episodes.ForEach(c => db.Insert(c)); //Act mocker.Resolve().SetEpisodeIgnore(1, true); //Assert var episodesInDb = db.Fetch(@"SELECT * FROM Episodes"); episodesInDb.Should().HaveCount(4); episodesInDb.Where(e => e.Ignored).Should().HaveCount(1); mocker.VerifyAllMocks(); } [Test] public void IgnoreEpisode_RemoveIgnore() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.Ignored = true) .Build().ToList(); episodes.ForEach(c => db.Insert(c)); //Act mocker.Resolve().SetEpisodeIgnore(1, false); //Assert var episodesInDb = db.Fetch(@"SELECT * FROM Episodes"); episodesInDb.Should().HaveCount(4); episodesInDb.Where(e => !e.Ignored).Should().HaveCount(1); mocker.VerifyAllMocks(); } [Test] public void IgnoreSeason_Ignore() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.Ignored = false) .Build().ToList(); episodes.ForEach(c => db.Insert(c)); //Act mocker.Resolve().SetSeasonIgnore(10, 1, true); //Assert var episodesInDb = db.Fetch(@"SELECT * FROM Episodes"); episodesInDb.Should().HaveCount(4); episodesInDb.Where(e => e.Ignored).Should().HaveCount(4); mocker.VerifyAllMocks(); } [Test] public void IgnoreSeason_RemoveIgnore() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.Ignored = true) .Build().ToList(); episodes.ForEach(c => db.Insert(c)); //Act mocker.Resolve().SetSeasonIgnore(10, 1, false); //Assert var episodesInDb = db.Fetch(@"SELECT * FROM Episodes"); episodesInDb.Should().HaveCount(4); episodesInDb.Where(e => !e.Ignored).Should().HaveCount(4); mocker.VerifyAllMocks(); } [Test] public void IgnoreSeason_Ignore_Half() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .WhereTheFirst(2) .Have(c => c.Ignored = false) .AndTheRemaining() .Have(c => c.Ignored = true) .Build().ToList(); episodes.ForEach(c => db.Insert(c)); //Act mocker.Resolve().SetSeasonIgnore(10, 1, true); //Assert var episodesInDb = db.Fetch(@"SELECT * FROM Episodes"); episodesInDb.Should().HaveCount(4); episodesInDb.Where(e => e.Ignored).Should().HaveCount(4); mocker.VerifyAllMocks(); } [Test] public void EpisodesWithoutFiles_no_specials() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var series = Builder.CreateNew() .With(s => s.SeriesId = 10) .Build(); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.AirDate = DateTime.Today.AddDays(-4)) .Have(c => c.Ignored = true) .WhereTheFirst(2) .Have(c => c.EpisodeFileId = 0) .WhereSection(1, 2) .Have(c => c.Ignored = false) .Build().ToList(); var specials = Builder.CreateListOfSize(2) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 0) .Have(c => c.AirDate = DateTime.Today.AddDays(-4)) .Have(c => c.EpisodeFileId = 0) .WhereTheFirst(1) .Have(c => c.Ignored = true) .AndTheRemaining() .Have(c => c.Ignored = false) .Build().ToList(); db.Insert(series); db.InsertMany(episodes); db.InsertMany(specials); //Act var missingFiles = mocker.Resolve().EpisodesWithoutFiles(false); //Assert missingFiles.Should().HaveCount(1); missingFiles.Where(e => e.EpisodeFileId == 0).Should().HaveCount(1); mocker.VerifyAllMocks(); } [Test] public void EpisodesWithoutFiles_with_specials() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var series = Builder.CreateNew() .With(s => s.SeriesId = 10) .Build(); var episodes = Builder.CreateListOfSize(4) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.AirDate = DateTime.Today.AddDays(-4)) .Have(c => c.Ignored = true) .WhereTheFirst(2) .Have(c => c.EpisodeFileId = 0) .WhereSection(1, 2) .Have(c => c.Ignored = false) .Build().ToList(); var specials = Builder.CreateListOfSize(2) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 0) .Have(c => c.AirDate = DateTime.Today.AddDays(-4)) .Have(c => c.EpisodeFileId = 0) .WhereTheFirst(1) .Have(c => c.Ignored = true) .AndTheRemaining() .Have(c => c.Ignored = false) .Build().ToList(); db.Insert(series); db.InsertMany(episodes); db.InsertMany(specials); //Act var missingFiles = mocker.Resolve().EpisodesWithoutFiles(true); //Assert missingFiles.Should().HaveCount(2); missingFiles.Where(e => e.EpisodeFileId == 0).Should().HaveCount(2); mocker.VerifyAllMocks(); } [Test] public void EpisodesWithFiles_success() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var series = Builder.CreateNew() .With(s => s.SeriesId = 10) .Build(); var episodeFile = Builder.CreateNew() .With(c => c.EpisodeFileId = 1) .Build(); var episodes = Builder.CreateListOfSize(2) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.AirDate = DateTime.Today.AddDays(-4)) .Have(c => c.Ignored = true) .Have(c => c.EpisodeFile = episodeFile) .Have(c => c.EpisodeFileId = episodeFile.EpisodeFileId) .Build().ToList(); db.Insert(series); db.Insert(episodeFile); db.InsertMany(episodes); //Act var withFiles = mocker.Resolve().EpisodesWithFiles(); //Assert withFiles.Should().HaveCount(2); withFiles.Where(e => e.EpisodeFileId == 0).Should().HaveCount(0); withFiles.Where(e => e.EpisodeFile == null).Should().HaveCount(0); foreach (var withFile in withFiles) { withFile.EpisodeFile.Should().NotBeNull(); withFile.Series.Title.Should().NotBeNullOrEmpty(); } mocker.VerifyAllMocks(); } [Test] public void EpisodesWithFiles_no_files() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var series = Builder.CreateNew() .With(s => s.SeriesId = 10) .Build(); var episodes = Builder.CreateListOfSize(2) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.AirDate = DateTime.Today.AddDays(-4)) .Have(c => c.Ignored = true) .Have(c => c.EpisodeFileId = 0) .Build().ToList(); db.Insert(series); db.InsertMany(episodes); //Act var withFiles = mocker.Resolve().EpisodesWithFiles(); //Assert withFiles.Should().HaveCount(0); mocker.VerifyAllMocks(); } [Test] public void GetEpisodesByFileId_multi_episodes() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var series = Builder.CreateNew() .With(s => s.SeriesId = 10) .Build(); var fakeEpisodes = Builder.CreateListOfSize(2) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Have(c => c.EpisodeFileId = 12345) .Build(); db.Insert(series); db.InsertMany(fakeEpisodes); //Act var episodes = mocker.Resolve().GetEpisodesByFileId(12345); //Assert episodes.Should().HaveCount(2); mocker.VerifyAllMocks(); } [Test] public void GetEpisodesByFileId_single_episode() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var series = Builder.CreateNew() .With(s => s.SeriesId = 10) .Build(); var fakeEpisode = Builder.CreateNew() .With(c => c.SeriesId = 10) .With(c => c.SeasonNumber = 1) .With(c => c.EpisodeFileId = 12345) .Build(); db.Insert(series); db.Insert(fakeEpisode); //Act var episodes = mocker.Resolve().GetEpisodesByFileId(12345); //Assert episodes.Should().HaveCount(1); episodes.First().ShouldHave().AllPropertiesBut(e => e.Series).EqualTo(fakeEpisode); mocker.VerifyAllMocks(); } [Test] public void IsFirstOrLastEpisodeInSeason_false() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var fakeEpisodes = Builder.CreateListOfSize(10) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Build(); db.InsertMany(fakeEpisodes); //Act var result = mocker.Resolve().IsFirstOrLastEpisodeOfSeason(10, 1, 5); //Assert result.Should().BeFalse(); } [Test] public void IsFirstOrLastEpisodeInSeason_true_first() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var fakeEpisodes = Builder.CreateListOfSize(10) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Build(); db.InsertMany(fakeEpisodes); //Act var result = mocker.Resolve().IsFirstOrLastEpisodeOfSeason(10, 1, 1); //Assert result.Should().BeFalse(); } [Test] public void IsFirstOrLastEpisodeInSeason_true_last() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var fakeEpisodes = Builder.CreateListOfSize(10) .WhereAll() .Have(c => c.SeriesId = 10) .Have(c => c.SeasonNumber = 1) .Build(); db.InsertMany(fakeEpisodes); //Act var result = mocker.Resolve().IsFirstOrLastEpisodeOfSeason(10, 1, 10); //Assert result.Should().BeFalse(); } [TestCase("The Office (US) - S01E01 - Episode Title", PostDownloadStatusType.Unpacking, 1)] [TestCase("The Office (US) - S01E01 - Episode Title", PostDownloadStatusType.Failed, 1)] [TestCase("The Office (US) - S01E01E02 - Episode Title", PostDownloadStatusType.Unpacking, 2)] [TestCase("The Office (US) - S01E01E02 - Episode Title", PostDownloadStatusType.Failed, 2)] [TestCase("The Office (US) - Season 01 - Episode Title", PostDownloadStatusType.Unpacking, 10)] [TestCase("The Office (US) - Season 01 - Episode Title", PostDownloadStatusType.Failed, 10)] public void SetPostDownloadStatus(string folderName, PostDownloadStatusType postDownloadStatus, int episodeCount) { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew() .With(s => s.SeriesId = 12345) .With(s => s.CleanTitle = "officeus") .Build(); var fakeEpisodes = Builder.CreateListOfSize(episodeCount) .WhereAll() .Have(c => c.SeriesId = 12345) .Have(c => c.SeasonNumber = 1) .Have(c => c.PostDownloadStatus = PostDownloadStatusType.Unknown) .Build(); db.Insert(fakeSeries); db.InsertMany(fakeEpisodes); mocker.GetMock().Setup(s => s.FindSeries("officeus")).Returns(fakeSeries); //Act mocker.Resolve().SetPostDownloadStatus(fakeEpisodes.Select(e => e.EpisodeId).ToList(), postDownloadStatus); //Assert var result = db.Fetch(); result.Where(e => e.PostDownloadStatus == postDownloadStatus).Count().Should().Be(episodeCount); } [Test] public void SetPostDownloadStatus_Invalid_EpisodeId() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var postDownloadStatus = PostDownloadStatusType.Failed; var fakeSeries = Builder.CreateNew() .With(s => s.SeriesId = 12345) .With(s => s.CleanTitle = "officeus") .Build(); var fakeEpisodes = Builder.CreateListOfSize(1) .WhereAll() .Have(c => c.SeriesId = 12345) .Have(c => c.SeasonNumber = 1) .Have(c => c.PostDownloadStatus = PostDownloadStatusType.Unknown) .Build(); db.Insert(fakeSeries); db.InsertMany(fakeEpisodes); mocker.GetMock().Setup(s => s.FindSeries("officeus")).Returns(fakeSeries); //Act mocker.Resolve().SetPostDownloadStatus(new List{300}, postDownloadStatus); //Assert var result = db.Fetch(); result.Where(e => e.PostDownloadStatus == postDownloadStatus).Count().Should().Be(0); } [Test] [ExpectedException(typeof(ArgumentException))] public void SetPostDownloadStatus_should_throw_if_episode_list_is_empty() { var mocker = new AutoMoqer(); mocker.Resolve().SetPostDownloadStatus(new List(), PostDownloadStatusType.Failed); } } }