From 334d4d1e9b2f77d4ca2cb73514b55d4747b4fc03 Mon Sep 17 00:00:00 2001 From: Mark McDowall Date: Sat, 23 Feb 2013 19:54:50 -0800 Subject: [PATCH] Dear gdrive, you suck! --- .../CleanUpDatabaseFixture.cs | 139 +++ .../GetNewFilenameFixture.cs | 842 ++++++++++++++++++ 2 files changed, 981 insertions(+) diff --git a/NzbDrone.Core.Test/ProviderTests/MediaFileProviderTests/CleanUpDatabaseFixture.cs b/NzbDrone.Core.Test/ProviderTests/MediaFileProviderTests/CleanUpDatabaseFixture.cs index e69de29bb..90ba51765 100644 --- a/NzbDrone.Core.Test/ProviderTests/MediaFileProviderTests/CleanUpDatabaseFixture.cs +++ b/NzbDrone.Core.Test/ProviderTests/MediaFileProviderTests/CleanUpDatabaseFixture.cs @@ -0,0 +1,139 @@ +using System; +using System.Collections.Generic; +using System.Linq; +using System.Text; +using FizzWare.NBuilder; +using FluentAssertions; +using NUnit.Framework; +using NzbDrone.Core.Tv; +using NzbDrone.Core.Model; +using NzbDrone.Core.Providers; +using NzbDrone.Core.Providers.Core; +using NzbDrone.Core.Repository; +using NzbDrone.Core.Repository.Quality; +using NzbDrone.Core.Test.Framework; + +namespace NzbDrone.Core.Test.ProviderTests.MediaFileProviderTests +{ + [TestFixture] + public class CleanUpDatabaseFixture : SqlCeTest + { + + [SetUp] + public void Setup() + { + WithRealDb(); + } + + private void WithAutoIgnore(bool autoIgnore) + { + + Mocker.GetMock() + .SetupGet(c => c.AutoIgnorePreviouslyDownloadedEpisodes).Returns(autoIgnore); + } + + + + [Test] + public void CleanUpDatabse_should_detach_none_existing_file_from_episodes_with_auto_ignore() + { + WithAutoIgnore(true); + + var episodes = Builder.CreateListOfSize(3) + .All().With(c => c.GrabDate = DateTime.Now) + .And(c => c.Ignored = false) + .And(c => c.PostDownloadStatus = PostDownloadStatusType.NoError) + .Build(); + + + Db.InsertMany(episodes); + + //Act + Mocker.Resolve().CleanUpDatabase(); + var result = Db.Fetch(); + + //Assert + result.Should().HaveSameCount(episodes); + result.Should().OnlyContain(e => e.EpisodeFileId == 0); + result.Should().OnlyContain(e => e.PostDownloadStatus == PostDownloadStatusType.Unknown); + result.Should().OnlyContain(e => e.Ignored); + result.Should().OnlyContain(e => e.GrabDate == null); + } + + [Test] + public void CleanUpDatabse_should_detach_none_existing_file_from_episodes_with_no_auto_ignore() + { + WithAutoIgnore(false); + + var episodes = Builder.CreateListOfSize(3) + .All().With(c => c.GrabDate = DateTime.Now) + .And(c => c.PostDownloadStatus = PostDownloadStatusType.NoError) + .TheFirst(2).With(c => c.Ignored = true) + .TheLast(1).With(c => c.Ignored = false) + .Build(); + + + Db.InsertMany(episodes); + + //Act + Mocker.Resolve().CleanUpDatabase(); + var result = Db.Fetch(); + + //Assert + result.Should().HaveSameCount(episodes); + result.Should().OnlyContain(e => e.EpisodeFileId == 0); + result.Should().OnlyContain(e => e.PostDownloadStatus == PostDownloadStatusType.Unknown); + result.Should().OnlyContain(e => e.GrabDate == null); + result.Should().Contain(c => c.Ignored == true); + result.Should().Contain(c => c.Ignored == false); + } + + [Test] + public void CleanUpDatabse_should_not_change_episodes_with_no_file_id() + { + //Setup + var episodes = Builder.CreateListOfSize(3) + .All().With(c => c.GrabDate = DateTime.Now) + .And(c => c.Ignored = false) + .And(c => c.PostDownloadStatus = PostDownloadStatusType.NoError) + .Build(); + + Db.InsertMany(episodes); + + //Act + Mocker.Resolve().CleanUpDatabase(); + var result = Db.Fetch(); + + //Assert + result.Should().HaveSameCount(episodes); + result.Should().OnlyContain(e => e.EpisodeFileId == 0); + result.Should().NotContain(e => e.PostDownloadStatus == PostDownloadStatusType.Unknown); + result.Should().NotContain(e => e.Ignored); + result.Should().NotContain(e => e.GrabDate == null); + } + + + [Test] + public void DeleteOrphanedEpisodeFiles() + { + //Setup + var episodeFiles = Builder + .CreateListOfSize(10) + .All() + .With(e => e.Quality = QualityTypes.DVD) + .Build(); + var episodes = Builder.CreateListOfSize(5).Build(); + + Db.InsertMany(episodes); + Db.InsertMany(episodeFiles); + + //Act + Mocker.Resolve().CleanUpDatabase(); + var result = Db.Fetch(); + + //Assert + result.Should().HaveCount(5); + result.Should().OnlyContain(e => e.EpisodeFileId > 0); + } + } +} diff --git a/NzbDrone.Core.Test/ProviderTests/MediaFileProviderTests/GetNewFilenameFixture.cs b/NzbDrone.Core.Test/ProviderTests/MediaFileProviderTests/GetNewFilenameFixture.cs index e69de29bb..bb6b22854 100644 --- a/NzbDrone.Core.Test/ProviderTests/MediaFileProviderTests/GetNewFilenameFixture.cs +++ b/NzbDrone.Core.Test/ProviderTests/MediaFileProviderTests/GetNewFilenameFixture.cs @@ -0,0 +1,842 @@ +// ReSharper disable RedundantUsingDirective + +using System; +using System.Collections.Generic; +using System.IO; +using System.Linq; +using FizzWare.NBuilder; +using FluentAssertions; +using NUnit.Framework; +using NzbDrone.Core.Tv; +using NzbDrone.Core.Providers; +using NzbDrone.Core.Providers.Core; +using NzbDrone.Core.Repository; +using NzbDrone.Core.Repository.Quality; +using NzbDrone.Core.Test.Framework; +using NzbDrone.Test.Common.AutoMoq; + +namespace NzbDrone.Core.Test.ProviderTests.MediaFileProviderTests +{ + [TestFixture] + // ReSharper disable InconsistentNaming + public class MediaFileProvider_GetNewFilenameTest : CoreTest + { + private Series _series; + + [SetUp] + public void Setup() + { + _series = Builder + .CreateNew() + .With(s => s.Title = "South Park") + .Build(); + } + + [Test] + public void GetNewFilename_Series_Episode_Quality_S01E05_Dash() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("South Park - S15E06 - City Sushi [HDTV-720p]", result); + } + + [Test] + public void GetNewFilename_Episode_Quality_1x05_Dash() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(false); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("15x06 - City Sushi [HDTV-720p]", result); + } + + [Test] + public void GetNewFilename_Series_Quality_01x05_Space() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(false); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(1); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(1); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 5) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("South Park 05x06 [HDTV-720p]", result); + } + + [Test] + public void GetNewFilename_Series_s01e05_Space() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(false); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(1); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(3); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 5) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("South Park s05e06", result); + } + + [Test] + public void GetNewFilename_Series_Episode_s01e05_Periods() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(1); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(3); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(true); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 5) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("South.Park.s05e06.City.Sushi", result); + } + + [Test] + public void GetNewFilename_Series_Episode_s01e05_Dash_Periods_Quality() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(3); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(true); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 5) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("South.Park.-.s05e06.-.City.Sushi.[HDTV-720p]", result); + } + + [Test] + public void GetNewFilename_S01E05_Dash() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(false); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(false); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("S15E06", result); + } + + [Test] + public void GetNewFilename_multi_Series_Episode_Quality_S01E05_Scene_Dash() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(3); + + var episodeOne = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (1)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 23) + .Build(); + + var episodeTwo = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (2)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 24) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episodeOne, episodeTwo }, new Series { Title = "The Mentalist" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("The Mentalist - S03E23-E24 - Strawberries and Cream [HDTV-720p]", result); + } + + [Test] + public void GetNewFilename_multi_Episode_Quality_1x05_Repeat_Dash() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(false); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(2); + + var episodeOne = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (1)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 23) + .Build(); + + var episodeTwo = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (2)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 24) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episodeOne, episodeTwo }, new Series { Title = "The Mentalist" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("3x23x24 - Strawberries and Cream [HDTV-720p]", result); + } + + [Test] + public void GetNewFilename_multi_Episode_Quality_01x05_Repeat_Space() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(false); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(1); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(2); + + var episodeOne = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (1)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 23) + .Build(); + + var episodeTwo = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (2)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 24) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episodeOne, episodeTwo }, new Series { Title = "The Mentalist" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("3x23x24 Strawberries and Cream [HDTV-720p]", result); + } + + [Test] + public void GetNewFilename_multi_Series_Episode_s01e05_Duplicate_Period() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(1); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(3); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(true); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(1); + + var episodeOne = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (1)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 23) + .Build(); + + var episodeTwo = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (2)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 24) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episodeOne, episodeTwo }, new Series { Title = "The Mentalist" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("The.Mentalist.s03e23.s03e24.Strawberries.and.Cream", result); + } + + [Test] + public void GetNewFilename_multi_Series_S01E05_Extend_Dash_Period() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(false); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(true); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(0); + + var episodeOne = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (1)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 23) + .Build(); + + var episodeTwo = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (2)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 24) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episodeOne, episodeTwo }, new Series { Title = "The Mentalist" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("The.Mentalist.-.S03E23-24", result); + } + + [Test] + public void GetNewFilename_multi_1x05_Repeat_Dash_Period() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(false); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(false); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(true); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(2); + + var episodeOne = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (1)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 23) + .Build(); + + var episodeTwo = Builder.CreateNew() + .With(e => e.Title = "Strawberries and Cream (2)") + .With(e => e.SeasonNumber = 3) + .With(e => e.EpisodeNumber = 24) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episodeOne, episodeTwo }, new Series { Title = "The Mentalist" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("3x23x24", result); + } + + [Test] + public void GetNewFilename_should_append_proper_when_proper_and_append_quality_is_true() + { + //Setup + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, true, new EpisodeFile()); + + //Assert + result.Should().Be("South Park - S15E06 - City Sushi [HDTV-720p] [Proper]"); + } + + [Test] + public void GetNewFilename_should_not_append_proper_when_not_proper_and_append_quality_is_true() + { + //Setup + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + result.Should().Be("South Park - S15E06 - City Sushi [HDTV-720p]"); + } + + [Test] + public void GetNewFilename_should_not_append_proper_when_proper_and_append_quality_is_false() + { + //Setup + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, true, new EpisodeFile()); + + //Assert + result.Should().Be("South Park - S15E06 - City Sushi"); + } + + [Test] + public void GetNewFilename_should_order_multiple_episode_files_in_numerical_order() + { + //Setup + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(3); + + var episode = Builder.CreateNew() + .With(e => e.Title = "Hey, Baby, What's Wrong? (1)") + .With(e => e.SeasonNumber = 6) + .With(e => e.EpisodeNumber = 6) + .Build(); + + var episode2 = Builder.CreateNew() + .With(e => e.Title = "Hey, Baby, What's Wrong? (2)") + .With(e => e.SeasonNumber = 6) + .With(e => e.EpisodeNumber = 7) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode2, episode }, new Series { Title = "30 Rock" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + result.Should().Be("30 Rock - S06E06-E07 - Hey, Baby, What's Wrong!"); + } + + [Test] + public void GetNewFilename_Series_Episode_Quality_S01E05_Period() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("South Park.S15E06.City Sushi [HDTV-720p]", result); + } + + [Test] + public void GetNewFilename_Episode_Quality_1x05_Period() + { + //Setup + + + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(false); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + Assert.AreEqual("15x06.City Sushi [HDTV-720p]", result); + } + + [Test] + public void GetNewFilename_UseSceneName_when_sceneName_isNull() + { + //Setup + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(false); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + fakeConfig.SetupGet(c => c.SortingUseSceneName).Returns(true); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + var episodeFile = Builder.CreateNew() + .With(e => e.SceneName = null) + .With(e => e.Path = @"C:\Test\TV\30 Rock - S01E01 - Test") + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, episodeFile); + + //Assert + result.Should().Be(Path.GetFileNameWithoutExtension(episodeFile.Path)); + } + + [Test] + public void GetNewFilename_UseSceneName_when_sceneName_isNotNull() + { + //Setup + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(false); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + fakeConfig.SetupGet(c => c.SortingUseSceneName).Returns(true); + + var episode = Builder.CreateNew() + .With(e => e.Title = "City Sushi") + .With(e => e.SeasonNumber = 15) + .With(e => e.EpisodeNumber = 6) + .Build(); + + var episodeFile = Builder.CreateNew() + .With(e => e.SceneName = "30.Rock.S01E01.xvid-LOL") + .With(e => e.Path = @"C:\Test\TV\30 Rock - S01E01 - Test") + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode }, _series, QualityTypes.HDTV720p, false, episodeFile); + + //Assert + result.Should().Be(episodeFile.SceneName); + } + + [Test] + public void should_only_have_one_episodeTitle_when_episode_titles_are_the_same() + { + //Setup + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(3); + + var episode = Builder.CreateNew() + .With(e => e.Title = "Hey, Baby, What's Wrong? (1)") + .With(e => e.SeasonNumber = 6) + .With(e => e.EpisodeNumber = 6) + .Build(); + + var episode2 = Builder.CreateNew() + .With(e => e.Title = "Hey, Baby, What's Wrong? (2)") + .With(e => e.SeasonNumber = 6) + .With(e => e.EpisodeNumber = 7) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode2, episode }, new Series { Title = "30 Rock" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + result.Should().Be("30 Rock - S06E06-E07 - Hey, Baby, What's Wrong!"); + } + + [Test] + public void should_have_two_episodeTitles_when_episode_titles_are_not_the_same() + { + //Setup + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(3); + + var episode = Builder.CreateNew() + .With(e => e.Title = "Hello") + .With(e => e.SeasonNumber = 6) + .With(e => e.EpisodeNumber = 6) + .Build(); + + var episode2 = Builder.CreateNew() + .With(e => e.Title = "World") + .With(e => e.SeasonNumber = 6) + .With(e => e.EpisodeNumber = 7) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode2, episode }, new Series { Title = "30 Rock" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + result.Should().Be("30 Rock - S06E06-E07 - Hello + World"); + } + + [Test] + public void should_have_two_episodeTitles_when_distinct_count_is_two() + { + //Setup + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + fakeConfig.SetupGet(c => c.SortingMultiEpisodeStyle).Returns(3); + + var episode = Builder.CreateNew() + .With(e => e.Title = "Hello (3)") + .With(e => e.SeasonNumber = 6) + .With(e => e.EpisodeNumber = 6) + .Build(); + + var episode2 = Builder.CreateNew() + .With(e => e.Title = "Hello (2)") + .With(e => e.SeasonNumber = 6) + .With(e => e.EpisodeNumber = 7) + .Build(); + + var episode3 = Builder.CreateNew() + .With(e => e.Title = "World") + .With(e => e.SeasonNumber = 6) + .With(e => e.EpisodeNumber = 8) + .Build(); + + //Act + string result = Mocker.Resolve().GetNewFilename(new List { episode, episode2, episode3 }, new Series { Title = "30 Rock" }, QualityTypes.HDTV720p, false, new EpisodeFile()); + + //Assert + result.Should().Be("30 Rock - S06E06-E07-E08 - Hello + World"); + } + + [Test] + public void should_use_airDate_if_series_isDaily() + { + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(true); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var series = Builder + .CreateNew() + .With(s => s.SeriesType = SeriesType.Daily) + .With(s => s.Title = "The Daily Show with Jon Stewart") + .Build(); + + var episodes = Builder + .CreateListOfSize(1) + .All() + .With(e => e.AirDate = new DateTime(2012, 12, 13)) + .With(e => e.Title = "Kristen Stewart") + .Build(); + + var result = Mocker.Resolve() + .GetNewFilename(episodes, series, QualityTypes.HDTV720p, false, new EpisodeFile()); + result.Should().Be("The Daily Show with Jon Stewart - 2012-12-13 - Kristen Stewart [HDTV-720p]"); + } + + [Test] + public void should_use_airDate_if_series_isDaily_no_episode_title() + { + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(false); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var series = Builder + .CreateNew() + .With(s => s.SeriesType = SeriesType.Daily) + .With(s => s.Title = "The Daily Show with Jon Stewart") + .Build(); + + var episodes = Builder + .CreateListOfSize(1) + .All() + .With(e => e.AirDate = new DateTime(2012, 12, 13)) + .With(e => e.Title = "Kristen Stewart") + .Build(); + + var result = Mocker.Resolve() + .GetNewFilename(episodes, series, QualityTypes.HDTV720p, false, new EpisodeFile()); + result.Should().Be("The Daily Show with Jon Stewart - 2012-12-13"); + } + + [Test] + public void should_set_airdate_to_unknown_if_not_available() + { + var fakeConfig = Mocker.GetMock(); + fakeConfig.SetupGet(c => c.SortingIncludeSeriesName).Returns(true); + fakeConfig.SetupGet(c => c.SortingIncludeEpisodeTitle).Returns(true); + fakeConfig.SetupGet(c => c.SortingAppendQuality).Returns(false); + fakeConfig.SetupGet(c => c.SortingSeparatorStyle).Returns(0); + fakeConfig.SetupGet(c => c.SortingNumberStyle).Returns(2); + fakeConfig.SetupGet(c => c.SortingReplaceSpaces).Returns(false); + + var series = Builder + .CreateNew() + .With(s => s.SeriesType = SeriesType.Daily) + .With(s => s.Title = "The Daily Show with Jon Stewart") + .Build(); + + var episodes = Builder + .CreateListOfSize(1) + .All() + .With(e => e.AirDate = null) + .With(e => e.Title = "Kristen Stewart") + .Build(); + + var result = Mocker.Resolve() + .GetNewFilename(episodes, series, QualityTypes.HDTV720p, false, new EpisodeFile()); + result.Should().Be("The Daily Show with Jon Stewart - Unknown - Kristen Stewart"); + } + } +} \ No newline at end of file