using System; using System.Linq; using FizzWare.NBuilder; using FluentAssertions; using NUnit.Framework; using NzbDrone.Common; using NzbDrone.Core.Providers; using NzbDrone.Core.Providers.Core; using NzbDrone.Core.Repository; using NzbDrone.Core.Test.Framework; using NzbDrone.Test.Common; namespace NzbDrone.Core.Test.ProviderTests { [TestFixture] // ReSharper disable InconsistentNaming public class ReferenceDataProviderTest : CoreTest { private const string validSeriesIds = "[1,2,3,4,5]"; private const string invalidSeriesIds = "[1,2,NaN,4,5]"; private const string url = "http://services.nzbdrone.com/DailySeries/AllIds"; [SetUp] public void Setup() { Mocker.GetMock().SetupGet(s => s.ServiceRootUrl) .Returns("http://services.nzbdrone.com"); } [Test] public void GetDailySeriesIds_should_return_list_of_int_when_all_are_valid() { //Setup Mocker.GetMock().Setup(s => s.DownloadString(url)) .Returns(validSeriesIds); //Act var result = Mocker.Resolve().GetDailySeriesIds(); //Assert result.Should().HaveCount(5); } [Test] public void GetDailySeriesIds_should_return_empty_list_when_unable_to_parse() { //Setup Mocker.GetMock().Setup(s => s.DownloadString(url)) .Returns(invalidSeriesIds); //Act var result = Mocker.Resolve().GetDailySeriesIds(); //Assert result.Should().BeEmpty(); ExceptionVerification.ExpectedWarns(1); } [Test] public void GetDailySeriesIds_should_return_empty_list_of_int_when_server_is_unavailable() { //Setup Mocker.GetMock().Setup(s => s.DownloadString(url)) .Throws(new Exception()); //Act var result = Mocker.Resolve().GetDailySeriesIds(); //Assert result.Should().HaveCount(0); ExceptionVerification.ExpectedWarns(1); } [Test] public void IsDailySeries_should_return_true() { //Setup Mocker.GetMock().Setup(s => s.DownloadString(url)) .Returns(validSeriesIds); //Act var result = Mocker.Resolve().IsSeriesDaily(1); //Assert result.Should().BeTrue(); } [Test] public void IsDailySeries_should_return_false() { //Setup Mocker.GetMock().Setup(s => s.DownloadString(url)) .Returns(validSeriesIds); //Act var result = Mocker.Resolve().IsSeriesDaily(10); //Assert result.Should().BeFalse(); } [Test] public void UpdateDailySeries_should_update_series_that_match_daily_series_list() { WithRealDb(); var fakeSeries = Builder.CreateListOfSize(5) .All() .With(s => s.IsDaily = false) .Build(); Db.InsertMany(fakeSeries); //Setup Mocker.GetMock().Setup(s => s.DownloadString(url)) .Returns(validSeriesIds); //Act Mocker.Resolve().UpdateDailySeries(); //Assert var result = Db.Fetch(); result.Where(s => s.IsDaily).Should().HaveCount(5); } [Test] public void UpdateDailySeries_should_update_series_should_skip_series_that_dont_match() { WithRealDb(); var fakeSeries = Builder.CreateListOfSize(5) .All() .With(s => s.IsDaily = false) .TheFirst(1) .With(s => s.SeriesId = 10) .TheNext(1) .With(s => s.SeriesId = 11) .TheNext(1) .With(s => s.SeriesId = 12) .Build(); Db.InsertMany(fakeSeries); //Setup Mocker.GetMock().Setup(s => s.DownloadString(url)) .Returns(validSeriesIds); //Act Mocker.Resolve().UpdateDailySeries(); //Assert var result = Db.Fetch(); result.Where(s => !s.IsDaily).Should().HaveCount(3); result.Where(s => s.IsDaily).Should().HaveCount(2); } [Test] public void UpdateDailySeries_should_update_series_should_not_overwrite_existing_isDaily() { WithRealDb(); var fakeSeries = Builder.CreateListOfSize(5) .All() .With(s => s.IsDaily = false) .TheFirst(1) .With(s => s.SeriesId = 10) .With(s => s.IsDaily = true) .TheNext(1) .With(s => s.SeriesId = 11) .TheNext(1) .With(s => s.SeriesId = 12) .Build(); Db.InsertMany(fakeSeries); //Setup Mocker.GetMock().Setup(s => s.DownloadString(url)) .Returns(validSeriesIds); //Act Mocker.Resolve().UpdateDailySeries(); //Assert var result = Db.Fetch(); result.Where(s => s.IsDaily).Should().HaveCount(3); result.Where(s => !s.IsDaily).Should().HaveCount(2); } } }