mirror of https://github.com/Sonarr/Sonarr
293 lines
11 KiB
C#
293 lines
11 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.Model;
|
|
using NzbDrone.Core.Providers;
|
|
using NzbDrone.Core.Providers.Core;
|
|
using NzbDrone.Core.Providers.DownloadClients;
|
|
using NzbDrone.Core.Repository;
|
|
using NzbDrone.Core.Repository.Quality;
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
// ReSharper disable InconsistentNaming
|
|
|
|
namespace NzbDrone.Core.Test.ProviderTests
|
|
{
|
|
[TestFixture]
|
|
public class DownloadProviderFixture : CoreTest
|
|
{
|
|
public static object[] SabNamingCases =
|
|
{
|
|
new object[] { 1, new[] { 2 }, "My Episode Title", QualityTypes.DVD, false, "My Series Name - 1x02 - My Episode Title [DVD]" },
|
|
new object[] { 1, new[] { 2 }, "My Episode Title", QualityTypes.DVD, true, "My Series Name - 1x02 - My Episode Title [DVD] [Proper]" },
|
|
new object[] { 1, new[] { 2 }, "", QualityTypes.DVD, true, "My Series Name - 1x02 - [DVD] [Proper]" },
|
|
new object[] { 1, new[] { 2, 4 }, "My Episode Title", QualityTypes.HDTV, false, "My Series Name - 1x02-1x04 - My Episode Title [HDTV]" },
|
|
new object[] { 1, new[] { 2, 4 }, "My Episode Title", QualityTypes.HDTV, true, "My Series Name - 1x02-1x04 - My Episode Title [HDTV] [Proper]" },
|
|
new object[] { 1, new[] { 2, 4 }, "", QualityTypes.HDTV, true, "My Series Name - 1x02-1x04 - [HDTV] [Proper]" },
|
|
};
|
|
|
|
private void SetDownloadClient(DownloadClientType clientType)
|
|
{
|
|
Mocker.GetMock<ConfigProvider>()
|
|
.Setup(c => c.DownloadClient)
|
|
.Returns(clientType);
|
|
}
|
|
|
|
private EpisodeParseResult SetupParseResult()
|
|
{
|
|
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().ToList();
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Setup(c => c.GetEpisodesByParseResult(It.IsAny<EpisodeParseResult>())).Returns(episodes);
|
|
|
|
return Builder<EpisodeParseResult>.CreateNew()
|
|
.With(c => c.Quality = new QualityModel(QualityTypes.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 WithSuccessfullAdd()
|
|
{
|
|
Mocker.GetMock<SabProvider>()
|
|
.Setup(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), It.IsAny<bool>()))
|
|
.Returns(true);
|
|
|
|
Mocker.GetMock<BlackholeProvider>()
|
|
.Setup(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), It.IsAny<bool>()))
|
|
.Returns(true);
|
|
}
|
|
|
|
private void WithFailedAdd()
|
|
{
|
|
Mocker.GetMock<SabProvider>()
|
|
.Setup(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), false))
|
|
.Returns(false);
|
|
|
|
Mocker.GetMock<BlackholeProvider>()
|
|
.Setup(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), false))
|
|
.Returns(false);
|
|
}
|
|
|
|
[Test]
|
|
public void Download_report_should_send_to_sab_add_to_history_mark_as_grabbed()
|
|
{
|
|
WithSuccessfullAdd();
|
|
SetDownloadClient(DownloadClientType.Sabnzbd);
|
|
|
|
var parseResult = SetupParseResult();
|
|
|
|
//Act
|
|
Mocker.Resolve<DownloadProvider>().DownloadReport(parseResult);
|
|
|
|
|
|
//Assert
|
|
Mocker.GetMock<SabProvider>()
|
|
.Verify(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), true), Times.Once());
|
|
|
|
Mocker.GetMock<BlackholeProvider>()
|
|
.Verify(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), true), Times.Never());
|
|
|
|
Mocker.GetMock<HistoryProvider>()
|
|
.Verify(s => s.Add(It.Is<History>(h => h.EpisodeId == 12 && h.SeriesId == 5)), Times.Once());
|
|
|
|
Mocker.GetMock<HistoryProvider>()
|
|
.Verify(s => s.Add(It.Is<History>(h => h.EpisodeId == 99 && h.SeriesId == 5)), Times.Once());
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Verify(c => c.MarkEpisodeAsFetched(12));
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Verify(c => c.MarkEpisodeAsFetched(99));
|
|
|
|
Mocker.GetMock<ExternalNotificationProvider>()
|
|
.Verify(c => c.OnGrab(It.IsAny<string>()));
|
|
}
|
|
|
|
[Test]
|
|
public void should_download_nzb_to_blackhole_add_to_history_mark_as_grabbed()
|
|
{
|
|
WithSuccessfullAdd();
|
|
SetDownloadClient(DownloadClientType.Blackhole);
|
|
|
|
var parseResult = SetupParseResult();
|
|
|
|
//Act
|
|
Mocker.Resolve<DownloadProvider>().DownloadReport(parseResult);
|
|
|
|
|
|
//Assert
|
|
Mocker.GetMock<SabProvider>()
|
|
.Verify(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), true), Times.Never());
|
|
|
|
Mocker.GetMock<BlackholeProvider>()
|
|
.Verify(s => s.DownloadNzb(It.IsAny<String>(), It.IsAny<String>(), true), Times.Once());
|
|
|
|
Mocker.GetMock<HistoryProvider>()
|
|
.Verify(s => s.Add(It.Is<History>(h => h.EpisodeId == 12 && h.SeriesId == 5)), Times.Once());
|
|
|
|
Mocker.GetMock<HistoryProvider>()
|
|
.Verify(s => s.Add(It.Is<History>(h => h.EpisodeId == 99 && h.SeriesId == 5)), Times.Once());
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Verify(c => c.MarkEpisodeAsFetched(12));
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Verify(c => c.MarkEpisodeAsFetched(99));
|
|
|
|
Mocker.GetMock<ExternalNotificationProvider>()
|
|
.Verify(c => c.OnGrab(It.IsAny<string>()));
|
|
}
|
|
|
|
[TestCase(DownloadClientType.Sabnzbd)]
|
|
[TestCase(DownloadClientType.Blackhole)]
|
|
public void Download_report_should_not_add_to_history_mark_as_grabbed_if_add_fails(DownloadClientType clientType)
|
|
{
|
|
WithFailedAdd();
|
|
SetDownloadClient(clientType);
|
|
|
|
var parseResult = SetupParseResult();
|
|
|
|
//Act
|
|
Mocker.Resolve<DownloadProvider>().DownloadReport(parseResult);
|
|
|
|
Mocker.GetMock<HistoryProvider>()
|
|
.Verify(s => s.Add(It.IsAny<History>()), Times.Never());
|
|
|
|
|
|
Mocker.GetMock<EpisodeProvider>()
|
|
.Verify(c => c.MarkEpisodeAsFetched(It.IsAny<int>()), Times.Never());
|
|
|
|
Mocker.GetMock<ExternalNotificationProvider>()
|
|
.Verify(c => c.OnGrab(It.IsAny<String>()), Times.Never());
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_sab_as_active_client()
|
|
{
|
|
SetDownloadClient(DownloadClientType.Sabnzbd);
|
|
Mocker.Resolve<DownloadProvider>().GetActiveDownloadClient().Should().BeAssignableTo<SabProvider>();
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_blackhole_as_active_client()
|
|
{
|
|
SetDownloadClient(DownloadClientType.Blackhole);
|
|
Mocker.Resolve<DownloadProvider>().GetActiveDownloadClient().Should().BeAssignableTo<BlackholeProvider>();
|
|
}
|
|
|
|
[Test, TestCaseSource("SabNamingCases")]
|
|
public void create_proper_sab_titles(int seasons, int[] episodes, string title, QualityTypes quality, bool proper, string expected)
|
|
{
|
|
var series = Builder<Series>.CreateNew()
|
|
.With(c => c.Title = "My Series Name")
|
|
.Build();
|
|
|
|
var fakeEpisodes = new List<Episode>();
|
|
|
|
foreach(var episode in episodes)
|
|
fakeEpisodes.Add(Builder<Episode>
|
|
.CreateNew()
|
|
.With(e => e.EpisodeNumber = episode)
|
|
.With(e => e.Title = title)
|
|
.Build());
|
|
|
|
var parsResult = new EpisodeParseResult()
|
|
{
|
|
AirDate = DateTime.Now,
|
|
EpisodeNumbers = episodes.ToList(),
|
|
Quality = new QualityModel(quality, proper),
|
|
SeasonNumber = seasons,
|
|
Series = series,
|
|
EpisodeTitle = title,
|
|
Episodes = fakeEpisodes
|
|
};
|
|
|
|
Mocker.Resolve<DownloadProvider>().GetDownloadTitle(parsResult).Should().Be(expected);
|
|
}
|
|
|
|
[TestCase(true, Result = "My Series Name - Season 1 [Bluray720p] [Proper]")]
|
|
[TestCase(false, Result = "My Series Name - Season 1 [Bluray720p]")]
|
|
public string create_proper_sab_season_title(bool proper)
|
|
{
|
|
var series = Builder<Series>.CreateNew()
|
|
.With(c => c.Title = "My Series Name")
|
|
.Build();
|
|
|
|
var parsResult = new EpisodeParseResult()
|
|
{
|
|
AirDate = DateTime.Now,
|
|
Quality = new QualityModel(QualityTypes.Bluray720p, proper),
|
|
SeasonNumber = 1,
|
|
Series = series,
|
|
EpisodeTitle = "My Episode Title",
|
|
FullSeason = true
|
|
};
|
|
|
|
return Mocker.Resolve<DownloadProvider>().GetDownloadTitle(parsResult);
|
|
}
|
|
|
|
[TestCase(true, Result = "My Series Name - 2011-12-01 - My Episode Title [Bluray720p] [Proper]")]
|
|
[TestCase(false, Result = "My Series Name - 2011-12-01 - My Episode Title [Bluray720p]")]
|
|
public string create_proper_sab_daily_titles(bool proper)
|
|
{
|
|
var series = Builder<Series>.CreateNew()
|
|
.With(c => c.IsDaily = true)
|
|
.With(c => c.Title = "My Series Name")
|
|
.Build();
|
|
|
|
var episode = Builder<Episode>.CreateNew()
|
|
.With(e => e.Title = "My Episode Title")
|
|
.Build();
|
|
|
|
var parsResult = new EpisodeParseResult
|
|
{
|
|
AirDate = new DateTime(2011, 12, 1),
|
|
Quality = new QualityModel(QualityTypes.Bluray720p, proper),
|
|
Series = series,
|
|
EpisodeTitle = "My Episode Title",
|
|
Episodes = new List<Episode>{ episode }
|
|
};
|
|
|
|
return Mocker.Resolve<DownloadProvider>().GetDownloadTitle(parsResult);
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_repeat_the_same_episode_title()
|
|
{
|
|
var series = Builder<Series>.CreateNew()
|
|
.With(c => c.Title = "My Series Name")
|
|
.Build();
|
|
|
|
var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
|
|
.All()
|
|
.With(e => e.SeasonNumber = 5)
|
|
.TheFirst(1)
|
|
.With(e => e.Title = "My Episode Title (1)")
|
|
.TheLast(1)
|
|
.With(e => e.Title = "My Episode Title (2)")
|
|
.Build();
|
|
|
|
var parsResult = new EpisodeParseResult
|
|
{
|
|
AirDate = DateTime.Now,
|
|
EpisodeNumbers = new List<int>{ 10, 11 },
|
|
Quality = new QualityModel(QualityTypes.HDTV, false),
|
|
SeasonNumber = 35,
|
|
Series = series,
|
|
Episodes = fakeEpisodes
|
|
};
|
|
|
|
Mocker.Resolve<DownloadProvider>().GetDownloadTitle(parsResult).Should().Be("My Series Name - 5x01-5x02 - My Episode Title [HDTV]");
|
|
}
|
|
}
|
|
} |