mirror of https://github.com/Radarr/Radarr
180 lines
7.5 KiB
C#
180 lines
7.5 KiB
C#
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Core.Extras.Subtitles;
|
|
using NzbDrone.Core.MediaFiles;
|
|
using NzbDrone.Core.MediaFiles.MovieImport;
|
|
using NzbDrone.Core.Movies;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Test.Common;
|
|
|
|
namespace NzbDrone.Core.Test.Extras.Subtitles
|
|
{
|
|
[TestFixture]
|
|
public class SubtitleServiceFixture : CoreTest<SubtitleService>
|
|
{
|
|
private Movie _movie;
|
|
private MovieFile _movieFile;
|
|
private LocalMovie _localMovie;
|
|
|
|
private string _MovieFolder;
|
|
private string _releaseFolder;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
_MovieFolder = @"C:\Test\Movies\Movie Title".AsOsAgnostic();
|
|
_releaseFolder = @"C:\Test\Unsorted Movies\Movie.Title.2022".AsOsAgnostic();
|
|
|
|
_movie = Builder<Movie>.CreateNew()
|
|
.With(s => s.Path = _MovieFolder)
|
|
.Build();
|
|
|
|
_movieFile = Builder<MovieFile>.CreateNew()
|
|
.With(f => f.Path = Path.Combine(_movie.Path, "Movie Title - 2022.mkv").AsOsAgnostic())
|
|
.With(f => f.RelativePath = @"Movie Title - 2022.mkv".AsOsAgnostic())
|
|
.Build();
|
|
|
|
_localMovie = Builder<LocalMovie>.CreateNew()
|
|
.With(l => l.Movie = _movie)
|
|
.With(l => l.Path = Path.Combine(_releaseFolder, "Movie.Title.2022.mkv").AsOsAgnostic())
|
|
.With(l => l.FileMovieInfo = new ParsedMovieInfo
|
|
{
|
|
MovieTitles = new List<string> { "Movie Title" },
|
|
Year = 2022
|
|
})
|
|
.Build();
|
|
|
|
Mocker.GetMock<IDiskProvider>().Setup(s => s.GetParentFolder(It.IsAny<string>()))
|
|
.Returns((string path) => Directory.GetParent(path).FullName);
|
|
|
|
Mocker.GetMock<IDetectSample>().Setup(s => s.IsSample(It.IsAny<MovieMetadata>(), It.IsAny<string>()))
|
|
.Returns(DetectSampleResult.NotSample);
|
|
}
|
|
|
|
[Test]
|
|
[TestCase("Movie.Title.2022.en.nfo")]
|
|
public void should_not_import_non_subtitle_file(string filePath)
|
|
{
|
|
var files = new List<string> { Path.Combine(_releaseFolder, filePath).AsOsAgnostic() };
|
|
|
|
var results = Subject.ImportFiles(_localMovie, _movieFile, files, true).ToList();
|
|
|
|
results.Count.Should().Be(0);
|
|
}
|
|
|
|
[Test]
|
|
[TestCase("Movie Title - 2022.srt", "Movie Title - 2022.srt")]
|
|
[TestCase("Movie.Title.2022.en.srt", "Movie Title - 2022.en.srt")]
|
|
[TestCase("Movie.Title.2022.english.srt", "Movie Title - 2022.en.srt")]
|
|
[TestCase("Movie Title 2022_en_sdh_forced.srt", "Movie Title - 2022.en.sdh.forced.srt")]
|
|
[TestCase("Movie_Title_2022 en.srt", "Movie Title - 2022.en.srt")]
|
|
[TestCase(@"Subs\Movie.Title.2022\2_en.srt", "Movie Title - 2022.en.srt")]
|
|
[TestCase("sub.srt", "Movie Title - 2022.srt")]
|
|
public void should_import_matching_subtitle_file(string filePath, string expectedOutputPath)
|
|
{
|
|
var files = new List<string> { Path.Combine(_releaseFolder, filePath).AsOsAgnostic() };
|
|
|
|
var results = Subject.ImportFiles(_localMovie, _movieFile, files, true).ToList();
|
|
|
|
results.Count.Should().Be(1);
|
|
|
|
results[0].RelativePath.AsOsAgnostic().PathEquals(expectedOutputPath.AsOsAgnostic()).Should().Be(true);
|
|
}
|
|
|
|
[Test]
|
|
public void should_import_multiple_subtitle_files_per_language()
|
|
{
|
|
var files = new List<string>
|
|
{
|
|
Path.Combine(_releaseFolder, "Movie.Title.2022.en.srt").AsOsAgnostic(),
|
|
Path.Combine(_releaseFolder, "Movie.Title.2022.eng.srt").AsOsAgnostic(),
|
|
Path.Combine(_releaseFolder, "Subs", "Movie_Title_2022_en_forced.srt").AsOsAgnostic(),
|
|
Path.Combine(_releaseFolder, "Subs", "Movie.Title.2022", "2_fr.srt").AsOsAgnostic()
|
|
};
|
|
|
|
var expectedOutputs = new string[]
|
|
{
|
|
"Movie Title - 2022.1.en.srt",
|
|
"Movie Title - 2022.2.en.srt",
|
|
"Movie Title - 2022.en.forced.srt",
|
|
"Movie Title - 2022.fr.srt",
|
|
};
|
|
|
|
var results = Subject.ImportFiles(_localMovie, _movieFile, files, true).ToList();
|
|
|
|
results.Count.Should().Be(expectedOutputs.Length);
|
|
|
|
for (var i = 0; i < expectedOutputs.Length; i++)
|
|
{
|
|
results[i].RelativePath.AsOsAgnostic().PathEquals(expectedOutputs[i].AsOsAgnostic()).Should().Be(true);
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
public void should_import_multiple_subtitle_files_per_language_with_tags()
|
|
{
|
|
var files = new List<string>
|
|
{
|
|
Path.Combine(_releaseFolder, "Movie.Title.2022.en.forced.cc.srt").AsOsAgnostic(),
|
|
Path.Combine(_releaseFolder, "Movie.Title.2022.other.en.forced.cc.srt").AsOsAgnostic(),
|
|
Path.Combine(_releaseFolder, "Movie.Title.2022.en.forced.sdh.srt").AsOsAgnostic(),
|
|
Path.Combine(_releaseFolder, "Movie.Title.2022.en.forced.default.srt").AsOsAgnostic(),
|
|
};
|
|
|
|
var expectedOutputs = new[]
|
|
{
|
|
"Movie Title - 2022.1.en.forced.cc.srt",
|
|
"Movie Title - 2022.2.en.forced.cc.srt",
|
|
"Movie Title - 2022.en.forced.sdh.srt",
|
|
"Movie Title - 2022.en.forced.default.srt"
|
|
};
|
|
|
|
var results = Subject.ImportFiles(_localMovie, _movieFile, files, true).ToList();
|
|
|
|
results.Count.Should().Be(expectedOutputs.Length);
|
|
|
|
for (var i = 0; i < expectedOutputs.Length; i++)
|
|
{
|
|
results[i].RelativePath.AsOsAgnostic().PathEquals(expectedOutputs[i].AsOsAgnostic()).Should().Be(true);
|
|
}
|
|
}
|
|
|
|
[Test]
|
|
[TestCase(@"Subs\2_en.srt", "Movie Title - 2022.en.srt")]
|
|
public void should_import_unmatching_subtitle_file_if_only_episode(string filePath, string expectedOutputPath)
|
|
{
|
|
var subtitleFile = Path.Combine(_releaseFolder, filePath).AsOsAgnostic();
|
|
|
|
var sampleFile = Path.Combine(_movie.Path, "Movie Title - 2022.sample.mkv").AsOsAgnostic();
|
|
|
|
var videoFiles = new string[]
|
|
{
|
|
_localMovie.Path,
|
|
sampleFile
|
|
};
|
|
|
|
Mocker.GetMock<IDiskProvider>().Setup(s => s.GetFiles(It.IsAny<string>(), SearchOption.AllDirectories))
|
|
.Returns(videoFiles);
|
|
|
|
Mocker.GetMock<IDetectSample>().Setup(s => s.IsSample(It.IsAny<MovieMetadata>(), sampleFile))
|
|
.Returns(DetectSampleResult.Sample);
|
|
|
|
var results = Subject.ImportFiles(_localMovie, _movieFile, new List<string> { subtitleFile }, true).ToList();
|
|
|
|
results.Count.Should().Be(1);
|
|
|
|
results[0].RelativePath.AsOsAgnostic().PathEquals(expectedOutputPath.AsOsAgnostic()).Should().Be(true);
|
|
|
|
ExceptionVerification.ExpectedWarns(1);
|
|
}
|
|
}
|
|
}
|