New: Prevent grabbing season packs if full season hasn't aired yet

Closes #743
2026-04-27 22:56:45 -04:00 · 2015-10-30 14:18:43 +01:00
parent f2c8156c00
commit 2fabe2d198
4 changed files with 117 additions and 0 deletions
@@ -0,0 +1,75 @@
+
+using System;
+using NUnit.Framework;
+using NzbDrone.Core.DecisionEngine.Specifications;
+using NzbDrone.Core.Parser.Model;
+using NzbDrone.Core.Test.Framework;
+using FizzWare.NBuilder;
+using System.Linq;
+using FluentAssertions;
+using NzbDrone.Core.Tv;
+using Moq;
+using System.Collections.Generic;
+
+namespace NzbDrone.Core.Test.DecisionEngineTests
+{
+    [TestFixture]
+    public class FullSeasonSpecificationFixture : CoreTest<FullSeasonSpecification>
+    {
+        private RemoteEpisode _remoteEpisode;
+
+        [SetUp]
+        public void Setup()
+        {
+            var show = Builder<Series>.CreateNew().With(s => s.Id = 1234).Build();
+            _remoteEpisode = new RemoteEpisode
+            {
+                ParsedEpisodeInfo = new ParsedEpisodeInfo
+                {
+                    FullSeason = true
+                },
+                Episodes = Builder<Episode>.CreateListOfSize(3)
+                                           .All()
+                                           .With(e => e.AirDateUtc = DateTime.UtcNow.AddDays(-8))
+                                           .With(s => s.SeriesId = show.Id)
+                                           .BuildList(),
+                Series = show,
+                Release = new ReleaseInfo
+                {
+                    Title = "Series.Title.S01.720p.BluRay.X264-RlsGrp"
+                }
+            };
+
+            Mocker.GetMock<IEpisodeService>().Setup(s => s.EpisodesBetweenDates(It.IsAny<DateTime>(), It.IsAny<DateTime>(), false))
+                                             .Returns(new List<Episode>());
+        }
+
+        [Test]
+        public void should_return_true_if_is_not_a_full_season()
+        {
+            _remoteEpisode.ParsedEpisodeInfo.FullSeason = false;
+            _remoteEpisode.Episodes.Last().AirDateUtc = DateTime.UtcNow.AddDays(+2);
+            Mocker.Resolve<FullSeasonSpecification>().IsSatisfiedBy(_remoteEpisode, null).Accepted.Should().BeTrue();
+        }
+
+        [Test]
+        public void should_return_true_if_all_episodes_have_aired()
+        {
+            Mocker.Resolve<FullSeasonSpecification>().IsSatisfiedBy(_remoteEpisode, null).Accepted.Should().BeTrue();
+        }
+
+        [Test]
+        public void should_return_false_if_one_episode_has_not_aired()
+        {
+            _remoteEpisode.Episodes.Last().AirDateUtc = DateTime.UtcNow.AddDays(+2);
+            Mocker.Resolve<FullSeasonSpecification>().IsSatisfiedBy(_remoteEpisode, null).Accepted.Should().BeFalse();
+        }
+
+        [Test]
+        public void should_return_false_if_an_episode_does_not_have_an_air_date()
+        {
+            _remoteEpisode.Episodes.Last().AirDateUtc = null;
+            Mocker.Resolve<FullSeasonSpecification>().IsSatisfiedBy(_remoteEpisode, null).Accepted.Should().BeFalse();
+        }
+    }
+}