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.Model.Notification; using NzbDrone.Core.Providers; using NzbDrone.Core.Providers.Indexer; using NzbDrone.Core.Providers.Jobs; using NzbDrone.Core.Repository; using NzbDrone.Core.Repository.Quality; using NzbDrone.Core.Test.Framework; namespace NzbDrone.Core.Test { [TestFixture] // ReSharper disable InconsistentNaming public class EpisodeSearchJobTest : TestBase { [Test] public void processResults_ParseResult_should_return_after_match() { var parseResults = Builder.CreateListOfSize(5) .Build(); var episode = Builder.CreateNew().Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(c => c.IsQualityNeeded(It.IsAny())).Returns(true); mocker.GetMock() .Setup(c => c.DownloadReport(It.IsAny())).Returns(true); //Act mocker.Resolve().ProcessResults(new ProgressNotification("Test"), episode, parseResults); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.IsQualityNeeded(It.IsAny()), Times.Once()); mocker.GetMock().Verify(c => c.DownloadReport(It.IsAny()), Times.Once()); } [Test] public void processResults_higher_quality_should_be_called_first() { var parseResults = Builder.CreateListOfSize(10) .WhereAll().Have(c => c.Quality = new Quality(QualityTypes.DVD, true)) .WhereRandom(1).Has(c => c.Quality = new Quality(QualityTypes.Bluray1080p, true)) .Build(); var episode = Builder.CreateNew().Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup( c => c.IsQualityNeeded(It.Is(d => d.Quality.QualityType == QualityTypes.Bluray1080p))) .Returns(true); mocker.GetMock() .Setup( c => c.DownloadReport(It.Is(d => d.Quality.QualityType == QualityTypes.Bluray1080p))) .Returns(true); //Act mocker.Resolve().ProcessResults(new ProgressNotification("Test"), episode, parseResults); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.IsQualityNeeded(It.IsAny()), Times.Once()); mocker.GetMock().Verify(c => c.DownloadReport(It.IsAny()), Times.Once()); } [Test] public void processResults_when_same_quality_proper_should_be_called_first() { var parseResults = Builder.CreateListOfSize(20) .WhereAll().Have(c => c.Quality = new Quality(QualityTypes.DVD, false)) .WhereRandom(1).Has(c => c.Quality = new Quality(QualityTypes.DVD, true)) .Build(); parseResults.Where(c => c.Quality.Proper).Should().HaveCount(1); var episode = Builder.CreateNew().Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(c => c.IsQualityNeeded(It.Is(p => p.Quality.Proper))).Returns(true); mocker.GetMock() .Setup(c => c.DownloadReport(It.Is(p => p.Quality.Proper))).Returns(true); //Act mocker.Resolve().ProcessResults(new ProgressNotification("Test"), episode, parseResults); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.IsQualityNeeded(It.IsAny()), Times.Once()); mocker.GetMock().Verify(c => c.DownloadReport(It.IsAny()), Times.Once()); } [Test] public void processResults_when_not_needed_should_check_the_rest() { var parseResults = Builder.CreateListOfSize(4) .Build(); var episode = Builder.CreateNew().Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(c => c.IsQualityNeeded(It.IsAny())).Returns(false); //Act mocker.Resolve().ProcessResults(new ProgressNotification("Test"), episode, parseResults); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.IsQualityNeeded(It.IsAny()), Times.Exactly(4)); ExceptionVerification.ExcpectedWarns(1); } [Test] public void processResults_failed_IsNeeded_should_check_the_rest() { var parseResults = Builder.CreateListOfSize(4) .Build(); var episode = Builder.CreateNew().Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(c => c.IsQualityNeeded(It.IsAny())).Throws(new Exception()); //Act mocker.Resolve().ProcessResults(new ProgressNotification("Test"), episode, parseResults); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.IsQualityNeeded(It.IsAny()), Times.Exactly(4)); ExceptionVerification.ExcpectedErrors(4); ExceptionVerification.ExcpectedWarns(1); } [Test] public void processResults_failed_download_should_not_check_the_rest() { var parseResults = Builder.CreateListOfSize(4) .Build(); var episode = Builder.CreateNew().Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(c => c.IsQualityNeeded(It.IsAny())).Returns(true); mocker.GetMock() .Setup(c => c.DownloadReport(It.IsAny())).Throws(new Exception()); //Act mocker.Resolve().ProcessResults(new ProgressNotification("Test"), episode, parseResults); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.IsQualityNeeded(It.IsAny()), Times.Exactly(1)); mocker.GetMock().Verify(c => c.DownloadReport(It.IsAny()), Times.Exactly(1)); ExceptionVerification.ExcpectedErrors(1); } [Test] [TestCase(0)] [TestCase(-1)] [TestCase(-100)] [ExpectedException(typeof(ArgumentOutOfRangeException))] public void start_target_id_less_than_0_throws_exception(int target) { var mocker = new AutoMoqer(MockBehavior.Strict); mocker.Resolve().Start(new ProgressNotification("Test"), target); } [Test] public void start_should_search_all_providers() { var parseResults = Builder.CreateListOfSize(4) .Build(); var episode = Builder.CreateNew() .With(c => c.Series = Builder.CreateNew().Build()) .With(c => c.SeasonNumber = 12) .Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(c => c.GetEpisode(episode.EpisodeId)) .Returns(episode); var indexer1 = new Mock(); indexer1.Setup(c => c.FetchEpisode(episode.Series.Title, episode.SeasonNumber, episode.EpisodeNumber)) .Returns(parseResults).Verifiable(); var indexer2 = new Mock(); indexer2.Setup(c => c.FetchEpisode(episode.Series.Title, episode.SeasonNumber, episode.EpisodeNumber)) .Returns(parseResults).Verifiable(); var indexers = new List { indexer1.Object, indexer2.Object }; mocker.GetMock() .Setup(c => c.GetEnabledIndexers()) .Returns(indexers); mocker.GetMock() .Setup(c => c.IsQualityNeeded(It.Is(d => d.Series != null && d.Episodes.Count != 0))).Returns(false); //Act mocker.Resolve().Start(new ProgressNotification("Test"), episode.EpisodeId); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.IsQualityNeeded(It.IsAny()), Times.Exactly(8)); ExceptionVerification.ExcpectedWarns(1); indexer1.VerifyAll(); indexer2.VerifyAll(); } [Test] public void start_failed_indexer_should_not_break_job() { var parseResults = Builder.CreateListOfSize(4) .Build(); var episode = Builder.CreateNew() .With(c => c.Series = Builder.CreateNew().Build()) .With(c => c.SeasonNumber = 12) .Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(c => c.GetEpisode(episode.EpisodeId)) .Returns(episode); var indexer1 = new Mock(); indexer1.Setup(c => c.FetchEpisode(episode.Series.Title, episode.SeasonNumber, episode.EpisodeNumber)) .Returns(parseResults).Verifiable(); var indexer2 = new Mock(); indexer2.Setup(c => c.FetchEpisode(episode.Series.Title, episode.SeasonNumber, episode.EpisodeNumber)) .Throws(new Exception()).Verifiable(); var indexer3 = new Mock(); indexer2.Setup(c => c.FetchEpisode(episode.Series.Title, episode.SeasonNumber, episode.EpisodeNumber)) .Returns(parseResults).Verifiable(); var indexers = new List { indexer1.Object, indexer2.Object, indexer3.Object }; mocker.GetMock() .Setup(c => c.GetEnabledIndexers()) .Returns(indexers); mocker.GetMock() .Setup(c => c.IsQualityNeeded(It.Is(d => d.Series != null && d.Episodes.Count != 0))).Returns(false); //Act mocker.Resolve().Start(new ProgressNotification("Test"), episode.EpisodeId); //Assert mocker.VerifyAllMocks(); mocker.GetMock().Verify(c => c.IsQualityNeeded(It.IsAny()), Times.Exactly(8)); ExceptionVerification.ExcpectedWarns(1); ExceptionVerification.ExcpectedErrors(1); indexer1.VerifyAll(); indexer2.VerifyAll(); indexer3.VerifyAll(); } [Test] public void start_no_episode_found_should_return_with_error_logged() { var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(c => c.GetEpisode(It.IsAny())) .Returns(null); //Act mocker.Resolve().Start(new ProgressNotification("Test"), 12); //Assert mocker.VerifyAllMocks(); ExceptionVerification.ExcpectedErrors(1); } } }