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

using System;
using System.Collections.Generic;
using System.Linq;
using FizzWare.NBuilder;
using Moq;
using NUnit.Framework;
using NzbDrone.Core.Download;
using NzbDrone.Core.Model;
using NzbDrone.Core.Qualities;
using NzbDrone.Core.Test.Framework;
using NzbDrone.Core.Tv;

namespace NzbDrone.Core.Test.Download
{
    [TestFixture]
    public class DownloadServiceFixture : CoreTest<DownloadService>
    {
        private EpisodeParseResult _parseResult;

        [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<EpisodeParseResult>.CreateNew()
                   .With(c => c.Quality = new QualityModel(Quality.DVD, false))
                   .With(c => c.Series = Builder<Series>.CreateNew().Build())
                   .With(c => c.EpisodeNumbers = new List<int> { 2 })
                   .With(c => c.Episodes = episodes)
                   .Build();
        }

        private void WithSuccessfulAdd()
        {
            Mocker.GetMock<IDownloadClient>()
                .Setup(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), It.IsAny<bool>()))
                .Returns(true);
        }

        private void WithFailedAdd()
        {
            Mocker.GetMock<IDownloadClient>()
                .Setup(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), It.IsAny<bool>()))
                .Returns(false);
        }

        [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<String>(), It.IsAny<String>(), true), Times.Once());
        }

        [Test]
        public void Download_report_should_not_publish_on_failed_grab_event()
        {
            WithFailedAdd();

            Subject.DownloadReport(_parseResult);
            VerifyEventNotPublished<EpisodeGrabbedEvent>();
        }



    }
}