You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
263 lines
12 KiB
263 lines
12 KiB
using System.Linq;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.MediaFiles.EpisodeImport.Specifications;
|
|
using NzbDrone.Core.Parser;
|
|
using NzbDrone.Core.Parser.Model;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Core.Tv;
|
|
using NzbDrone.Test.Common;
|
|
|
|
namespace NzbDrone.Core.Test.MediaFiles.EpisodeImport.Specifications
|
|
{
|
|
[TestFixture]
|
|
public class MatchesFolderSpecificationFixture : CoreTest<MatchesFolderSpecification>
|
|
{
|
|
private LocalEpisode _localEpisode;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
_localEpisode = Builder<LocalEpisode>.CreateNew()
|
|
.With(l => l.Path = @"C:\Test\Unsorted\Series.Title.S01E01.720p.HDTV-Sonarr\S01E05.mkv".AsOsAgnostic())
|
|
.With(l => l.FileEpisodeInfo =
|
|
Builder<ParsedEpisodeInfo>.CreateNew()
|
|
.With(p => p.EpisodeNumbers = new[] { 5 })
|
|
.With(p => p.SeasonNumber == 1)
|
|
.With(p => p.FullSeason = false)
|
|
.Build())
|
|
.With(l => l.FolderEpisodeInfo =
|
|
Builder<ParsedEpisodeInfo>.CreateNew()
|
|
.With(p => p.EpisodeNumbers = new[] { 1 })
|
|
.With(p => p.SeasonNumber == 1)
|
|
.With(p => p.FullSeason = false)
|
|
.Build())
|
|
.Build();
|
|
|
|
}
|
|
|
|
private void GivenEpisodes(ParsedEpisodeInfo parsedEpisodeInfo, int[] episodeNumbers)
|
|
{
|
|
var seasonNumber = parsedEpisodeInfo.SeasonNumber;
|
|
|
|
var episodes = episodeNumbers.Select(n =>
|
|
Builder<Episode>.CreateNew()
|
|
.With(e => e.Id = seasonNumber * 10 + n)
|
|
.With(e => e.SeasonNumber = seasonNumber)
|
|
.With(e => e.EpisodeNumber = n)
|
|
.Build()
|
|
).ToList();
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
.Setup(s => s.GetEpisodes(parsedEpisodeInfo, It.IsAny<Series>(), true, null))
|
|
.Returns(episodes);
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_accepted_for_existing_file()
|
|
{
|
|
_localEpisode.ExistingFile = true;
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_accepted_if_folder_name_is_not_parseable()
|
|
{
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title\S01E01.mkv".AsOsAgnostic();
|
|
_localEpisode.FolderEpisodeInfo = null;
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_accepted_if_file_name_is_not_parseable()
|
|
{
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01E01\AFDAFD.mkv".AsOsAgnostic();
|
|
_localEpisode.FileEpisodeInfo = null;
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_should_be_accepted_for_full_season()
|
|
{
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01\S01E01.mkv".AsOsAgnostic();
|
|
_localEpisode.FolderEpisodeInfo.EpisodeNumbers = new int[0];
|
|
_localEpisode.FolderEpisodeInfo.FullSeason = true;
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, _localEpisode.FileEpisodeInfo.EpisodeNumbers);
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, new []{ 1, 2, 3, 4, 5 });
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_accepted_if_file_and_folder_have_the_same_episode()
|
|
{
|
|
_localEpisode.FileEpisodeInfo.EpisodeNumbers = new[] { 1 };
|
|
_localEpisode.FolderEpisodeInfo.EpisodeNumbers = new[] { 1 };
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01E01.720p.HDTV-Sonarr\S01E01.mkv".AsOsAgnostic();
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, _localEpisode.FileEpisodeInfo.EpisodeNumbers);
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, _localEpisode.FolderEpisodeInfo.EpisodeNumbers);
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_accepted_if_file_is_one_episode_in_folder()
|
|
{
|
|
_localEpisode.FileEpisodeInfo.EpisodeNumbers = new[] { 1 };
|
|
_localEpisode.FolderEpisodeInfo.EpisodeNumbers = new[] { 1 };
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01E01E02.720p.HDTV-Sonarr\S01E01.mkv".AsOsAgnostic();
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, _localEpisode.FileEpisodeInfo.EpisodeNumbers);
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, _localEpisode.FolderEpisodeInfo.EpisodeNumbers);
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_disregard_subfolder()
|
|
{
|
|
_localEpisode.FileEpisodeInfo.EpisodeNumbers = new[] { 5, 6 };
|
|
_localEpisode.FolderEpisodeInfo.EpisodeNumbers = new[] { 1, 2 };
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01E01E02.720p.HDTV-Sonarr\S01E05E06.mkv".AsOsAgnostic();
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, _localEpisode.FileEpisodeInfo.EpisodeNumbers);
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, _localEpisode.FolderEpisodeInfo.EpisodeNumbers);
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_rejected_if_file_and_folder_do_not_have_same_episode()
|
|
{
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01E01.720p.HDTV-Sonarr\S01E05.mkv".AsOsAgnostic();
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, _localEpisode.FileEpisodeInfo.EpisodeNumbers);
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, _localEpisode.FolderEpisodeInfo.EpisodeNumbers);
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_rejected_if_file_and_folder_do_not_have_the_same_episodes()
|
|
{
|
|
_localEpisode.FileEpisodeInfo.EpisodeNumbers = new[] { 5, 6 };
|
|
_localEpisode.FolderEpisodeInfo.EpisodeNumbers = new[] { 1, 2 };
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01E01E02.720p.HDTV-Sonarr\S01E05E06.mkv".AsOsAgnostic();
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, _localEpisode.FileEpisodeInfo.EpisodeNumbers);
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, _localEpisode.FolderEpisodeInfo.EpisodeNumbers);
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_rejected_if_file_and_folder_do_not_have_episodes_from_the_same_season()
|
|
{
|
|
_localEpisode.FileEpisodeInfo.SeasonNumber = 2;
|
|
_localEpisode.FileEpisodeInfo.EpisodeNumbers = new[] { 1 };
|
|
|
|
_localEpisode.FolderEpisodeInfo.FullSeason = true;
|
|
_localEpisode.FolderEpisodeInfo.SeasonNumber = 1;
|
|
_localEpisode.FolderEpisodeInfo.EpisodeNumbers = new[] { 1, 2 };
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, _localEpisode.FileEpisodeInfo.EpisodeNumbers);
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, _localEpisode.FolderEpisodeInfo.EpisodeNumbers);
|
|
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01.720p.HDTV-Sonarr\S02E01.mkv".AsOsAgnostic();
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_rejected_if_file_and_folder_do_not_have_episodes_from_the_same_partial_season()
|
|
{
|
|
_localEpisode.FileEpisodeInfo.SeasonNumber = 2;
|
|
_localEpisode.FileEpisodeInfo.EpisodeNumbers = new[] { 1 };
|
|
|
|
_localEpisode.FolderEpisodeInfo.SeasonNumber = 1;
|
|
_localEpisode.FolderEpisodeInfo.EpisodeNumbers = new[] { 1, 2 };
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, _localEpisode.FileEpisodeInfo.EpisodeNumbers);
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, _localEpisode.FolderEpisodeInfo.EpisodeNumbers);
|
|
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01.720p.HDTV-Sonarr\S02E01.mkv".AsOsAgnostic();
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeFalse();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_accepted_if_file_and_folder_have_episodes_from_the_same_season()
|
|
{
|
|
_localEpisode.FileEpisodeInfo.SeasonNumber = 1;
|
|
_localEpisode.FileEpisodeInfo.EpisodeNumbers = new[] { 1 };
|
|
|
|
_localEpisode.FolderEpisodeInfo.FullSeason = true;
|
|
_localEpisode.FolderEpisodeInfo.SeasonNumber = 1;
|
|
_localEpisode.FolderEpisodeInfo.EpisodeNumbers = new[] { 1, 2 };
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, _localEpisode.FileEpisodeInfo.EpisodeNumbers);
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, _localEpisode.FolderEpisodeInfo.EpisodeNumbers);
|
|
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01.720p.HDTV-Sonarr\S01E01.mkv".AsOsAgnostic();
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_accepted_if_both_file_and_folder_info_map_to_same_special()
|
|
{
|
|
var title = "Some.Special.S12E00.WEB-DL.1080p-GoodNightTV";
|
|
var actualInfo = Parser.Parser.ParseTitle("Some.Special.S0E100.WEB-DL.1080p-GoodNightTV.mkv");
|
|
|
|
var folderInfo = Parser.Parser.ParseTitle(title);
|
|
var fileInfo = Parser.Parser.ParseTitle(title + ".mkv");
|
|
var localEpisode = new LocalEpisode
|
|
{
|
|
FileEpisodeInfo = fileInfo,
|
|
FolderEpisodeInfo = folderInfo,
|
|
Series = new Tv.Series
|
|
{
|
|
Id = 1,
|
|
Title = "Some Special"
|
|
}
|
|
};
|
|
|
|
GivenEpisodes(actualInfo, actualInfo.EpisodeNumbers);
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
.Setup(v => v.ParseSpecialEpisodeTitle(fileInfo, It.IsAny<string>(), 0, 0, null))
|
|
.Returns(actualInfo);
|
|
|
|
Mocker.GetMock<IParsingService>()
|
|
.Setup(v => v.ParseSpecialEpisodeTitle(folderInfo, It.IsAny<string>(), 0, 0, null))
|
|
.Returns(actualInfo);
|
|
|
|
Subject.IsSatisfiedBy(localEpisode, null).Accepted.Should().BeTrue();
|
|
}
|
|
|
|
[Test]
|
|
public void should_be_accepted_if_file_has_absolute_episode_number_and_folder_uses_standard()
|
|
{
|
|
_localEpisode.FileEpisodeInfo.SeasonNumber = 1;
|
|
_localEpisode.FileEpisodeInfo.AbsoluteEpisodeNumbers = new[] { 1 };
|
|
|
|
_localEpisode.FolderEpisodeInfo.SeasonNumber = 1;
|
|
_localEpisode.FolderEpisodeInfo.EpisodeNumbers = new[] { 1, 2 };
|
|
|
|
GivenEpisodes(_localEpisode.FileEpisodeInfo, new []{ 1 });
|
|
GivenEpisodes(_localEpisode.FolderEpisodeInfo, _localEpisode.FolderEpisodeInfo.EpisodeNumbers);
|
|
|
|
_localEpisode.Path = @"C:\Test\Unsorted\Series.Title.S01.720p.HDTV-Sonarr\S02E01.mkv".AsOsAgnostic();
|
|
|
|
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue();
|
|
}
|
|
}
|
|
}
|