New: Use absolute format if absolute episode number is not required and is missing

Closes #5124
This commit is contained in:
Mark McDowall 2022-08-18 23:06:49 -07:00
parent 89b0b04e08
commit 6660e0c3f3
5 changed files with 97 additions and 13 deletions

View File

@ -41,7 +41,7 @@ namespace NzbDrone.Core.Test.MediaFiles.EpisodeImport.Specifications
}; };
Mocker.GetMock<IBuildFileNames>() Mocker.GetMock<IBuildFileNames>()
.Setup(s => s.RequiresAbsoluteEpisodeNumber(_series, episodes)) .Setup(s => s.RequiresAbsoluteEpisodeNumber())
.Returns(true); .Returns(true);
} }
@ -68,7 +68,7 @@ namespace NzbDrone.Core.Test.MediaFiles.EpisodeImport.Specifications
_localEpisode.Episodes.First().AbsoluteEpisodeNumber = null; _localEpisode.Episodes.First().AbsoluteEpisodeNumber = null;
Mocker.GetMock<IBuildFileNames>() Mocker.GetMock<IBuildFileNames>()
.Setup(s => s.RequiresAbsoluteEpisodeNumber(_series, _localEpisode.Episodes)) .Setup(s => s.RequiresAbsoluteEpisodeNumber())
.Returns(false); .Returns(false);
Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue(); Subject.IsSatisfiedBy(_localEpisode, null).Accepted.Should().BeTrue();

View File

@ -0,0 +1,87 @@
using System.Collections.Generic;
using System.Linq;
using FizzWare.NBuilder;
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Core.CustomFormats;
using NzbDrone.Core.MediaFiles;
using NzbDrone.Core.Organizer;
using NzbDrone.Core.Qualities;
using NzbDrone.Core.Test.Framework;
using NzbDrone.Core.Tv;
namespace NzbDrone.Core.Test.OrganizerTests.FileNameBuilderTests
{
[TestFixture]
public class AbsoluteEpisodeFormatFixture : CoreTest<FileNameBuilder>
{
private Series _series;
private Episode _episode;
private EpisodeFile _episodeFile;
private NamingConfig _namingConfig;
[SetUp]
public void Setup()
{
_series = Builder<Series>
.CreateNew()
.With(s => s.SeriesType = SeriesTypes.Anime)
.With(s => s.Title = "Anime Series")
.Build();
_episode = Builder<Episode>.CreateNew()
.With(e => e.Title = "City Sushi")
.With(e => e.SeasonNumber = 15)
.With(e => e.EpisodeNumber = 6)
.With(e => e.AbsoluteEpisodeNumber = 100)
.Build();
_episodeFile = new EpisodeFile { Id = 5, Quality = new QualityModel(Quality.HDTV720p), ReleaseGroup = "SonarrTest" };
_namingConfig = NamingConfig.Default;
_namingConfig.RenameEpisodes = true;
Mocker.GetMock<INamingConfigService>()
.Setup(c => c.GetConfig()).Returns(_namingConfig);
Mocker.GetMock<IQualityDefinitionService>()
.Setup(v => v.Get(Moq.It.IsAny<Quality>()))
.Returns<Quality>(v => Quality.DefaultQualityDefinitions.First(c => c.Quality == v));
Mocker.GetMock<ICustomFormatService>()
.Setup(v => v.All())
.Returns(new List<CustomFormat>());
}
[Test]
public void should_use_standard_format_if_absolute_format_requires_absolute_episode_number_and_it_is_missing()
{
_episode.AbsoluteEpisodeNumber = null;
_namingConfig.StandardEpisodeFormat = "{Series Title} S{season:00}E{episode:00}";
_namingConfig.AnimeEpisodeFormat = "{Series Title} {absolute:00} [{ReleaseGroup}]";
Subject.BuildFileName(new List<Episode> { _episode }, _series, _episodeFile)
.Should().Be("Anime Series S15E06");
}
[Test]
public void should_use_absolute_format_if_absolute_format_requires_absolute_episode_number_and_it_is_available()
{
_namingConfig.StandardEpisodeFormat = "{Series Title} S{season:00}E{episode:00}";
_namingConfig.AnimeEpisodeFormat = "{Series Title} {absolute:00} [{ReleaseGroup}]";
Subject.BuildFileName(new List<Episode> { _episode }, _series, _episodeFile)
.Should().Be("Anime Series 100 [SonarrTest]");
}
[Test]
public void should_use_absolute_format_if_absolute_format_does_not_require_absolute_episode_number_and_it_is_not_available()
{
_namingConfig.StandardEpisodeFormat = "{Series Title} S{season:00}E{episode:00}";
_namingConfig.AnimeEpisodeFormat = "{Series Title} S{season:00}E{episode:00} [{ReleaseGroup}]";
Subject.BuildFileName(new List<Episode> { _episode }, _series, _episodeFile)
.Should().Be("Anime Series S15E06 [SonarrTest]");
}
}
}

View File

@ -42,14 +42,14 @@ namespace NzbDrone.Core.Test.OrganizerTests.FileNameBuilderTests
public void should_return_false_when_absolute_episode_number_is_not_part_of_the_pattern() public void should_return_false_when_absolute_episode_number_is_not_part_of_the_pattern()
{ {
_namingConfig.AnimeEpisodeFormat = "{Series Title} S{season:00}E{episode:00}"; _namingConfig.AnimeEpisodeFormat = "{Series Title} S{season:00}E{episode:00}";
Subject.RequiresAbsoluteEpisodeNumber(_series, new List<Episode> { _episode }).Should().BeFalse(); Subject.RequiresAbsoluteEpisodeNumber().Should().BeFalse();
} }
[Test] [Test]
public void should_return_true_when_absolute_episode_number_is_part_of_the_pattern() public void should_return_true_when_absolute_episode_number_is_part_of_the_pattern()
{ {
_namingConfig.AnimeEpisodeFormat = "{Series Title} {absolute:00}"; _namingConfig.AnimeEpisodeFormat = "{Series Title} {absolute:00}";
Subject.RequiresAbsoluteEpisodeNumber(_series, new List<Episode> { _episode }).Should().BeTrue(); Subject.RequiresAbsoluteEpisodeNumber().Should().BeTrue();
} }
} }
} }

View File

@ -28,7 +28,7 @@ namespace NzbDrone.Core.MediaFiles.EpisodeImport.Specifications
return Decision.Accept(); return Decision.Accept();
} }
if (!_buildFileNames.RequiresAbsoluteEpisodeNumber(localEpisode.Series, localEpisode.Episodes)) if (!_buildFileNames.RequiresAbsoluteEpisodeNumber())
{ {
_logger.Debug("File name format does not require absolute episode number, skipping check"); _logger.Debug("File name format does not require absolute episode number, skipping check");
return Decision.Accept(); return Decision.Accept();

View File

@ -27,7 +27,7 @@ namespace NzbDrone.Core.Organizer
string GetSeriesFolder(Series series, NamingConfig namingConfig = null); string GetSeriesFolder(Series series, NamingConfig namingConfig = null);
string GetSeasonFolder(Series series, int seasonNumber, NamingConfig namingConfig = null); string GetSeasonFolder(Series series, int seasonNumber, NamingConfig namingConfig = null);
bool RequiresEpisodeTitle(Series series, List<Episode> episodes); bool RequiresEpisodeTitle(Series series, List<Episode> episodes);
bool RequiresAbsoluteEpisodeNumber(Series series, List<Episode> episodes); bool RequiresAbsoluteEpisodeNumber();
} }
public class FileNameBuilder : IBuildFileNames public class FileNameBuilder : IBuildFileNames
@ -166,7 +166,9 @@ namespace NzbDrone.Core.Organizer
pattern = namingConfig.DailyEpisodeFormat; pattern = namingConfig.DailyEpisodeFormat;
} }
if (series.SeriesType == SeriesTypes.Anime && episodes.All(e => e.AbsoluteEpisodeNumber.HasValue)) if (series.SeriesType == SeriesTypes.Anime &&
(episodes.All(e => e.AbsoluteEpisodeNumber.HasValue) ||
!RequiresAbsoluteEpisodeNumber()))
{ {
pattern = namingConfig.AnimeEpisodeFormat; pattern = namingConfig.AnimeEpisodeFormat;
} }
@ -433,13 +435,8 @@ namespace NzbDrone.Core.Organizer
}); });
} }
public bool RequiresAbsoluteEpisodeNumber(Series series, List<Episode> episodes) public bool RequiresAbsoluteEpisodeNumber()
{ {
if (series.SeriesType != SeriesTypes.Anime)
{
return false;
}
var namingConfig = _namingConfigService.GetConfig(); var namingConfig = _namingConfigService.GetConfig();
var pattern = namingConfig.AnimeEpisodeFormat; var pattern = namingConfig.AnimeEpisodeFormat;