New: Rewrite of download decision engine.

2026-04-20 22:14:34 -04:00 · 2012-02-06 21:08:07 -08:00
parent a168bdfa00
commit 5717b7f596
60 changed files with 2013 additions and 1745 deletions
@@ -0,0 +1,118 @@
+// ReSharper disable RedundantUsingDirective
+
+using System.Collections.Generic;
+using System.Linq;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using NUnit.Framework;
+using NzbDrone.Core.Model;
+using NzbDrone.Core.Providers;
+using NzbDrone.Core.Providers.DecisionEngine;
+using NzbDrone.Core.Repository;
+using NzbDrone.Core.Repository.Quality;
+using NzbDrone.Core.Test.Framework;
+
+namespace NzbDrone.Core.Test.ProviderTests.DecisionEngineTests
+{
+    [TestFixture]
+    // ReSharper disable InconsistentNaming
+    public class UpgradeDiskSpecificationFixtrue : CoreTest
+    {
+        private UpgradeDiskSpecification _upgradeDisk;
+
+        private EpisodeParseResult parseResultMulti;
+        private EpisodeParseResult parseResultSingle;
+        private EpisodeFile firstFile;
+        private EpisodeFile secondFile;
+
+        [SetUp]
+        public void Setup()
+        {
+            Mocker.Resolve<QualityUpgradeSpecification>();
+            _upgradeDisk = Mocker.Resolve<UpgradeDiskSpecification>();
+
+            var fakeSeries = Builder<Series>.CreateNew()
+                         .With(c => c.QualityProfile = new QualityProfile { Cutoff = QualityTypes.Bluray1080p })
+                         .Build();
+
+            parseResultMulti = new EpisodeParseResult
+            {
+                Series = fakeSeries,
+                Quality = new Quality(QualityTypes.DVD, true),
+                EpisodeNumbers = new List<int> { 3, 4 },
+                SeasonNumber = 12,
+            };
+
+            parseResultSingle = new EpisodeParseResult
+            {
+                Series = fakeSeries,
+                Quality = new Quality(QualityTypes.DVD, true),
+                EpisodeNumbers = new List<int> { 3 },
+                SeasonNumber = 12,
+            };
+
+
+            firstFile = new EpisodeFile { Quality = QualityTypes.Bluray1080p, Proper = true };
+            secondFile = new EpisodeFile { Quality = QualityTypes.Bluray1080p, Proper = true };
+
+            var singleEpisodeList = new List<Episode> { new Episode { EpisodeFile = firstFile }, new Episode { EpisodeFile = null } };
+            var doubleEpisodeList = new List<Episode> { new Episode { EpisodeFile = firstFile }, new Episode { EpisodeFile = secondFile }, new Episode { EpisodeFile = null } };
+
+            Mocker.GetMock<EpisodeProvider>().Setup(c => c.GetEpisodesByParseResult(parseResultSingle)).Returns(singleEpisodeList);
+            Mocker.GetMock<EpisodeProvider>().Setup(c => c.GetEpisodesByParseResult(parseResultMulti)).Returns(doubleEpisodeList);
+        }
+
+        private void WithFirstFileUpgradable()
+        {
+            firstFile.Quality = QualityTypes.SDTV;
+        }
+
+        private void WithSecondFileUpgradable()
+        {
+            secondFile.Quality = QualityTypes.SDTV;
+        }
+
+        [Test]
+        public void should_return_false_if_single_episode_doesnt_exist_on_disk()
+        {
+            Mocker.GetMock<EpisodeProvider>().Setup(c => c.GetEpisodesByParseResult(parseResultSingle)).Returns(new List<Episode>());
+
+            _upgradeDisk.IsSatisfiedBy(parseResultSingle).Should().BeTrue();
+        }
+
+        [Test]
+        public void should_be_upgradable_if_only_episode_is_upgradable()
+        {
+            WithFirstFileUpgradable();
+            _upgradeDisk.IsSatisfiedBy(parseResultSingle).Should().BeTrue();
+        }
+
+        [Test]
+        public void should_be_upgradable_if_both_episodes_are_upgradable()
+        {
+            WithFirstFileUpgradable();
+            WithSecondFileUpgradable();
+            _upgradeDisk.IsSatisfiedBy(parseResultMulti).Should().BeTrue();
+        }
+
+        [Test]
+        public void should_be_not_upgradable_if_both_episodes_are_not_upgradable()
+        {
+            _upgradeDisk.IsSatisfiedBy(parseResultMulti).Should().BeFalse();
+        }
+
+        [Test]
+        public void should_be_not_upgradable_if_only_first_episodes_is_upgradable()
+        {
+            WithFirstFileUpgradable();
+            _upgradeDisk.IsSatisfiedBy(parseResultMulti).Should().BeFalse();
+        }
+
+        [Test]
+        public void should_be_not_upgradable_if_only_second_episodes_is_upgradable()
+        {
+            WithSecondFileUpgradable();
+            _upgradeDisk.IsSatisfiedBy(parseResultMulti).Should().BeFalse();
+        }
+    }
+}