|
|
@ -1,303 +0,0 @@
|
|
|
|
using System;
|
|
|
|
|
|
|
|
using System.Linq;
|
|
|
|
|
|
|
|
using FizzWare.NBuilder;
|
|
|
|
|
|
|
|
using FluentAssertions;
|
|
|
|
|
|
|
|
using Moq;
|
|
|
|
|
|
|
|
using NUnit.Framework;
|
|
|
|
|
|
|
|
using NzbDrone.Common;
|
|
|
|
|
|
|
|
using NzbDrone.Core.MediaFiles;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Parser;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Qualities;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Tv;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Providers;
|
|
|
|
|
|
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.ProviderTests.DiskScanProviderTests
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
public class ImportFileFixture : CoreTest<DiskScanService>
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private long _fileSize = 80.Megabytes();
|
|
|
|
|
|
|
|
private Series _fakeSeries;
|
|
|
|
|
|
|
|
private Episode[] _fakeEpisodes;
|
|
|
|
|
|
|
|
private Episode _fakeEpisode;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[SetUp]
|
|
|
|
|
|
|
|
public void Setup()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
_fakeSeries = Builder<Series>
|
|
|
|
|
|
|
|
.CreateNew()
|
|
|
|
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_fakeEpisode = Builder<Episode>
|
|
|
|
|
|
|
|
.CreateNew()
|
|
|
|
|
|
|
|
.With(c => c.EpisodeFileId = 0)
|
|
|
|
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
|
|
|
|
|
|
|
|
.All()
|
|
|
|
|
|
|
|
.With(c => c.SeasonNumber = 3)
|
|
|
|
|
|
|
|
.With(c => c.EpisodeFileId = 1)
|
|
|
|
|
|
|
|
.With(e => e.EpisodeFile = new EpisodeFile())
|
|
|
|
|
|
|
|
.BuildList().ToArray();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenNewFile();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenVideoDuration(TimeSpan.FromMinutes(20));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenFileSize(_fileSize);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void GivenFileSize(long size)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
_fileSize = size;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IDiskProvider>()
|
|
|
|
|
|
|
|
.Setup(d => d.GetFileSize(It.IsAny<String>()))
|
|
|
|
|
|
|
|
.Returns(size);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void GivenVideoDuration(TimeSpan duration)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
Mocker.GetMock<IVideoFileInfoReader>()
|
|
|
|
|
|
|
|
.Setup(d => d.GetRunTime(It.IsAny<String>()))
|
|
|
|
|
|
|
|
.Returns(duration);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void GivenEpisodes(Episode[] episodes, QualityModel quality)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
foreach (var episode in episodes)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
if (episode.EpisodeFile == null)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
episode.EpisodeFileId = 0;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
else
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
episode.EpisodeFileId = episode.EpisodeFile.Value.Id;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
|
|
|
|
|
|
.Setup(c => c.GetEpisodes(It.IsAny<string>(), It.IsAny<Series>()))
|
|
|
|
|
|
|
|
.Returns(new LocalEpisode
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
Episodes = episodes.ToList(),
|
|
|
|
|
|
|
|
Quality = quality
|
|
|
|
|
|
|
|
});
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void GivenNewFile()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
Mocker.GetMock<IMediaFileService>()
|
|
|
|
|
|
|
|
.Setup(p => p.Exists(It.IsAny<String>()))
|
|
|
|
|
|
|
|
.Returns(false);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void import_new_file_should_succeed()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
GivenEpisodes(new[] { _fakeEpisode }, new QualityModel());
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
VerifyFileImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void import_new_file_with_same_quality_should_succeed()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
_fakeEpisode.EpisodeFile = new EpisodeFile { Quality = new QualityModel(Quality.SDTV) };
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(new[] { _fakeEpisode }, new QualityModel(Quality.SDTV));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
VerifyFileImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void import_new_file_with_better_quality_should_succeed()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
_fakeEpisode.EpisodeFile = new EpisodeFile { Quality = new QualityModel(Quality.SDTV) };
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(new[] { _fakeEpisode }, new QualityModel(Quality.HDTV1080p));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
VerifyFileImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void import_new_file_episode_has_better_quality_should_skip()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
_fakeEpisode.EpisodeFile = new EpisodeFile { Quality = new QualityModel(Quality.HDTV1080p), Id = 1 };
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(new[] { _fakeEpisode }, new QualityModel(Quality.SDTV));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifySkipImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void import_unparsable_file_should_skip()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
|
|
|
|
|
|
.Setup(c => c.GetEpisodes(It.IsAny<string>(), It.IsAny<Series>()))
|
|
|
|
|
|
|
|
.Returns<LocalEpisode>(null);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifySkipImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void import_existing_file_should_skip()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
Mocker.GetMock<IMediaFileService>()
|
|
|
|
|
|
|
|
.Setup(p => p.Exists(It.IsAny<String>()))
|
|
|
|
|
|
|
|
.Returns(true);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifySkipImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void import_file_with_no_episode_in_db_should_skip()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
GivenEpisodes(new Episode[0], new QualityModel());
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifySkipImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void import_new_multi_part_file_episode_with_better_quality_than_existing()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
_fakeEpisodes[0].EpisodeFile = new EpisodeFile();
|
|
|
|
|
|
|
|
_fakeEpisodes[1].EpisodeFile = new EpisodeFile();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_fakeEpisodes[0].EpisodeFile = new EpisodeFile { Quality = new QualityModel(Quality.SDTV) };
|
|
|
|
|
|
|
|
_fakeEpisodes[1].EpisodeFile = new EpisodeFile { Quality = new QualityModel(Quality.SDTV) };
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(_fakeEpisodes, new QualityModel(Quality.HDTV1080p));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifyFileImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void skip_import_new_multi_part_file_episode_existing_has_better_quality()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
_fakeEpisodes[0].EpisodeFile = new EpisodeFile { Quality = new QualityModel(Quality.HDTV1080p), Id = 1 };
|
|
|
|
|
|
|
|
_fakeEpisodes[1].EpisodeFile = new EpisodeFile { Quality = new QualityModel(Quality.HDTV1080p), Id = 1 };
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(_fakeEpisodes, new QualityModel(Quality.SDTV));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifySkipImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void should_skip_if_file_size_is_under_70MB_and_runTime_under_3_minutes()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
GivenFileSize(50.Megabytes());
|
|
|
|
|
|
|
|
GivenVideoDuration(TimeSpan.FromMinutes(1));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(new[] { _fakeEpisode }, new QualityModel(Quality.HDTV1080p));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifySkipImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void should_import_if_file_size_is_under_70MB_but_runTime_over_3_minutes()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
GivenFileSize(50.Megabytes());
|
|
|
|
|
|
|
|
GivenVideoDuration(TimeSpan.FromMinutes(20));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(new[] { _fakeEpisode }, new QualityModel(Quality.HDTV1080p));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifyFileImport(result);
|
|
|
|
|
|
|
|
Mocker.GetMock<IDiskProvider>().Verify(p => p.DeleteFile(It.IsAny<string>()), Times.Never());
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void should_import_if_file_size_is_over_70MB_but_runTime_under_3_minutes()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
GivenFileSize(100.Megabytes());
|
|
|
|
|
|
|
|
GivenVideoDuration(TimeSpan.FromMinutes(1));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(new[] { _fakeEpisode }, new QualityModel(Quality.HDTV1080p));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifyFileImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void should_import_special_even_if_file_size_is_under_70MB_and_runTime_under_3_minutes()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
GivenFileSize(10.Megabytes());
|
|
|
|
|
|
|
|
GivenVideoDuration(TimeSpan.FromMinutes(1));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_fakeEpisode.SeasonNumber = 0;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(new[] { _fakeEpisode }, new QualityModel(Quality.HDTV1080p));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifyFileImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
|
|
|
|
public void should_skip_if_daily_series_with_file_size_is_under_70MB_and_runTime_under_3_minutes()
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
GivenFileSize(10.Megabytes());
|
|
|
|
|
|
|
|
GivenVideoDuration(TimeSpan.FromMinutes(1));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_fakeEpisode.SeasonNumber = 0;
|
|
|
|
|
|
|
|
_fakeSeries.SeriesType = SeriesTypes.Daily;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
GivenEpisodes(new[] { _fakeEpisode }, new QualityModel(Quality.HDTV1080p));
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
var result = Subject.ImportFile(_fakeSeries, "file.ext");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
VerifySkipImport(result);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void VerifyFileImport(EpisodeFile result)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
result.Should().NotBeNull();
|
|
|
|
|
|
|
|
result.SeriesId.Should().Be(_fakeSeries.Id);
|
|
|
|
|
|
|
|
result.Size.Should().Be(_fileSize);
|
|
|
|
|
|
|
|
result.DateAdded.Should().HaveDay(DateTime.UtcNow.Day);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Mocker.GetMock<IMediaFileService>().Verify(c => c.Add(result), Times.Once());
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
private void VerifySkipImport(EpisodeFile result)
|
|
|
|
|
|
|
|
{
|
|
|
|
|
|
|
|
result.Should().BeNull();
|
|
|
|
|
|
|
|
Mocker.GetMock<IMediaFileService>().Verify(p => p.Add(It.IsAny<EpisodeFile>()), Times.Never());
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|