mirror of https://github.com/Radarr/Radarr
92 lines
3.6 KiB
C#
92 lines
3.6 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using FizzWare.NBuilder;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.IndexerSearch;
|
|
using NzbDrone.Core.Indexers;
|
|
using NzbDrone.Core.Model;
|
|
using NzbDrone.Core.Model.Notification;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Core.Tv;
|
|
|
|
namespace NzbDrone.Core.Test.IndexerSearchTests
|
|
{
|
|
public abstract class IndexerSearchTestBase<TSearch> : CoreTest<TSearch>
|
|
where TSearch : IndexerSearchBase
|
|
{
|
|
protected Series _series;
|
|
protected Episode _episode;
|
|
protected ProgressNotification notification = new ProgressNotification("Testing");
|
|
|
|
protected Mock<IndexerBase> _indexer1;
|
|
protected Mock<IndexerBase> _indexer2;
|
|
protected List<IndexerBase> _indexers;
|
|
protected IList<EpisodeParseResult> _parseResults;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
_series = Builder<Series>
|
|
.CreateNew()
|
|
.Build();
|
|
|
|
_episode = Builder<Episode>
|
|
.CreateNew()
|
|
.With(e => e.SeriesId = _series.Id)
|
|
.With(e => e.Series = _series)
|
|
.Build();
|
|
|
|
|
|
_parseResults = Builder<EpisodeParseResult>
|
|
.CreateListOfSize(10)
|
|
.Build();
|
|
|
|
_indexer1 = new Mock<IndexerBase>();
|
|
_indexer2 = new Mock<IndexerBase>();
|
|
_indexers = new List<IndexerBase> { _indexer1.Object, _indexer2.Object };
|
|
|
|
Mocker.GetMock<IIndexerService>()
|
|
.Setup(c => c.GetEnabledIndexers())
|
|
.Returns(_indexers);
|
|
}
|
|
|
|
protected void WithValidIndexers()
|
|
{
|
|
_indexer1.Setup(c => c.FetchEpisode(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(_parseResults);
|
|
_indexer1.Setup(c => c.FetchDailyEpisode(It.IsAny<string>(), It.IsAny<DateTime>()))
|
|
.Returns(_parseResults);
|
|
_indexer1.Setup(c => c.FetchPartialSeason(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(_parseResults);
|
|
|
|
_indexer2.Setup(c => c.FetchEpisode(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(_parseResults);
|
|
_indexer2.Setup(c => c.FetchDailyEpisode(It.IsAny<string>(), It.IsAny<DateTime>()))
|
|
.Returns(_parseResults);
|
|
_indexer2.Setup(c => c.FetchPartialSeason(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Returns(_parseResults);
|
|
}
|
|
|
|
protected void WithBrokenIndexers()
|
|
{
|
|
_indexer1.Setup(c => c.FetchEpisode(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Throws(new Exception());
|
|
_indexer1.Setup(c => c.FetchDailyEpisode(It.IsAny<string>(), It.IsAny<DateTime>()))
|
|
.Throws(new Exception());
|
|
_indexer1.Setup(c => c.FetchPartialSeason(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Throws(new Exception());
|
|
|
|
_indexer2.Setup(c => c.FetchEpisode(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Throws(new Exception());
|
|
_indexer2.Setup(c => c.FetchDailyEpisode(It.IsAny<string>(), It.IsAny<DateTime>()))
|
|
.Throws(new Exception());
|
|
_indexer2.Setup(c => c.FetchPartialSeason(It.IsAny<string>(), It.IsAny<int>(), It.IsAny<int>()))
|
|
.Throws(new Exception());
|
|
|
|
_indexer1.SetupGet(c => c.Name).Returns("Indexer1");
|
|
_indexer1.SetupGet(c => c.Name).Returns("Indexer2");
|
|
}
|
|
}
|
|
}
|