Fixed: Language Augment from Folder and Filename

src/NzbDrone.Core.Test/MediaFiles/EpisodeImport/Aggregation/Aggregators/Augmenters/Language/AugmentLanguageFromMediaInfoFixture.cs

@@ -0,0 +1,110 @@
+using System.Collections.Generic;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using NUnit.Framework;
+using NzbDrone.Core.MediaFiles.EpisodeImport.Aggregation.Aggregators.Augmenters.Language;
+using NzbDrone.Core.MediaFiles.MediaInfo;
+using NzbDrone.Core.Parser.Model;
+using NzbDrone.Core.Test.Framework;
+
+namespace NzbDrone.Core.Test.MediaFiles.EpisodeImport.Aggregation.Aggregators.Augmenters.Language
+{
+    [TestFixture]
+    public class AugmentLanguageFromMediaInfoFixture : CoreTest<AugmentLanguageFromMediaInfo>
+    {
+        [Test]
+        public void should_return_null_if_media_info_is_null()
+        {
+            var localEpisode = Builder<LocalEpisode>.CreateNew()
+                                                    .With(l => l.MediaInfo = null)
+                                                    .Build();
+
+            Subject.AugmentLanguage(localEpisode, null).Should().Be(null);
+        }
+
+        [Test]
+        public void should_return_language_for_single_known_language()
+        {
+            var mediaInfo = Builder<MediaInfoModel>.CreateNew()
+                                                   .With(m => m.AudioLanguages = new List<string> { "eng" })
+                                                   .Build();
+
+            var localEpisode = Builder<LocalEpisode>.CreateNew()
+                                                    .With(l => l.MediaInfo = mediaInfo)
+                                                    .Build();
+
+            var result = Subject.AugmentLanguage(localEpisode, null);
+
+            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 = new List<string> { "eng", "eng" })
+                                                   .Build();
+
+            var localEpisode = Builder<LocalEpisode>.CreateNew()
+                                                    .With(l => l.MediaInfo = mediaInfo)
+                                                    .Build();
+
+            var result = Subject.AugmentLanguage(localEpisode, null);
+
+            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 = new List<string> { "pirate", "pirate" })
+                                                   .Build();
+
+            var localEpisode = Builder<LocalEpisode>.CreateNew()
+                                                    .With(l => l.MediaInfo = mediaInfo)
+                                                    .Build();
+
+            var result = Subject.AugmentLanguage(localEpisode, null);
+
+            result.Should().BeNull();
+        }
+
+        [Test]
+        public void should_return_known_languages_only()
+        {
+            var mediaInfo = Builder<MediaInfoModel>.CreateNew()
+                                                   .With(m => m.AudioLanguages = new List<string> { "eng", "pirate" })
+                                                   .Build();
+
+            var localEpisode = Builder<LocalEpisode>.CreateNew()
+                                                    .With(l => l.MediaInfo = mediaInfo)
+                                                    .Build();
+
+            var result = Subject.AugmentLanguage(localEpisode, null);
+
+            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 = new List<string> { "eng", "ger" })
+                                                   .Build();
+
+            var localMovie = Builder<LocalEpisode>.CreateNew()
+                                                    .With(l => l.MediaInfo = mediaInfo)
+                                                    .Build();
+
+            var result = Subject.AugmentLanguage(localMovie, null);
+
+            result.Languages.Count.Should().Be(2);
+            result.Languages.Should().Contain(Core.Languages.Language.English);
+            result.Languages.Should().Contain(Core.Languages.Language.German);
+        }
+    }
+}

src/NzbDrone.Core/MediaFiles/EpisodeImport/Aggregation/Aggregators/Augmenters/Language/AugmentLanguageFromDownloadClientItem.cs

@@ -25,7 +25,7 @@ namespace NzbDrone.Core.MediaFiles.EpisodeImport.Aggregation.Aggregators.Augment
             {
                 var episodeTitleLanguage = LanguageParser.ParseLanguages(episode.Title, false);
 
-                languages = localEpisode.DownloadClientEpisodeInfo.Languages.Except(episodeTitleLanguage).ToList();
+                languages = languages.Except(episodeTitleLanguage).ToList();
             }
 
             return new AugmentLanguageResult(languages, Confidence.DownloadClientItem);

src/NzbDrone.Core/MediaFiles/EpisodeImport/Aggregation/Aggregators/Augmenters/Language/AugmentLanguageFromFileName.cs

@@ -23,7 +23,7 @@ namespace NzbDrone.Core.MediaFiles.EpisodeImport.Aggregation.Aggregators.Augment
             {
                 var episodeTitleLanguage = LanguageParser.ParseLanguages(episode.Title, false);
 
-                languages = localEpisode.DownloadClientEpisodeInfo.Languages.Except(episodeTitleLanguage).ToList();
+                languages = languages.Except(episodeTitleLanguage).ToList();
             }
 
             return new AugmentLanguageResult(languages, Confidence.Filename);

src/NzbDrone.Core/MediaFiles/EpisodeImport/Aggregation/Aggregators/Augmenters/Language/AugmentLanguageFromFolder.cs

@@ -23,7 +23,7 @@ namespace NzbDrone.Core.MediaFiles.EpisodeImport.Aggregation.Aggregators.Augment
             {
                 var episodeTitleLanguage = LanguageParser.ParseLanguages(episode.Title, false);
 
-                languages = localEpisode.DownloadClientEpisodeInfo.Languages.Except(episodeTitleLanguage).ToList();
+                languages = languages.Except(episodeTitleLanguage).ToList();
             }
 
             return new AugmentLanguageResult(languages, Confidence.Foldername);