You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
420 lines
15 KiB
420 lines
15 KiB
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.Download;
|
|
using NzbDrone.Core.Download.Clients.Transmission;
|
|
|
|
namespace NzbDrone.Core.Test.Download.DownloadClientTests.TransmissionTests
|
|
{
|
|
[TestFixture]
|
|
public class TransmissionFixture : TransmissionFixtureBase<Transmission>
|
|
{
|
|
[Test]
|
|
public void queued_item_should_have_required_properties()
|
|
{
|
|
PrepareClientToReturnQueuedItem();
|
|
var item = Subject.GetItems().Single();
|
|
VerifyQueued(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();
|
|
VerifyWarning(item);
|
|
}
|
|
|
|
[Test]
|
|
public void completed_download_should_have_required_properties()
|
|
{
|
|
PrepareClientToReturnCompletedItem();
|
|
var item = Subject.GetItems().Single();
|
|
VerifyCompleted(item);
|
|
|
|
item.CanBeRemoved.Should().BeFalse();
|
|
item.CanMoveFiles.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void magnet_download_should_not_return_the_item()
|
|
{
|
|
PrepareClientToReturnMagnetItem();
|
|
Subject.GetItems().Count().Should().Be(0);
|
|
}
|
|
|
|
[Test]
|
|
public void Download_should_return_unique_id()
|
|
{
|
|
GivenSuccessfulDownload();
|
|
|
|
var remoteBook = CreateRemoteBook();
|
|
|
|
var id = Subject.Download(remoteBook);
|
|
|
|
id.Should().NotBeNullOrEmpty();
|
|
}
|
|
|
|
[Test]
|
|
public void Download_with_TvDirectory_should_force_directory()
|
|
{
|
|
GivenTvDirectory();
|
|
GivenSuccessfulDownload();
|
|
|
|
var remoteBook = CreateRemoteBook();
|
|
|
|
var id = Subject.Download(remoteBook);
|
|
|
|
id.Should().NotBeNullOrEmpty();
|
|
|
|
Mocker.GetMock<ITransmissionProxy>()
|
|
.Verify(v => v.AddTorrentFromData(It.IsAny<byte[]>(), @"C:/Downloads/Finished/Readarr", It.IsAny<TransmissionSettings>()), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void Download_with_category_should_force_directory()
|
|
{
|
|
GivenMusicCategory();
|
|
GivenSuccessfulDownload();
|
|
|
|
var remoteBook = CreateRemoteBook();
|
|
|
|
var id = Subject.Download(remoteBook);
|
|
|
|
id.Should().NotBeNullOrEmpty();
|
|
|
|
Mocker.GetMock<ITransmissionProxy>()
|
|
.Verify(v => v.AddTorrentFromData(It.IsAny<byte[]>(), @"C:/Downloads/Finished/transmission/Readarr", It.IsAny<TransmissionSettings>()), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void Download_with_category_should_not_have_double_slashes()
|
|
{
|
|
GivenMusicCategory();
|
|
GivenSuccessfulDownload();
|
|
|
|
_transmissionConfigItems["download-dir"] += "/";
|
|
|
|
var remoteBook = CreateRemoteBook();
|
|
|
|
var id = Subject.Download(remoteBook);
|
|
|
|
id.Should().NotBeNullOrEmpty();
|
|
|
|
Mocker.GetMock<ITransmissionProxy>()
|
|
.Verify(v => v.AddTorrentFromData(It.IsAny<byte[]>(), @"C:/Downloads/Finished/transmission/Readarr", It.IsAny<TransmissionSettings>()), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void Download_without_TvDirectory_and_Category_should_use_default()
|
|
{
|
|
GivenSuccessfulDownload();
|
|
|
|
var remoteBook = CreateRemoteBook();
|
|
|
|
var id = Subject.Download(remoteBook);
|
|
|
|
id.Should().NotBeNullOrEmpty();
|
|
|
|
Mocker.GetMock<ITransmissionProxy>()
|
|
.Verify(v => v.AddTorrentFromData(It.IsAny<byte[]>(), null, It.IsAny<TransmissionSettings>()), Times.Once());
|
|
}
|
|
|
|
[TestCase("magnet:?xt=urn:btih:ZPBPA2P6ROZPKRHK44D5OW6NHXU5Z6KR&tr=udp", "CBC2F069FE8BB2F544EAE707D75BCD3DE9DCF951")]
|
|
public void Download_should_get_hash_from_magnet_url(string magnetUrl, string expectedHash)
|
|
{
|
|
GivenSuccessfulDownload();
|
|
|
|
var remoteBook = CreateRemoteBook();
|
|
remoteBook.Release.DownloadUrl = magnetUrl;
|
|
|
|
var id = Subject.Download(remoteBook);
|
|
|
|
id.Should().Be(expectedHash);
|
|
}
|
|
|
|
[TestCase(TransmissionTorrentStatus.Stopped, DownloadItemStatus.Downloading)]
|
|
[TestCase(TransmissionTorrentStatus.CheckWait, DownloadItemStatus.Downloading)]
|
|
[TestCase(TransmissionTorrentStatus.Check, DownloadItemStatus.Downloading)]
|
|
[TestCase(TransmissionTorrentStatus.Queued, DownloadItemStatus.Queued)]
|
|
[TestCase(TransmissionTorrentStatus.Downloading, DownloadItemStatus.Downloading)]
|
|
[TestCase(TransmissionTorrentStatus.SeedingWait, DownloadItemStatus.Downloading)]
|
|
[TestCase(TransmissionTorrentStatus.Seeding, DownloadItemStatus.Downloading)]
|
|
public void GetItems_should_return_queued_item_as_downloadItemStatus(TransmissionTorrentStatus apiStatus, DownloadItemStatus expectedItemStatus)
|
|
{
|
|
_queued.Status = apiStatus;
|
|
|
|
PrepareClientToReturnQueuedItem();
|
|
|
|
var item = Subject.GetItems().Single();
|
|
|
|
item.Status.Should().Be(expectedItemStatus);
|
|
}
|
|
|
|
[TestCase(TransmissionTorrentStatus.Queued, DownloadItemStatus.Queued)]
|
|
[TestCase(TransmissionTorrentStatus.Downloading, DownloadItemStatus.Downloading)]
|
|
[TestCase(TransmissionTorrentStatus.Seeding, DownloadItemStatus.Downloading)]
|
|
public void GetItems_should_return_downloading_item_as_downloadItemStatus(TransmissionTorrentStatus apiStatus, DownloadItemStatus expectedItemStatus)
|
|
{
|
|
_downloading.Status = apiStatus;
|
|
|
|
PrepareClientToReturnDownloadingItem();
|
|
|
|
var item = Subject.GetItems().Single();
|
|
|
|
item.Status.Should().Be(expectedItemStatus);
|
|
}
|
|
|
|
[TestCase(TransmissionTorrentStatus.Stopped, DownloadItemStatus.Completed, false)]
|
|
[TestCase(TransmissionTorrentStatus.CheckWait, DownloadItemStatus.Downloading, false)]
|
|
[TestCase(TransmissionTorrentStatus.Check, DownloadItemStatus.Downloading, false)]
|
|
[TestCase(TransmissionTorrentStatus.Queued, DownloadItemStatus.Completed, false)]
|
|
[TestCase(TransmissionTorrentStatus.SeedingWait, DownloadItemStatus.Completed, false)]
|
|
[TestCase(TransmissionTorrentStatus.Seeding, DownloadItemStatus.Completed, false)]
|
|
public void GetItems_should_return_completed_item_as_downloadItemStatus(TransmissionTorrentStatus apiStatus, DownloadItemStatus expectedItemStatus, bool expectedValue)
|
|
{
|
|
_completed.Status = apiStatus;
|
|
|
|
PrepareClientToReturnCompletedItem();
|
|
|
|
var item = Subject.GetItems().Single();
|
|
|
|
item.Status.Should().Be(expectedItemStatus);
|
|
item.CanBeRemoved.Should().Be(expectedValue);
|
|
item.CanMoveFiles.Should().Be(expectedValue);
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_status_with_outputdirs()
|
|
{
|
|
var result = Subject.GetStatus();
|
|
|
|
result.IsLocalhost.Should().BeTrue();
|
|
result.OutputRootFolders.Should().NotBeNull();
|
|
result.OutputRootFolders.First().Should().Be(@"C:\Downloads\Finished\transmission");
|
|
}
|
|
|
|
[Test]
|
|
public void should_exclude_items_not_in_category()
|
|
{
|
|
GivenMusicCategory();
|
|
|
|
_downloading.DownloadDir = @"C:/Downloads/Finished/transmission/Readarr";
|
|
|
|
GivenTorrents(new List<TransmissionTorrent>
|
|
{
|
|
_downloading,
|
|
_queued
|
|
});
|
|
|
|
var items = Subject.GetItems().ToList();
|
|
|
|
items.Count.Should().Be(1);
|
|
items.First().Status.Should().Be(DownloadItemStatus.Downloading);
|
|
}
|
|
|
|
[Test]
|
|
public void should_exclude_items_not_in_TvDirectory()
|
|
{
|
|
GivenTvDirectory();
|
|
|
|
_downloading.DownloadDir = @"C:/Downloads/Finished/Readarr/subdir";
|
|
|
|
GivenTorrents(new List<TransmissionTorrent>
|
|
{
|
|
_downloading,
|
|
_queued
|
|
});
|
|
|
|
var items = Subject.GetItems().ToList();
|
|
|
|
items.Count.Should().Be(1);
|
|
items.First().Status.Should().Be(DownloadItemStatus.Downloading);
|
|
}
|
|
|
|
[Test]
|
|
public void should_fix_forward_slashes()
|
|
{
|
|
WindowsOnly();
|
|
|
|
_downloading.DownloadDir = @"C:/Downloads/Finished/transmission";
|
|
|
|
GivenTorrents(new List<TransmissionTorrent>
|
|
{
|
|
_downloading
|
|
});
|
|
|
|
var items = Subject.GetItems().ToList();
|
|
|
|
items.Should().HaveCount(1);
|
|
items.First().OutputPath.Should().Be(@"C:\Downloads\Finished\transmission\" + _title);
|
|
}
|
|
|
|
[TestCase("2.84 ()")]
|
|
[TestCase("2.84+ ()")]
|
|
[TestCase("2.84 (other info)")]
|
|
[TestCase("2.84 (2.84)")]
|
|
public void should_only_check_version_number(string version)
|
|
{
|
|
Mocker.GetMock<ITransmissionProxy>()
|
|
.Setup(s => s.GetClientVersion(It.IsAny<TransmissionSettings>()))
|
|
.Returns(version);
|
|
|
|
Subject.Test().IsValid.Should().BeTrue();
|
|
}
|
|
|
|
[TestCase(-1)] // Infinite/Unknown
|
|
[TestCase(-2)] // Magnet Downloading
|
|
public void should_ignore_negative_eta(int eta)
|
|
{
|
|
_completed.Eta = eta;
|
|
|
|
PrepareClientToReturnCompletedItem();
|
|
var item = Subject.GetItems().Single();
|
|
item.RemainingTime.Should().NotHaveValue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_be_removable_and_should_not_allow_move_files_if_max_ratio_reached_and_not_stopped()
|
|
{
|
|
GivenGlobalSeedLimits(1.0);
|
|
PrepareClientToReturnCompletedItem(false, ratio: 1.0);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeFalse();
|
|
item.CanMoveFiles.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_be_removable_and_should_not_allow_move_files_if_max_ratio_is_not_set()
|
|
{
|
|
GivenGlobalSeedLimits();
|
|
PrepareClientToReturnCompletedItem(true, ratio: 1.0);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeFalse();
|
|
item.CanMoveFiles.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_removable_and_should_allow_move_files_if_max_ratio_reached_and_paused()
|
|
{
|
|
GivenGlobalSeedLimits(1.0);
|
|
PrepareClientToReturnCompletedItem(true, ratio: 1.0);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeTrue();
|
|
item.CanMoveFiles.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_removable_and_should_allow_move_files_if_overridden_max_ratio_reached_and_paused()
|
|
{
|
|
GivenGlobalSeedLimits(2.0);
|
|
PrepareClientToReturnCompletedItem(true, ratio: 1.0, ratioLimit: 0.8);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeTrue();
|
|
item.CanMoveFiles.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_be_removable_if_overridden_max_ratio_not_reached_and_paused()
|
|
{
|
|
GivenGlobalSeedLimits(0.2);
|
|
PrepareClientToReturnCompletedItem(true, ratio: 0.5, ratioLimit: 0.8);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeFalse();
|
|
item.CanMoveFiles.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_be_removable_and_should_not_allow_move_files_if_max_idletime_reached_and_not_paused()
|
|
{
|
|
GivenGlobalSeedLimits(null, 20);
|
|
PrepareClientToReturnCompletedItem(false, ratio: 2.0, seedingTime: 30);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeFalse();
|
|
item.CanMoveFiles.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_removable_and_should_allow_move_files_if_max_idletime_reached_and_paused()
|
|
{
|
|
GivenGlobalSeedLimits(null, 20);
|
|
PrepareClientToReturnCompletedItem(true, ratio: 2.0, seedingTime: 20);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeTrue();
|
|
item.CanMoveFiles.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_removable_and_should_allow_move_files_if_overridden_max_idletime_reached_and_paused()
|
|
{
|
|
GivenGlobalSeedLimits(null, 40);
|
|
PrepareClientToReturnCompletedItem(true, ratio: 2.0, seedingTime: 20, idleLimit: 10);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeTrue();
|
|
item.CanMoveFiles.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_removable_and_should_not_allow_move_files_if_overridden_max_idletime_reached_and_not_paused()
|
|
{
|
|
GivenGlobalSeedLimits(null, 40);
|
|
PrepareClientToReturnCompletedItem(false, ratio: 2.0, seedingTime: 20, idleLimit: 10);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeTrue();
|
|
item.CanMoveFiles.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_be_removable_if_overridden_max_idletime_not_reached_and_paused()
|
|
{
|
|
GivenGlobalSeedLimits(null, 20);
|
|
PrepareClientToReturnCompletedItem(true, ratio: 2.0, seedingTime: 30, idleLimit: 40);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeFalse();
|
|
item.CanMoveFiles.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_be_removable_if_max_idletime_reached_but_ratio_not_and_not_paused()
|
|
{
|
|
GivenGlobalSeedLimits(2.0, 20);
|
|
PrepareClientToReturnCompletedItem(false, ratio: 1.0, seedingTime: 30);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeFalse();
|
|
item.CanMoveFiles.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_removable_and_should_allow_move_files_if_max_idletime_configured_and_paused()
|
|
{
|
|
GivenGlobalSeedLimits(2.0, 20);
|
|
PrepareClientToReturnCompletedItem(true, ratio: 1.0, seedingTime: 30);
|
|
|
|
var item = Subject.GetItems().Single();
|
|
item.CanBeRemoved.Should().BeTrue();
|
|
item.CanMoveFiles.Should().BeTrue();
|
|
}
|
|
}
|
|
}
|