|
|
|
using System;
|
|
|
|
using System.IO;
|
|
|
|
using System.Linq;
|
|
|
|
using System.Collections.Generic;
|
|
|
|
using FizzWare.NBuilder;
|
|
|
|
using FluentAssertions;
|
|
|
|
using Moq;
|
|
|
|
using NUnit.Framework;
|
|
|
|
using NzbDrone.Common;
|
|
|
|
using NzbDrone.Core.Download;
|
|
|
|
using NzbDrone.Core.Download.Clients.Sabnzbd;
|
|
|
|
using NzbDrone.Core.Download.Clients.Sabnzbd.Responses;
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
using NzbDrone.Core.Tv;
|
|
|
|
using NzbDrone.Test.Common;
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.Download.DownloadClientTests.SabnzbdTests
|
|
|
|
{
|
|
|
|
[TestFixture]
|
|
|
|
public class SabnzbdFixture : DownloadClientFixtureBase<Sabnzbd>
|
|
|
|
{
|
|
|
|
private SabnzbdQueue _queued;
|
|
|
|
private SabnzbdHistory _failed;
|
|
|
|
private SabnzbdHistory _completed;
|
|
|
|
private SabnzbdConfig _config;
|
|
|
|
|
|
|
|
[SetUp]
|
|
|
|
public void Setup()
|
|
|
|
{
|
|
|
|
Subject.Definition = new DownloadClientDefinition();
|
|
|
|
Subject.Definition.Settings = new SabnzbdSettings
|
|
|
|
{
|
|
|
|
Host = "127.0.0.1",
|
|
|
|
Port = 2222,
|
|
|
|
ApiKey = "5c770e3197e4fe763423ee7c392c25d1",
|
|
|
|
Username = "admin",
|
|
|
|
Password = "pass",
|
|
|
|
TvCategory = "tv",
|
|
|
|
RecentTvPriority = (int)SabnzbdPriority.High
|
|
|
|
};
|
|
|
|
_queued = new SabnzbdQueue
|
|
|
|
{
|
|
|
|
DefaultRootFolder = @"Y:\nzbget\root".AsOsAgnostic(),
|
|
|
|
Paused = false,
|
|
|
|
Items = new List<SabnzbdQueueItem>()
|
|
|
|
{
|
|
|
|
new SabnzbdQueueItem
|
|
|
|
{
|
|
|
|
Status = SabnzbdDownloadStatus.Downloading,
|
|
|
|
Size = 1000,
|
|
|
|
Sizeleft = 10,
|
|
|
|
Timeleft = TimeSpan.FromSeconds(10),
|
|
|
|
Category = "tv",
|
|
|
|
Id = "sabnzbd_nzb12345",
|
|
|
|
Title = "Droned.S01E01.Pilot.1080p.WEB-DL-DRONE"
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
_failed = new SabnzbdHistory
|
|
|
|
{
|
|
|
|
Items = new List<SabnzbdHistoryItem>()
|
|
|
|
{
|
|
|
|
new SabnzbdHistoryItem
|
|
|
|
{
|
|
|
|
Status = SabnzbdDownloadStatus.Failed,
|
|
|
|
Size = 1000,
|
|
|
|
Category = "tv",
|
|
|
|
Id = "sabnzbd_nzb12345",
|
|
|
|
Title = "Droned.S01E01.Pilot.1080p.WEB-DL-DRONE"
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
_completed = new SabnzbdHistory
|
|
|
|
{
|
|
|
|
Items = new List<SabnzbdHistoryItem>()
|
|
|
|
{
|
|
|
|
new SabnzbdHistoryItem
|
|
|
|
{
|
|
|
|
Status = SabnzbdDownloadStatus.Completed,
|
|
|
|
Size = 1000,
|
|
|
|
Category = "tv",
|
|
|
|
Id = "sabnzbd_nzb12345",
|
|
|
|
Title = "Droned.S01E01.Pilot.1080p.WEB-DL-DRONE",
|
|
|
|
Storage = "/remote/mount/vv/Droned.S01E01.Pilot.1080p.WEB-DL-DRONE"
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
_config = new SabnzbdConfig
|
|
|
|
{
|
|
|
|
Misc = new SabnzbdConfigMisc
|
|
|
|
{
|
|
|
|
complete_dir = @"/remote/mount"
|
|
|
|
},
|
|
|
|
Categories = new List<SabnzbdCategory>
|
|
|
|
{
|
|
|
|
new SabnzbdCategory { Name = "tv", Dir = "vv" }
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
Mocker.GetMock<ISabnzbdProxy>()
|
|
|
|
.Setup(s => s.GetConfig(It.IsAny<SabnzbdSettings>()))
|
|
|
|
.Returns(_config);
|
|
|
|
}
|
|
|
|
|
|
|
|
protected void GivenMountPoint(String mountPath)
|
|
|
|
{
|
|
|
|
(Subject.Definition.Settings as SabnzbdSettings).TvCategoryLocalPath = mountPath;
|
|
|
|
}
|
|
|
|
|
|
|
|
protected void GivenFailedDownload()
|
|
|
|
{
|
|
|
|
Mocker.GetMock<ISabnzbdProxy>()
|
|
|
|
.Setup(s => s.DownloadNzb(It.IsAny<Byte[]>(), It.IsAny<String>(), It.IsAny<String>(), It.IsAny<int>(), It.IsAny<SabnzbdSettings>()))
|
|
|
|
.Returns((SabnzbdAddResponse)null);
|
|
|
|
}
|
|
|
|
|
|
|
|
protected void GivenSuccessfulDownload()
|
|
|
|
{
|
|
|
|
Mocker.GetMock<ISabnzbdProxy>()
|
|
|
|
.Setup(s => s.DownloadNzb(It.IsAny<Byte[]>(), It.IsAny<String>(), It.IsAny<String>(), It.IsAny<int>(), It.IsAny<SabnzbdSettings>()))
|
|
|
|
.Returns(new SabnzbdAddResponse()
|
|
|
|
{
|
|
|
|
Status = true,
|
|
|
|
Ids = new List<string> { "sabznbd_nzo12345" }
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
protected virtual void GivenQueue(SabnzbdQueue queue)
|
|
|
|
{
|
|
|
|
if (queue == null)
|
|
|
|
{
|
|
|
|
queue = new SabnzbdQueue()
|
|
|
|
{
|
|
|
|
DefaultRootFolder = _queued.DefaultRootFolder,
|
|
|
|
Items = new List<SabnzbdQueueItem>()
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
Mocker.GetMock<ISabnzbdProxy>()
|
|
|
|
.Setup(s => s.GetQueue(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<SabnzbdSettings>()))
|
|
|
|
.Returns(queue);
|
|
|
|
}
|
|
|
|
|
|
|
|
protected virtual void GivenHistory(SabnzbdHistory history)
|
|
|
|
{
|
|
|
|
if (history == null)
|
|
|
|
history = new SabnzbdHistory() { Items = new List<SabnzbdHistoryItem>() };
|
|
|
|
|
|
|
|
Mocker.GetMock<ISabnzbdProxy>()
|
|
|
|
.Setup(s => s.GetHistory(It.IsAny<int>(), It.IsAny<int>(), It.IsAny<SabnzbdSettings>()))
|
|
|
|
.Returns(history);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void GetItems_should_return_no_items_when_queue_is_empty()
|
|
|
|
{
|
|
|
|
GivenQueue(null);
|
|
|
|
GivenHistory(null);
|
|
|
|
|
|
|
|
Subject.GetItems().Should().BeEmpty();
|
|
|
|
}
|
|
|
|
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Grabbing)]
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Queued)]
|
|
|
|
public void queued_item_should_have_required_properties(SabnzbdDownloadStatus status)
|
|
|
|
{
|
|
|
|
_queued.Items.First().Status = status;
|
|
|
|
|
|
|
|
GivenQueue(_queued);
|
|
|
|
GivenHistory(null);
|
|
|
|
|
|
|
|
var result = Subject.GetItems().Single();
|
|
|
|
|
|
|
|
VerifyQueued(result);
|
|
|
|
result.RemainingTime.Should().NotBe(TimeSpan.Zero);
|
|
|
|
}
|
|
|
|
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Paused)]
|
|
|
|
public void paused_item_should_have_required_properties(SabnzbdDownloadStatus status)
|
|
|
|
{
|
|
|
|
_queued.Items.First().Status = status;
|
|
|
|
|
|
|
|
GivenQueue(_queued);
|
|
|
|
GivenHistory(null);
|
|
|
|
|
|
|
|
var result = Subject.GetItems().Single();
|
|
|
|
|
|
|
|
VerifyPaused(result);
|
|
|
|
}
|
|
|
|
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Checking)]
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Downloading)]
|
|
|
|
[TestCase(SabnzbdDownloadStatus.QuickCheck)]
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Verifying)]
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Repairing)]
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Fetching)]
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Extracting)]
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Moving)]
|
|
|
|
[TestCase(SabnzbdDownloadStatus.Running)]
|
|
|
|
public void downloading_item_should_have_required_properties(SabnzbdDownloadStatus status)
|
|
|
|
{
|
|
|
|
_queued.Items.First().Status = status;
|
|
|
|
|
|
|
|
GivenQueue(_queued);
|
|
|
|
GivenHistory(null);
|
|
|
|
|
|
|
|
var result = Subject.GetItems().Single();
|
|
|
|
|
|
|
|
VerifyDownloading(result);
|
|
|
|
result.RemainingTime.Should().NotBe(TimeSpan.Zero);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void completed_download_should_have_required_properties()
|
|
|
|
{
|
|
|
|
GivenQueue(null);
|
|
|
|
GivenHistory(_completed);
|
|
|
|
|
|
|
|
var result = Subject.GetItems().Single();
|
|
|
|
|
|
|
|
VerifyCompleted(result);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void failed_item_should_have_required_properties()
|
|
|
|
{
|
|
|
|
_completed.Items.First().Status = SabnzbdDownloadStatus.Failed;
|
|
|
|
|
|
|
|
GivenQueue(null);
|
|
|
|
GivenHistory(_completed);
|
|
|
|
|
|
|
|
var result = Subject.GetItems().Single();
|
|
|
|
|
|
|
|
VerifyFailed(result);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void Download_should_return_unique_id()
|
|
|
|
{
|
|
|
|
GivenSuccessfulDownload();
|
|
|
|
|
|
|
|
var remoteEpisode = CreateRemoteEpisode();
|
|
|
|
|
|
|
|
var id = Subject.Download(remoteEpisode);
|
|
|
|
|
|
|
|
id.Should().NotBeNullOrEmpty();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void GetItems_should_ignore_downloads_from_other_categories()
|
|
|
|
{
|
|
|
|
_completed.Items.First().Category = "myowncat";
|
|
|
|
|
|
|
|
GivenQueue(null);
|
|
|
|
GivenHistory(_completed);
|
|
|
|
|
|
|
|
var items = Subject.GetItems();
|
|
|
|
|
|
|
|
items.Should().BeEmpty();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void Download_should_use_sabRecentTvPriority_when_recentEpisode_is_true()
|
|
|
|
{
|
|
|
|
Mocker.GetMock<ISabnzbdProxy>()
|
|
|
|
.Setup(s => s.DownloadNzb(It.IsAny<Byte[]>(), It.IsAny<String>(), It.IsAny<String>(), (int)SabnzbdPriority.High, It.IsAny<SabnzbdSettings>()))
|
|
|
|
.Returns(new SabnzbdAddResponse());
|
|
|
|
|
|
|
|
var remoteEpisode = CreateRemoteEpisode();
|
|
|
|
remoteEpisode.Episodes = Builder<Episode>.CreateListOfSize(1)
|
|
|
|
.All()
|
|
|
|
.With(e => e.AirDate = DateTime.Today.ToString(Episode.AIR_DATE_FORMAT))
|
|
|
|
.Build()
|
|
|
|
.ToList();
|
|
|
|
|
|
|
|
Subject.Download(remoteEpisode);
|
|
|
|
|
|
|
|
Mocker.GetMock<ISabnzbdProxy>()
|
|
|
|
.Verify(v => v.DownloadNzb(It.IsAny<Byte[]>(), It.IsAny<String>(), It.IsAny<String>(), (int)SabnzbdPriority.High, It.IsAny<SabnzbdSettings>()), Times.Once());
|
|
|
|
}
|
|
|
|
|
|
|
|
[TestCase(@"Droned.S01E01.Pilot.1080p.WEB-DL-DRONE", @"Droned.S01E01_Pilot_1080p_WEB-DL-DRONE.mkv")]
|
|
|
|
[TestCase(@"Droned.S01E01.Pilot.1080p.WEB-DL-DRONE", @"SubDir\Droned.S01E01_Pilot_1080p_WEB-DL-DRONE.mkv")]
|
|
|
|
[TestCase(@"Droned.S01E01.Pilot.1080p.WEB-DL-DRONE.mkv", @"SubDir\Droned.S01E01_Pilot_1080p_WEB-DL-DRONE.mkv")]
|
|
|
|
[TestCase(@"Droned.S01E01.Pilot.1080p.WEB-DL-DRONE.mkv", @"SubDir\SubDir\Droned.S01E01_Pilot_1080p_WEB-DL-DRONE.mkv")]
|
|
|
|
public void should_return_path_to_jobfolder(String title, String storage)
|
|
|
|
{
|
|
|
|
_completed.Items.First().Title = title;
|
|
|
|
_completed.Items.First().Storage = (@"C:\sorted\" + title + @"\" + storage).AsOsAgnostic();
|
|
|
|
|
|
|
|
GivenQueue(null);
|
|
|
|
GivenHistory(_completed);
|
|
|
|
|
|
|
|
var result = Subject.GetItems().Single();
|
|
|
|
|
|
|
|
result.OutputPath.Should().Be((@"C:\sorted\" + title).AsOsAgnostic());
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_remap_storage_if_mounted()
|
|
|
|
{
|
|
|
|
GivenMountPoint(@"O:\mymount".AsOsAgnostic());
|
|
|
|
|
|
|
|
GivenQueue(null);
|
|
|
|
GivenHistory(_completed);
|
|
|
|
|
|
|
|
var result = Subject.GetItems().Single();
|
|
|
|
|
|
|
|
result.OutputPath.Should().Be(@"O:\mymount\Droned.S01E01.Pilot.1080p.WEB-DL-DRONE".AsOsAgnostic());
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_not_blow_up_if_storage_is_drive_root()
|
|
|
|
{
|
|
|
|
_completed.Items.First().Storage = @"C:\".AsOsAgnostic();
|
|
|
|
|
|
|
|
GivenQueue(null);
|
|
|
|
GivenHistory(_completed);
|
|
|
|
|
|
|
|
var result = Subject.GetItems().Single();
|
|
|
|
|
|
|
|
result.OutputPath.Should().Be(@"C:\".AsOsAgnostic());
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_not_blow_up_if_storage_doesnt_have_jobfolder()
|
|
|
|
{
|
|
|
|
_completed.Items.First().Storage = @"C:\sorted\somewhere\asdfasdf\asdfasdf.mkv".AsOsAgnostic();
|
|
|
|
|
|
|
|
GivenQueue(null);
|
|
|
|
GivenHistory(_completed);
|
|
|
|
|
|
|
|
var result = Subject.GetItems().Single();
|
|
|
|
|
|
|
|
result.OutputPath.Should().Be(@"C:\sorted\somewhere\asdfasdf\asdfasdf.mkv".AsOsAgnostic());
|
|
|
|
}
|
|
|
|
|
|
|
|
[TestCase(@"Y:\nzbget\root", @"completed\downloads", @"vv", @"Y:\nzbget\root\completed\downloads\vv")]
|
|
|
|
[TestCase(@"Y:\nzbget\root", @"completed", @"vv", @"Y:\nzbget\root\completed\vv")]
|
|
|
|
[TestCase(@"/nzbget/root", @"completed/downloads", @"vv", @"/nzbget/root/completed/downloads/vv")]
|
|
|
|
[TestCase(@"/nzbget/root", @"completed", @"vv", @"/nzbget/root/completed/vv")]
|
|
|
|
public void should_return_status_with_outputdir(String rootFolder, String completeDir, String categoryDir, String expectedDir)
|
|
|
|
{
|
|
|
|
_queued.DefaultRootFolder = rootFolder;
|
|
|
|
_config.Misc.complete_dir = completeDir;
|
|
|
|
_config.Categories.First().Dir = categoryDir;
|
|
|
|
|
|
|
|
GivenQueue(null);
|
|
|
|
|
|
|
|
var result = Subject.GetStatus();
|
|
|
|
|
|
|
|
result.IsLocalhost.Should().BeTrue();
|
|
|
|
result.OutputRootFolders.Should().NotBeNull();
|
|
|
|
result.OutputRootFolders.First().Should().Be(expectedDir);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_return_status_with_mounted_outputdir()
|
|
|
|
{
|
|
|
|
GivenMountPoint(@"O:\mymount".AsOsAgnostic());
|
|
|
|
|
|
|
|
GivenQueue(null);
|
|
|
|
|
|
|
|
var result = Subject.GetStatus();
|
|
|
|
|
|
|
|
result.IsLocalhost.Should().BeTrue();
|
|
|
|
result.OutputRootFolders.Should().NotBeNull();
|
|
|
|
result.OutputRootFolders.First().Should().Be(@"O:\mymount".AsOsAgnostic());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|