using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Net; using System.Text; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Common; using NzbDrone.Core.Providers.Core; using NzbDrone.Core.Providers.DownloadClients; using NzbDrone.Core.Test.Framework; using NzbDrone.Test.Common; namespace NzbDrone.Core.Test.ProviderTests.DownloadClientTests { [TestFixture] public class PneumaticProviderFixture : CoreTest { private const string nzbUrl = "http://www.nzbs.com/url"; private const string title = "30.Rock.S01E05.hdtv.xvid-LoL"; private const string pneumaticFolder = @"d:\nzb\pneumatic\"; private const string sabDrop = @"d:\unsorted tv\"; private string nzbPath; [SetUp] public void Setup() { nzbPath = pneumaticFolder + title + ".nzb"; Mocker.GetMock().SetupGet(c => c.PneumaticDirectory).Returns(pneumaticFolder); Mocker.GetMock().SetupGet(c => c.SabDropDirectory).Returns(sabDrop); } 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 should_download_file_if_it_doesnt_exist() { Mocker.Resolve().DownloadNzb(nzbUrl, title).Should().BeTrue(); Mocker.GetMock().Verify(c => c.DownloadFile(nzbUrl, nzbPath),Times.Once()); } [Test] public void should_not_download_file_if_it_doesn_exist() { WithExistingFile(); Mocker.Resolve().DownloadNzb(nzbUrl, title).Should().BeTrue(); Mocker.GetMock().Verify(c => c.DownloadFile(It.IsAny(), It.IsAny()), Times.Never()); } [Test] public void should_return_false_on_failed_download() { WithFailedDownload(); Mocker.Resolve().DownloadNzb(nzbUrl, title).Should().BeFalse(); ExceptionVerification.ExpectedWarns(1); } [Test] public void should_skip_if_full_season_download() { Mocker.Resolve().DownloadNzb(nzbUrl, "30 Rock - Season 1").Should().BeFalse(); } [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"); Mocker.Resolve().DownloadNzb(nzbUrl, illegalTitle).Should().BeTrue(); Mocker.GetMock().Verify(c => c.DownloadFile(It.IsAny(), expectedFilename), Times.Once()); } } }