mirror of https://github.com/lidarr/Lidarr
218 lines
7.4 KiB
C#
218 lines
7.4 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Net;
|
|
using System.ServiceModel.Syndication;
|
|
using System.Text;
|
|
using AutoMoq;
|
|
using FizzWare.NBuilder;
|
|
using Gallio.Framework;
|
|
using MbUnit.Framework;
|
|
using MbUnit.Framework.ContractVerifiers;
|
|
using Moq;
|
|
using NzbDrone.Core.Providers;
|
|
using NzbDrone.Core.Providers.Core;
|
|
using NzbDrone.Core.Providers.Indexer;
|
|
using NzbDrone.Core.Repository;
|
|
using NzbDrone.Core.Repository.Quality;
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
namespace NzbDrone.Core.Test
|
|
{
|
|
[TestFixture]
|
|
// ReSharper disable InconsistentNaming
|
|
public class IndexerTests : TestBase
|
|
{
|
|
[Test]
|
|
[Row("nzbsorg.xml", 0)]
|
|
[Row("nzbsrus.xml", 6)]
|
|
[Row("newzbin.xml", 1)]
|
|
[Row("nzbmatrix.xml", 1)]
|
|
public void parse_feed_xml(string fileName, int warns)
|
|
{
|
|
var mocker = new AutoMoqer();
|
|
|
|
mocker.GetMock<HttpProvider>()
|
|
.Setup(h => h.DownloadStream(It.IsAny<String>(), It.IsAny<NetworkCredential>()))
|
|
.Returns(File.OpenRead(".\\Files\\Rss\\" + fileName));
|
|
|
|
var fakeSettings = Builder<IndexerSetting>.CreateNew().Build();
|
|
mocker.GetMock<IndexerProvider>()
|
|
.Setup(c => c.GetSettings(It.IsAny<Type>()))
|
|
.Returns(fakeSettings);
|
|
|
|
var mockIndexer = mocker.Resolve<MockIndexer>();
|
|
var parseResults = mockIndexer.FetchRss();
|
|
|
|
foreach (var episodeParseResult in parseResults)
|
|
{
|
|
var Uri = new Uri(episodeParseResult.NzbUrl);
|
|
Assert.DoesNotContain(Uri.PathAndQuery, "//");
|
|
}
|
|
|
|
|
|
Assert.IsNotEmpty(parseResults);
|
|
|
|
Assert.ForAll(parseResults, s => Assert.AreEqual(mockIndexer.Name, s.Indexer));
|
|
Assert.ForAll(parseResults, s => Assert.AreNotEqual("", s.NzbTitle));
|
|
Assert.ForAll(parseResults, s => Assert.AreNotEqual(null, s.NzbTitle));
|
|
|
|
ExceptionVerification.ExcpectedWarns(warns);
|
|
}
|
|
|
|
[Test]
|
|
public void newzbin()
|
|
{
|
|
var mocker = new AutoMoqer();
|
|
|
|
mocker.GetMock<HttpProvider>()
|
|
.Setup(h => h.DownloadStream(It.IsAny<String>(), It.IsAny<NetworkCredential>()))
|
|
.Returns(File.OpenRead(".\\Files\\Rss\\newzbin.xml"));
|
|
|
|
var fakeSettings = Builder<IndexerSetting>.CreateNew().Build();
|
|
mocker.GetMock<IndexerProvider>()
|
|
.Setup(c => c.GetSettings(It.IsAny<Type>()))
|
|
.Returns(fakeSettings);
|
|
|
|
var newzbinProvider = mocker.Resolve<Newzbin>();
|
|
var parseResults = newzbinProvider.FetchRss();
|
|
|
|
foreach (var episodeParseResult in parseResults)
|
|
{
|
|
var Uri = new Uri(episodeParseResult.NzbUrl);
|
|
Assert.DoesNotContain(Uri.PathAndQuery, "//");
|
|
}
|
|
|
|
|
|
Assert.IsNotEmpty(parseResults);
|
|
Assert.ForAll(parseResults, s => Assert.AreEqual(newzbinProvider.Name, s.Indexer));
|
|
Assert.ForAll(parseResults, s => Assert.AreNotEqual("", s.NzbTitle));
|
|
Assert.ForAll(parseResults, s => Assert.AreNotEqual(null, s.NzbTitle));
|
|
|
|
ExceptionVerification.ExcpectedWarns(1);
|
|
}
|
|
|
|
|
|
[Test]
|
|
[Row("Adventure.Inc.S03E19.DVDRip.XviD-OSiTV", 3, 19, QualityTypes.DVD)]
|
|
public void custome_parser_partial_success(string title, int season, int episode, QualityTypes quality)
|
|
{
|
|
var mocker = new AutoMoqer();
|
|
|
|
const string summary = "My fake summary";
|
|
|
|
var fakeSettings = Builder<IndexerSetting>.CreateNew().Build();
|
|
mocker.GetMock<IndexerProvider>()
|
|
.Setup(c => c.GetSettings(It.IsAny<Type>()))
|
|
.Returns(fakeSettings);
|
|
|
|
var fakeRssItem = Builder<SyndicationItem>.CreateNew()
|
|
.With(c => c.Title = new TextSyndicationContent(title))
|
|
.With(c => c.Summary = new TextSyndicationContent(summary))
|
|
.Build();
|
|
|
|
var result = mocker.Resolve<CustomParserIndexer>().ParseFeed(fakeRssItem);
|
|
|
|
Assert.IsNotNull(result);
|
|
Assert.AreEqual(summary, result.EpisodeTitle);
|
|
Assert.AreEqual(season, result.SeasonNumber);
|
|
Assert.AreEqual(episode, result.EpisodeNumbers[0]);
|
|
Assert.AreEqual(quality, result.Quality);
|
|
}
|
|
|
|
|
|
[Test]
|
|
[Row("Adventure.Inc.DVDRip.XviD-OSiTV")]
|
|
public void custome_parser_full_parse(string title)
|
|
{
|
|
var mocker = new AutoMoqer();
|
|
|
|
const string summary = "My fake summary";
|
|
|
|
var fakeSettings = Builder<IndexerSetting>.CreateNew().Build();
|
|
mocker.GetMock<IndexerProvider>()
|
|
.Setup(c => c.GetSettings(It.IsAny<Type>()))
|
|
.Returns(fakeSettings);
|
|
|
|
var fakeRssItem = Builder<SyndicationItem>.CreateNew()
|
|
.With(c => c.Title = new TextSyndicationContent(title))
|
|
.With(c => c.Summary = new TextSyndicationContent(summary))
|
|
.Build();
|
|
|
|
var result = mocker.Resolve<CustomParserIndexer>().ParseFeed(fakeRssItem);
|
|
|
|
Assert.IsNotNull(result);
|
|
Assert.AreEqual(summary, result.EpisodeTitle);
|
|
ExceptionVerification.ExcpectedWarns(1);
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
public void downloadFeed()
|
|
{
|
|
var mocker = new AutoMoqer();
|
|
mocker.SetConstant(new HttpProvider());
|
|
|
|
var fakeSettings = Builder<IndexerSetting>.CreateNew().Build();
|
|
mocker.GetMock<IndexerProvider>()
|
|
.Setup(c => c.GetSettings(It.IsAny<Type>()))
|
|
.Returns(fakeSettings);
|
|
|
|
mocker.Resolve<TestUrlIndexer>().FetchRss();
|
|
|
|
ExceptionVerification.IgnoreWarns();
|
|
}
|
|
|
|
|
|
[Test]
|
|
public void nzbsorg_search_returns_valid_results()
|
|
{
|
|
var mocker = new AutoMoqer();
|
|
|
|
mocker.GetMock<ConfigProvider>()
|
|
.SetupGet(c => c.NzbsOrgUId)
|
|
.Returns("43516");
|
|
|
|
mocker.GetMock<ConfigProvider>()
|
|
.SetupGet(c => c.NzbsOrgHash)
|
|
.Returns("bc8edb4cc49d4ae440775adec5ac001f");
|
|
|
|
|
|
mocker.Resolve<HttpProvider>();
|
|
|
|
var result = mocker.Resolve<NzbsOrg>().FetchEpisode("Simpsons", 21, 23);
|
|
|
|
Assert.IsNotEmpty(result);
|
|
Assert.ForAll(result, r => r.CleanTitle == "simpsons");
|
|
Assert.ForAll(result, r => r.SeasonNumber == 21);
|
|
Assert.ForAll(result, r => r.EpisodeNumbers.Contains(23));
|
|
}
|
|
|
|
|
|
[Test]
|
|
public void nzbsorg_multi_word_search_returns_valid_results()
|
|
{
|
|
var mocker = new AutoMoqer();
|
|
|
|
mocker.GetMock<ConfigProvider>()
|
|
.SetupGet(c => c.NzbsOrgUId)
|
|
.Returns("43516");
|
|
|
|
mocker.GetMock<ConfigProvider>()
|
|
.SetupGet(c => c.NzbsOrgHash)
|
|
.Returns("bc8edb4cc49d4ae440775adec5ac001f");
|
|
|
|
|
|
mocker.Resolve<HttpProvider>();
|
|
|
|
var result = mocker.Resolve<NzbsOrg>().FetchEpisode("Blue Bloods", 1, 19);
|
|
|
|
Assert.IsNotEmpty(result);
|
|
Assert.ForAll(result, r => r.CleanTitle == "bluebloods");
|
|
Assert.ForAll(result, r => r.SeasonNumber == 1);
|
|
Assert.ForAll(result, r => r.EpisodeNumbers.Contains(19));
|
|
}
|
|
}
|
|
}
|