mirror of https://github.com/lidarr/Lidarr
406 lines
18 KiB
C#
406 lines
18 KiB
C#
using System.Collections.Generic;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Core.DecisionEngine;
|
|
using NzbDrone.Core.Download;
|
|
using NzbDrone.Core.Download.TrackedDownloads;
|
|
using NzbDrone.Core.History;
|
|
using NzbDrone.Core.MediaFiles;
|
|
using NzbDrone.Core.MediaFiles.TrackImport;
|
|
using NzbDrone.Core.Messaging.Events;
|
|
using NzbDrone.Core.Music;
|
|
using NzbDrone.Core.Parser;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Test.Common;
|
|
|
|
namespace NzbDrone.Core.Test.Download.CompletedDownloadServiceTests
|
|
{
|
|
[TestFixture]
|
|
public class ImportFixture : CoreTest<CompletedDownloadService>
|
|
{
|
|
private TrackedDownload _trackedDownload;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
var completed = Builder<DownloadClientItem>.CreateNew()
|
|
.With(h => h.Status = DownloadItemStatus.Completed)
|
|
.With(h => h.OutputPath = new OsPath(@"C:\DropFolder\MyDownload".AsOsAgnostic()))
|
|
.With(h => h.Title = "Drone.S01E01.HDTV")
|
|
.Build();
|
|
|
|
var remoteAlbum = BuildRemoteAlbum();
|
|
|
|
_trackedDownload = Builder<TrackedDownload>.CreateNew()
|
|
.With(c => c.State = TrackedDownloadState.Downloading)
|
|
.With(c => c.DownloadItem = completed)
|
|
.With(c => c.RemoteAlbum = remoteAlbum)
|
|
.Build();
|
|
|
|
Mocker.GetMock<IDownloadClient>()
|
|
.SetupGet(c => c.Definition)
|
|
.Returns(new DownloadClientDefinition { Id = 1, Name = "testClient" });
|
|
|
|
Mocker.GetMock<IProvideDownloadClient>()
|
|
.Setup(c => c.Get(It.IsAny<int>()))
|
|
.Returns(Mocker.GetMock<IDownloadClient>().Object);
|
|
|
|
Mocker.GetMock<IHistoryService>()
|
|
.Setup(s => s.MostRecentForDownloadId(_trackedDownload.DownloadItem.DownloadId))
|
|
.Returns(new EntityHistory());
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
.Setup(s => s.GetArtist("Drone.S01E01.HDTV"))
|
|
.Returns(remoteAlbum.Artist);
|
|
|
|
Mocker.GetMock<IHistoryService>()
|
|
.Setup(s => s.FindByDownloadId(It.IsAny<string>()))
|
|
.Returns(new List<EntityHistory>());
|
|
|
|
Mocker.GetMock<IProvideImportItemService>()
|
|
.Setup(s => s.ProvideImportItem(It.IsAny<DownloadClientItem>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns<DownloadClientItem, DownloadClientItem>((i, p) => i);
|
|
}
|
|
|
|
private Album CreateAlbum(int id, int trackCount)
|
|
{
|
|
return new Album
|
|
{
|
|
Id = id,
|
|
AlbumReleases = new List<AlbumRelease>
|
|
{
|
|
new AlbumRelease
|
|
{
|
|
Monitored = true,
|
|
TrackCount = trackCount
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
private RemoteAlbum BuildRemoteAlbum()
|
|
{
|
|
return new RemoteAlbum
|
|
{
|
|
Artist = new Artist(),
|
|
Albums = new List<Album> { CreateAlbum(1, 1) }
|
|
};
|
|
}
|
|
|
|
private void GivenABadlyNamedDownload()
|
|
{
|
|
_trackedDownload.DownloadItem.DownloadId = "1234";
|
|
_trackedDownload.DownloadItem.Title = "Droned Pilot"; // Set a badly named download
|
|
Mocker.GetMock<IHistoryService>()
|
|
.Setup(s => s.MostRecentForDownloadId(It.Is<string>(i => i == "1234")))
|
|
.Returns(new EntityHistory() { SourceTitle = "Droned S01E01" });
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
.Setup(s => s.GetArtist(It.IsAny<string>()))
|
|
.Returns((Artist)null);
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
.Setup(s => s.GetArtist("Droned S01E01"))
|
|
.Returns(BuildRemoteAlbum().Artist);
|
|
}
|
|
|
|
private void GivenArtistMatch()
|
|
{
|
|
Mocker.GetMock<IParsingService>()
|
|
.Setup(s => s.GetArtist(It.IsAny<string>()))
|
|
.Returns(_trackedDownload.RemoteAlbum.Artist);
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_mark_as_imported_if_all_files_were_rejected()
|
|
{
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Setup(v => v.ProcessPath(It.IsAny<string>(), It.IsAny<ImportMode>(), It.IsAny<Artist>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns(new List<ImportResult>
|
|
{
|
|
new ImportResult(
|
|
new ImportDecision<LocalTrack>(
|
|
new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() }, new Rejection("Rejected!")), "Test Failure"),
|
|
|
|
new ImportResult(
|
|
new ImportDecision<LocalTrack>(
|
|
new LocalTrack { Path = @"C:\TestPath\Droned.S01E02.mkv".AsOsAgnostic() }, new Rejection("Rejected!")), "Test Failure")
|
|
});
|
|
|
|
Subject.Import(_trackedDownload);
|
|
|
|
Mocker.GetMock<IEventAggregator>()
|
|
.Verify(v => v.PublishEvent<DownloadCompletedEvent>(It.IsAny<DownloadCompletedEvent>()), Times.Never());
|
|
|
|
AssertNotImported();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_mark_as_imported_if_no_tracks_were_parsed()
|
|
{
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Setup(v => v.ProcessPath(It.IsAny<string>(), It.IsAny<ImportMode>(), It.IsAny<Artist>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns(new List<ImportResult>
|
|
{
|
|
new ImportResult(
|
|
new ImportDecision<LocalTrack>(
|
|
new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() }, new Rejection("Rejected!")), "Test Failure"),
|
|
|
|
new ImportResult(
|
|
new ImportDecision<LocalTrack>(
|
|
new LocalTrack { Path = @"C:\TestPath\Droned.S01E02.mkv".AsOsAgnostic() }, new Rejection("Rejected!")), "Test Failure")
|
|
});
|
|
|
|
_trackedDownload.RemoteAlbum.Albums.Clear();
|
|
|
|
Subject.Import(_trackedDownload);
|
|
|
|
AssertNotImported();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_mark_as_imported_if_all_files_were_skipped()
|
|
{
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Setup(v => v.ProcessPath(It.IsAny<string>(), It.IsAny<ImportMode>(), It.IsAny<Artist>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns(new List<ImportResult>
|
|
{
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() }), "Test Failure"),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() }), "Test Failure")
|
|
});
|
|
|
|
Subject.Import(_trackedDownload);
|
|
|
|
AssertNotImported();
|
|
}
|
|
|
|
[Test]
|
|
public void should_mark_as_imported_if_all_tracks_were_imported_but_extra_files_were_not()
|
|
{
|
|
GivenArtistMatch();
|
|
|
|
_trackedDownload.RemoteAlbum.Albums = new List<Album>
|
|
{
|
|
CreateAlbum(1, 3)
|
|
};
|
|
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Setup(v => v.ProcessPath(It.IsAny<string>(), It.IsAny<ImportMode>(), It.IsAny<Artist>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns(new List<ImportResult>
|
|
{
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() })),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() })),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() })),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() }), "Test Failure")
|
|
});
|
|
|
|
Mocker.GetMock<IHistoryService>()
|
|
.Setup(s => s.FindByDownloadId(It.IsAny<string>()))
|
|
.Returns(new List<EntityHistory>());
|
|
|
|
Subject.Import(_trackedDownload);
|
|
|
|
AssertImported();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_mark_as_imported_if_some_tracks_were_not_imported()
|
|
{
|
|
_trackedDownload.RemoteAlbum.Albums = new List<Album>
|
|
{
|
|
CreateAlbum(1, 1),
|
|
CreateAlbum(1, 2),
|
|
CreateAlbum(1, 1)
|
|
};
|
|
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Setup(v => v.ProcessPath(It.IsAny<string>(), It.IsAny<ImportMode>(), It.IsAny<Artist>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns(new List<ImportResult>
|
|
{
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() })),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() })),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() })),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() }), "Test Failure"),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() }), "Test Failure")
|
|
});
|
|
|
|
var history = Builder<EntityHistory>.CreateListOfSize(2)
|
|
.BuildList();
|
|
|
|
Mocker.GetMock<IHistoryService>()
|
|
.Setup(s => s.FindByDownloadId(It.IsAny<string>()))
|
|
.Returns(history);
|
|
|
|
Mocker.GetMock<ITrackedDownloadAlreadyImported>()
|
|
.Setup(s => s.IsImported(_trackedDownload, history))
|
|
.Returns(true);
|
|
|
|
Subject.Import(_trackedDownload);
|
|
|
|
AssertNotImported();
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_mark_as_imported_if_some_of_episodes_were_not_imported_including_history()
|
|
{
|
|
var tracks = Builder<Track>.CreateListOfSize(3).BuildList();
|
|
|
|
var releases = Builder<AlbumRelease>.CreateListOfSize(3).All().With(x => x.Monitored = true).With(x => x.TrackCount = 1).BuildList();
|
|
releases[0].Tracks = new List<Track> { tracks[0] };
|
|
releases[1].Tracks = new List<Track> { tracks[1] };
|
|
releases[2].Tracks = new List<Track> { tracks[2] };
|
|
|
|
var albums = Builder<Album>.CreateListOfSize(3).BuildList();
|
|
|
|
albums[0].AlbumReleases = new List<AlbumRelease> { releases[0] };
|
|
albums[1].AlbumReleases = new List<AlbumRelease> { releases[1] };
|
|
albums[2].AlbumReleases = new List<AlbumRelease> { releases[2] };
|
|
|
|
_trackedDownload.RemoteAlbum.Albums = albums;
|
|
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Setup(v => v.ProcessPath(It.IsAny<string>(), It.IsAny<ImportMode>(), It.IsAny<Artist>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns(new List<ImportResult>
|
|
{
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv" })),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv" }), "Test Failure"),
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv" }), "Test Failure")
|
|
});
|
|
|
|
var history = Builder<EntityHistory>.CreateListOfSize(2)
|
|
.BuildList();
|
|
|
|
Mocker.GetMock<IHistoryService>()
|
|
.Setup(s => s.FindByDownloadId(It.IsAny<string>()))
|
|
.Returns(history);
|
|
|
|
Mocker.GetMock<ITrackedDownloadAlreadyImported>()
|
|
.Setup(s => s.IsImported(It.IsAny<TrackedDownload>(), It.IsAny<List<EntityHistory>>()))
|
|
.Returns(false);
|
|
|
|
Subject.Import(_trackedDownload);
|
|
|
|
AssertNotImported();
|
|
}
|
|
|
|
[Test]
|
|
public void should_mark_as_imported_if_all_tracks_were_imported()
|
|
{
|
|
_trackedDownload.RemoteAlbum.Albums = new List<Album>
|
|
{
|
|
CreateAlbum(1, 2)
|
|
};
|
|
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Setup(v => v.ProcessPath(It.IsAny<string>(), It.IsAny<ImportMode>(), It.IsAny<Artist>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns(new List<ImportResult>
|
|
{
|
|
new ImportResult(
|
|
new ImportDecision<LocalTrack>(
|
|
new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() })),
|
|
|
|
new ImportResult(
|
|
new ImportDecision<LocalTrack>(
|
|
new LocalTrack { Path = @"C:\TestPath\Droned.S01E02.mkv".AsOsAgnostic() }))
|
|
});
|
|
|
|
Subject.Import(_trackedDownload);
|
|
|
|
AssertImported();
|
|
}
|
|
|
|
[Test]
|
|
public void should_mark_as_imported_if_all_episodes_were_imported_including_history()
|
|
{
|
|
var track1 = new Track { Id = 1 };
|
|
var track2 = new Track { Id = 2 };
|
|
|
|
var releases = Builder<AlbumRelease>.CreateListOfSize(2).All().With(x => x.Monitored = true).With(x => x.TrackCount = 1).BuildList();
|
|
releases[0].Tracks = new List<Track> { track1 };
|
|
releases[1].Tracks = new List<Track> { track2 };
|
|
|
|
var albums = Builder<Album>.CreateListOfSize(2).BuildList();
|
|
|
|
albums[0].AlbumReleases = new List<AlbumRelease> { releases[0] };
|
|
albums[1].AlbumReleases = new List<AlbumRelease> { releases[1] };
|
|
|
|
_trackedDownload.RemoteAlbum.Albums = albums;
|
|
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Setup(v => v.ProcessPath(It.IsAny<string>(), It.IsAny<ImportMode>(), It.IsAny<Artist>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns(new List<ImportResult>
|
|
{
|
|
new ImportResult(
|
|
new ImportDecision<LocalTrack>(
|
|
new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv", Tracks = new List<Track> { track1 } })),
|
|
|
|
new ImportResult(
|
|
new ImportDecision<LocalTrack>(
|
|
new LocalTrack { Path = @"C:\TestPath\Droned.S01E02.mkv", Tracks = new List<Track> { track2 } }), "Test Failure")
|
|
});
|
|
|
|
var history = Builder<EntityHistory>.CreateListOfSize(2)
|
|
.All()
|
|
.With(x => x.EventType = EntityHistoryEventType.TrackFileImported)
|
|
.With(x => x.ArtistId = 1)
|
|
.BuildList();
|
|
|
|
Mocker.GetMock<IHistoryService>()
|
|
.Setup(s => s.FindByDownloadId(It.IsAny<string>()))
|
|
.Returns(history);
|
|
|
|
Mocker.GetMock<ITrackedDownloadAlreadyImported>()
|
|
.Setup(s => s.IsImported(It.IsAny<TrackedDownload>(), It.IsAny<List<EntityHistory>>()))
|
|
.Returns(true);
|
|
|
|
Subject.Import(_trackedDownload);
|
|
|
|
AssertImported();
|
|
}
|
|
|
|
[Test]
|
|
public void should_mark_as_imported_if_the_download_can_be_tracked_using_the_source_seriesid()
|
|
{
|
|
GivenABadlyNamedDownload();
|
|
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Setup(v => v.ProcessPath(It.IsAny<string>(), It.IsAny<ImportMode>(), It.IsAny<Artist>(), It.IsAny<DownloadClientItem>()))
|
|
.Returns(new List<ImportResult>
|
|
{
|
|
new ImportResult(new ImportDecision<LocalTrack>(new LocalTrack { Path = @"C:\TestPath\Droned.S01E01.mkv".AsOsAgnostic() }))
|
|
});
|
|
|
|
Mocker.GetMock<IArtistService>()
|
|
.Setup(v => v.GetArtist(It.IsAny<int>()))
|
|
.Returns(BuildRemoteAlbum().Artist);
|
|
|
|
Subject.Import(_trackedDownload);
|
|
|
|
AssertImported();
|
|
}
|
|
|
|
private void AssertNotImported()
|
|
{
|
|
Mocker.GetMock<IEventAggregator>()
|
|
.Verify(v => v.PublishEvent(It.IsAny<DownloadCompletedEvent>()), Times.Never());
|
|
|
|
_trackedDownload.State.Should().Be(TrackedDownloadState.ImportFailed);
|
|
}
|
|
|
|
private void AssertImported()
|
|
{
|
|
Mocker.GetMock<IDownloadedTracksImportService>()
|
|
.Verify(v => v.ProcessPath(_trackedDownload.DownloadItem.OutputPath.FullPath, ImportMode.Auto, _trackedDownload.RemoteAlbum.Artist, _trackedDownload.DownloadItem), Times.Once());
|
|
|
|
Mocker.GetMock<IEventAggregator>()
|
|
.Verify(v => v.PublishEvent(It.IsAny<DownloadCompletedEvent>()), Times.Once());
|
|
|
|
_trackedDownload.State.Should().Be(TrackedDownloadState.Imported);
|
|
}
|
|
}
|
|
}
|