mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-12-21 01:49:57 +02:00
75 lines
2.9 KiB
C#
75 lines
2.9 KiB
C#
using System;
|
|
using System.Linq;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Core.Configuration;
|
|
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.NzbgetProviderTests
|
|
{
|
|
public class DownloadNzbFixture : CoreTest
|
|
{
|
|
private const string _url = "http://www.nzbdrone.com";
|
|
private const string _title = "30 Rock - S01E01 - Pilot [HDTV-720p]";
|
|
private RemoteEpisode _remoteEpisode;
|
|
|
|
[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.NzbgetRecentTvPriority).Returns(PriorityType.High);
|
|
|
|
_remoteEpisode = new RemoteEpisode();
|
|
_remoteEpisode.Report = new ReportInfo();
|
|
_remoteEpisode.Report.Title = _title;
|
|
_remoteEpisode.Report.NzbUrl = _url;
|
|
|
|
_remoteEpisode.Episodes = Builder<Episode>.CreateListOfSize(1)
|
|
.All()
|
|
.With(e => e.AirDate = DateTime.Today.ToString(Episode.AIR_DATE_FORMAT))
|
|
.Build()
|
|
.ToList();
|
|
}
|
|
|
|
private void WithFailResponse()
|
|
{
|
|
Mocker.GetMock<IHttpProvider>()
|
|
.Setup(s => s.PostCommand("192.168.5.55:6789", "nzbget", "pass", It.IsAny<String>()))
|
|
.Returns(ReadAllText("Files", "Nzbget", "JsonError.txt"));
|
|
}
|
|
|
|
[Test]
|
|
public void should_add_item_to_queue()
|
|
{
|
|
Mocker.GetMock<IHttpProvider>()
|
|
.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\",50,false,\"http://www.nzbdrone.com\"]}"))))
|
|
.Returns("{\"version\": \"1.1\",\"result\": true}");
|
|
|
|
Mocker.Resolve<NzbgetClient>()
|
|
.DownloadNzb(_remoteEpisode)
|
|
.Should()
|
|
.BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_throw_when_error_is_returned()
|
|
{
|
|
WithFailResponse();
|
|
|
|
Assert.Throws<ApplicationException>(() => Mocker.Resolve<NzbgetClient>().DownloadNzb(_remoteEpisode));
|
|
}
|
|
}
|
|
}
|