Sonarr/NzbDrone.Core.Test/DownloadProviderTest.cs

using System;
using System.Collections.Generic;
using System.Text;
using AutoMoq;
using FizzWare.NBuilder;
using Gallio.Framework;
using MbUnit.Framework;
using MbUnit.Framework.ContractVerifiers;
using Moq;
using NzbDrone.Core.Model;
using NzbDrone.Core.Providers;
using NzbDrone.Core.Repository;

// ReSharper disable InconsistentNaming
namespace NzbDrone.Core.Test
{
    [TestFixture]
    public class DownloadProviderTest
    {
        [Test]
        public void Download_report_should_send_to_sab_and_add_to_history()
        {
            var mocker = new AutoMoqer(MockBehavior.Strict);
            var parseResult = Builder<EpisodeParseResult>.CreateNew()
                .With(e => e.Episodes = Builder<Episode>.CreateListOfSize(2)
                    .WhereTheFirst(1).Has(s => s.EpisodeId = 12)
                    .AndTheNext(1).Has(s => s.EpisodeId = 99)
                    .Build())
                    .Build();

            const string sabTitle = "My fake sab title";
            mocker.GetMock<SabProvider>()
                .Setup(s => s.IsInQueue(It.IsAny<String>()))
                .Returns(false);

            mocker.GetMock<SabProvider>()
                .Setup(s => s.AddByUrl(parseResult.NzbUrl, sabTitle))
                .Returns(true);

            mocker.GetMock<SabProvider>()
               .Setup(s => s.GetSabTitle(parseResult))
               .Returns(sabTitle);

            mocker.GetMock<HistoryProvider>()
                .Setup(s => s.Add(It.Is<History>(h=>h.EpisodeId == 12)));
            mocker.GetMock<HistoryProvider>()
    .Setup(s => s.Add(It.Is<History>(h => h.EpisodeId == 99)));

            mocker.Resolve<DownloadProvider>().DownloadReport(parseResult);

            
            mocker.VerifyAllMocks();
        }
    }
}