Sonarr/NzbDrone.Core.Test/Download/DownloadServiceFixture.cs

101 lines
3.0 KiB
C#
Raw Normal View History

2013-04-01 06:22:16 +00:00
using System.Linq;
2013-08-17 23:27:18 +00:00
using System.Net;
2013-04-01 06:22:16 +00:00
using FizzWare.NBuilder;
using Moq;
using NUnit.Framework;
using NzbDrone.Core.Download;
using NzbDrone.Core.Parser.Model;
2013-04-01 06:22:16 +00:00
using NzbDrone.Core.Test.Framework;
using NzbDrone.Core.Tv;
using NzbDrone.Test.Common;
2013-04-01 06:22:16 +00:00
namespace NzbDrone.Core.Test.Download
{
[TestFixture]
public class DownloadServiceFixture : CoreTest<DownloadService>
{
private RemoteEpisode _parseResult;
2013-04-01 06:22:16 +00:00
[SetUp]
public void Setup()
{
Mocker.GetMock<IProvideDownloadClient>()
.Setup(c => c.GetDownloadClient()).Returns(Mocker.GetMock<IDownloadClient>().Object);
var episodes = Builder<Episode>.CreateListOfSize(2)
.TheFirst(1).With(s => s.Id = 12)
.TheNext(1).With(s => s.Id = 99)
.All().With(s => s.SeriesId = 5)
.Build().ToList();
_parseResult = Builder<RemoteEpisode>.CreateNew()
2013-04-01 06:22:16 +00:00
.With(c => c.Series = Builder<Series>.CreateNew().Build())
.With(c => c.Report = Builder<ReportInfo>.CreateNew().Build())
2013-04-01 06:22:16 +00:00
.With(c => c.Episodes = episodes)
.Build();
Mocker.GetMock<IDownloadClient>().Setup(c => c.IsConfigured).Returns(true);
2013-04-01 06:22:16 +00:00
}
private void WithSuccessfulAdd()
{
Mocker.GetMock<IDownloadClient>()
2013-08-17 23:27:18 +00:00
.Setup(s => s.DownloadNzb(It.IsAny<RemoteEpisode>()));
2013-04-01 06:22:16 +00:00
}
private void WithFailedAdd()
{
Mocker.GetMock<IDownloadClient>()
.Setup(s => s.DownloadNzb(It.IsAny<RemoteEpisode>()))
2013-08-17 23:27:18 +00:00
.Throws(new WebException());
2013-04-01 06:22:16 +00:00
}
[Test]
public void Download_report_should_publish_on_grab_event()
{
WithSuccessfulAdd();
Subject.DownloadReport(_parseResult);
VerifyEventPublished<EpisodeGrabbedEvent>();
}
[Test]
public void Download_report_should_grab_using_client()
{
WithSuccessfulAdd();
Subject.DownloadReport(_parseResult);
Mocker.GetMock<IDownloadClient>()
.Verify(s => s.DownloadNzb(It.IsAny<RemoteEpisode>()), Times.Once());
2013-04-01 06:22:16 +00:00
}
[Test]
public void Download_report_should_not_publish_on_failed_grab_event()
{
WithFailedAdd();
2013-08-17 23:27:18 +00:00
Assert.Throws<WebException>(() => Subject.DownloadReport(_parseResult));
2013-04-01 06:22:16 +00:00
VerifyEventNotPublished<EpisodeGrabbedEvent>();
}
[Test]
public void should_not_attempt_download_if_client_isnt_configure()
{
Mocker.GetMock<IDownloadClient>().Setup(c => c.IsConfigured).Returns(false);
Subject.DownloadReport(_parseResult);
2013-08-17 23:27:18 +00:00
Mocker.GetMock<IDownloadClient>().Verify(c => c.DownloadNzb(It.IsAny<RemoteEpisode>()), Times.Never());
VerifyEventNotPublished<EpisodeGrabbedEvent>();
ExceptionVerification.ExpectedWarns(1);
}
2013-04-01 06:22:16 +00:00
}
}