Sonarr/NzbDrone.Core.Test/ProviderTests/MediaFileProviderTests/GetNewFilenameFixture.cs

// ReSharper disable RedundantUsingDirective

using System.Collections.Generic;

using FizzWare.NBuilder;
using FluentAssertions;
using NUnit.Framework;
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
    {
        [Test]
        public void GetNewFilename_Series_Episode_Quality_S01E05_Dash()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 15)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("South Park - S15E06 - City Sushi [HDTV]", result);
        }

        [Test]
        public void GetNewFilename_Episode_Quality_1x05_Dash()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 15)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("15x06 - City Sushi [HDTV]", result);
        }

        [Test]
        public void GetNewFilename_Series_Quality_01x05_Space()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 5)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("South Park 05x06 [HDTV]", result);
        }

        [Test]
        public void GetNewFilename_Series_s01e05_Space()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 5)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("South Park s05e06", result);
        }

        [Test]
        public void GetNewFilename_Series_Episode_s01e05_Periods()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 5)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("South.Park.s05e06.City.Sushi", result);
        }

        [Test]
        public void GetNewFilename_Series_Episode_s01e05_Dash_Periods_Quality()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 5)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("South.Park.-.s05e06.-.City.Sushi.[HDTV]", result);
        }

        [Test]
        public void GetNewFilename_S01E05_Dash()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 15)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("S15E06", result);
        }

        [Test]
        public void GetNewFilename_multi_Series_Episode_Quality_S01E05_Scene_Dash()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "Strawberries and Cream (1)")
                            .With(e => e.SeasonNumber = 3)
                            .With(e => e.EpisodeNumber = 23)
                            .Build();

            var episodeTwo = Builder<Episode>.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<MediaFileProvider>().GetNewFilename(new List<Episode> { episodeOne, episodeTwo }, "The Mentalist", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("The Mentalist - S03E23-E24 - Strawberries and Cream (1) + Strawberries and Cream (2) [HDTV]", result);
        }

        [Test]
        public void GetNewFilename_multi_Episode_Quality_1x05_Repeat_Dash()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "Strawberries and Cream (1)")
                            .With(e => e.SeasonNumber = 3)
                            .With(e => e.EpisodeNumber = 23)
                            .Build();

            var episodeTwo = Builder<Episode>.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<MediaFileProvider>().GetNewFilename(new List<Episode> { episodeOne, episodeTwo }, "The Mentalist", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("3x23x24 - Strawberries and Cream (1) + Strawberries and Cream (2) [HDTV]", result);
        }

        [Test]
        public void GetNewFilename_multi_Episode_Quality_01x05_Repeat_Space()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "Strawberries and Cream (1)")
                            .With(e => e.SeasonNumber = 3)
                            .With(e => e.EpisodeNumber = 23)
                            .Build();

            var episodeTwo = Builder<Episode>.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<MediaFileProvider>().GetNewFilename(new List<Episode> { episodeOne, episodeTwo }, "The Mentalist", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("3x23x24 Strawberries and Cream (1) + Strawberries and Cream (2) [HDTV]", result);
        }

        [Test]
        public void GetNewFilename_multi_Series_Episode_s01e05_Duplicate_Period()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "Strawberries and Cream (1)")
                            .With(e => e.SeasonNumber = 3)
                            .With(e => e.EpisodeNumber = 23)
                            .Build();

            var episodeTwo = Builder<Episode>.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<MediaFileProvider>().GetNewFilename(new List<Episode> { episodeOne, episodeTwo }, "The Mentalist", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("The.Mentalist.s03e23.s03e24.Strawberries.and.Cream.(1).+.Strawberries.and.Cream.(2)", result);
        }

        [Test]
        public void GetNewFilename_multi_Series_S01E05_Extend_Dash_Period()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "Strawberries and Cream (1)")
                            .With(e => e.SeasonNumber = 3)
                            .With(e => e.EpisodeNumber = 23)
                            .Build();

            var episodeTwo = Builder<Episode>.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<MediaFileProvider>().GetNewFilename(new List<Episode> { episodeOne, episodeTwo }, "The Mentalist", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("The.Mentalist.-.S03E23-24", result);
        }

        [Test]
        public void GetNewFilename_multi_1x05_Repeat_Dash_Period()
        {
            //Setup
            

            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "Strawberries and Cream (1)")
                            .With(e => e.SeasonNumber = 3)
                            .With(e => e.EpisodeNumber = 23)
                            .Build();

            var episodeTwo = Builder<Episode>.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<MediaFileProvider>().GetNewFilename(new List<Episode> { episodeOne, episodeTwo }, "The Mentalist", QualityTypes.HDTV, false);

            //Assert
            Assert.AreEqual("3x23x24", result);
        }

        [Test]
        public void GetNewFilename_should_append_proper_when_proper_and_append_quality_is_true()
        {
            //Setup
            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 15)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, true);

            //Assert
            result.Should().Be("South Park - S15E06 - City Sushi [HDTV] [Proper]");
        }

        [Test]
        public void GetNewFilename_should_not_append_proper_when_not_proper_and_append_quality_is_true()
        {
            //Setup
            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 15)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, false);

            //Assert
            result.Should().Be("South Park - S15E06 - City Sushi [HDTV]");
        }

        [Test]
        public void GetNewFilename_should_not_append_proper_when_proper_and_append_quality_is_false()
        {
            //Setup
            var fakeConfig = Mocker.GetMock<ConfigProvider>();
            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<Episode>.CreateNew()
                            .With(e => e.Title = "City Sushi")
                            .With(e => e.SeasonNumber = 15)
                            .With(e => e.EpisodeNumber = 6)
                            .Build();

            //Act
            string result = Mocker.Resolve<MediaFileProvider>().GetNewFilename(new List<Episode> { episode }, "South Park", QualityTypes.HDTV, true);

            //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<ConfigProvider>();
            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<Episode>.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<Episode>.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<MediaFileProvider>().GetNewFilename(new List<Episode> { episode2, episode }, "30 Rock", QualityTypes.HDTV, false);

            //Assert
            result.Should().Be("30 Rock - S06E06-E07 - Hey, Baby, What's Wrong! (1) + Hey, Baby, What's Wrong! (2)");
        }
    }
}