// ReSharper disable RedundantUsingDirective using System; using System.Collections.Generic; 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_GetEpisodesByParseResult : TestBase { [Test] public void Single_GetSeason_Episode_Exists() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var fakeEpisode = Builder.CreateNew() .With(e => e.SeriesId = 1) .With(e => e.SeasonNumber = 2) .With(e => e.EpisodeNumber = 10) .Build(); var fakeSeries = Builder.CreateNew().Build(); db.Insert(fakeEpisode); db.Insert(fakeSeries); var parseResult = new EpisodeParseResult { Series = fakeSeries, SeasonNumber = 2, EpisodeNumbers = new List { 10 } }; var ep = mocker.Resolve().GetEpisodesByParseResult(parseResult); ep.Should().HaveCount(1); parseResult.EpisodeTitle.Should().Be(fakeEpisode.Title); ep.First().ShouldHave().AllPropertiesBut(e => e.Series); } [Test] public void Single_GetSeason_Episode_Doesnt_exists_should_not_add() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var parseResult = new EpisodeParseResult { Series = fakeSeries, SeasonNumber = 2, EpisodeNumbers = new List { 10 } }; var ep = mocker.Resolve().GetEpisodesByParseResult(parseResult); ep.Should().BeEmpty(); db.Fetch().Should().HaveCount(0); } [Test] public void Single_GetSeason_Episode_Doesnt_exists_should_add() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var parseResult = new EpisodeParseResult { Series = fakeSeries, SeasonNumber = 2, EpisodeNumbers = new List { 10 } }; var ep = mocker.Resolve().GetEpisodesByParseResult(parseResult, true); ep.Should().HaveCount(1); db.Fetch().Should().HaveCount(1); } [Test] public void Multi_GetSeason_Episode_Exists() { var db = MockLib.GetEmptyDatabase(); var mocker = new AutoMoqer(); mocker.SetConstant(db); var fakeEpisode = Builder.CreateNew() .With(e => e.SeriesId = 1) .With(e => e.SeasonNumber = 2) .With(e => e.EpisodeNumber = 10) .Build(); var fakeEpisode2 = Builder.CreateNew() .With(e => e.SeriesId = 1) .With(e => e.SeasonNumber = 2) .With(e => e.EpisodeNumber = 11) .Build(); var fakeSeries = Builder.CreateNew().Build(); db.Insert(fakeEpisode); db.Insert(fakeEpisode2); db.Insert(fakeSeries); var parseResult = new EpisodeParseResult { Series = fakeSeries, SeasonNumber = 2, EpisodeNumbers = new List { 10, 11 } }; var ep = mocker.Resolve().GetEpisodesByParseResult(parseResult); ep.Should().HaveCount(2); db.Fetch().Should().HaveCount(2); ep.First().ShouldHave().AllPropertiesBut(e => e.Series); parseResult.EpisodeTitle.Should().Be(fakeEpisode.Title); } [Test] public void Multi_GetSeason_Episode_Doesnt_exists_should_not_add() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var parseResult = new EpisodeParseResult { Series = fakeSeries, SeasonNumber = 2, EpisodeNumbers = new List { 10, 11 } }; var ep = mocker.Resolve().GetEpisodesByParseResult(parseResult); ep.Should().BeEmpty(); db.Fetch().Should().HaveCount(0); } [Test] public void Multi_GetSeason_Episode_Doesnt_exists_should_add() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var parseResult = new EpisodeParseResult { Series = fakeSeries, SeasonNumber = 2, EpisodeNumbers = new List { 10, 11 } }; var ep = mocker.Resolve().GetEpisodesByParseResult(parseResult, true); ep.Should().HaveCount(2); db.Fetch().Should().HaveCount(2); } [Test] public void Get_Episode_Zero_Doesnt_Exist_Should_add_ignored() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var parseResult = new EpisodeParseResult { Series = fakeSeries, SeasonNumber = 2, EpisodeNumbers = new List { 0 } }; var ep = mocker.Resolve().GetEpisodesByParseResult(parseResult, true); ep.Should().HaveCount(1); db.Fetch().Should().HaveCount(1); ep.First().Ignored.Should().BeTrue(); } [Test] public void Get_Multi_Episode_Zero_Doesnt_Exist_Should_not_add_ignored() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeSeries = Builder.CreateNew().Build(); var parseResult = new EpisodeParseResult { Series = fakeSeries, SeasonNumber = 2, EpisodeNumbers = new List { 0, 1 } }; var ep = mocker.Resolve().GetEpisodesByParseResult(parseResult, true); ep.Should().HaveCount(2); db.Fetch().Should().HaveCount(2); ep.First().Ignored.Should().BeFalse(); } [Test] [Description("GetEpisodeParseResult should return empty list if episode list is null")] public void GetEpisodeParseResult_should_return_empty_list_if_episode_list_is_null() { var mocker = new AutoMoqer(); //Act var episodes = mocker.Resolve().GetEpisodesByParseResult(new EpisodeParseResult()); //Assert episodes.Should().NotBeNull(); episodes.Should().BeEmpty(); } [Test] [Description("GetEpisodeParseResult should return empty list if episode list is empty")] public void GetEpisodeParseResult_should_return_empty_list_if_episode_list_is_empty() { var mocker = new AutoMoqer(); //Act var episodes = mocker.Resolve().GetEpisodesByParseResult(new EpisodeParseResult{EpisodeNumbers = new List()}); //Assert episodes.Should().NotBeNull(); episodes.Should().BeEmpty(); } } }