using System.Collections.Generic; using System.Linq; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Core.DecisionEngine; using NzbDrone.Core.DecisionEngine.Specifications; using NzbDrone.Core.IndexerSearch.Definitions; using NzbDrone.Core.Movies; using NzbDrone.Core.Parser; using NzbDrone.Core.Parser.Model; using NzbDrone.Core.Test.Framework; using NzbDrone.Test.Common; namespace NzbDrone.Core.Test.DecisionEngineTests { [TestFixture] public class DownloadDecisionMakerFixture : CoreTest { private List _reports; private RemoteMovie _remoteEpisode; private MappingResult _mappingResult; private Mock _pass1; private Mock _pass2; private Mock _pass3; private Mock _fail1; private Mock _fail2; private Mock _fail3; [SetUp] public void Setup() { ParseMovieTitle(); _pass1 = new Mock(); _pass2 = new Mock(); _pass3 = new Mock(); _fail1 = new Mock(); _fail2 = new Mock(); _fail3 = new Mock(); _pass1.Setup(c => c.IsSatisfiedBy(It.IsAny(), null)).Returns(Decision.Accept); _pass2.Setup(c => c.IsSatisfiedBy(It.IsAny(), null)).Returns(Decision.Accept); _pass3.Setup(c => c.IsSatisfiedBy(It.IsAny(), null)).Returns(Decision.Accept); _fail1.Setup(c => c.IsSatisfiedBy(It.IsAny(), null)).Returns(Decision.Reject("fail1")); _fail2.Setup(c => c.IsSatisfiedBy(It.IsAny(), null)).Returns(Decision.Reject("fail2")); _fail3.Setup(c => c.IsSatisfiedBy(It.IsAny(), null)).Returns(Decision.Reject("fail3")); _reports = new List { new ReleaseInfo { Title = "Trolls.2016.720p.WEB-DL.DD5.1.H264-FGT" } }; _remoteEpisode = new RemoteMovie { Movie = new Movie(), ParsedMovieInfo = new ParsedMovieInfo() }; _mappingResult = new MappingResult { Movie = new Movie(), MappingResultType = MappingResultType.Success }; _mappingResult.RemoteMovie = _remoteEpisode; Mocker.GetMock() .Setup(c => c.Map(It.IsAny(), It.IsAny(), It.IsAny())).Returns(_mappingResult); } private void GivenSpecifications(params Mock[] mocks) { Mocker.SetConstant>(mocks.Select(c => c.Object)); } [Test] public void should_call_all_specifications() { GivenSpecifications(_pass1, _pass2, _pass3, _fail1, _fail2, _fail3); Subject.GetRssDecision(_reports).ToList(); _fail1.Verify(c => c.IsSatisfiedBy(_remoteEpisode, null), Times.Once()); _fail2.Verify(c => c.IsSatisfiedBy(_remoteEpisode, null), Times.Once()); _fail3.Verify(c => c.IsSatisfiedBy(_remoteEpisode, null), Times.Once()); _pass1.Verify(c => c.IsSatisfiedBy(_remoteEpisode, null), Times.Once()); _pass2.Verify(c => c.IsSatisfiedBy(_remoteEpisode, null), Times.Once()); _pass3.Verify(c => c.IsSatisfiedBy(_remoteEpisode, null), Times.Once()); } [Test] public void should_return_rejected_if_single_specs_fail() { GivenSpecifications(_fail1); var result = Subject.GetRssDecision(_reports); result.Single().Approved.Should().BeFalse(); } [Test] public void should_return_rejected_if_one_of_specs_fail() { GivenSpecifications(_pass1, _fail1, _pass2, _pass3); var result = Subject.GetRssDecision(_reports); result.Single().Approved.Should().BeFalse(); } [Test] public void should_return_pass_if_all_specs_pass() { GivenSpecifications(_pass1, _pass2, _pass3); var result = Subject.GetRssDecision(_reports); result.Single().Approved.Should().BeTrue(); } [Test] public void should_have_same_number_of_rejections_as_specs_that_failed() { GivenSpecifications(_pass1, _pass2, _pass3, _fail1, _fail2, _fail3); var result = Subject.GetRssDecision(_reports); result.Single().Rejections.Should().HaveCount(3); } [Test] public void should_not_attempt_to_map_episode_if_not_parsable() { GivenSpecifications(_pass1, _pass2, _pass3); _reports[0].Title = "Not parsable"; _mappingResult.MappingResultType = MappingResultType.NotParsable; Subject.GetRssDecision(_reports).ToList(); Mocker.GetMock().Verify(c => c.Map(It.IsAny(), It.IsAny(), It.IsAny()), Times.Never()); _pass1.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); _pass2.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); _pass3.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); } [Test] public void should_not_attempt_to_map_episode_if_series_title_is_blank() { GivenSpecifications(_pass1, _pass2, _pass3); _reports[0].Title = "1937 - Snow White and the Seven Dwarves"; _mappingResult.MappingResultType = MappingResultType.NotParsable; var results = Subject.GetRssDecision(_reports).ToList(); Mocker.GetMock().Verify(c => c.Map(It.IsAny(), It.IsAny(), It.IsAny()), Times.Never()); _pass1.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); _pass2.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); _pass3.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); results.Should().NotBeEmpty(); } [Test] public void should_return_rejected_result_for_unparsable_search() { GivenSpecifications(_pass1, _pass2, _pass3); _reports[0].Title = "1937 - Snow White and the Seven Dwarves"; _mappingResult.MappingResultType = MappingResultType.NotParsable; Subject.GetSearchDecision(_reports, new MovieSearchCriteria()).ToList(); Mocker.GetMock().Verify(c => c.Map(It.IsAny(), It.IsAny(), It.IsAny()), Times.Never()); _pass1.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); _pass2.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); _pass3.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); } [Test] public void should_not_attempt_to_make_decision_if_series_is_unknown() { GivenSpecifications(_pass1, _pass2, _pass3); _remoteEpisode.Movie = null; _mappingResult.MappingResultType = MappingResultType.TitleNotFound; Subject.GetRssDecision(_reports); _pass1.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); _pass2.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); _pass3.Verify(c => c.IsSatisfiedBy(It.IsAny(), null), Times.Never()); } [Test] public void broken_report_shouldnt_blowup_the_process() { GivenSpecifications(_pass1); Mocker.GetMock().Setup(c => c.Map(It.IsAny(), It.IsAny(), It.IsAny())) .Throws(); _reports = new List { new ReleaseInfo { Title = "Trolls.2016.720p.WEB-DL.DD5.1.H264-FGT" }, new ReleaseInfo { Title = "Trolls.2016.720p.WEB-DL.DD5.1.H264-FGT" }, new ReleaseInfo { Title = "Trolls.2016.720p.WEB-DL.DD5.1.H264-FGT" } }; Subject.GetRssDecision(_reports); Mocker.GetMock().Verify(c => c.Map(It.IsAny(), It.IsAny(), It.IsAny()), Times.Exactly(_reports.Count)); ExceptionVerification.ExpectedErrors(3); } [Test] public void should_return_unknown_series_rejection_if_series_is_unknown() { GivenSpecifications(_pass1, _pass2, _pass3); _remoteEpisode.Movie = null; var result = Subject.GetRssDecision(_reports); result.Should().HaveCount(1); } [Test] public void should_not_allow_download_if_series_is_unknown() { GivenSpecifications(_pass1, _pass2, _pass3); _remoteEpisode.Movie = null; _mappingResult.MappingResultType = MappingResultType.TitleNotFound; var result = Subject.GetRssDecision(_reports); result.Should().HaveCount(1); // result.First().RemoteMovie.DownloadAllowed.Should().BeFalse(); } [Test] [Ignore("Series")] public void should_not_allow_download_if_no_episodes_found() { GivenSpecifications(_pass1, _pass2, _pass3); _remoteEpisode.Movie = null; var result = Subject.GetRssDecision(_reports); result.Should().HaveCount(1); // result.First().RemoteMovie.DownloadAllowed.Should().BeFalse(); } [Test] public void should_return_a_decision_when_exception_is_caught() { GivenSpecifications(_pass1); Mocker.GetMock().Setup(c => c.Map(It.IsAny(), It.IsAny(), It.IsAny())) .Throws(); _reports = new List { new ReleaseInfo { Title = "Trolls.2016.720p.WEB-DL.DD5.1.H264-FGT" }, }; Subject.GetRssDecision(_reports).Should().HaveCount(1); ExceptionVerification.ExpectedErrors(1); } } }