Fixed: Don't duplicate languages from MediaInfo

pull/4588/head^2
Qstick 4 years ago
parent f8f58c2eda
commit a552389ee8

@ -0,0 +1,111 @@
using FizzWare.NBuilder;
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Core.Languages;
using NzbDrone.Core.MediaFiles.MediaInfo;
using NzbDrone.Core.MediaFiles.MovieImport.Aggregation.Aggregators.Augmenters.Language;
using NzbDrone.Core.Parser;
using NzbDrone.Core.Parser.Model;
using NzbDrone.Core.Test.Framework;
namespace NzbDrone.Core.Test.MediaFiles.MovieImport.Aggregation.Aggregators.Augmenters.Language
{
[TestFixture]
public class AugmentLanguageFromMediaInfoFixture : CoreTest<AugmentLanguageFromMediaInfo>
{
[Test]
public void should_return_null_if_media_info_is_null()
{
var localMovie = Builder<LocalMovie>.CreateNew()
.With(l => l.MediaInfo = null)
.Build();
Subject.AugmentLanguage(localMovie).Should().Be(null);
}
[Test]
public void should_return_language_for_single_known_language()
{
var mediaInfo = Builder<MediaInfoModel>.CreateNew()
.With(m => m.AudioLanguages = "English")
.Build();
var localMovie = Builder<LocalMovie>.CreateNew()
.With(l => l.MediaInfo = mediaInfo)
.Build();
var result = Subject.AugmentLanguage(localMovie);
result.Languages.Count.Should().Be(1);
result.Languages.Should().Contain(Core.Languages.Language.English);
}
[Test]
public void should_only_return_one_when_language_duplicated()
{
var mediaInfo = Builder<MediaInfoModel>.CreateNew()
.With(m => m.AudioLanguages = "English / English")
.Build();
var localMovie = Builder<LocalMovie>.CreateNew()
.With(l => l.MediaInfo = mediaInfo)
.Build();
var result = Subject.AugmentLanguage(localMovie);
result.Languages.Count.Should().Be(1);
result.Languages.Should().Contain(Core.Languages.Language.English);
}
[Test]
public void should_return_null_if_all_unknown()
{
var mediaInfo = Builder<MediaInfoModel>.CreateNew()
.With(m => m.AudioLanguages = "Pirate / Pirate")
.Build();
var localMovie = Builder<LocalMovie>.CreateNew()
.With(l => l.MediaInfo = mediaInfo)
.Build();
var result = Subject.AugmentLanguage(localMovie);
result.Should().BeNull();
}
[Test]
public void should_return_known_languages_only()
{
var mediaInfo = Builder<MediaInfoModel>.CreateNew()
.With(m => m.AudioLanguages = "English / Pirate")
.Build();
var localMovie = Builder<LocalMovie>.CreateNew()
.With(l => l.MediaInfo = mediaInfo)
.Build();
var result = Subject.AugmentLanguage(localMovie);
result.Languages.Count.Should().Be(1);
result.Languages.Should().Contain(Core.Languages.Language.English);
}
[Test]
public void should_return_multiple_known_languages()
{
var mediaInfo = Builder<MediaInfoModel>.CreateNew()
.With(m => m.AudioLanguages = "English / German")
.Build();
var localMovie = Builder<LocalMovie>.CreateNew()
.With(l => l.MediaInfo = mediaInfo)
.Build();
var result = Subject.AugmentLanguage(localMovie);
result.Languages.Count.Should().Be(2);
result.Languages.Should().Contain(Core.Languages.Language.English);
result.Languages.Should().Contain(Core.Languages.Language.German);
}
}
}

@ -15,7 +15,7 @@ namespace NzbDrone.Core.MediaFiles.MovieImport.Aggregation.Aggregators.Augmenter
return null; return null;
} }
var audioLanguages = localMovie.MediaInfo.AudioLanguages.Split('/').Distinct().ToList(); var audioLanguages = localMovie.MediaInfo.AudioLanguages.Split('/').Select(l => l.Trim()).Distinct().ToList();
var languages = new List<Languages.Language>(); var languages = new List<Languages.Language>();

Loading…
Cancel
Save