// ReSharper disable RedundantUsingDirective using System; using System.Collections.Generic; using AutoMoq; using FizzWare.NBuilder; using Moq; using NUnit.Framework; using NzbDrone.Core.Model.Notification; using NzbDrone.Core.Providers; using NzbDrone.Core.Providers.Jobs; using NzbDrone.Core.Repository; using NzbDrone.Core.Test.Framework; using NzbDrone.Test.Common; namespace NzbDrone.Core.Test.JobTests { [TestFixture] // ReSharper disable InconsistentNaming public class DiskScanJobTest : CoreTest { [Test] public void series_specific_scan_should_scan_series() { var series = Builder.CreateNew() .With(s => s.SeriesId = 12) .Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(p => p.GetSeries(series.SeriesId)) .Returns(series); mocker.GetMock() .Setup(p => p.Scan(series)) .Returns(new List()); //Act mocker.Resolve().Start(new ProgressNotification("Test"), series.SeriesId, 0); //Assert mocker.VerifyAllMocks(); } [Test] public void job_with_no_target_should_scan_all_series() { var series = Builder.CreateListOfSize(2) .TheFirst(1).With(s => s.SeriesId = 12) .TheNext(1).With(s => s.SeriesId = 15) .Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(p => p.GetAllSeries()) .Returns(series); mocker.GetMock() .Setup(s => s.Scan(series[0])) .Returns(new List()); mocker.GetMock() .Setup(s => s.Scan(series[1])) .Returns(new List()); mocker.Resolve().Start(new ProgressNotification("Test"), 0, 0); mocker.VerifyAllMocks(); } [Test] public void failed_scan_should_not_terminated_job() { var series = Builder.CreateListOfSize(2) .TheFirst(1).With(s => s.SeriesId = 12) .TheNext(1).With(s => s.SeriesId = 15) .Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(p => p.GetAllSeries()) .Returns(series); mocker.GetMock() .Setup(s => s.Scan(series[0])) .Throws(new InvalidOperationException("Bad Job")); mocker.GetMock() .Setup(s => s.Scan(series[1])) .Throws(new InvalidOperationException("Bad Job")); mocker.Resolve().Start(new ProgressNotification("Test"), 0, 0); mocker.VerifyAllMocks(); ExceptionVerification.ExcpectedErrors(2); } [Test] public void job_with_no_target_should_scan_series_with_episodes() { var series = Builder.CreateListOfSize(2) .TheFirst(1).With(s => s.SeriesId = 12) .TheNext(1).With(s => s.SeriesId = 15) .Build(); var mocker = new AutoMoqer(MockBehavior.Strict); mocker.GetMock() .Setup(p => p.GetAllSeries()) .Returns(series); mocker.GetMock() .Setup(s => s.Scan(series[0])) .Returns(new List()); mocker.GetMock() .Setup(s => s.Scan(series[1])) .Returns(new List()); mocker.Resolve().Start(new ProgressNotification("Test"), 0, 0); mocker.VerifyAllMocks(); mocker.GetMock().Verify(s => s.Scan(It.IsAny()), Times.Exactly(2)); } } }