Add tests for normalizing IMDb IDs in ReleaseSearchService

pull/1682/head v1.5.0.3390
Bogdan 2 years ago
parent eba5413250
commit 051930455e

@ -0,0 +1,91 @@
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using FluentAssertions;
using Moq;
using NUnit.Framework;
using NzbDrone.Core.Indexers;
using NzbDrone.Core.IndexerSearch;
using NzbDrone.Core.IndexerSearch.Definitions;
using NzbDrone.Core.Test.Framework;
namespace NzbDrone.Core.Test.IndexerSearchTests
{
public class ReleaseSearchServiceFixture : CoreTest<ReleaseSearchService>
{
private Mock<IIndexer> _mockIndexer;
[SetUp]
public void SetUp()
{
_mockIndexer = Mocker.GetMock<IIndexer>();
_mockIndexer.SetupGet(s => s.Definition).Returns(new IndexerDefinition { Id = 1 });
_mockIndexer.SetupGet(s => s.SupportsSearch).Returns(true);
Mocker.GetMock<IIndexerFactory>()
.Setup(s => s.Enabled(It.IsAny<bool>()))
.Returns(new List<IIndexer> { _mockIndexer.Object });
}
private List<SearchCriteriaBase> WatchForSearchCriteria()
{
var result = new List<SearchCriteriaBase>();
_mockIndexer.Setup(v => v.Fetch(It.IsAny<MovieSearchCriteria>()))
.Callback<MovieSearchCriteria>(s => result.Add(s))
.Returns(Task.FromResult(new IndexerPageableQueryResult()));
_mockIndexer.Setup(v => v.Fetch(It.IsAny<TvSearchCriteria>()))
.Callback<TvSearchCriteria>(s => result.Add(s))
.Returns(Task.FromResult(new IndexerPageableQueryResult()));
return result;
}
[TestCase("tt0183790", "0183790")]
[TestCase("0183790", "0183790")]
[TestCase("183790", "0183790")]
[TestCase("tt10001870", "10001870")]
[TestCase("10001870", "10001870")]
public void should_normalize_imdbid_movie_search_criteria(string input, string expected)
{
var allCriteria = WatchForSearchCriteria();
var request = new NewznabRequest
{
t = "movie",
imdbid = input
};
Subject.Search(request, new List<int> { 1 }, false);
var criteria = allCriteria.OfType<MovieSearchCriteria>().ToList();
criteria.Count.Should().Be(1);
criteria[0].ImdbId.Should().Be(expected);
}
[TestCase("tt0183790", "0183790")]
[TestCase("0183790", "0183790")]
[TestCase("183790", "0183790")]
[TestCase("tt10001870", "10001870")]
[TestCase("10001870", "10001870")]
public void should_normalize_imdbid_tv_search_criteria(string input, string expected)
{
var allCriteria = WatchForSearchCriteria();
var request = new NewznabRequest
{
t = "tvsearch",
imdbid = input
};
Subject.Search(request, new List<int> { 1 }, false);
var criteria = allCriteria.OfType<TvSearchCriteria>().ToList();
criteria.Count.Should().Be(1);
criteria[0].ImdbId.Should().Be(expected);
}
}
}
Loading…
Cancel
Save