You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
143 lines
5.1 KiB
143 lines
5.1 KiB
using System.Collections.Generic;
|
|
using FizzWare.NBuilder;
|
|
using Marr.Data;
|
|
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.Parser;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Profiles;
|
|
using NzbDrone.Core.Qualities;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Core.Tv;
|
|
|
|
namespace NzbDrone.Core.Test.Download.Pending.PendingReleaseServiceTests
|
|
{
|
|
[TestFixture]
|
|
public class RemoveGrabbedFixture : CoreTest<PendingReleaseService>
|
|
{
|
|
private DownloadDecision _temporarilyRejected;
|
|
private Series _series;
|
|
private Episode _episode;
|
|
private Profile _profile;
|
|
private ReleaseInfo _release;
|
|
private ParsedEpisodeInfo _parsedEpisodeInfo;
|
|
private RemoteEpisode _remoteEpisode;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
_series = Builder<Series>.CreateNew()
|
|
.Build();
|
|
|
|
_episode = Builder<Episode>.CreateNew()
|
|
.Build();
|
|
|
|
_profile = new Profile
|
|
{
|
|
Name = "Test",
|
|
Cutoff = Quality.MP3_256,
|
|
Items = new List<ProfileQualityItem>
|
|
{
|
|
new ProfileQualityItem { Allowed = true, Quality = Quality.MP3_256 },
|
|
new ProfileQualityItem { Allowed = true, Quality = Quality.MP3_320 },
|
|
new ProfileQualityItem { Allowed = true, Quality = Quality.MP3_512 }
|
|
},
|
|
};
|
|
|
|
_series.Profile = new LazyLoaded<Profile>(_profile);
|
|
|
|
_release = Builder<ReleaseInfo>.CreateNew().Build();
|
|
|
|
_parsedEpisodeInfo = Builder<ParsedEpisodeInfo>.CreateNew().Build();
|
|
_parsedEpisodeInfo.Quality = new QualityModel(Quality.MP3_256);
|
|
|
|
_remoteEpisode = new RemoteEpisode();
|
|
_remoteEpisode.Episodes = new List<Episode>{ _episode };
|
|
_remoteEpisode.Series = _series;
|
|
_remoteEpisode.ParsedEpisodeInfo = _parsedEpisodeInfo;
|
|
_remoteEpisode.Release = _release;
|
|
|
|
_temporarilyRejected = new DownloadDecision(_remoteEpisode, new Rejection("Temp Rejected", RejectionType.Temporary));
|
|
|
|
Mocker.GetMock<IPendingReleaseRepository>()
|
|
.Setup(s => s.All())
|
|
.Returns(new List<PendingRelease>());
|
|
|
|
Mocker.GetMock<ISeriesService>()
|
|
.Setup(s => s.GetSeries(It.IsAny<int>()))
|
|
.Returns(_series);
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
.Setup(s => s.GetEpisodes(It.IsAny<ParsedEpisodeInfo>(), _series, true, null))
|
|
.Returns(new List<Episode> {_episode});
|
|
|
|
Mocker.GetMock<IPrioritizeDownloadDecision>()
|
|
.Setup(s => s.PrioritizeDecisions(It.IsAny<List<DownloadDecision>>()))
|
|
.Returns((List<DownloadDecision> d) => d);
|
|
}
|
|
|
|
private void GivenHeldRelease(QualityModel quality)
|
|
{
|
|
var parsedEpisodeInfo = _parsedEpisodeInfo.JsonClone();
|
|
parsedEpisodeInfo.Quality = quality;
|
|
|
|
var heldReleases = Builder<PendingRelease>.CreateListOfSize(1)
|
|
.All()
|
|
.With(h => h.SeriesId = _series.Id)
|
|
.With(h => h.Release = _release.JsonClone())
|
|
.With(h => h.ParsedEpisodeInfo = parsedEpisodeInfo)
|
|
.Build();
|
|
|
|
Mocker.GetMock<IPendingReleaseRepository>()
|
|
.Setup(s => s.All())
|
|
.Returns(heldReleases);
|
|
}
|
|
|
|
[Test]
|
|
public void should_delete_if_the_grabbed_quality_is_the_same()
|
|
{
|
|
GivenHeldRelease(_parsedEpisodeInfo.Quality);
|
|
|
|
Subject.Handle(new EpisodeGrabbedEvent(_remoteEpisode));
|
|
|
|
VerifyDelete();
|
|
}
|
|
|
|
[Test]
|
|
public void should_delete_if_the_grabbed_quality_is_the_higher()
|
|
{
|
|
GivenHeldRelease(new QualityModel(Quality.MP3_192));
|
|
|
|
Subject.Handle(new EpisodeGrabbedEvent(_remoteEpisode));
|
|
|
|
VerifyDelete();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_delete_if_the_grabbed_quality_is_the_lower()
|
|
{
|
|
GivenHeldRelease(new QualityModel(Quality.MP3_512));
|
|
|
|
Subject.Handle(new EpisodeGrabbedEvent(_remoteEpisode));
|
|
|
|
VerifyNoDelete();
|
|
}
|
|
|
|
private void VerifyDelete()
|
|
{
|
|
Mocker.GetMock<IPendingReleaseRepository>()
|
|
.Verify(v => v.Delete(It.IsAny<PendingRelease>()), Times.Once());
|
|
}
|
|
|
|
private void VerifyNoDelete()
|
|
{
|
|
Mocker.GetMock<IPendingReleaseRepository>()
|
|
.Verify(v => v.Delete(It.IsAny<PendingRelease>()), Times.Never());
|
|
}
|
|
}
|
|
}
|