using System.Collections.Generic; using System.Linq; using FizzWare.NBuilder; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Core.DecisionEngine; using NzbDrone.Core.Download; using NzbDrone.Core.MediaFiles; using NzbDrone.Core.MediaFiles.MovieImport; using NzbDrone.Core.MediaFiles.MovieImport.Aggregation; using NzbDrone.Core.Movies; using NzbDrone.Core.Parser; using NzbDrone.Core.Parser.Model; using NzbDrone.Core.Profiles; using NzbDrone.Core.Qualities; using NzbDrone.Core.Test.Framework; using NzbDrone.Test.Common; namespace NzbDrone.Core.Test.MediaFiles.MovieImport { [TestFixture] // TODO: Add tests to ensure helpers for augmenters are correctly passed. public class ImportDecisionMakerFixture : CoreTest { private List _videoFiles; private LocalMovie _localMovie; private Movie _movie; private QualityModel _quality; private ParsedMovieInfo _fileInfo; 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(), It.IsAny())).Returns(Decision.Accept()); _pass2.Setup(c => c.IsSatisfiedBy(It.IsAny(), It.IsAny())).Returns(Decision.Accept()); _pass3.Setup(c => c.IsSatisfiedBy(It.IsAny(), It.IsAny())).Returns(Decision.Accept()); _fail1.Setup(c => c.IsSatisfiedBy(It.IsAny(), It.IsAny())).Returns(Decision.Reject("_fail1")); _fail2.Setup(c => c.IsSatisfiedBy(It.IsAny(), It.IsAny())).Returns(Decision.Reject("_fail2")); _fail3.Setup(c => c.IsSatisfiedBy(It.IsAny(), It.IsAny())).Returns(Decision.Reject("_fail3")); _movie = Builder.CreateNew() .With(e => e.Path = @"C:\Test\Movie".AsOsAgnostic()) .With(e => e.Profile = new Profile { Items = Qualities.QualityFixture.GetDefaultQualities() }) .Build(); _quality = new QualityModel(Quality.DVD); _localMovie = new LocalMovie { Movie = _movie, Quality = _quality, Path = @"C:\Test\Unsorted\The.Movie.2018.DVDRip.XviD-OSiTV.avi" }; _fileInfo = new ParsedMovieInfo { MovieTitles = new List { "The Movie" }, Year = 2018, Quality = _quality }; Mocker.GetMock() .Setup(c => c.ParseMinimalPathMovieInfo(It.IsAny())) .Returns(_fileInfo); GivenVideoFiles(new List { @"C:\Test\Unsorted\The.Movie.2018.DVDRip.XviD-OSiTV.avi".AsOsAgnostic() }); } private void GivenSpecifications(params Mock[] mocks) { Mocker.SetConstant(mocks.Select(c => c.Object)); } private void GivenVideoFiles(IEnumerable videoFiles) { _videoFiles = videoFiles.ToList(); Mocker.GetMock() .Setup(c => c.FilterExistingFiles(_videoFiles, It.IsAny())) .Returns(_videoFiles); } private void GivenAugmentationSuccess() { Mocker.GetMock() .Setup(s => s.Augment(It.IsAny(), It.IsAny())) .Callback((localMovie, downloadClientItem) => { localMovie.Movie = _localMovie.Movie; }); } [Test] public void should_call_all_specifications() { var downloadClientItem = Builder.CreateNew().Build(); GivenAugmentationSuccess(); GivenSpecifications(_pass1, _pass2, _pass3, _fail1, _fail2, _fail3); Subject.GetImportDecisions(_videoFiles, _movie, downloadClientItem, null, false, true); _fail1.Verify(c => c.IsSatisfiedBy(It.IsAny(), downloadClientItem), Times.Once()); _fail2.Verify(c => c.IsSatisfiedBy(It.IsAny(), downloadClientItem), Times.Once()); _fail3.Verify(c => c.IsSatisfiedBy(It.IsAny(), downloadClientItem), Times.Once()); _pass1.Verify(c => c.IsSatisfiedBy(It.IsAny(), downloadClientItem), Times.Once()); _pass2.Verify(c => c.IsSatisfiedBy(It.IsAny(), downloadClientItem), Times.Once()); _pass3.Verify(c => c.IsSatisfiedBy(It.IsAny(), downloadClientItem), Times.Once()); } [Test] public void should_return_rejected_if_single_specs_fail() { GivenSpecifications(_fail1); var result = Subject.GetImportDecisions(_videoFiles, _movie); 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, _movie); result.Single().Approved.Should().BeFalse(); } [Test] public void should_return_approved_if_all_specs_pass() { GivenAugmentationSuccess(); GivenSpecifications(_pass1, _pass2, _pass3); var result = Subject.GetImportDecisions(_videoFiles, _movie); result.Single().Approved.Should().BeTrue(); } [Test] public void should_have_same_number_of_rejections_as_specs_that_failed() { GivenAugmentationSuccess(); GivenSpecifications(_pass1, _pass2, _pass3, _fail1, _fail2, _fail3); var result = Subject.GetImportDecisions(_videoFiles, _movie); result.Single().Rejections.Should().HaveCount(3); } [Test] public void should_not_blowup_the_process_due_to_failed_parse() { GivenSpecifications(_pass1); Mocker.GetMock() .Setup(c => c.Augment(It.IsAny(), It.IsAny())) .Throws(); _videoFiles = new List { "The.Movie.2021.DVDRip.XviD-OSiTV", "The.Movie.2021.DVDRip.XviD-OSiTV", "The.Movie.2021.DVDRip.XviD-OSiTV" }; GivenVideoFiles(_videoFiles); Subject.GetImportDecisions(_videoFiles, _movie); Mocker.GetMock() .Verify(c => c.Augment(It.IsAny(), It.IsAny()), Times.Exactly(_videoFiles.Count)); ExceptionVerification.ExpectedErrors(3); } [Test] public void should_call_parsing_service_with_filename_as_simpletitle() { GivenSpecifications(_pass1, _pass2, _pass3); Mocker.GetMock() .Setup(c => c.ParseMinimalPathMovieInfo(It.IsAny())) .Returns(null); var folderInfo = new ParsedMovieInfo { SimpleReleaseTitle = "A Movie Folder 2018", Quality = _quality }; var result = Subject.GetImportDecisions(_videoFiles, _movie, null, folderInfo, true); var fileNames = _videoFiles.Select(System.IO.Path.GetFileName); Mocker.GetMock() .Setup(c => c.Augment(It.IsAny(), It.IsAny())) .Throws(); } [Test] public void should_return_a_decision_when_exception_is_caught() { Mocker.GetMock() .Setup(c => c.Augment(It.IsAny(), It.IsAny())) .Throws(); _videoFiles = new List { "The.Movie.2021.DVDRip.XviD-OSiTV" }; GivenVideoFiles(_videoFiles); Subject.GetImportDecisions(_videoFiles, _movie).Should().HaveCount(1); ExceptionVerification.ExpectedErrors(1); } } }