New: Drone now uses the Download Client API to determine if a download is ready for import. (User configuration is required to replace the drone factory with this feature)

2026-04-20 21:54:25 -04:00 · 2014-04-19 17:09:22 +02:00
parent dcb586b937
commit 2035fe8578
196 changed files with 3961 additions and 2223 deletions
@@ -0,0 +1,95 @@
+using System;
+using System.Linq;
+using System.Collections.Generic;
+using FluentAssertions;
+using Moq;
+using FizzWare.NBuilder;
+using NUnit.Framework;
+using NzbDrone.Test.Common;
+using NzbDrone.Common.Disk;
+using NzbDrone.Core.Configuration;
+using NzbDrone.Core.HealthCheck.Checks;
+using NzbDrone.Core.Test.Framework;
+using NzbDrone.Core.Download;
+
+namespace NzbDrone.Core.Test.HealthCheck.Checks
+{
+    [TestFixture]
+    public class ImportMechanismCheckFixture : CoreTest<ImportMechanismCheck>
+    {
+        private const string DRONE_FACTORY_FOLDER = @"C:\Test\Unsorted";
+
+        private IList<TrackedDownload> _completed;
+
+        private void GivenCompletedDownloadHandling(bool? enabled = null)
+        {
+            if (enabled.HasValue)
+            {
+                Mocker.GetMock<IConfigService>()
+                      .Setup(s => s.IsDefined("EnableCompletedDownloadHandling"))
+                      .Returns(true);
+
+                Mocker.GetMock<IConfigService>()
+                      .SetupGet(s => s.EnableCompletedDownloadHandling)
+                      .Returns(enabled.Value);
+            }
+
+            _completed = Builder<TrackedDownload>.CreateListOfSize(1)
+                .All()
+                .With(v => v.State == TrackedDownloadState.Downloading)
+                .With(v => v.DownloadItem = new DownloadClientItem())
+                .With(v => v.DownloadItem.Status = DownloadItemStatus.Completed)
+                .With(v => v.DownloadItem.OutputPath = @"C:\Test\DropFolder\myfile.mkv".AsOsAgnostic())
+                .Build();
+
+            Mocker.GetMock<IDownloadTrackingService>()
+                .Setup(v => v.GetCompletedDownloads())
+                .Returns(_completed.ToList());
+        }
+
+        private void GivenDroneFactoryFolder(bool exists = false)
+        {
+            Mocker.GetMock<IConfigService>()
+                  .SetupGet(s => s.DownloadedEpisodesFolder)
+                  .Returns(DRONE_FACTORY_FOLDER.AsOsAgnostic());
+
+            Mocker.GetMock<IDiskProvider>()
+                  .Setup(s => s.FolderExists(DRONE_FACTORY_FOLDER.AsOsAgnostic()))
+                  .Returns(exists);
+        }
+
+        [Test]
+        public void should_return_warning_when_completed_download_handling_not_configured()
+        {
+            Subject.Check().ShouldBeWarning();
+        }
+
+        [Test]
+        public void should_return_warning_when_both_completeddownloadhandling_and_dronefactory_are_not_configured()
+        {
+            GivenCompletedDownloadHandling(false);
+            
+            Subject.Check().ShouldBeWarning();
+        }
+
+        [Test]
+        public void should_return_warning_when_downloadclient_drops_in_dronefactory_folder()
+        {
+            GivenCompletedDownloadHandling(true);
+            GivenDroneFactoryFolder(true);
+
+            _completed.First().DownloadItem.OutputPath = (DRONE_FACTORY_FOLDER + @"\myfile.mkv").AsOsAgnostic();
+
+            Subject.Check().ShouldBeWarning();
+        }
+        
+        [Test]
+        public void should_return_ok_when_no_issues_found()
+        {
+            GivenCompletedDownloadHandling(true);
+            GivenDroneFactoryFolder(true);
+
+            Subject.Check().ShouldBeOk();
+        }
+    }
+}