New: RQBit download client

Co-authored-by: Mark Mendoza <markolo25@gmail.com>
2026-04-20 21:54:58 -04:00 · 2025-09-02 00:08:03 +02:00
parent 7fdc4d6638
commit 37cb978f18
12 changed files with 937 additions and 0 deletions
@@ -0,0 +1,234 @@
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using FluentAssertions;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Core.Download;
+using NzbDrone.Core.Download.Clients.RQBit;
+using NzbDrone.Core.MediaFiles.TorrentInfo;
+
+namespace NzbDrone.Core.Test.Download.DownloadClientTests.RQBitTests
+{
+    [TestFixture]
+    public class RQBitFixture : DownloadClientFixtureBase<RQBit>
+    {
+        protected RQBitTorrent _queued;
+        protected RQBitTorrent _downloading;
+        protected RQBitTorrent _failed;
+        protected RQBitTorrent _completed;
+
+        [SetUp]
+        public void Setup()
+        {
+            Subject.Definition = new DownloadClientDefinition();
+            Subject.Definition.Settings = new RQbitSettings
+            {
+                Host = "127.0.0.1",
+                Port = 3030,
+                UseSsl = false
+            };
+
+            _queued = new RQBitTorrent
+            {
+                Hash = "HASH",
+                IsFinished = false,
+                IsActive = false,
+                Name = _title,
+                TotalSize = 1000,
+                RemainingSize = 1000,
+                Path = "somepath"
+            };
+
+            _downloading = new RQBitTorrent
+            {
+                Hash = "HASH",
+                IsFinished = false,
+                IsActive = true,
+                Name = _title,
+                TotalSize = 1000,
+                RemainingSize = 100,
+                Path = "somepath"
+            };
+
+            _failed = new RQBitTorrent
+            {
+                Hash = "HASH",
+                IsFinished = false,
+                IsActive = false,
+                Name = _title,
+                TotalSize = 1000,
+                RemainingSize = 1000,
+                Path = "somepath"
+            };
+
+            _completed = new RQBitTorrent
+            {
+                Hash = "HASH",
+                IsFinished = true,
+                IsActive = false,
+                Name = _title,
+                TotalSize = 1000,
+                RemainingSize = 0,
+                Path = "somepath"
+            };
+
+            Mocker.GetMock<ITorrentFileInfoReader>()
+                  .Setup(s => s.GetHashFromTorrentFile(It.IsAny<byte[]>()))
+                  .Returns("CBC2F069FE8BB2F544EAE707D75BCD3DE9DCF951");
+        }
+
+        protected void GivenSuccessfulDownload()
+        {
+            Mocker.GetMock<IRQbitProxy>()
+                  .Setup(s => s.AddTorrentFromUrl(It.IsAny<string>(), It.IsAny<RQbitSettings>()))
+                  .Returns("CBC2F069FE8BB2F544EAE707D75BCD3DE9DCF951");
+            Mocker.GetMock<IRQbitProxy>()
+                  .Setup(s => s.AddTorrentFromFile(It.IsAny<string>(), It.IsAny<byte[]>(), It.IsAny<RQbitSettings>()))
+                  .Returns("CBC2F069FE8BB2F544EAE707D75BCD3DE9DCF951");
+        }
+
+        protected virtual void GivenTorrents(List<RQBitTorrent> torrents)
+        {
+            if (torrents == null)
+            {
+                torrents = new List<RQBitTorrent>();
+            }
+
+            Mocker.GetMock<IRQbitProxy>()
+                  .Setup(s => s.GetTorrents(It.IsAny<RQbitSettings>()))
+                  .Returns(torrents);
+        }
+
+        protected void PrepareClientToReturnQueuedItem()
+        {
+            GivenTorrents(new List<RQBitTorrent>
+                {
+                    _queued
+                });
+        }
+
+        protected void PrepareClientToReturnDownloadingItem()
+        {
+            GivenTorrents(new List<RQBitTorrent>
+                {
+                    _downloading
+                });
+        }
+
+        protected void PrepareClientToReturnFailedItem()
+        {
+            GivenTorrents(new List<RQBitTorrent>
+                {
+                    _failed
+                });
+        }
+
+        protected void PrepareClientToReturnCompletedItem()
+        {
+            GivenTorrents(new List<RQBitTorrent>
+                {
+                    _completed
+                });
+        }
+
+        [Test]
+        public void queued_item_should_have_required_properties()
+        {
+            PrepareClientToReturnQueuedItem();
+            var item = Subject.GetItems().Single();
+            VerifyPaused(item);
+        }
+
+        [Test]
+        public void downloading_item_should_have_required_properties()
+        {
+            PrepareClientToReturnDownloadingItem();
+            var item = Subject.GetItems().Single();
+            VerifyDownloading(item);
+        }
+
+        [Test]
+        public void failed_item_should_have_required_properties()
+        {
+            PrepareClientToReturnFailedItem();
+            var item = Subject.GetItems().Single();
+            VerifyPaused(item);
+        }
+
+        [Test]
+        public void completed_download_should_have_required_properties()
+        {
+            PrepareClientToReturnCompletedItem();
+            var item = Subject.GetItems().Single();
+            VerifyCompleted(item);
+        }
+
+        [Test]
+        public async Task Download_should_return_unique_id()
+        {
+            GivenSuccessfulDownload();
+
+            var remoteEpisode = CreateRemoteEpisode();
+
+            var id = await Subject.Download(remoteEpisode, CreateIndexer());
+
+            id.Should().NotBeNullOrEmpty();
+        }
+
+        [TestCase("magnet:?xt=urn:btih:ZPBPA2P6ROZPKRHK44D5OW6NHXU5Z6KR&tr=udp", "CBC2F069FE8BB2F544EAE707D75BCD3DE9DCF951")]
+        public async Task Download_should_get_hash_from_magnet_url(string magnetUrl, string expectedHash)
+        {
+            GivenSuccessfulDownload();
+
+            var remoteEpisode = CreateRemoteEpisode();
+            remoteEpisode.Release.DownloadUrl = magnetUrl;
+
+            var id = await Subject.Download(remoteEpisode, CreateIndexer());
+
+            id.Should().Be(expectedHash);
+        }
+
+        [Test]
+        public void should_return_status_with_outputdirs()
+        {
+            var result = Subject.GetStatus();
+
+            result.IsLocalhost.Should().BeTrue();
+        }
+
+        [Test]
+        public void GetItems_should_ignore_torrents_with_empty_path()
+        {
+            var torrents = new List<RQBitTorrent>
+            {
+                new RQBitTorrent { Name = "Test1", Hash = "Hash1", Path = "" },
+                new RQBitTorrent { Name = "Test2", Hash = "Hash2", Path = "/valid/path" }
+            };
+
+            GivenTorrents(torrents);
+
+            var items = Subject.GetItems();
+
+            items.Should().HaveCount(1);
+            items.First().Title.Should().Be("Test2");
+        }
+
+        [Test]
+        public void GetItems_should_ignore_torrents_with_relative_path()
+        {
+            var torrents = new List<RQBitTorrent>
+            {
+                new RQBitTorrent { Name = "Test1", Hash = "Hash1", Path = "./relative/path" },
+                new RQBitTorrent { Name = "Test2", Hash = "Hash2", Path = "/absolute/path" }
+            };
+
+            GivenTorrents(torrents);
+
+            var items = Subject.GetItems();
+
+            items.Should().HaveCount(1);
+            items.First().Title.Should().Be("Test2");
+        }
+    }
+}