// ReSharper disable RedundantUsingDirective using System.Linq; using FizzWare.NBuilder; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Core.Tv; using NzbDrone.Core.Providers; using NzbDrone.Core.DecisionEngine; using NzbDrone.Core.Repository; using NzbDrone.Core.Repository.Quality; using NzbDrone.Core.Test.Framework; namespace NzbDrone.Core.Test.DecisionEngineTests { [TestFixture] // ReSharper disable InconsistentNaming public class UpgradePossibleSpecificationFixture : CoreTest { private void WithWebdlCutoff() { var profile = new QualityProfile { Cutoff = QualityTypes.WEBDL720p }; Mocker.GetMock().Setup(s => s.Get(It.IsAny())).Returns(profile); } private Series _series; private EpisodeFile _episodeFile; private Episode _episode; [SetUp] public void SetUp() { _series = Builder.CreateNew() .Build(); _episodeFile = Builder.CreateNew() .With(f => f.Quality = QualityTypes.SDTV) .Build(); _episode = Builder.CreateNew() .With(e => e.SeriesId = _series.Id) .With(e => e.Series = _series) .With(e => e.EpisodeFile = _episodeFile) .Build(); } [Test] public void IsUpgradePossible_should_return_true_if_no_episode_file_exists() { var episode = Builder.CreateNew() .With(e => e.EpisodeFile = null) .Build(); //Act bool result = Mocker.Resolve().IsSatisfiedBy(episode); //Assert result.Should().BeTrue(); } [Test] public void IsUpgradePossible_should_return_true_if_current_episode_is_less_than_cutoff() { WithWebdlCutoff(); //Act bool result = Mocker.Resolve().IsSatisfiedBy(_episode); //Assert result.Should().BeTrue(); } [Test] public void IsUpgradePossible_should_return_false_if_current_episode_is_equal_to_cutoff() { WithWebdlCutoff(); _episodeFile.Quality = QualityTypes.WEBDL720p; //Act bool result = Mocker.Resolve().IsSatisfiedBy(_episode); //Assert result.Should().BeFalse(); } [Test] public void IsUpgradePossible_should_return_false_if_current_episode_is_greater_than_cutoff() { WithWebdlCutoff(); _episodeFile.Quality = QualityTypes.Bluray720p; //Act bool result = Mocker.Resolve().IsSatisfiedBy(_episode); //Assert result.Should().BeFalse(); } } }