Sonarr/NzbDrone.Core.Test/ProviderTests/SeriesProviderTest.cs

using System;
using System.Linq;

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;

// ReSharper disable InconsistentNaming
namespace NzbDrone.Core.Test.ProviderTests
{
    [TestFixture]
    public class SeriesProviderTest : CoreTest
    {
        [TestCase(true)]
        [TestCase(false)]
        public void Add_new_series(bool useSeasonFolder)
        {
            WithRealDb();

            Mocker.GetMock<ConfigProvider>()
                .Setup(c => c.UseSeasonFolder).Returns(useSeasonFolder);

            var fakeProfiles = Builder<QualityProfile>.CreateListOfSize(2).Build();

            Db.InsertMany(fakeProfiles);

            const string path = "C:\\Test\\";
            const int tvDbId = 1234;
            const int qualityProfileId = 2;

            //Act
            var seriesProvider = Mocker.Resolve<SeriesProvider>();
            seriesProvider.AddSeries(path, tvDbId, qualityProfileId);

            //Assert
            var series = seriesProvider.GetAllSeries();
            series.Should().HaveCount(1);
            Assert.AreEqual(path, series.First().Path);
            Assert.AreEqual(tvDbId, series.First().SeriesId);
            Assert.AreEqual(qualityProfileId, series.First().QualityProfileId);
            series.First().SeasonFolder.Should().Be(useSeasonFolder);
        }


        [TestCase(0)]
        [TestCase(-1)]
        public void add_series_should_fail_if_series_is_less_than_zero(int seriesId)
        {
            WithRealDb();
            Assert.Throws<ArgumentOutOfRangeException>(() => Mocker.Resolve<SeriesProvider>().AddSeries("C:\\Test", seriesId, 1));
        }

        [Test]
        public void find_series_empty_repo()
        {
            WithRealDb();

            //Act
            var seriesProvider = Mocker.Resolve<SeriesProvider>();
            var series = seriesProvider.FindSeries("My Title");

            //Assert
            Assert.IsNull(series);
        }

        [Test]
        [ExpectedException(typeof(InvalidOperationException), ExpectedMessage = "Sequence contains no elements")]
        public void Get_series_invalid_series_id_should_return_null()
        {
            WithRealDb();

            //Act
            var seriesProvider = Mocker.Resolve<SeriesProvider>();
            var series = seriesProvider.GetSeries(2);


            //Assert
            Assert.IsNull(series);
        }

        [Test]
        public void Get_series_by_id()
        {
            WithRealDb();

            var fakeSeries = Builder<Series>.CreateNew()
                .With(c => c.QualityProfileId = 1)
                .With(c => c.EpisodeCount = 0)
                .With(c => c.EpisodeFileCount = 0)
                .With(c => c.SeasonCount = 0)
                .Build();
            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetSeries(1);

            //Assert
            series.ShouldHave().AllPropertiesBut(s => s.QualityProfile, s => s.SeriesId, s => s.NextAiring).EqualTo(fakeSeries);
            series.QualityProfile.Should().NotBeNull();
            series.QualityProfile.ShouldHave().Properties(q => q.Name, q => q.SonicAllowed, q => q.Cutoff, q => q.SonicAllowed).EqualTo(fakeQuality);

        }

        [Test]
        public void Find_series_by_cleanName_mapped()
        {
            WithRealDb();

            var fakeSeries = Builder<Series>.CreateNew()
                .With(c => c.QualityProfileId = 1)
                .With(c => c.CleanTitle = "laworder")
                .Build();
            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();

            var id = Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            Mocker.Resolve<QualityProvider>();
            Mocker.GetMock<SceneMappingProvider>().Setup(s => s.GetSeriesId("laworder")).Returns(1);

            var series = Mocker.Resolve<SeriesProvider>().FindSeries("laworder");

            //Assert
            series.ShouldHave().AllPropertiesBut(s => s.QualityProfile, s => s.SeriesId);
            series.QualityProfile.Should().NotBeNull();
            series.QualityProfile.ShouldHave().Properties(q => q.Name, q => q.SonicAllowed, q => q.Cutoff, q => q.AllowedString);
        }

        [Test]
        public void find_series_empty_match()
        {
            WithRealDb();

            var fakeSEries = Builder<Series>.CreateNew()
                                                .Build();

            Db.Insert(fakeSEries);

            //Act
            var seriesProvider = Mocker.Resolve<SeriesProvider>();

            //Assert
            seriesProvider.FindSeries("WrongTitle").Should().BeNull();
        }

        [TestCase("The Test", "Test")]
        [TestCase("Through the Wormhole", "Through.the.Wormhole")]
        public void find_series_match(string title, string searchTitle)
        {
            WithRealDb();

            var fakeSeries = Builder<Series>.CreateNew()
                .With(c => c.Title = title)
                .With(c => c.CleanTitle = Parser.NormalizeTitle(title))
                .Build();

            var fakeQuality = Builder<QualityProfile>.CreateNew()
                                    .With(c => c.QualityProfileId = fakeSeries.QualityProfileId)
                                    .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            var series = Mocker.Resolve<SeriesProvider>().FindSeries(searchTitle);

            //Assert
            series.Should().NotBeNull();
            series.SeriesId.Should().Be(series.SeriesId);
            series.Title.Should().BeEquivalentTo(series.Title);
        }

        [Test]
        public void is_monitored()
        {
            WithRealDb();

            Db.Insert(Builder<Series>.CreateNew()
                                                  .With(c => c.Monitored = true)
                                                  .With(c => c.SeriesId = 12)
                                                  .Build());

            Db.Insert(Builder<Series>.CreateNew()
                                                 .With(c => c.Monitored = false)
                                                 .With(c => c.SeriesId = 11)
                                                 .Build());

            Db.InsertMany(Builder<QualityProfile>.CreateListOfSize(3).Build());

            //Act, Assert
            var provider = Mocker.Resolve<SeriesProvider>();
            provider.IsMonitored(12).Should().BeTrue();
            Assert.IsFalse(provider.IsMonitored(11));
            Assert.IsFalse(provider.IsMonitored(1));
        }

        [Test]
        public void Get_Series_With_Count()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
            var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
                .All().With(e => e.SeriesId = fakeSeries.SeriesId)
                .With(e => e.Ignored = false)
                .With(e => e.AirDate = DateTime.Today)
                .TheFirst(5)
                .With(e => e.EpisodeFileId = 0)
                .TheLast(2)
                .With(e => e.AirDate = DateTime.Today.AddDays(1))
                .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);
            Db.InsertMany(fakeEpisodes);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();

            //Assert
            series.Should().HaveCount(1);
            Assert.AreEqual(8, series[0].EpisodeCount);
            Assert.AreEqual(3, series[0].EpisodeFileCount);
        }

        [Test]
        public void Get_Series_With_Count_AllIgnored()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
            var fakeEpisodes = Builder<Episode>.CreateListOfSize(10).All().With(e => e.SeriesId = fakeSeries.SeriesId).With(e => e.Ignored = true).Random(5).With(e => e.EpisodeFileId = 0).Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);
            Db.InsertMany(fakeEpisodes);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();

            //Assert
            series.Should().HaveCount(1);
            Assert.AreEqual(0, series[0].EpisodeCount);
            Assert.AreEqual(0, series[0].EpisodeFileCount);
        }

        [Test]
        public void Get_Series_With_Count_AllDownloaded()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
            var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
                .All()
                .With(e => e.SeriesId = fakeSeries.SeriesId)
                .With(e => e.Ignored = false)
                .With(e => e.AirDate = DateTime.Today.AddDays(-1))
                .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);
            Db.InsertMany(fakeEpisodes);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();

            //Assert
            series.Should().HaveCount(1);
            Assert.AreEqual(10, series[0].EpisodeCount);
            Assert.AreEqual(10, series[0].EpisodeFileCount);
        }

        [Test]
        public void Get_Series_With_Count_Half_Ignored()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
            var fakeEpisodes = Builder<Episode>.CreateListOfSize(10)
                .All()
                .With(e => e.SeriesId = fakeSeries.SeriesId)
                .With(e => e.AirDate = DateTime.Today.AddDays(-1))
                .TheFirst(5)
                .With(e => e.Ignored = false)
                .TheLast(5)
                .With(e => e.Ignored = true)
                .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);
            Db.InsertMany(fakeEpisodes);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();

            //Assert
            series.Should().HaveCount(1);
            Assert.AreEqual(5, series[0].EpisodeCount);
            Assert.AreEqual(5, series[0].EpisodeFileCount);
        }

        [Test]
        public void Get_Single_Series()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew()
                .With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
                .With(e => e.SeriesId = 1)
                .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetSeries(1);

            //Assert
            series.QualityProfile.Should().NotBeNull();
            series.QualityProfileId.Should().Be(fakeQuality.QualityProfileId);
        }

        [Test]
        public void SeriesPathExists_exact_match()
        {
            WithRealDb();

            var path = @"C:\Test\TV\30 Rock";

            var fakeSeries = Builder<Series>.CreateListOfSize(10)
                .All()
                .With(c => c.QualityProfileId = 1)
                .TheFirst(1)
                .With(c => c.Path = path)
                .Build();
            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();

            Db.InsertMany(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            Mocker.Resolve<QualityProvider>();
            //Mocker.GetMock<IDatabase>().Setup(s => s.Fetch<Series, QualityProfile>(It.IsAny<string>())).Returns(
            //fakeSeries.ToList());

            var result = Mocker.Resolve<SeriesProvider>().SeriesPathExists(path);

            //Assert
            result.Should().BeTrue();
        }

        [Test]
        public void SeriesPathExists_match()
        {
            WithRealDb();

            var path = @"C:\Test\TV\30 Rock";

            var fakeSeries = Builder<Series>.CreateListOfSize(10)
                .All()
                .With(c => c.QualityProfileId = 1)
                .TheFirst(1)
                .With(c => c.Path = path)
                .Build();
            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();

            Db.InsertMany(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            Mocker.Resolve<QualityProvider>();
            //Mocker.GetMock<IDatabase>().Setup(s => s.Fetch<Series, QualityProfile>(It.IsAny<string>())).Returns(
            //fakeSeries.ToList());

            var result = Mocker.Resolve<SeriesProvider>().SeriesPathExists(path.ToUpper());

            //Assert
            result.Should().BeTrue();
        }

        [Test]
        public void SeriesPathExists_match_alt()
        {
            WithRealDb();

            var path = @"C:\Test\TV\The Simpsons";

            var fakeSeries = Builder<Series>.CreateListOfSize(10)
                .All()
                .With(c => c.QualityProfileId = 1)
                .TheFirst(1)
                .With(c => c.Path = path)
                .Build();
            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();

            Db.InsertMany(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            Mocker.Resolve<QualityProvider>();
            //Mocker.GetMock<IDatabase>().Setup(s => s.Fetch<Series, QualityProfile>(It.IsAny<string>())).Returns(
            //fakeSeries.ToList());

            var result = Mocker.Resolve<SeriesProvider>().SeriesPathExists(@"c:\Test\Tv\the sIMpsons");

            //Assert
            result.Should().BeTrue();
        }

        [Test]
        public void SeriesPathExists_match_false()
        {
            WithRealDb();

            var path = @"C:\Test\TV\30 Rock";

            var fakeSeries = Builder<Series>.CreateListOfSize(10)
                .All()
                .With(c => c.QualityProfileId = 1)
                .TheFirst(1)
                .With(c => c.Path = path)
                .Build();
            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();

            Db.InsertMany(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            Mocker.Resolve<QualityProvider>();
            //Mocker.GetMock<IDatabase>().Setup(s => s.Fetch<Series, QualityProfile>(It.IsAny<string>())).Returns(
            //fakeSeries.ToList());

            var result = Mocker.Resolve<SeriesProvider>().SeriesPathExists(@"C:\Test\TV\Not A match");

            //Assert
            result.Should().BeFalse();
        }

        [Test]
        public void Get_Series_NextAiring_Today()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
            var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
                .All()
                .With(e => e.SeriesId = fakeSeries.SeriesId)
                .With(e => e.Ignored = false)
                .With(e => e.AirDate = DateTime.Today.AddDays(1))
                .TheFirst(1)
                .With(e => e.AirDate = DateTime.Today)
                .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);
            Db.InsertMany(fakeEpisodes);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();

            //Assert
            series.Should().HaveCount(1);
            series[0].NextAiring.Should().Be(DateTime.Today);
        }

        [Test]
        public void Get_Series_NextAiring_Tomorrow_Last_Aired_Yesterday()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
            var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
                .All()
                .With(e => e.SeriesId = fakeSeries.SeriesId)
                .With(e => e.Ignored = false)
                .With(e => e.AirDate = DateTime.Today.AddDays(1))
                .TheFirst(1)
                .With(e => e.AirDate = DateTime.Today.AddDays(-1))
                .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);
            Db.InsertMany(fakeEpisodes);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();

            //Assert
            series.Should().HaveCount(1);
            series[0].NextAiring.Should().Be(DateTime.Today.AddDays(1));
        }

        [Test]
        public void Get_Series_NextAiring_Unknown()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
            var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
                .All()
                .With(e => e.SeriesId = fakeSeries.SeriesId)
                .With(e => e.AirDate = null)
                .With(e => e.Ignored = false)
                .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);
            Db.InsertMany(fakeEpisodes);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();

            //Assert
            series.Should().HaveCount(1);
            series[0].NextAiring.Should().NotHaveValue();
        }

        [Test]
        public void Get_Series_NextAiring_1_month()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
            var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
                .All()
                .With(e => e.SeriesId = fakeSeries.SeriesId)
                .With(e => e.Ignored = false)
                .With(e => e.AirDate = DateTime.Today.AddMonths(1))
                .TheFirst(1)
                .With(e => e.AirDate = DateTime.Today.AddDays(-1))
                .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);
            Db.InsertMany(fakeEpisodes);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();

            //Assert
            series.Should().HaveCount(1);
            series[0].NextAiring.Should().Be(DateTime.Today.AddMonths(1));
        }

        [Test]
        public void Get_Series_NextAiring_skip_ignored()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateNew().With(e => e.QualityProfileId = fakeQuality.QualityProfileId).Build();
            var fakeEpisodes = Builder<Episode>.CreateListOfSize(2)
                .All()
                .With(e => e.SeriesId = fakeSeries.SeriesId)
                .With(e => e.AirDate = DateTime.Today.AddMonths(1))
                .With(e => e.Ignored = false)
                .TheFirst(1)
                .With(e => e.AirDate = DateTime.Today.AddDays(1))
                .With(e => e.Ignored = true)
                .Build();

            Db.Insert(fakeSeries);
            Db.Insert(fakeQuality);
            Db.InsertMany(fakeEpisodes);

            //Act
            Mocker.Resolve<QualityProvider>();
            var series = Mocker.Resolve<SeriesProvider>().GetAllSeriesWithEpisodeCount();

            //Assert
            series.Should().HaveCount(1);
            series[0].NextAiring.Should().Be(DateTime.Today.AddMonths(1));
        }

        [Test]
        public void SearchForSeries_should_return_results_that_start_with_query()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateListOfSize(10)
                .All()
                .With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
                .Build();

            Db.InsertMany(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("Titl");

            //Assert
            series.Should().HaveCount(10);
        }

        [Test]
        public void SearchForSeries_should_return_results_that_contain_the_query()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateListOfSize(10)
                .All()
                .With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
                .Build();

            Db.InsertMany(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("itl");

            //Assert
            series.Should().HaveCount(10);
        }

        [Test]
        public void SearchForSeries_should_return_results_that_end_with_the_query()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateListOfSize(10)
                .All()
                .With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
                .Build();

            Db.InsertMany(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("2");

            //Assert
            series.Should().HaveCount(1);
        }

        [Test]
        public void SearchForSeries_should_not_return_results_that_do_not_contain_the_query()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateListOfSize(10)
                .All()
                .With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
                .Build();

            Db.InsertMany(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("NotATitle");

            //Assert
            series.Should().HaveCount(0);
        }

        [Test]
        public void SearchForSeries_should_return_results_when_query_has_special_characters()
        {
            WithRealDb();

            var fakeQuality = Builder<QualityProfile>.CreateNew().Build();
            var fakeSeries = Builder<Series>.CreateListOfSize(10)
                .All()
                .With(e => e.QualityProfileId = fakeQuality.QualityProfileId)
                .TheLast(1)
                .With(s => s.Title = "It's Always Sunny")
                .Build();

            Db.InsertMany(fakeSeries);
            Db.Insert(fakeQuality);

            //Act
            var series = Mocker.Resolve<SeriesProvider>().SearchForSeries("it's");

            //Assert
            series.Should().HaveCount(1);
        }
    }
}