|
|
|
@ -1,4 +1,5 @@
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using FluentAssertions;
|
|
|
|
|
using Moq;
|
|
|
|
|
using NUnit.Framework;
|
|
|
|
@ -7,6 +8,7 @@ using NzbDrone.Core.Download.TrackedDownloads;
|
|
|
|
|
using NzbDrone.Core.History;
|
|
|
|
|
using NzbDrone.Core.Indexers;
|
|
|
|
|
using NzbDrone.Core.Movies;
|
|
|
|
|
using NzbDrone.Core.Movies.Events;
|
|
|
|
|
using NzbDrone.Core.Parser;
|
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
@ -83,5 +85,123 @@ namespace NzbDrone.Core.Test.Download.TrackedDownloads
|
|
|
|
|
trackedDownload.RemoteMovie.Movie.Should().NotBeNull();
|
|
|
|
|
trackedDownload.RemoteMovie.Movie.Id.Should().Be(3);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
public void should_unmap_tracked_download_if_movie_deleted()
|
|
|
|
|
{
|
|
|
|
|
GivenDownloadHistory();
|
|
|
|
|
|
|
|
|
|
var remoteMovie = new RemoteMovie
|
|
|
|
|
{
|
|
|
|
|
Movie = new Movie() { Id = 3 },
|
|
|
|
|
|
|
|
|
|
ParsedMovieInfo = new ParsedMovieInfo()
|
|
|
|
|
{
|
|
|
|
|
MovieTitle = "A Movie",
|
|
|
|
|
Year = 1998
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
|
|
|
.Setup(s => s.Map(It.IsAny<ParsedMovieInfo>(), It.IsAny<string>(), null))
|
|
|
|
|
.Returns(new MappingResult { RemoteMovie = remoteMovie });
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IHistoryService>()
|
|
|
|
|
.Setup(s => s.FindByDownloadId(It.IsAny<string>()))
|
|
|
|
|
.Returns(new List<MovieHistory>());
|
|
|
|
|
|
|
|
|
|
ParseMovieTitle();
|
|
|
|
|
|
|
|
|
|
var client = new DownloadClientDefinition()
|
|
|
|
|
{
|
|
|
|
|
Id = 1,
|
|
|
|
|
Protocol = DownloadProtocol.Torrent
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
var item = new DownloadClientItem()
|
|
|
|
|
{
|
|
|
|
|
Title = "A Movie 1998",
|
|
|
|
|
DownloadId = "12345",
|
|
|
|
|
DownloadClientInfo = new DownloadClientItemClientInfo
|
|
|
|
|
{
|
|
|
|
|
Id = 1,
|
|
|
|
|
Type = "Blackhole",
|
|
|
|
|
Name = "Blackhole Client",
|
|
|
|
|
Protocol = DownloadProtocol.Torrent
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
Subject.TrackDownload(client, item);
|
|
|
|
|
Subject.GetTrackedDownloads().Should().HaveCount(1);
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
|
|
|
.Setup(s => s.Map(It.IsAny<ParsedMovieInfo>(), It.IsAny<string>(), null))
|
|
|
|
|
.Returns(new MappingResult { MappingResultType = MappingResultType.Unknown });
|
|
|
|
|
|
|
|
|
|
Subject.Handle(new MoviesDeletedEvent(new List<Movie> { remoteMovie.Movie }, false, false));
|
|
|
|
|
|
|
|
|
|
var trackedDownloads = Subject.GetTrackedDownloads();
|
|
|
|
|
trackedDownloads.Should().HaveCount(1);
|
|
|
|
|
trackedDownloads.First().RemoteMovie.Should().BeNull();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
public void should_not_throw_when_processing_deleted_movie()
|
|
|
|
|
{
|
|
|
|
|
GivenDownloadHistory();
|
|
|
|
|
|
|
|
|
|
var remoteMovie = new RemoteMovie
|
|
|
|
|
{
|
|
|
|
|
Movie = new Movie() { Id = 3 },
|
|
|
|
|
|
|
|
|
|
ParsedMovieInfo = new ParsedMovieInfo()
|
|
|
|
|
{
|
|
|
|
|
MovieTitle = "A Movie",
|
|
|
|
|
Year = 1998
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
|
|
|
.Setup(s => s.Map(It.IsAny<ParsedMovieInfo>(), It.IsAny<string>(), null))
|
|
|
|
|
.Returns(new MappingResult { MappingResultType = MappingResultType.Unknown });
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IHistoryService>()
|
|
|
|
|
.Setup(s => s.FindByDownloadId(It.IsAny<string>()))
|
|
|
|
|
.Returns(new List<MovieHistory>());
|
|
|
|
|
|
|
|
|
|
ParseMovieTitle();
|
|
|
|
|
|
|
|
|
|
var client = new DownloadClientDefinition()
|
|
|
|
|
{
|
|
|
|
|
Id = 1,
|
|
|
|
|
Protocol = DownloadProtocol.Torrent
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
var item = new DownloadClientItem()
|
|
|
|
|
{
|
|
|
|
|
Title = "A Movie 1998",
|
|
|
|
|
DownloadId = "12345",
|
|
|
|
|
DownloadClientInfo = new DownloadClientItemClientInfo
|
|
|
|
|
{
|
|
|
|
|
Id = 1,
|
|
|
|
|
Type = "Blackhole",
|
|
|
|
|
Name = "Blackhole Client",
|
|
|
|
|
Protocol = DownloadProtocol.Torrent
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
Subject.TrackDownload(client, item);
|
|
|
|
|
Subject.GetTrackedDownloads().Should().HaveCount(1);
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
|
|
|
.Setup(s => s.Map(It.IsAny<ParsedMovieInfo>(), It.IsAny<string>(), null))
|
|
|
|
|
.Returns(new MappingResult { MappingResultType = MappingResultType.Unknown });
|
|
|
|
|
|
|
|
|
|
Subject.Handle(new MoviesDeletedEvent(new List<Movie> { remoteMovie.Movie }, false, false));
|
|
|
|
|
|
|
|
|
|
var trackedDownloads = Subject.GetTrackedDownloads();
|
|
|
|
|
trackedDownloads.Should().HaveCount(1);
|
|
|
|
|
trackedDownloads.First().RemoteMovie.Should().BeNull();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|