Lidarr/NzbDrone.Core.Test/Download/DownloadClientTests/PneumaticProviderFixture.cs

105 lines
3.6 KiB
C#
Raw Normal View History

using System.IO;
2012-08-30 00:20:48 +00:00
using System.Net;
using FluentAssertions;
using Moq;
using NUnit.Framework;
using NzbDrone.Common;
2013-02-24 06:48:52 +00:00
using NzbDrone.Core.Configuration;
2013-03-05 05:33:34 +00:00
using NzbDrone.Core.Download.Clients;
using NzbDrone.Core.Parser.Model;
2012-08-30 00:20:48 +00:00
using NzbDrone.Core.Test.Framework;
using NzbDrone.Test.Common;
2013-04-01 06:22:16 +00:00
namespace NzbDrone.Core.Test.Download.DownloadClientTests
2012-08-30 00:20:48 +00:00
{
[TestFixture]
public class PneumaticProviderFixture : CoreTest<PneumaticClient>
2012-08-30 00:20:48 +00:00
{
private const string _nzbUrl = "http://www.nzbs.com/url";
private const string _title = "30.Rock.S01E05.hdtv.xvid-LoL";
2013-07-26 05:29:59 +00:00
private string _pneumaticFolder;
private string _sabDrop;
private string _nzbPath;
private RemoteEpisode _remoteEpisode;
2012-08-30 00:20:48 +00:00
[SetUp]
public void Setup()
{
2013-07-26 05:29:59 +00:00
_pneumaticFolder = @"d:\nzb\pneumatic\".AsOsAgnostic();
_nzbPath = Path.Combine(_pneumaticFolder, _title + ".nzb").AsOsAgnostic();
_sabDrop = @"d:\unsorted tv\".AsOsAgnostic();
2012-08-30 15:33:09 +00:00
Mocker.GetMock<IConfigService>().SetupGet(c => c.PneumaticFolder).Returns(_pneumaticFolder);
Mocker.GetMock<IConfigService>().SetupGet(c => c.DownloadedEpisodesFolder).Returns(_sabDrop);
_remoteEpisode = new RemoteEpisode();
_remoteEpisode.Report = new ReportInfo();
_remoteEpisode.Report.Title = _title;
_remoteEpisode.Report.NzbUrl = _nzbUrl;
_remoteEpisode.ParsedEpisodeInfo = new ParsedEpisodeInfo();
_remoteEpisode.ParsedEpisodeInfo.FullSeason = false;
2012-08-30 00:20:48 +00:00
}
private void WithExistingFile()
{
Mocker.GetMock<IDiskProvider>().Setup(c => c.FileExists(_nzbPath)).Returns(true);
2012-08-30 00:20:48 +00:00
}
private void WithFailedDownload()
{
2013-04-10 23:41:45 +00:00
Mocker.GetMock<IHttpProvider>().Setup(c => c.DownloadFile(It.IsAny<string>(), It.IsAny<string>())).Throws(new WebException());
2012-08-30 00:20:48 +00:00
}
[Test]
public void should_download_file_if_it_doesnt_exist()
{
Subject.DownloadNzb(_remoteEpisode).Should().BeTrue();
2012-08-30 00:20:48 +00:00
2013-07-26 05:29:59 +00:00
Mocker.GetMock<IHttpProvider>().Verify(c => c.DownloadFile(_nzbUrl, _nzbPath), Times.Once());
2012-08-30 00:20:48 +00:00
}
[Test]
public void should_not_download_file_if_it_doesn_exist()
{
WithExistingFile();
Subject.DownloadNzb(_remoteEpisode).Should().BeTrue();
2012-08-30 00:20:48 +00:00
2013-04-10 23:41:45 +00:00
Mocker.GetMock<IHttpProvider>().Verify(c => c.DownloadFile(It.IsAny<string>(), It.IsAny<string>()), Times.Never());
2012-08-30 00:20:48 +00:00
}
[Test]
public void should_return_false_on_failed_download()
{
WithFailedDownload();
Subject.DownloadNzb(_remoteEpisode).Should().BeFalse();
2013-07-26 05:29:59 +00:00
2012-08-30 00:20:48 +00:00
ExceptionVerification.ExpectedWarns(1);
}
[Test]
public void should_skip_if_full_season_download()
{
_remoteEpisode.Report.Title = "30 Rock - Season 1";
_remoteEpisode.ParsedEpisodeInfo.FullSeason = true;
Mocker.Resolve<PneumaticClient>().DownloadNzb(_remoteEpisode).Should().BeFalse();
2012-08-30 00:20:48 +00:00
}
[Test]
public void should_replace_illegal_characters_in_title()
{
var illegalTitle = "Saturday Night Live - S38E08 - Jeremy Renner/Maroon 5 [SDTV]";
var expectedFilename = Path.Combine(_pneumaticFolder, "Saturday Night Live - S38E08 - Jeremy Renner+Maroon 5 [SDTV].nzb");
_remoteEpisode.Report.Title = illegalTitle;
Subject.DownloadNzb(_remoteEpisode).Should().BeTrue();
2013-04-10 23:41:45 +00:00
Mocker.GetMock<IHttpProvider>().Verify(c => c.DownloadFile(It.IsAny<string>(), expectedFilename), Times.Once());
}
2012-08-30 00:20:48 +00:00
}
}