|
|
|
using System.Collections.Generic;
|
|
|
|
using FizzWare.NBuilder;
|
|
|
|
using FluentAssertions;
|
|
|
|
using Moq;
|
|
|
|
using NUnit.Framework;
|
|
|
|
using NzbDrone.Core.Download;
|
|
|
|
using NzbDrone.Core.History;
|
|
|
|
using NzbDrone.Core.MediaFiles.MovieImport.Specifications;
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
using NzbDrone.Core.Qualities;
|
|
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.MediaFiles.MovieImport.Specifications
|
|
|
|
{
|
|
|
|
[TestFixture]
|
|
|
|
public class GrabbedReleaseQualityFixture : CoreTest<GrabbedReleaseQualitySpecification>
|
|
|
|
{
|
|
|
|
private LocalMovie _localMovie;
|
|
|
|
private DownloadClientItem _downloadClientItem;
|
|
|
|
|
|
|
|
[SetUp]
|
|
|
|
public void Setup()
|
|
|
|
{
|
|
|
|
_localMovie = Builder<LocalMovie>.CreateNew()
|
|
|
|
.With(l => l.Quality = new QualityModel(Quality.Bluray720p))
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
_downloadClientItem = Builder<DownloadClientItem>.CreateNew()
|
|
|
|
.Build();
|
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenHistory(List<MovieHistory> history)
|
|
|
|
{
|
|
|
|
Mocker.GetMock<IHistoryService>()
|
|
|
|
.Setup(s => s.FindByDownloadId(It.IsAny<string>()))
|
|
|
|
.Returns(history);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_be_accepted_when_downloadClientItem_is_null()
|
|
|
|
{
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_be_accepted_if_no_history_for_downloadId()
|
|
|
|
{
|
|
|
|
GivenHistory(new List<MovieHistory>());
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, _downloadClientItem).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_be_accepted_if_no_grabbed_history_for_downloadId()
|
|
|
|
{
|
|
|
|
var history = Builder<MovieHistory>.CreateListOfSize(1)
|
|
|
|
.All()
|
|
|
|
.With(h => h.EventType = MovieHistoryEventType.Unknown)
|
|
|
|
.BuildList();
|
|
|
|
|
|
|
|
GivenHistory(history);
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, _downloadClientItem).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_be_accepted_if_grabbed_history_quality_is_unknown()
|
|
|
|
{
|
|
|
|
var history = Builder<MovieHistory>.CreateListOfSize(1)
|
|
|
|
.All()
|
|
|
|
.With(h => h.EventType = MovieHistoryEventType.Grabbed)
|
|
|
|
.With(h => h.Quality = new QualityModel(Quality.Unknown))
|
|
|
|
.BuildList();
|
|
|
|
|
|
|
|
GivenHistory(history);
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, _downloadClientItem).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_be_accepted_if_grabbed_history_quality_matches()
|
|
|
|
{
|
|
|
|
var history = Builder<MovieHistory>.CreateListOfSize(1)
|
|
|
|
.All()
|
|
|
|
.With(h => h.EventType = MovieHistoryEventType.Grabbed)
|
|
|
|
.With(h => h.Quality = _localMovie.Quality)
|
|
|
|
.BuildList();
|
|
|
|
|
|
|
|
GivenHistory(history);
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, _downloadClientItem).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_be_accepted_if_grabbed_history_quality_does_not_match_log_only()
|
|
|
|
{
|
|
|
|
var history = Builder<MovieHistory>.CreateListOfSize(1)
|
|
|
|
.All()
|
|
|
|
.With(h => h.EventType = MovieHistoryEventType.Grabbed)
|
|
|
|
.With(h => h.Quality = new QualityModel(Quality.HDTV720p))
|
|
|
|
.BuildList();
|
|
|
|
|
|
|
|
GivenHistory(history);
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, _downloadClientItem).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|