using System; using System.Collections.Generic; using System.Linq; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Core.DecisionEngine; using NzbDrone.Core.MediaFiles; using NzbDrone.Core.MediaFiles.EpisodeImport; using NzbDrone.Core.Parser; using NzbDrone.Core.Parser.Model; using NzbDrone.Core.Test.Framework; using NzbDrone.Core.Tv; using NzbDrone.Test.Common; namespace NzbDrone.Core.Test.MediaFileTests.EpisodeImportTests { [TestFixture] public class ImportDecisionMakerFixture : CoreTest { private List _videoFiles; private LocalEpisode _localEpisode; private Series _series; private Mock _pass1; private Mock _pass2; private Mock _pass3; private Mock _fail1; private Mock _fail2; private Mock _fail3; [SetUp] public void Setup() { _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())).Returns(true); _pass1.Setup(c => c.RejectionReason).Returns("_pass1"); _pass2.Setup(c => c.IsSatisfiedBy(It.IsAny())).Returns(true); _pass2.Setup(c => c.RejectionReason).Returns("_pass2"); _pass3.Setup(c => c.IsSatisfiedBy(It.IsAny())).Returns(true); _pass3.Setup(c => c.RejectionReason).Returns("_pass3"); _fail1.Setup(c => c.IsSatisfiedBy(It.IsAny())).Returns(false); _fail1.Setup(c => c.RejectionReason).Returns("_fail1"); _fail2.Setup(c => c.IsSatisfiedBy(It.IsAny())).Returns(false); _fail2.Setup(c => c.RejectionReason).Returns("_fail2"); _fail3.Setup(c => c.IsSatisfiedBy(It.IsAny())).Returns(false); _fail3.Setup(c => c.RejectionReason).Returns("_fail3"); _videoFiles = new List { @"C:\Test\Unsorted\The.Office.S03E115.DVDRip.XviD-OSiTV.avi" }; _series = new Series(); _localEpisode = new LocalEpisode { Series = _series, Path = @"C:\Test\Unsorted\The.Office.S03E115.DVDRip.XviD-OSiTV.avi" }; Mocker.GetMock() .Setup(c => c.GetEpisodes(It.IsAny(), It.IsAny(), It.IsAny())) .Returns(_localEpisode); Mocker.GetMock() .Setup(c => c.FilterExistingFiles(_videoFiles, It.IsAny())) .Returns(_videoFiles); } 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.GetImportDecisions(_videoFiles, new Series(), false); _fail1.Verify(c => c.IsSatisfiedBy(_localEpisode), Times.Once()); _fail2.Verify(c => c.IsSatisfiedBy(_localEpisode), Times.Once()); _fail3.Verify(c => c.IsSatisfiedBy(_localEpisode), Times.Once()); _pass1.Verify(c => c.IsSatisfiedBy(_localEpisode), Times.Once()); _pass2.Verify(c => c.IsSatisfiedBy(_localEpisode), Times.Once()); _pass3.Verify(c => c.IsSatisfiedBy(_localEpisode), Times.Once()); } [Test] public void should_return_rejected_if_single_specs_fail() { GivenSpecifications(_fail1); var result = Subject.GetImportDecisions(_videoFiles, new Series(), false); result.Single().Approved.Should().BeFalse(); } [Test] public void should_return_rejected_if_one_of_specs_fail() { GivenSpecifications(_pass1, _fail1, _pass2, _pass3); var result = Subject.GetImportDecisions(_videoFiles, new Series(), false); result.Single().Approved.Should().BeFalse(); } [Test] public void should_return_pass_if_all_specs_pass() { GivenSpecifications(_pass1, _pass2, _pass3); var result = Subject.GetImportDecisions(_videoFiles, new Series(), false); 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.GetImportDecisions(_videoFiles, new Series(), false); result.Single().Rejections.Should().HaveCount(3); } [Test] public void failed_parse_shouldnt_blowup_the_process() { GivenSpecifications(_pass1); Mocker.GetMock().Setup(c => c.GetEpisodes(It.IsAny(), It.IsAny(), It.IsAny())) .Throws(); _videoFiles = new List { "The.Office.S03E115.DVDRip.XviD-OSiTV", "The.Office.S03E115.DVDRip.XviD-OSiTV", "The.Office.S03E115.DVDRip.XviD-OSiTV" }; Mocker.GetMock() .Setup(c => c.FilterExistingFiles(_videoFiles, It.IsAny())) .Returns(_videoFiles); Subject.GetImportDecisions(_videoFiles, new Series(), false); Mocker.GetMock() .Verify(c => c.GetEpisodes(It.IsAny(), It.IsAny(), It.IsAny()), Times.Exactly(_videoFiles.Count)); ExceptionVerification.ExpectedErrors(3); } } }