Download clients now use thingy provider

2026-04-25 22:36:59 -04:00 · 2014-02-13 21:31:49 -08:00
parent ba22600412
commit 606d78f5e1
123 changed files with 2076 additions and 1820 deletions
@@ -0,0 +1,61 @@
+using System;
+using System.Linq;
+using FizzWare.NBuilder;
+using Moq;
+using NUnit.Framework;
+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()
+        {
+            var p = new object[] {"30.Rock.S01E01.Pilot.720p.hdtv.nzb", "TV", 50, false, "http://www.nzbdrone.com"};
+
+            Mocker.GetMock<INzbgetProxy>()
+                    .Setup(s => s.AddNzb(It.IsAny<NzbgetSettings>(), p))
+                    .Returns(true);
+
+            Subject.DownloadNzb(_remoteEpisode);
+
+            Mocker.GetMock<INzbgetProxy>()
+                    .Verify(v => v.AddNzb(It.IsAny<NzbgetSettings>(), It.IsAny<object []>()), Times.Once());
+        }
+    }
+}
@@ -0,0 +1,80 @@
+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")
+                                             .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);
+        }
+    }
+}