using System; using System.Collections.Generic; using FizzWare.NBuilder; using Moq; using NUnit.Framework; using NzbDrone.Common.Extensions; using NzbDrone.Core.DecisionEngine; using NzbDrone.Core.Download; using NzbDrone.Core.Download.Pending; using NzbDrone.Core.Indexers; 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; namespace NzbDrone.Core.Test.Download.Pending.PendingReleaseServiceTests { [TestFixture] public class RemoveRejectedFixture : CoreTest { private DownloadDecision _temporarilyRejected; private Movie _movie; private Profile _profile; private ReleaseInfo _release; private ParsedMovieInfo _parsedMovieInfo; private RemoteMovie _remoteMovie; [SetUp] public void Setup() { _movie = Builder.CreateNew() .Build(); _profile = new Profile { Name = "Test", Cutoff = Quality.HDTV720p.Id, Items = new List { new ProfileQualityItem { Allowed = true, Quality = Quality.HDTV720p }, new ProfileQualityItem { Allowed = true, Quality = Quality.WEBDL720p }, new ProfileQualityItem { Allowed = true, Quality = Quality.Bluray720p } }, }; _movie.Profile = _profile; _release = Builder.CreateNew().Build(); _parsedMovieInfo = Builder.CreateNew().Build(); _parsedMovieInfo.Quality = new QualityModel(Quality.HDTV720p); _remoteMovie = new RemoteMovie(); // _remoteEpisode.Episodes = new List{ _episode }; _remoteMovie.Movie = _movie; _remoteMovie.ParsedMovieInfo = _parsedMovieInfo; _remoteMovie.Release = _release; _temporarilyRejected = new DownloadDecision(_remoteMovie, new Rejection("Temp Rejected", RejectionType.Temporary)); Mocker.GetMock() .Setup(s => s.All()) .Returns(new List()); Mocker.GetMock() .Setup(s => s.GetMovie(It.IsAny())) .Returns(_movie); Mocker.GetMock() .Setup(s => s.GetMovies(It.IsAny>())) .Returns(new List { _movie }); Mocker.GetMock() .Setup(s => s.GetMovie(It.IsAny())) .Returns(_movie); Mocker.GetMock() .Setup(s => s.PrioritizeDecisionsForMovies(It.IsAny>())) .Returns((List d) => d); } private void GivenHeldRelease(string title, string indexer, DateTime publishDate) { var release = _release.JsonClone(); release.Indexer = indexer; release.PublishDate = publishDate; var heldReleases = Builder.CreateListOfSize(1) .All() .With(h => h.MovieId = _movie.Id) .With(h => h.Title = title) .With(h => h.Release = release) .Build(); Mocker.GetMock() .Setup(s => s.All()) .Returns(heldReleases); } [Test] public void should_remove_if_it_is_the_same_release_from_the_same_indexer() { GivenHeldRelease(_release.Title, _release.Indexer, _release.PublishDate); Subject.Handle(new RssSyncCompleteEvent(new ProcessedDecisions(new List(), new List(), new List { _temporarilyRejected }))); VerifyDelete(); } [Test] public void should_not_remove_if_title_is_different() { GivenHeldRelease(_release.Title + "-RP", _release.Indexer, _release.PublishDate); Subject.Handle(new RssSyncCompleteEvent(new ProcessedDecisions(new List(), new List(), new List { _temporarilyRejected }))); VerifyNoDelete(); } [Test] public void should_not_remove_if_indexer_is_different() { GivenHeldRelease(_release.Title, "AnotherIndexer", _release.PublishDate); Subject.Handle(new RssSyncCompleteEvent(new ProcessedDecisions(new List(), new List(), new List { _temporarilyRejected }))); VerifyNoDelete(); } [Test] public void should_not_remove_if_publish_date_is_different() { GivenHeldRelease(_release.Title, _release.Indexer, _release.PublishDate.AddHours(1)); Subject.Handle(new RssSyncCompleteEvent(new ProcessedDecisions(new List(), new List(), new List { _temporarilyRejected }))); VerifyNoDelete(); } private void VerifyDelete() { Mocker.GetMock() .Verify(v => v.Delete(It.IsAny()), Times.Once()); } private void VerifyNoDelete() { Mocker.GetMock() .Verify(v => v.Delete(It.IsAny()), Times.Never()); } } }