mirror of https://github.com/Radarr/Radarr
121 lines
3.6 KiB
C#
121 lines
3.6 KiB
C#
using System.Linq;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.Indexers.Newznab;
|
|
using NzbDrone.Core.IndexerSearch.Definitions;
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
namespace NzbDrone.Core.Test.IndexerTests.NewznabTests
|
|
{
|
|
public class NewznabRequestGeneratorFixture : CoreTest<NewznabRequestGenerator>
|
|
{
|
|
private MovieSearchCriteria _movieSearchCriteria;
|
|
private NewznabCapabilities _capabilities;
|
|
|
|
[SetUp]
|
|
public void SetUp()
|
|
{
|
|
Subject.Settings = new NewznabSettings()
|
|
{
|
|
BaseUrl = "http://127.0.0.1:1234/",
|
|
Categories = new [] { 1, 2 },
|
|
AnimeCategories = new [] { 3, 4 },
|
|
ApiKey = "abcd",
|
|
};
|
|
|
|
_movieSearchCriteria = new MovieSearchCriteria
|
|
{
|
|
Movie = new Movies.Movie { ImdbId = "tt0076759", Title = "Star Wars", Year = 1977 }
|
|
};
|
|
|
|
_capabilities = new NewznabCapabilities();
|
|
|
|
Mocker.GetMock<INewznabCapabilitiesProvider>()
|
|
.Setup(v => v.GetCapabilities(It.IsAny<NewznabSettings>()))
|
|
.Returns(_capabilities);
|
|
}
|
|
|
|
[Test]
|
|
public void should_use_all_categories_for_feed()
|
|
{
|
|
var results = Subject.GetRecentRequests();
|
|
|
|
results.GetAllTiers().Should().HaveCount(1);
|
|
|
|
var page = results.GetAllTiers().First().First();
|
|
|
|
page.Url.Query.Should().Contain("&cat=1,2,3,4&");
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_have_duplicate_categories()
|
|
{
|
|
Subject.Settings.Categories = new[] { 1, 2, 3 };
|
|
|
|
var results = Subject.GetRecentRequests();
|
|
|
|
results.GetAllTiers().Should().HaveCount(1);
|
|
|
|
var page = results.GetAllTiers().First().First();
|
|
|
|
page.Url.FullUri.Should().Contain("&cat=1,2,3,4&");
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_subsequent_pages()
|
|
{
|
|
var results = Subject.GetSearchRequests(_movieSearchCriteria);
|
|
|
|
results.GetAllTiers().Should().HaveCount(1);
|
|
|
|
var pages = results.GetAllTiers().First().Take(3).ToList();
|
|
|
|
pages[0].Url.FullUri.Should().Contain("&offset=0&");
|
|
pages[1].Url.FullUri.Should().Contain("&offset=100&");
|
|
pages[2].Url.FullUri.Should().Contain("&offset=200&");
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_get_unlimited_pages()
|
|
{
|
|
var results = Subject.GetSearchRequests(_movieSearchCriteria);
|
|
|
|
results.GetAllTiers().Should().HaveCount(1);
|
|
|
|
var pages = results.GetAllTiers().First().Take(500).ToList();
|
|
|
|
pages.Count.Should().BeLessThan(500);
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_search_by_imdbid_if_not_supported()
|
|
{
|
|
_capabilities.SupportedMovieSearchParameters = new[] { "q" };
|
|
|
|
var results = Subject.GetSearchRequests(_movieSearchCriteria);
|
|
|
|
results.GetAllTiers().Should().HaveCount(1);
|
|
|
|
var page = results.GetAllTiers().First().First();
|
|
|
|
page.Url.Query.Should().NotContain("imdbid=0076759");
|
|
page.Url.Query.Should().Contain("q=star");
|
|
}
|
|
|
|
[Test]
|
|
public void should_search_by_imdbid_if_supported()
|
|
{
|
|
_capabilities.SupportedMovieSearchParameters = new[] { "q", "imdbid" };
|
|
|
|
var results = Subject.GetSearchRequests(_movieSearchCriteria);
|
|
results.GetTier(0).Should().HaveCount(1);
|
|
|
|
var page = results.GetAllTiers().First().First();
|
|
|
|
page.Url.Query.Should().Contain("imdbid=0076759");
|
|
}
|
|
|
|
}
|
|
}
|