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-25 22:37:27 -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,118 @@
+using System.IO;
+using System.Net;
+using System.Linq;
+using Moq;
+using NUnit.Framework;
+using FluentAssertions;
+using NzbDrone.Test.Common;
+using NzbDrone.Core.Test.Framework;
+using NzbDrone.Common;
+using NzbDrone.Common.Disk;
+using NzbDrone.Common.Http;
+using NzbDrone.Core.Parser.Model;
+using NzbDrone.Core.Download;
+using NzbDrone.Core.Download.Clients;
+using NzbDrone.Core.Download.Clients.UsenetBlackhole;
+
+namespace NzbDrone.Core.Test.Download.DownloadClientTests.Blackhole
+{
+
+    [TestFixture]
+    public class UsenetBlackholeFixture : DownloadClientFixtureBase<UsenetBlackhole>
+    {
+        protected string _completedDownloadFolder;
+        protected string _blackholeFolder;
+        protected string _filePath;
+
+        [SetUp]
+        public void Setup()
+        {
+            _completedDownloadFolder = @"c:\blackhole\completed".AsOsAgnostic();
+            _blackholeFolder = @"c:\blackhole\nzb".AsOsAgnostic();
+            _filePath = (@"c:\blackhole\nzb\" + _title + ".nzb").AsOsAgnostic();
+
+            Subject.Definition = new DownloadClientDefinition();
+            Subject.Definition.Settings = new UsenetBlackholeSettings
+            {
+                NzbFolder = _blackholeFolder,
+                WatchFolder = _completedDownloadFolder
+            };
+        }
+
+        protected void WithSuccessfulDownload()
+        {
+
+        }
+
+        protected void WithFailedDownload()
+        {
+            Mocker.GetMock<IHttpProvider>()
+                .Setup(c => c.DownloadFile(It.IsAny<string>(), It.IsAny<string>()))
+                .Throws(new WebException());
+        }
+
+        protected void GivenCompletedItem()
+        {
+            var targetDir = Path.Combine(_completedDownloadFolder, _title);
+            Mocker.GetMock<IDiskProvider>()
+                .Setup(c => c.GetDirectories(_completedDownloadFolder))
+                .Returns(new[] { targetDir });
+
+            Mocker.GetMock<IDiskProvider>()
+                .Setup(c => c.GetFiles(targetDir, SearchOption.AllDirectories))
+                .Returns(new[] { Path.Combine(_completedDownloadFolder, "somefile.mkv") });
+
+            Mocker.GetMock<IDiskProvider>()
+                .Setup(c => c.GetFileSize(It.IsAny<string>()))
+                .Returns(1000000);
+        }
+
+        [Test]
+        public void completed_download_should_have_required_properties()
+        {
+            GivenCompletedItem();
+
+            var result = Subject.GetItems().Single();
+
+            VerifyCompleted(result);
+        }
+
+        [Test]
+        public void Download_should_download_file_if_it_doesnt_exist()
+        {
+            var remoteEpisode = CreateRemoteEpisode();
+
+            Subject.Download(remoteEpisode);
+
+            Mocker.GetMock<IHttpProvider>().Verify(c => c.DownloadFile(_downloadUrl, _filePath), Times.Once());
+        }
+
+        [Test]
+        public void Download_should_replace_illegal_characters_in_title()
+        {
+            var illegalTitle = "Saturday Night Live - S38E08 - Jeremy Renner/Maroon 5 [SDTV]";
+            var expectedFilename = Path.Combine(_blackholeFolder, "Saturday Night Live - S38E08 - Jeremy Renner+Maroon 5 [SDTV]" + Path.GetExtension(_filePath));
+
+            var remoteEpisode = CreateRemoteEpisode();
+            remoteEpisode.Release.Title = illegalTitle;
+
+            Subject.Download(remoteEpisode);
+
+            Mocker.GetMock<IHttpProvider>().Verify(c => c.DownloadFile(It.IsAny<string>(), expectedFilename), Times.Once());
+        }
+
+        [Test]
+        public void GetItems_should_considered_locked_files_downloading()
+        {
+            GivenCompletedItem();
+
+            Mocker.GetMock<IDiskProvider>()
+                .Setup(c => c.IsFileLocked(It.IsAny<string>()))
+                .Returns(true);
+
+            var result = Subject.GetItems().Single();
+
+            result.Status.Should().Be(DownloadItemStatus.Downloading);
+        }
+    }
+}