2018-03-14 20:41:36 +00:00
|
|
|
using System;
|
2014-04-04 00:08:51 +00:00
|
|
|
using FizzWare.NBuilder;
|
|
|
|
using FluentAssertions;
|
|
|
|
using Moq;
|
|
|
|
using NUnit.Framework;
|
|
|
|
using NzbDrone.Core.MediaFiles.MediaInfo;
|
2019-12-22 22:08:53 +00:00
|
|
|
using NzbDrone.Core.MediaFiles.MovieImport;
|
|
|
|
using NzbDrone.Core.Movies;
|
2014-04-04 00:08:51 +00:00
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
using NzbDrone.Core.Qualities;
|
|
|
|
using NzbDrone.Core.Test.Framework;
|
2019-07-16 02:27:35 +00:00
|
|
|
using NzbDrone.Test.Common;
|
2014-04-04 00:08:51 +00:00
|
|
|
|
2018-03-14 20:41:36 +00:00
|
|
|
namespace NzbDrone.Core.Test.MediaFiles.MovieImport
|
2014-04-04 00:08:51 +00:00
|
|
|
{
|
|
|
|
[TestFixture]
|
2019-07-16 02:27:35 +00:00
|
|
|
public class DetectSampleFixture : CoreTest<DetectSample>
|
2014-04-04 00:08:51 +00:00
|
|
|
{
|
2022-03-20 15:55:47 +00:00
|
|
|
private MovieMetadata _movie;
|
2018-03-14 20:41:36 +00:00
|
|
|
private LocalMovie _localMovie;
|
2014-04-04 00:08:51 +00:00
|
|
|
|
|
|
|
[SetUp]
|
|
|
|
public void Setup()
|
|
|
|
{
|
2022-03-20 15:55:47 +00:00
|
|
|
_movie = Builder<MovieMetadata>.CreateNew()
|
2014-11-18 01:07:27 +00:00
|
|
|
.With(s => s.Runtime = 30)
|
2014-04-04 00:08:51 +00:00
|
|
|
.Build();
|
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
_localMovie = new LocalMovie
|
|
|
|
{
|
|
|
|
Path = @"C:\Test\30 Rock\30.rock.s01e01.avi",
|
2022-03-20 15:55:47 +00:00
|
|
|
Movie = new Movie { MovieMetadata = _movie },
|
2019-07-16 02:27:35 +00:00
|
|
|
Quality = new QualityModel(Quality.HDTV720p)
|
|
|
|
};
|
2014-04-04 00:08:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenRuntime(int seconds)
|
|
|
|
{
|
|
|
|
Mocker.GetMock<IVideoFileInfoReader>()
|
2015-10-03 17:45:26 +00:00
|
|
|
.Setup(s => s.GetRunTime(It.IsAny<string>()))
|
2014-04-04 00:08:51 +00:00
|
|
|
.Returns(new TimeSpan(0, 0, seconds));
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
2014-04-08 14:12:21 +00:00
|
|
|
public void should_return_false_for_flv()
|
2014-04-04 00:08:51 +00:00
|
|
|
{
|
2018-03-14 20:41:36 +00:00
|
|
|
_localMovie.Path = @"C:\Test\some.show.s01e01.flv";
|
2014-04-04 00:08:51 +00:00
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
ShouldBeNotSample();
|
2014-04-04 00:08:51 +00:00
|
|
|
|
|
|
|
Mocker.GetMock<IVideoFileInfoReader>().Verify(c => c.GetRunTime(It.IsAny<string>()), Times.Never());
|
|
|
|
}
|
|
|
|
|
2014-09-01 18:14:54 +00:00
|
|
|
[Test]
|
|
|
|
public void should_return_false_for_strm()
|
|
|
|
{
|
2018-03-14 20:41:36 +00:00
|
|
|
_localMovie.Path = @"C:\Test\some.show.s01e01.strm";
|
2014-09-01 18:14:54 +00:00
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
ShouldBeNotSample();
|
2014-09-01 18:14:54 +00:00
|
|
|
|
|
|
|
Mocker.GetMock<IVideoFileInfoReader>().Verify(c => c.GetRunTime(It.IsAny<string>()), Times.Never());
|
|
|
|
}
|
|
|
|
|
2020-06-24 01:41:16 +00:00
|
|
|
[Test]
|
|
|
|
public void should_return_false_for_iso()
|
|
|
|
{
|
|
|
|
_localMovie.Path = @"C:\Test\some movie (2000).iso";
|
|
|
|
|
|
|
|
ShouldBeNotSample();
|
|
|
|
|
|
|
|
Mocker.GetMock<IVideoFileInfoReader>().Verify(c => c.GetRunTime(It.IsAny<string>()), Times.Never());
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_return_false_for_img()
|
|
|
|
{
|
|
|
|
_localMovie.Path = @"C:\Test\some movie (2000).img";
|
|
|
|
|
|
|
|
ShouldBeNotSample();
|
|
|
|
|
|
|
|
Mocker.GetMock<IVideoFileInfoReader>().Verify(c => c.GetRunTime(It.IsAny<string>()), Times.Never());
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_return_false_for_m2ts()
|
|
|
|
{
|
|
|
|
_localMovie.Path = @"C:\Test\some movie (2000).m2ts";
|
|
|
|
|
|
|
|
ShouldBeNotSample();
|
|
|
|
|
|
|
|
Mocker.GetMock<IVideoFileInfoReader>().Verify(c => c.GetRunTime(It.IsAny<string>()), Times.Never());
|
|
|
|
}
|
|
|
|
|
2014-04-04 00:08:51 +00:00
|
|
|
[Test]
|
|
|
|
public void should_use_runtime()
|
|
|
|
{
|
|
|
|
GivenRuntime(120);
|
|
|
|
|
2022-03-20 15:55:47 +00:00
|
|
|
Subject.IsSample(_localMovie.Movie.MovieMetadata,
|
2020-07-19 05:45:23 +00:00
|
|
|
_localMovie.Path);
|
2014-04-04 00:08:51 +00:00
|
|
|
|
2015-10-03 17:45:26 +00:00
|
|
|
Mocker.GetMock<IVideoFileInfoReader>().Verify(v => v.GetRunTime(It.IsAny<string>()), Times.Once());
|
2014-04-04 00:08:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
2014-04-08 14:12:21 +00:00
|
|
|
public void should_return_true_if_runtime_is_less_than_minimum()
|
2014-04-04 00:08:51 +00:00
|
|
|
{
|
|
|
|
GivenRuntime(60);
|
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
ShouldBeSample();
|
2014-04-04 00:08:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
2016-06-25 20:50:16 +00:00
|
|
|
public void should_return_false_if_runtime_greater_than_minimum()
|
2014-04-04 00:08:51 +00:00
|
|
|
{
|
2014-11-18 01:07:27 +00:00
|
|
|
GivenRuntime(600);
|
2014-04-04 00:08:51 +00:00
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
ShouldBeNotSample();
|
2014-04-04 00:08:51 +00:00
|
|
|
}
|
|
|
|
|
2016-06-25 20:50:16 +00:00
|
|
|
[Test]
|
|
|
|
public void should_return_false_if_runtime_greater_than_webisode_minimum()
|
|
|
|
{
|
2018-03-14 20:41:36 +00:00
|
|
|
_movie.Runtime = 6;
|
2016-06-25 20:50:16 +00:00
|
|
|
GivenRuntime(299);
|
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
ShouldBeNotSample();
|
2016-06-25 20:50:16 +00:00
|
|
|
}
|
|
|
|
|
2014-04-04 00:08:51 +00:00
|
|
|
[Test]
|
2019-07-16 02:27:35 +00:00
|
|
|
public void should_return_false_if_runtime_greater_than_anime_short_minimum()
|
2014-04-04 00:08:51 +00:00
|
|
|
{
|
2019-07-16 02:27:35 +00:00
|
|
|
_movie.Runtime = 2;
|
|
|
|
GivenRuntime(60);
|
2014-04-04 00:08:51 +00:00
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
ShouldBeNotSample();
|
2014-04-04 00:08:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
2019-07-16 02:27:35 +00:00
|
|
|
public void should_return_true_if_runtime_less_than_anime_short_minimum()
|
2014-04-04 00:08:51 +00:00
|
|
|
{
|
2019-07-16 02:27:35 +00:00
|
|
|
_movie.Runtime = 2;
|
|
|
|
GivenRuntime(10);
|
2014-04-04 00:08:51 +00:00
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
ShouldBeSample();
|
2014-04-04 00:08:51 +00:00
|
|
|
}
|
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
[Test]
|
|
|
|
public void should_return_indeterminate_if_mediainfo_result_is_null()
|
|
|
|
{
|
|
|
|
Mocker.GetMock<IVideoFileInfoReader>()
|
|
|
|
.Setup(s => s.GetRunTime(It.IsAny<string>()))
|
|
|
|
.Returns((TimeSpan?)null);
|
2014-04-11 01:32:19 +00:00
|
|
|
|
2022-03-20 15:55:47 +00:00
|
|
|
Subject.IsSample(_localMovie.Movie.MovieMetadata,
|
2020-07-19 05:45:23 +00:00
|
|
|
_localMovie.Path).Should().Be(DetectSampleResult.Indeterminate);
|
2014-12-12 15:39:51 +00:00
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
ExceptionVerification.ExpectedErrors(1);
|
|
|
|
}
|
2018-03-14 20:41:36 +00:00
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
private void ShouldBeSample()
|
2014-04-04 00:08:51 +00:00
|
|
|
{
|
2022-03-20 15:55:47 +00:00
|
|
|
Subject.IsSample(_localMovie.Movie.MovieMetadata,
|
2020-07-19 05:45:23 +00:00
|
|
|
_localMovie.Path).Should().Be(DetectSampleResult.Sample);
|
2014-04-04 00:08:51 +00:00
|
|
|
}
|
|
|
|
|
2019-07-16 02:27:35 +00:00
|
|
|
private void ShouldBeNotSample()
|
2014-04-04 00:08:51 +00:00
|
|
|
{
|
2022-03-20 15:55:47 +00:00
|
|
|
Subject.IsSample(_localMovie.Movie.MovieMetadata,
|
2020-07-19 05:45:23 +00:00
|
|
|
_localMovie.Path).Should().Be(DetectSampleResult.NotSample);
|
2014-04-04 00:08:51 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|