|
|
@ -0,0 +1,267 @@
|
|
|
|
|
|
|
|
using System;
|
|
|
|
|
|
|
|
using System.IO;
|
|
|
|
|
|
|
|
using System.Linq;
|
|
|
|
|
|
|
|
using System.Net;
|
|
|
|
|
|
|
|
using FizzWare.NBuilder;
|
|
|
|
|
|
|
|
using FluentAssertions;
|
|
|
|
|
|
|
|
using Moq;
|
|
|
|
|
|
|
|
using NUnit.Framework;
|
|
|
|
|
|
|
|
using NzbDrone.Common;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Model;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Model.Notification;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Providers;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Repository;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Repository.Quality;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Repository.Search;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
|
|
|
|
using NzbDrone.Test.Common.AutoMoq;
|
|
|
|
|
|
|
|
using NzbDrone.Test.Common;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.ProviderTests
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
[TestFixture]
|
|
|
|
|
|
|
|
// ReSharper disable InconsistentNaming
|
|
|
|
|
|
|
|
public class SearchHistoryProviderTest : CoreTest
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
private SearchHistory _searchHistory;
|
|
|
|
|
|
|
|
private Series _series;
|
|
|
|
|
|
|
|
private Episode _episode;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[SetUp]
|
|
|
|
|
|
|
|
public void Setup()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
_series = Builder<Series>.CreateNew()
|
|
|
|
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_episode = Builder<Episode>.CreateNew()
|
|
|
|
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var items = Builder<SearchHistoryItem>.CreateListOfSize(10)
|
|
|
|
|
|
|
|
.Build().ToList();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_searchHistory = Builder<SearchHistory>.CreateNew()
|
|
|
|
|
|
|
|
.With(h => h.EpisodeId = _episode.EpisodeId)
|
|
|
|
|
|
|
|
.With(h => h.SeriesId - _series.SeriesId)
|
|
|
|
|
|
|
|
.With(h => h.SearchHistoryItems = items)
|
|
|
|
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void WithUnsuccessfulSearch()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
foreach(var item in _searchHistory.SearchHistoryItems)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
item.Success = false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void WithSuccessfulSearch()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
foreach(var item in _searchHistory.SearchHistoryItems)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
item.Success = false;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var i = _searchHistory.SearchHistoryItems.Last();
|
|
|
|
|
|
|
|
i.Success = true;
|
|
|
|
|
|
|
|
i.SearchError = ReportRejectionType.None;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void Add_should_add_history_and_history_items()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Db.Fetch<SearchHistory>().Should().HaveCount(1);
|
|
|
|
|
|
|
|
Db.Fetch<SearchHistoryItem>().Should().HaveCount(10);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void Add_should_add_return_id()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.Should().NotBe(0);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void Delete_should_delete_history_and_history_items()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var history = Db.Fetch<SearchHistory>();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Delete(history.First().Id);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Db.Fetch<SearchHistory>().Should().HaveCount(0);
|
|
|
|
|
|
|
|
Db.Fetch<SearchHistoryItem>().Should().HaveCount(0);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void AllSearchHistory_should_return_all_items()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
Db.Insert(_episode);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().AllSearchHistory();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.Count.Should().Be(1);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void AllSearchHistory_should_have_series_title()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
Db.Insert(_episode);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().AllSearchHistory();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.Count.Should().Be(1);
|
|
|
|
|
|
|
|
result.First().SeriesTitle.Should().Be(_series.Title);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void AllSearchHistory_should_have_episode_information()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
Db.Insert(_episode);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().AllSearchHistory();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.Count.Should().Be(1);
|
|
|
|
|
|
|
|
result.First().EpisodeTitle.Should().Be(_episode.Title);
|
|
|
|
|
|
|
|
result.First().EpisodeNumber.Should().Be(_episode.EpisodeNumber);
|
|
|
|
|
|
|
|
result.First().SeasonNumber.Should().Be(_episode.SeasonNumber);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void AllSearchHistory_should_have_totalItems_count()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
Db.Insert(_episode);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().AllSearchHistory();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.Count.Should().Be(1);
|
|
|
|
|
|
|
|
result.First().TotalItems.Should().Be(_searchHistory.SearchHistoryItems.Count);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void AllSearchHistory_should_have_successfulCount_equal_zero_when_all_failed()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
Db.Insert(_episode);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
WithUnsuccessfulSearch();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().AllSearchHistory();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.Count.Should().Be(1);
|
|
|
|
|
|
|
|
result.First().SuccessfulCount.Should().Be(0);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void AllSearchHistory_should_have_successfulCount_equal_one_when_one_was_downloaded()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
Db.Insert(_episode);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
WithSuccessfulSearch();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().AllSearchHistory();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.Count.Should().Be(1);
|
|
|
|
|
|
|
|
result.First().SuccessfulCount.Should().Be(1);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void GetSearchHistory_should_return_searchHistory_with_items()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
Db.Insert(_episode);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
WithSuccessfulSearch();
|
|
|
|
|
|
|
|
var id = Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().GetSearchHistory(id);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.SearchHistoryItems.Should().HaveCount(_searchHistory.SearchHistoryItems.Count);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void GetSearchHistory_should_have_episodeDetails()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
Db.Insert(_episode);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
WithSuccessfulSearch();
|
|
|
|
|
|
|
|
var id = Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().GetSearchHistory(id);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.EpisodeNumber.Should().Be(_episode.EpisodeNumber);
|
|
|
|
|
|
|
|
result.SeasonNumber.Should().Be(_episode.SeasonNumber);
|
|
|
|
|
|
|
|
result.EpisodeTitle.Should().Be(_episode.Title);
|
|
|
|
|
|
|
|
result.AirDate.Should().Be(_episode.AirDate.Value);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void GetSearchHistory_should_not_have_episode_info_if_it_was_a_full_season_search()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_searchHistory.EpisodeId = 0;
|
|
|
|
|
|
|
|
var id = Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Mocker.Resolve<SearchHistoryProvider>().GetSearchHistory(id);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
result.EpisodeNumber.Should().Be(null);
|
|
|
|
|
|
|
|
result.SeasonNumber.Should().Be(_searchHistory.SeasonNumber);
|
|
|
|
|
|
|
|
result.EpisodeTitle.Should().Be(null);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void ForceDownload_should_download_report()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
Db.Insert(_series);
|
|
|
|
|
|
|
|
Db.Insert(_episode);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var reportTitle = String.Format("{0} - S{1:00}E{2:00}", _series.Title, _episode.SeasonNumber, _episode.EpisodeNumber);
|
|
|
|
|
|
|
|
_searchHistory.SearchHistoryItems.First().ReportTitle = reportTitle;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().Add(_searchHistory);
|
|
|
|
|
|
|
|
var items = Db.Fetch<SearchHistoryItem>();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.Resolve<SearchHistoryProvider>().ForceDownload(items.First().Id);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<DownloadProvider>().Verify(v => v.DownloadReport(It.IsAny<EpisodeParseResult>()), Times.Once());
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|