using System.IO; using System.Net; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Common; using NzbDrone.Core.Configuration; using NzbDrone.Core.Download.Clients; using NzbDrone.Core.Parser.Model; using NzbDrone.Core.Test.Framework; using NzbDrone.Test.Common; namespace NzbDrone.Core.Test.Download.DownloadClientTests { [TestFixture] public class BlackholeProviderFixture : CoreTest { private const string _nzbUrl = "http://www.nzbs.com/url"; private const string _title = "some_nzb_title"; private const string _blackHoleFolder = @"d:\nzb\blackhole\"; private const string _nzbPath = @"d:\nzb\blackhole\some_nzb_title.nzb"; private RemoteEpisode _remoteEpisode; [SetUp] public void Setup() { Mocker.GetMock().SetupGet(c => c.BlackholeFolder).Returns(_blackHoleFolder); _remoteEpisode = new RemoteEpisode(); _remoteEpisode.Report = new ReportInfo(); _remoteEpisode.Report.Title = _title; _remoteEpisode.Report.NzbUrl = _nzbUrl; } private void WithExistingFile() { Mocker.GetMock().Setup(c => c.FileExists(_nzbPath)).Returns(true); } private void WithFailedDownload() { Mocker.GetMock().Setup(c => c.DownloadFile(It.IsAny(), It.IsAny())).Throws(new WebException()); } [Test] public void DownloadNzb_should_download_file_if_it_doesnt_exist() { Subject.DownloadNzb(_remoteEpisode).Should().BeTrue(); Mocker.GetMock().Verify(c => c.DownloadFile(_nzbUrl, _nzbPath), Times.Once()); } [Test] public void DownloadNzb_not_download_file_if_it_doesn_exist() { WithExistingFile(); Subject.DownloadNzb(_remoteEpisode).Should().BeTrue(); Mocker.GetMock().Verify(c => c.DownloadFile(It.IsAny(), It.IsAny()), Times.Never()); } [Test] public void should_return_false_on_failed_download() { WithFailedDownload(); Subject.DownloadNzb(_remoteEpisode).Should().BeFalse(); ExceptionVerification.ExpectedWarns(1); } [Test] public void should_replace_illegal_characters_in_title() { var illegalTitle = "Saturday Night Live - S38E08 - Jeremy Renner/Maroon 5 [SDTV]"; var expectedFilename = Path.Combine(_blackHoleFolder, "Saturday Night Live - S38E08 - Jeremy Renner+Maroon 5 [SDTV].nzb"); _remoteEpisode.Report.Title = illegalTitle; Subject.DownloadNzb(_remoteEpisode).Should().BeTrue(); Mocker.GetMock().Verify(c => c.DownloadFile(It.IsAny(), expectedFilename), Times.Once()); } } }