Import process improvements

New: Post processing scripts can pass "Path" via API to scan a specific folder directly Fixed: Do not delete folder from drone factory when non-sample video files exist
2026-04-23 22:25:09 -04:00 · 2014-04-03 17:08:51 -07:00
parent 037127163f
commit e5263f143d
9 changed files with 385 additions and 189 deletions
@@ -0,0 +1,151 @@
+using System;
+using System.Linq;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Core.MediaFiles.EpisodeImport;
+using NzbDrone.Core.MediaFiles.MediaInfo;
+using NzbDrone.Core.Parser.Model;
+using NzbDrone.Core.Qualities;
+using NzbDrone.Core.Test.Framework;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.Test.MediaFiles.EpisodeImport
+{
+    [TestFixture]
+    public class SampleServiceFixture : CoreTest<SampleService>
+    {
+        private Series _series;
+        private LocalEpisode _localEpisode;
+
+        [SetUp]
+        public void Setup()
+        {
+            _series = Builder<Series>.CreateNew()
+                                     .With(s => s.SeriesType = SeriesTypes.Standard)
+                                     .Build();
+
+            var episodes = Builder<Episode>.CreateListOfSize(1)
+                                           .All()
+                                           .With(e => e.SeasonNumber = 1)
+                                           .Build()
+                                           .ToList();
+
+            _localEpisode = new LocalEpisode
+                                {
+                                    Path = @"C:\Test\30 Rock\30.rock.s01e01.avi",
+                                    Episodes = episodes,
+                                    Series = _series,
+                                    Quality = new QualityModel(Quality.HDTV720p)
+                                };
+        }
+
+        private void GivenFileSize(long size)
+        {
+            _localEpisode.Size = size;
+        }
+
+        private void GivenRuntime(int seconds)
+        {
+            Mocker.GetMock<IVideoFileInfoReader>()
+                  .Setup(s => s.GetRunTime(It.IsAny<String>()))
+                  .Returns(new TimeSpan(0, 0, seconds));
+        }
+
+        [Test]
+        public void should_return_true_if_series_is_daily()
+        {
+            _series.SeriesType = SeriesTypes.Daily;
+            ShouldBeFalse();
+        }
+
+        [Test]
+        public void should_return_true_if_season_zero()
+        {
+            _localEpisode.Episodes[0].SeasonNumber = 0;
+            ShouldBeFalse();
+        }
+
+        [Test]
+        public void should_return_true_for_flv()
+        {
+            _localEpisode.Path = @"C:\Test\some.show.s01e01.flv";
+
+            ShouldBeFalse();
+
+            Mocker.GetMock<IVideoFileInfoReader>().Verify(c => c.GetRunTime(It.IsAny<string>()), Times.Never());
+        }
+
+        [Test]
+        public void should_use_runtime()
+        {
+            GivenRuntime(120);
+            GivenFileSize(1000.Megabytes());
+
+            Subject.IsSample(_localEpisode.Series,
+                             _localEpisode.Quality,
+                             _localEpisode.Path,
+                             _localEpisode.Size,
+                             _localEpisode.SeasonNumber);
+
+            Mocker.GetMock<IVideoFileInfoReader>().Verify(v => v.GetRunTime(It.IsAny<String>()), Times.Once());
+        }
+
+        [Test]
+        public void should_return_false_if_runtime_is_less_than_minimum()
+        {
+            GivenRuntime(60);
+
+            ShouldBeTrue();
+        }
+
+        [Test]
+        public void should_return_true_if_runtime_greater_than_than_minimum()
+        {
+            GivenRuntime(120);
+
+            ShouldBeFalse();
+        }
+
+        [Test]
+        public void should_fall_back_to_file_size_if_mediainfo_dll_not_found_acceptable_size()
+        {
+            Mocker.GetMock<IVideoFileInfoReader>()
+                  .Setup(s => s.GetRunTime(It.IsAny<String>()))
+                  .Throws<DllNotFoundException>();
+
+            GivenFileSize(1000.Megabytes());
+            ShouldBeFalse();
+        }
+
+        [Test]
+        public void should_fall_back_to_file_size_if_mediainfo_dll_not_found_undersize()
+        {
+            Mocker.GetMock<IVideoFileInfoReader>()
+                  .Setup(s => s.GetRunTime(It.IsAny<String>()))
+                  .Throws<DllNotFoundException>();
+
+            GivenFileSize(1.Megabytes());
+            ShouldBeTrue();
+        }
+
+        private void ShouldBeTrue()
+        {
+            Subject.IsSample(_localEpisode.Series,
+                                         _localEpisode.Quality,
+                                         _localEpisode.Path,
+                                         _localEpisode.Size,
+                                         _localEpisode.SeasonNumber).Should().BeTrue();
+        }
+
+        private void ShouldBeFalse()
+        {
+            Subject.IsSample(_localEpisode.Series,
+                             _localEpisode.Quality,
+                             _localEpisode.Path,
+                             _localEpisode.Size,
+                             _localEpisode.SeasonNumber).Should().BeFalse();
+        }
+    }
+}