New: Rewrite of download decision engine.

2026-04-25 22:36:59 -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,69 @@
+// 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.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 QualityAllowedByProfileSpecificationFixtrue : CoreTest
+    {
+        private QualityAllowedByProfileSpecification _qualityAllowedByProfile;
+
+
+        private EpisodeParseResult parseResult;
+
+
+        [SetUp]
+        public void Setup()
+        {
+            _qualityAllowedByProfile = Mocker.Resolve<QualityAllowedByProfileSpecification>();
+
+            var fakeSeries = Builder<Series>.CreateNew()
+                         .With(c => c.QualityProfile = new QualityProfile { Cutoff = QualityTypes.Bluray1080p })
+                         .Build();
+
+            parseResult = new EpisodeParseResult
+            {
+                Series = fakeSeries,
+                Quality = new Quality(QualityTypes.DVD, true),
+                EpisodeNumbers = new List<int> { 3 },
+                SeasonNumber = 12,
+            };
+        }
+
+
+
+        [TestCase(QualityTypes.DVD)]
+        [TestCase(QualityTypes.HDTV)]
+        [TestCase(QualityTypes.Bluray1080p)]
+        public void should_allow_if_quality_is_defined_in_profile(QualityTypes qualityType)
+        {
+            parseResult.Quality.QualityType = qualityType;
+            parseResult.Series.QualityProfile.Allowed = new List<QualityTypes> { QualityTypes.DVD, QualityTypes.HDTV, QualityTypes.Bluray1080p };
+
+            _qualityAllowedByProfile.IsSatisfiedBy(parseResult).Should().BeTrue();
+        }
+
+        [TestCase(QualityTypes.SDTV)]
+        [TestCase(QualityTypes.WEBDL)]
+        [TestCase(QualityTypes.Bluray720p)]
+        public void should_not_allow_if_quality_is_not_defined_in_profile(QualityTypes qualityType)
+        {
+            parseResult.Quality.QualityType = qualityType;
+            parseResult.Series.QualityProfile.Allowed = new List<QualityTypes> { QualityTypes.DVD, QualityTypes.HDTV, QualityTypes.Bluray1080p };
+
+            _qualityAllowedByProfile.IsSatisfiedBy(parseResult).Should().BeFalse();
+        }
+
+    }
+}