You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
128 lines
4.2 KiB
128 lines
4.2 KiB
using System.Collections.Generic;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.Books;
|
|
using NzbDrone.Core.Download.TrackedDownloads;
|
|
using NzbDrone.Core.History;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
namespace NzbDrone.Core.Test.Download.TrackedDownloads
|
|
{
|
|
[TestFixture]
|
|
public class TrackedDownloadAlreadyImportedFixture : CoreTest<TrackedDownloadAlreadyImported>
|
|
{
|
|
private List<Book> _books;
|
|
private TrackedDownload _trackedDownload;
|
|
private List<History.History> _historyItems;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
_books = new List<Book>();
|
|
|
|
var remoteBook = Builder<RemoteBook>.CreateNew()
|
|
.With(r => r.Books = _books)
|
|
.Build();
|
|
|
|
_trackedDownload = Builder<TrackedDownload>.CreateNew()
|
|
.With(t => t.RemoteBook = remoteBook)
|
|
.Build();
|
|
|
|
_historyItems = new List<History.History>();
|
|
}
|
|
|
|
public void GivenEpisodes(int count)
|
|
{
|
|
_books.AddRange(Builder<Book>.CreateListOfSize(count)
|
|
.BuildList());
|
|
}
|
|
|
|
public void GivenHistoryForEpisode(Book episode, params HistoryEventType[] eventTypes)
|
|
{
|
|
foreach (var eventType in eventTypes)
|
|
{
|
|
_historyItems.Add(
|
|
Builder<History.History>.CreateNew()
|
|
.With(h => h.BookId = episode.Id)
|
|
.With(h => h.EventType = eventType)
|
|
.Build());
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_false_if_there_is_no_history()
|
|
{
|
|
GivenEpisodes(1);
|
|
|
|
Subject.IsImported(_trackedDownload, _historyItems)
|
|
.Should()
|
|
.BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_false_if_single_episode_download_is_not_imported()
|
|
{
|
|
GivenEpisodes(1);
|
|
|
|
GivenHistoryForEpisode(_books[0], HistoryEventType.Grabbed);
|
|
|
|
Subject.IsImported(_trackedDownload, _historyItems)
|
|
.Should()
|
|
.BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_false_if_no_episode_in_multi_episode_download_is_imported()
|
|
{
|
|
GivenEpisodes(2);
|
|
|
|
GivenHistoryForEpisode(_books[0], HistoryEventType.Grabbed);
|
|
GivenHistoryForEpisode(_books[1], HistoryEventType.Grabbed);
|
|
|
|
Subject.IsImported(_trackedDownload, _historyItems)
|
|
.Should()
|
|
.BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_should_return_false_if_only_one_episode_in_multi_episode_download_is_imported()
|
|
{
|
|
GivenEpisodes(2);
|
|
|
|
GivenHistoryForEpisode(_books[0], HistoryEventType.DownloadImported, HistoryEventType.Grabbed);
|
|
GivenHistoryForEpisode(_books[1], HistoryEventType.Grabbed);
|
|
|
|
Subject.IsImported(_trackedDownload, _historyItems)
|
|
.Should()
|
|
.BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_true_if_single_episode_download_is_imported()
|
|
{
|
|
GivenEpisodes(1);
|
|
|
|
GivenHistoryForEpisode(_books[0], HistoryEventType.DownloadImported, HistoryEventType.Grabbed);
|
|
|
|
Subject.IsImported(_trackedDownload, _historyItems)
|
|
.Should()
|
|
.BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_return_true_if_multi_episode_download_is_imported()
|
|
{
|
|
GivenEpisodes(2);
|
|
|
|
GivenHistoryForEpisode(_books[0], HistoryEventType.DownloadImported, HistoryEventType.Grabbed);
|
|
GivenHistoryForEpisode(_books[1], HistoryEventType.DownloadImported, HistoryEventType.Grabbed);
|
|
|
|
Subject.IsImported(_trackedDownload, _historyItems)
|
|
.Should()
|
|
.BeTrue();
|
|
}
|
|
}
|
|
}
|