mirror of https://github.com/Sonarr/Sonarr
67 lines
2.5 KiB
C#
67 lines
2.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Model.Nzbget;
|
|
using NzbDrone.Core.Providers.Core;
|
|
using NzbDrone.Core.Providers.DownloadClients;
|
|
using NzbDrone.Test.Common;
|
|
|
|
namespace NzbDrone.Core.Test.ProviderTests.DownloadClientTests.NzbgetProviderTests
|
|
{
|
|
public class DownloadNzbFixture : TestBase
|
|
{
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
var fakeConfig = Mocker.GetMock<IConfigService>();
|
|
fakeConfig.SetupGet(c => c.NzbgetHost).Returns("192.168.5.55");
|
|
fakeConfig.SetupGet(c => c.NzbgetPort).Returns(6789);
|
|
fakeConfig.SetupGet(c => c.NzbgetUsername).Returns("nzbget");
|
|
fakeConfig.SetupGet(c => c.NzbgetPassword).Returns("pass");
|
|
fakeConfig.SetupGet(c => c.NzbgetTvCategory).Returns("TV");
|
|
fakeConfig.SetupGet(c => c.NzbgetBacklogTvPriority).Returns(PriorityType.Normal);
|
|
fakeConfig.SetupGet(c => c.NzbgetRecentTvPriority).Returns(PriorityType.High);
|
|
}
|
|
|
|
|
|
private void WithFailResponse()
|
|
{
|
|
Mocker.GetMock<HttpProvider>()
|
|
.Setup(s => s.PostCommand("192.168.5.55:6789", "nzbget", "pass", It.IsAny<String>()))
|
|
.Returns(File.ReadAllText(@".\Files\Nzbget\JsonError.txt"));
|
|
}
|
|
|
|
[Test]
|
|
public void should_add_item_to_queue()
|
|
{
|
|
const string url = "http://www.nzbdrone.com";
|
|
const string title = "30 Rock - S01E01 - Pilot [HDTV-720p]";
|
|
|
|
Mocker.GetMock<HttpProvider>()
|
|
.Setup(s => s.PostCommand("192.168.5.55:6789", "nzbget", "pass",
|
|
It.Is<String>(c => c.Equals("{\"method\":\"appendurl\",\"params\":[\"30 Rock - S01E01 - Pilot [HDTV-720p]\",\"TV\",0,false,\"http://www.nzbdrone.com\"]}"))))
|
|
.Returns("{\"version\": \"1.1\",\"result\": true}");
|
|
|
|
Mocker.Resolve<NzbgetProvider>()
|
|
.DownloadNzb(url, title, false)
|
|
.Should()
|
|
.BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_throw_when_error_is_returned()
|
|
{
|
|
WithFailResponse();
|
|
|
|
Assert.Throws<ApplicationException>(() => Mocker.Resolve<NzbgetProvider>().DownloadNzb("http://www.nzbdrone.com", "30 Rock - S01E01 - Pilot [HDTV-720p]", false));
|
|
}
|
|
}
|
|
}
|