// ReSharper disable RedundantUsingDirective using System; using System.Collections.Generic; using System.Linq; using AutoMoq; using FizzWare.NBuilder; using MbUnit.Framework; using Moq; using NzbDrone.Core.Providers; using NzbDrone.Core.Repository; using NzbDrone.Core.Repository.Quality; using NzbDrone.Core.Test.Framework; using SubSonic.Repository; using TvdbLib.Data; // ReSharper disable InconsistentNaming namespace NzbDrone.Core.Test { [TestFixture] // ReSharper disable InconsistentNaming public class SeriesProviderTest : TestBase { [Test] public void Map_path_to_series() { //Arrange var fakeSeries = Builder.CreateNew() .With(f => f.SeriesName = "The Simpsons") .Build(); var fakeSearch = Builder.CreateNew() .With(s => s.Id = fakeSeries.Id) .With(s => s.SeriesName = fakeSeries.SeriesName) .Build(); var mocker = new AutoMoqer(); mocker.GetMock() .Setup(f => f.Exists(c => c.SeriesId == It.IsAny())) .Returns(false); mocker.GetMock() .Setup(f => f.GetSeries(It.IsAny())) .Returns(fakeSearch); mocker.GetMock() .Setup(f => f.GetSeries(fakeSeries.Id, false)) .Returns(fakeSeries) .Verifiable(); //Act var mappedSeries = mocker.Resolve().MapPathToSeries(@"D:\TV Shows\The Simpsons"); //Assert mocker.GetMock().VerifyAll(); Assert.AreEqual(fakeSeries, mappedSeries); } [Test] public void Add_new_series() { var mocker = new AutoMoqer(); mocker.SetConstant(MockLib.GetEmptyRepository()); const string path = "C:\\Test\\"; const int tvDbId = 1234; const int qualityProfileId = 2; //Act var seriesProvider = mocker.Resolve(); seriesProvider.AddSeries(path, tvDbId, qualityProfileId); //Assert var series = seriesProvider.GetAllSeries(); Assert.IsNotEmpty(series); Assert.Count(1, series); Assert.AreEqual(path, series.First().Path); Assert.AreEqual(tvDbId, series.First().SeriesId); Assert.AreEqual(qualityProfileId, series.First().QualityProfileId); } [Test] public void find_series_empty_repo() { var mocker = new AutoMoqer(); mocker.SetConstant(MockLib.GetEmptyRepository()); //Act var seriesProvider = mocker.Resolve(); var series = seriesProvider.FindSeries("My Title"); //Assert Assert.IsNull(series); } [Test] public void find_series_empty_match() { var mocker = new AutoMoqer(); var emptyRepository = MockLib.GetEmptyRepository(); mocker.SetConstant(emptyRepository); emptyRepository.Add(MockLib.GetFakeSeries(1, "MyTitle")); //Act var seriesProvider = mocker.Resolve(); var series = seriesProvider.FindSeries("WrongTitle"); //Assert Assert.IsNull(series); } [Test] [Row("The Test", "Test")] [Row("Through the Wormhole", "Through.the.Wormhole")] public void find_series_match(string title, string searchTitle) { var mocker = new AutoMoqer(); var emptyRepository = MockLib.GetEmptyRepository(); mocker.SetConstant(emptyRepository); emptyRepository.Add(MockLib.GetFakeSeries(1, title)); //Act var seriesProvider = mocker.Resolve(); var series = seriesProvider.FindSeries(searchTitle); //Assert Assert.IsNotNull(series); Assert.AreEqual(title, series.Title); } [Test] public void is_monitored() { var mocker = new AutoMoqer(); mocker.SetConstant(MockLib.GetEmptyRepository()); mocker.Resolve().Add(Builder.CreateNew() .With(c => c.Monitored = true) .With(c => c.SeriesId = 12) .Build()); mocker.Resolve().Add(Builder.CreateNew() .With(c => c.Monitored = false) .With(c => c.SeriesId = 11) .Build()); //Act, Assert var provider = mocker.Resolve(); Assert.IsTrue(provider.IsMonitored(12)); Assert.IsFalse(provider.IsMonitored(11)); Assert.IsFalse(provider.IsMonitored(1)); } [Test] [Row(QualityTypes.SDTV, true)] [Row(QualityTypes.Unknown, false)] [Row(QualityTypes.Bluray1080p, false)] [Row(QualityTypes.Bluray720p, false)] [Row(QualityTypes.HDTV, false)] [Row(QualityTypes.WEBDL, false)] public void QualityWanted(QualityTypes qualityTypes, Boolean result) { var quality = Builder.CreateNew() .With(q => q.Allowed = new List { QualityTypes.DVD, QualityTypes.SDTV }) .With(q => q.Cutoff = QualityTypes.DVD) .Build(); var series = Builder.CreateNew() .With(c => c.SeriesId = 12) .With(c => c.QualityProfileId = quality.QualityProfileId) .Build(); var mocker = new AutoMoqer(); var emptyRepository = MockLib.GetEmptyRepository(); mocker.SetConstant(emptyRepository); mocker.GetMock() .Setup(c => c.Find(quality.QualityProfileId)).Returns(quality); emptyRepository.Add(series); emptyRepository.Add(quality); //Act var needed = mocker.Resolve().QualityWanted(12, qualityTypes); Assert.AreEqual(result, needed); } } }