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-22 22:14:44 -04:00 · 2014-04-19 17:09:22 +02:00
parent dcb586b937
commit 2035fe8578
196 changed files with 3961 additions and 2223 deletions
@@ -1,61 +0,0 @@
-using System;
-using System.IO;
-using System.Linq;
-using FizzWare.NBuilder;
-using Moq;
-using NUnit.Framework;
-using NzbDrone.Common;
-using NzbDrone.Core.Download;
-using NzbDrone.Core.Download.Clients.Nzbget;
-using NzbDrone.Core.Parser.Model;
-using NzbDrone.Core.Test.Framework;
-using NzbDrone.Core.Tv;
-
-namespace NzbDrone.Core.Test.Download.DownloadClientTests.NzbgetTests
-{
-    public class DownloadNzbFixture : CoreTest<Nzbget>
-    {
-        private const string _url = "http://www.nzbdrone.com";
-        private const string _title = "30.Rock.S01E01.Pilot.720p.hdtv";
-        private RemoteEpisode _remoteEpisode;
-
-        [SetUp]
-        public void Setup()
-        {
-            _remoteEpisode = new RemoteEpisode();
-            _remoteEpisode.Release = new ReleaseInfo();
-            _remoteEpisode.Release.Title = _title;
-            _remoteEpisode.Release.DownloadUrl = _url;
-
-            _remoteEpisode.Episodes = Builder<Episode>.CreateListOfSize(1)
-                                                      .All()
-                                                      .With(e => e.AirDate = DateTime.Today.ToString(Episode.AIR_DATE_FORMAT))
-                                                      .Build()
-                                                      .ToList();
-
-            Subject.Definition = new DownloadClientDefinition();
-            Subject.Definition.Settings = new NzbgetSettings
-            {
-                Host = "localhost",
-                Port = 6789,
-                Username = "nzbget",
-                Password = "pass",
-                TvCategory = "tv",
-                RecentTvPriority = (int)NzbgetPriority.High
-            };
-        }
-
-        [Test]
-        public void should_add_item_to_queue()
-        {
-            Mocker.GetMock<INzbgetProxy>()
-                  .Setup(s => s.DownloadNzb(It.IsAny<Stream>(), It.IsAny<String>(), It.IsAny<String>(), It.IsAny<Int32>(), It.IsAny<NzbgetSettings>()))
-                  .Returns("id");
-
-            Subject.DownloadNzb(_remoteEpisode);
-
-            Mocker.GetMock<INzbgetProxy>()
-                  .Verify(v => v.DownloadNzb(It.IsAny<Stream>(), It.IsAny<String>(), It.IsAny<String>(), It.IsAny<Int32>(), It.IsAny<NzbgetSettings>()), Times.Once());
-        }
-    }
-}
@@ -0,0 +1,207 @@
+using System;
+using System.IO;
+using System.Linq;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Common;
+using NzbDrone.Core.Download;
+using NzbDrone.Core.Download.Clients.Nzbget;
+using NzbDrone.Core.Parser.Model;
+using NzbDrone.Core.Test.Framework;
+using NzbDrone.Core.Tv;
+using System.Collections.Generic;
+
+namespace NzbDrone.Core.Test.Download.DownloadClientTests.NzbgetTests
+{
+    [TestFixture]
+    public class NzbgetFixture : DownloadClientFixtureBase<Nzbget>
+    {
+        private NzbgetQueueItem _queued;
+        private NzbgetHistoryItem _failed;
+        private NzbgetHistoryItem _completed;
+
+        [SetUp]
+        public void Setup()
+        {
+            Subject.Definition = new DownloadClientDefinition();
+            Subject.Definition.Settings = new NzbgetSettings
+                                          {
+                                              Host = "192.168.5.55",
+                                              Port = 2222,
+                                              Username = "admin",
+                                              Password = "pass",
+                                              TvCategory = "tv",
+                                              RecentTvPriority = (int)NzbgetPriority.High
+                                          };
+
+            _queued = new NzbgetQueueItem
+                {
+                    FileSizeLo = 1000,
+                    RemainingSizeLo = 10,
+                    Category = "tv",
+                    NzbName = "Droned.S01E01.Pilot.1080p.WEB-DL-DRONE",
+                    Parameters = new List<NzbgetParameter> { new NzbgetParameter { Name = "drone", Value = "id" } }
+                };
+
+            _failed = new NzbgetHistoryItem
+                {
+                    FileSizeLo = 1000,
+                    Category = "tv",
+                    Name = "Droned.S01E01.Pilot.1080p.WEB-DL-DRONE",
+                    DestDir = "somedirectory",
+                    Parameters = new List<NzbgetParameter> { new NzbgetParameter { Name = "drone", Value = "id" } },
+                    ParStatus = "Some Error"
+                };
+
+            _completed = new NzbgetHistoryItem
+                {
+                    FileSizeLo = 1000,
+                    Category = "tv",
+                    Name = "Droned.S01E01.Pilot.1080p.WEB-DL-DRONE",
+                    DestDir = "somedirectory",
+                    Parameters = new List<NzbgetParameter> { new NzbgetParameter { Name = "drone", Value = "id" } },
+                    ParStatus = "SUCCESS",
+                    ScriptStatus = "NONE"
+                };
+        }
+
+        protected void WithFailedDownload()
+        {
+            Mocker.GetMock<INzbgetProxy>()
+                .Setup(s => s.DownloadNzb(It.IsAny<Stream>(), It.IsAny<String>(), It.IsAny<String>(), It.IsAny<int>(), It.IsAny<NzbgetSettings>()))
+                .Returns((String)null);
+        }
+
+        protected void WithSuccessfulDownload()
+        {
+            Mocker.GetMock<INzbgetProxy>()
+                .Setup(s => s.DownloadNzb(It.IsAny<Stream>(), It.IsAny<String>(), It.IsAny<String>(), It.IsAny<int>(), It.IsAny<NzbgetSettings>()))
+                .Returns(Guid.NewGuid().ToString().Replace("-", ""));
+        }
+
+        protected virtual void WithQueue(NzbgetQueueItem queue)
+        {
+            var list = new List<NzbgetQueueItem>();
+
+            if (queue != null)
+            {
+                list.Add(queue);
+            }
+
+            Mocker.GetMock<INzbgetProxy>()
+                .Setup(s => s.GetQueue(It.IsAny<NzbgetSettings>()))
+                .Returns(list);
+        }
+
+        protected virtual void WithHistory(NzbgetHistoryItem history)
+        {
+            var list = new List<NzbgetHistoryItem>();
+
+            if (history != null)
+            {
+                list.Add(history);
+            }
+
+            Mocker.GetMock<INzbgetProxy>()
+                .Setup(s => s.GetHistory(It.IsAny<NzbgetSettings>()))
+                .Returns(list);
+        }
+
+        [Test]
+        public void GetItems_should_return_no_items_when_queue_is_empty()
+        {
+            WithQueue(null);
+            WithHistory(null);
+
+            Subject.GetItems().Should().BeEmpty();
+        }
+
+        [Test]
+        public void queued_item_should_have_required_properties()
+        {
+            _queued.ActiveDownloads = 0;
+
+            WithQueue(_queued);
+            WithHistory(null);
+            
+            var result = Subject.GetItems().Single();
+
+            VerifyQueued(result);
+        }
+
+        [Test]
+        public void paused_item_should_have_required_properties()
+        {
+            _queued.PausedSizeLo = _queued.FileSizeLo;
+
+            WithQueue(_queued);
+            WithHistory(null);
+
+            var result = Subject.GetItems().Single();
+
+            VerifyPaused(result);
+        }
+
+        [Test]
+        public void downloading_item_should_have_required_properties()
+        {
+            _queued.ActiveDownloads = 1;
+
+            WithQueue(_queued);
+            WithHistory(null);
+
+            var result = Subject.GetItems().Single();
+
+            VerifyDownloading(result);
+        }
+
+        [Test]
+        public void completed_download_should_have_required_properties()
+        {
+            WithQueue(null);
+            WithHistory(_completed);
+
+            var result = Subject.GetItems().Single();
+
+            VerifyCompleted(result);
+        }
+
+        [Test]
+        public void failed_item_should_have_required_properties()
+        {
+            WithQueue(null);
+            WithHistory(_failed);
+
+            var result = Subject.GetItems().Single();
+
+            VerifyFailed(result);
+        }
+
+        [Test]
+        public void Download_should_return_unique_id()
+        {
+            WithSuccessfulDownload();
+
+            var remoteEpisode = CreateRemoteEpisode();
+
+            var id = Subject.Download(remoteEpisode);
+
+            id.Should().NotBeNullOrEmpty();
+        }
+
+        [Test]
+        public void GetItems_should_ignore_downloads_from_other_categories()
+        {
+            _completed.Category = "mycat";
+
+            WithQueue(null);
+            WithHistory(_completed);
+
+            var items = Subject.GetItems();
+
+            items.Should().BeEmpty();
+        }
+    }
+}
@@ -1,84 +0,0 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using FizzWare.NBuilder;
-using FluentAssertions;
-using Moq;
-using NUnit.Framework;
-using NzbDrone.Core.Download;
-using NzbDrone.Core.Download.Clients.Nzbget;
-using NzbDrone.Core.Parser;
-using NzbDrone.Core.Parser.Model;
-using NzbDrone.Core.Test.Framework;
-using NzbDrone.Core.Tv;
-
-namespace NzbDrone.Core.Test.Download.DownloadClientTests.NzbgetTests
-{
-    public class QueueFixture : CoreTest<Nzbget>
-    {
-        private List<NzbgetQueueItem> _queue;
-
-        [SetUp]
-        public void Setup()
-        {
-            _queue = Builder<NzbgetQueueItem>.CreateListOfSize(5)
-                                             .All()
-                                             .With(q => q.NzbName = "30.Rock.S01E01.Pilot.720p.hdtv.nzb")
-                                             .With(q => q.Parameters = new List<NzbgetParameter>
-                                                                       {
-                                                                           new NzbgetParameter { Name = "drone", Value = "id" }
-                                                                       })
-                                             .Build()
-                                             .ToList();
-
-            Subject.Definition = new DownloadClientDefinition();
-            Subject.Definition.Settings = new NzbgetSettings
-            {
-                Host = "localhost",
-                Port = 6789,
-                Username = "nzbget",
-                Password = "pass",
-                TvCategory = "tv",
-                RecentTvPriority = (int)NzbgetPriority.High
-            };
-        }
-
-        private void WithFullQueue()
-        {
-            Mocker.GetMock<INzbgetProxy>()
-                  .Setup(s => s.GetQueue(It.IsAny<NzbgetSettings>()))
-                  .Returns(_queue);
-        }
-
-        private void WithEmptyQueue()
-        {
-            Mocker.GetMock<INzbgetProxy>()
-                  .Setup(s => s.GetQueue(It.IsAny<NzbgetSettings>()))
-                  .Returns(new List<NzbgetQueueItem>());
-        }
-
-        [Test]
-        public void should_return_no_items_when_queue_is_empty()
-        {
-            WithEmptyQueue();
-
-            Subject.GetQueue()
-                   .Should()
-                   .BeEmpty();
-        }
-
-        [Test]
-        public void should_return_item_when_queue_has_item()
-        {
-            WithFullQueue();
-
-            Mocker.GetMock<IParsingService>()
-                  .Setup(s => s.Map(It.IsAny<ParsedEpisodeInfo>(), 0, null))
-                  .Returns(new RemoteEpisode {Series = new Series()});
-
-            Subject.GetQueue()
-                   .Should()
-                   .HaveCount(_queue.Count);
-        }
-    }
-}