2014-01-14 03:47:45 +00:00
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using FizzWare.NBuilder;
|
2014-01-14 19:58:53 +00:00
|
|
|
|
using FluentValidation.Results;
|
2014-01-14 03:47:45 +00:00
|
|
|
|
using Moq;
|
2014-09-07 11:44:24 +00:00
|
|
|
|
using NLog;
|
2014-01-14 03:47:45 +00:00
|
|
|
|
using NUnit.Framework;
|
2014-05-11 04:37:07 +00:00
|
|
|
|
using NzbDrone.Common.Http;
|
2014-09-07 11:44:24 +00:00
|
|
|
|
using NzbDrone.Core.Configuration;
|
2014-01-14 03:47:45 +00:00
|
|
|
|
using NzbDrone.Core.Indexers;
|
|
|
|
|
using NzbDrone.Core.IndexerSearch.Definitions;
|
2014-09-07 11:44:24 +00:00
|
|
|
|
using NzbDrone.Core.Parser;
|
2014-01-14 03:47:45 +00:00
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
2014-01-14 19:58:53 +00:00
|
|
|
|
using NzbDrone.Core.ThingiProvider;
|
2014-01-14 03:47:45 +00:00
|
|
|
|
using NzbDrone.Core.Tv;
|
|
|
|
|
using NzbDrone.Test.Common;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.IndexerTests
|
|
|
|
|
{
|
|
|
|
|
[TestFixture]
|
2014-09-07 11:44:24 +00:00
|
|
|
|
public class SeasonSearchFixture : TestBase<TestIndexer>
|
2014-01-14 03:47:45 +00:00
|
|
|
|
{
|
|
|
|
|
private Series _series;
|
|
|
|
|
|
|
|
|
|
[SetUp]
|
|
|
|
|
public void Setup()
|
|
|
|
|
{
|
|
|
|
|
_series = Builder<Series>.CreateNew().Build();
|
|
|
|
|
|
2014-09-12 19:21:03 +00:00
|
|
|
|
Mocker.GetMock<IHttpClient>()
|
2014-09-07 11:44:24 +00:00
|
|
|
|
.Setup(o => o.Execute(It.Is<HttpRequest>(v => v.Method == HttpMethod.GET)))
|
2014-09-12 22:39:42 +00:00
|
|
|
|
.Returns<HttpRequest>(r => new HttpResponse(r, new HttpHeader(), "<xml></xml>"));
|
2014-01-14 03:47:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
private void WithIndexer(bool paging, int resultCount)
|
2014-01-14 03:47:45 +00:00
|
|
|
|
{
|
2014-09-07 11:44:24 +00:00
|
|
|
|
var definition = new IndexerDefinition();
|
|
|
|
|
definition.Name = "Test";
|
|
|
|
|
Subject.Definition = definition;
|
2014-01-14 03:47:45 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
Subject._supportedPageSize = paging ? 100 : 0;
|
2014-01-14 03:47:45 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
var requestGenerator = Mocker.GetMock<IIndexerRequestGenerator>();
|
|
|
|
|
Subject._requestGenerator = requestGenerator.Object;
|
|
|
|
|
|
|
|
|
|
var requests = Builder<IndexerRequest>.CreateListOfSize(paging ? 100 : 1)
|
|
|
|
|
.All()
|
|
|
|
|
.WithConstructor(() => new IndexerRequest("http://my.feed.local/", HttpAccept.Rss))
|
|
|
|
|
.With(v => v.HttpRequest.Method = HttpMethod.GET)
|
|
|
|
|
.Build();
|
2014-01-14 19:58:53 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
requestGenerator.Setup(s => s.GetSearchRequests(It.IsAny<SeasonSearchCriteria>()))
|
|
|
|
|
.Returns(new List<IEnumerable<IndexerRequest>> { requests });
|
2014-01-14 19:58:53 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
var parser = Mocker.GetMock<IParseIndexerResponse>();
|
|
|
|
|
Subject._parser = parser.Object;
|
2014-01-14 19:58:53 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
var results = Builder<ReleaseInfo>.CreateListOfSize(resultCount)
|
|
|
|
|
.Build();
|
2014-01-14 19:58:53 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
parser.Setup(s => s.ParseResponse(It.IsAny<IndexerResponse>()))
|
|
|
|
|
.Returns(results);
|
2014-01-14 03:47:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
public void should_not_use_offset_if_result_count_is_less_than_90()
|
|
|
|
|
{
|
2014-09-07 11:44:24 +00:00
|
|
|
|
WithIndexer(true, 25);
|
2014-01-14 03:47:45 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
Subject.Fetch(new SeasonSearchCriteria { Series = _series, SceneTitles = new List<string>{_series.Title} });
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IHttpClient>().Verify(v => v.Execute(It.IsAny<HttpRequest>()), Times.Once());
|
2014-01-14 03:47:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
public void should_not_use_offset_for_sites_that_do_not_support_it()
|
|
|
|
|
{
|
2014-09-07 11:44:24 +00:00
|
|
|
|
WithIndexer(false, 125);
|
|
|
|
|
|
|
|
|
|
Subject.Fetch(new SeasonSearchCriteria { Series = _series, SceneTitles = new List<string> { _series.Title } });
|
2014-01-14 03:47:45 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
Mocker.GetMock<IHttpClient>().Verify(v => v.Execute(It.IsAny<HttpRequest>()), Times.Once());
|
2014-01-14 03:47:45 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
public void should_not_use_offset_if_its_already_tried_10_times()
|
|
|
|
|
{
|
2014-09-07 11:44:24 +00:00
|
|
|
|
WithIndexer(true, 100);
|
2014-01-14 03:47:45 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
Subject.Fetch(new SeasonSearchCriteria { Series = _series, SceneTitles = new List<string> { _series.Title } });
|
2014-01-14 19:58:53 +00:00
|
|
|
|
|
2014-09-07 11:44:24 +00:00
|
|
|
|
Mocker.GetMock<IHttpClient>().Verify(v => v.Execute(It.IsAny<HttpRequest>()), Times.Exactly(10));
|
2014-01-14 19:58:53 +00:00
|
|
|
|
}
|
|
|
|
|
}
|
2014-01-14 03:47:45 +00:00
|
|
|
|
}
|