Sonarr/NzbDrone.Core.Test/EpisodeProviderTest.cs

using System;
using System.Linq;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq.Expressions;
using AutoMoq;
using FizzWare.NBuilder;
using MbUnit.Framework;
using Moq;
using NzbDrone.Core.Model;
using NzbDrone.Core.Providers;
using NzbDrone.Core.Repository;
using NzbDrone.Core.Repository.Quality;
using SubSonic.Repository;
using TvdbLib.Data;

namespace NzbDrone.Core.Test
{
    [TestFixture]
    // ReSharper disable InconsistentNaming
    public class EpisodeProviderTest
    {
        [Test]
        public void RefreshEpisodeInfo()
        {
            //Arrange
            const int seriesId = 71663;
            const int episodeCount = 10;

            var fakeEpisodes = Builder<TvdbSeries>.CreateNew().With(
                c => c.Episodes =
                     new List<TvdbEpisode>(Builder<TvdbEpisode>.CreateListOfSize(episodeCount).
                                               WhereAll()
                                               .Have(l => l.Language = new TvdbLanguage(0, "eng", "a"))
                                               .Build())
                ).With(c => c.Id = seriesId).Build();

            var mocker = new AutoMoqer();

            mocker.SetConstant(MockLib.GetEmptyRepository());

            mocker.GetMock<TvDbProvider>()
                .Setup(c => c.GetSeries(seriesId, true))
                .Returns(fakeEpisodes).Verifiable();

            //mocker.GetMock<IRepository>().SetReturnsDefault();


            //Act
            var sw = Stopwatch.StartNew();
            mocker.Resolve<EpisodeProvider>().RefreshEpisodeInfo(seriesId);
            var actualCount = mocker.Resolve<EpisodeProvider>().GetEpisodeBySeries(seriesId);
            //Assert
            mocker.GetMock<TvDbProvider>().VerifyAll();
            Assert.Count(episodeCount, actualCount);
            Console.WriteLine("Duration: " + sw.Elapsed);
        }

        [Test]

        //Should Download
        [Row(QualityTypes.TV, true, QualityTypes.HDTV, false, true)]
        [Row(QualityTypes.DVD, true, QualityTypes.Bluray720, true, true)]
        [Row(QualityTypes.HDTV, false, QualityTypes.HDTV, true, true)]


        [Row(QualityTypes.HDTV, false, QualityTypes.HDTV, false, false)]
        [Row(QualityTypes.Bluray720, true, QualityTypes.Bluray1080, false, false)]
        [Row(QualityTypes.HDTV, true, QualityTypes.Bluray720, true, false)]
        [Row(QualityTypes.Bluray1080, true, QualityTypes.Bluray720, true, false)]
        [Row(QualityTypes.Bluray1080, false, QualityTypes.Bluray720, true, false)]
        [Row(QualityTypes.Bluray1080, false, QualityTypes.Bluray720, true, false)]
        [Row(QualityTypes.HDTV, false, QualityTypes.Bluray720, true, false)]
        public void Is_Needed_Tv_Dvd_BluRay_BluRay720_Is_Cutoff(QualityTypes fileQuality, bool isFileProper, QualityTypes reportQuality, bool isReportProper, bool excpected)
        {
            //Setup
            var parseResult = new EpisodeParseResult
                                  {
                                      SeriesId = 12,
                                      SeasonNumber = 2,
                                      Episodes = new List<int> { 3 },
                                      Quality = reportQuality,
                                      Proper = isReportProper
                                  };

            var epFile = new EpisodeFile
                             {
                                 Proper = isFileProper,
                                 Quality = fileQuality
                             };

            var seriesQualityProfile = new QualityProfile
            {
                Name = "HD",
                Allowed = new List<QualityTypes> { QualityTypes.HDTV, QualityTypes.WEBDL, QualityTypes.BDRip, QualityTypes.Bluray720 },
                Cutoff = QualityTypes.HDTV
            };

            var episodeInfo = new Episode
                                  {
                                      SeriesId = 12,
                                      SeasonNumber = 2,
                                      EpisodeNumber = 3,
                                      Series = new Series { QualityProfileId = 1, QualityProfile = seriesQualityProfile },
                                      EpisodeFile = epFile

                                  };





            var mocker = new AutoMoqer();
            mocker.GetMock<IRepository>()
                .Setup(r => r.Single(It.IsAny<Expression<Func<Episode, Boolean>>>()))
                .Returns(episodeInfo);

            mocker.GetMock<QualityProvider>()
                .Setup(q => q.Find(1))
                .Returns(seriesQualityProfile);

            var result = mocker.Resolve<EpisodeProvider>().IsNeeded(parseResult);

            Assert.AreEqual(excpected, result);
        }

        [Test]
        public void get_episode_by_parse_result()
        {
            var mocker = new AutoMoqer();
            var repo = MockLib.GetEmptyRepository();
            var fakeEpisodes = MockLib.GetFakeEpisodes(2);
            repo.AddMany(fakeEpisodes);
            mocker.SetConstant(repo);

            var targetEpisode = fakeEpisodes[4];

            var parseResult1 = new EpisodeParseResult
            {
                SeriesId = targetEpisode.SeriesId,
                SeasonNumber = targetEpisode.SeasonNumber,
                Episodes = new List<int> { targetEpisode.EpisodeNumber },
                Quality = QualityTypes.DVD
            };

            var result = mocker.Resolve<EpisodeProvider>().GetEpisodeByParseResult(parseResult1);


            Assert.Count(1, result);
            Assert.AreEqual(targetEpisode.EpisodeId, result.First().EpisodeId);
            Assert.AreEqual(targetEpisode.EpisodeNumber, result.First().EpisodeNumber);
            Assert.AreEqual(targetEpisode.SeasonNumber, result.First().SeasonNumber);
            Assert.AreEqual(targetEpisode.SeriesId, result.First().SeriesId);
        }

        [Test]
        public void Missing_episode_should_be_added()
        {
            //Setup
            var parseResult1 = new EpisodeParseResult
            {
                SeriesId = 12,
                SeasonNumber = 2,
                Episodes = new List<int> { 3 },
                Quality = QualityTypes.DVD

            };

            var parseResult2 = new EpisodeParseResult
            {
                SeriesId = 12,
                SeasonNumber = 3,
                Episodes = new List<int> { 3 },
                Quality = QualityTypes.DVD

            };

            var mocker = new AutoMoqer();
            mocker.SetConstant(MockLib.GetEmptyRepository());

            var episodeProvider = mocker.Resolve<EpisodeProvider>();
            var result1 = episodeProvider.IsNeeded(parseResult1);
            var result2 = episodeProvider.IsNeeded(parseResult2);
            var episodes = episodeProvider.GetEpisodeBySeries(12);
            Assert.IsTrue(result1);
            Assert.IsTrue(result2);
            Assert.IsNotEmpty(episodes);
            Assert.Count(2, episodes);
        }




        [Test]
        [Explicit]
        public void Add_daily_show_episodes()
        {
            var mocker = new AutoMoqer();
            mocker.SetConstant(MockLib.GetEmptyRepository());
            mocker.Resolve<TvDbProvider>();
            const int tvDbSeriesId = 71256;
            //act
            var seriesProvider = mocker.Resolve<SeriesProvider>();

            seriesProvider.AddSeries("c:\\test\\", tvDbSeriesId, 0);
            var episodeProvider = mocker.Resolve<EpisodeProvider>();
            episodeProvider.RefreshEpisodeInfo(tvDbSeriesId);

            //assert
            var episodes = episodeProvider.GetEpisodeBySeries(tvDbSeriesId);
            Assert.IsNotEmpty(episodes);
        }
    }
}