You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
125 lines
4.7 KiB
125 lines
4.7 KiB
using System;
|
|
|
|
using FizzWare.NBuilder;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.Model;
|
|
using NzbDrone.Core.Providers;
|
|
using NzbDrone.Core.Providers.Core;
|
|
using NzbDrone.Core.Repository;
|
|
using NzbDrone.Core.Repository.Quality;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Test.Common.AutoMoq;
|
|
|
|
// ReSharper disable InconsistentNaming
|
|
|
|
namespace NzbDrone.Core.Test.ProviderTests
|
|
{
|
|
[TestFixture]
|
|
public class DownloadProviderTest : CoreTest
|
|
{
|
|
[Test]
|
|
public void Download_report_should_send_to_sab_add_to_history_mark_as_grabbed()
|
|
{
|
|
WithStrictMocker();
|
|
var parseResult = Builder<EpisodeParseResult>.CreateNew()
|
|
.With(c => c.Quality = new Quality(QualityTypes.DVD, false))
|
|
.Build();
|
|
|
|
var episodes = Builder<Episode>.CreateListOfSize(2)
|
|
.TheFirst(1).With(s => s.EpisodeId = 12)
|
|
.TheNext(1).With(s => s.EpisodeId = 99)
|
|
.All().With(s => s.SeriesId = 5)
|
|
.Build();
|
|
|
|
|
|
const string sabTitle = "My fake sab title";
|
|
Mocker.GetMock<SabProvider>()
|
|
.Setup(s => s.IsInQueue(It.IsAny<EpisodeParseResult>()))
|
|
.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 && h.SeriesId == 5)));
|
|
Mocker.GetMock<HistoryProvider>()
|
|
.Setup(s => s.Add(It.Is<History>(h => h.EpisodeId == 99 && h.SeriesId == 5)));
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Setup(c => c.GetEpisodesByParseResult(It.IsAny<EpisodeParseResult>(), false)).Returns(episodes);
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Setup(c => c.MarkEpisodeAsFetched(12));
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Setup(c => c.MarkEpisodeAsFetched(99));
|
|
|
|
Mocker.GetMock<ExternalNotificationProvider>()
|
|
.Setup(c => c.OnGrab(It.IsAny<string>()));
|
|
|
|
Mocker.GetMock<ConfigProvider>()
|
|
.Setup(c => c.DownloadClient)
|
|
.Returns(DownloadClientType.Sabnzbd);
|
|
|
|
Mocker.Resolve<DownloadProvider>().DownloadReport(parseResult);
|
|
|
|
Mocker.VerifyAllMocks();
|
|
}
|
|
|
|
[Test]
|
|
public void Download_report_should_download_nzb_add_to_history_mark_as_grabbed()
|
|
{
|
|
WithStrictMocker();
|
|
var parseResult = Builder<EpisodeParseResult>.CreateNew()
|
|
.With(c => c.Quality = new Quality(QualityTypes.DVD, false))
|
|
.Build();
|
|
|
|
var episodes = Builder<Episode>.CreateListOfSize(2)
|
|
.TheFirst(1).With(s => s.EpisodeId = 12)
|
|
.TheNext(1).With(s => s.EpisodeId = 99)
|
|
.All().With(s => s.SeriesId = 5)
|
|
.Build();
|
|
|
|
|
|
const string sabTitle = "My fake sab title";
|
|
Mocker.GetMock<BlackholeProvider>()
|
|
.Setup(s => s.DownloadNzb(It.IsAny<EpisodeParseResult>(), 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 && h.SeriesId == 5)));
|
|
Mocker.GetMock<HistoryProvider>()
|
|
.Setup(s => s.Add(It.Is<History>(h => h.EpisodeId == 99 && h.SeriesId == 5)));
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Setup(c => c.GetEpisodesByParseResult(It.IsAny<EpisodeParseResult>(), false)).Returns(episodes);
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Setup(c => c.MarkEpisodeAsFetched(12));
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Setup(c => c.MarkEpisodeAsFetched(99));
|
|
|
|
Mocker.GetMock<ExternalNotificationProvider>()
|
|
.Setup(c => c.OnGrab(It.IsAny<string>()));
|
|
|
|
Mocker.GetMock<ConfigProvider>()
|
|
.Setup(c => c.DownloadClient)
|
|
.Returns(DownloadClientType.Blackhole);
|
|
|
|
Mocker.Resolve<DownloadProvider>().DownloadReport(parseResult);
|
|
|
|
Mocker.VerifyAllMocks();
|
|
}
|
|
}
|
|
} |