using System.Linq; using System; using System.Collections.Generic; using System.Threading; using FizzWare.NBuilder; using FluentAssertions; using NCrunch.Framework; using NUnit.Framework; using NzbDrone.Common; using NzbDrone.Core.Jobs; using NzbDrone.Core.Model; using NzbDrone.Core.Test.Framework; using NzbDrone.Test.Common; namespace NzbDrone.Core.Test.JobTests { [TestFixture] public class JobRepositoryFixture : DbTest { FakeJob _fakeJob; DisabledJob _disabledJob; [SetUp] public void Setup() { _fakeJob = new FakeJob(); _disabledJob = new DisabledJob(); } [Test] public void Init_should_add_defintaions() { IEnumerable baseFakeJobs = new List { _fakeJob }; Mocker.SetConstant(baseFakeJobs); Subject.Init(); Storage.All().Should().HaveCount(1); StoredModel.Interval.Should().Be((Int32)_fakeJob.DefaultInterval.TotalMinutes); StoredModel.Name.Should().Be(_fakeJob.Name); StoredModel.Type.Should().Be(_fakeJob.GetType().ToString()); StoredModel.LastExecution.Should().HaveYear(DateTime.Now.Year); StoredModel.LastExecution.Should().HaveMonth(DateTime.Now.Month); StoredModel.LastExecution.Should().HaveDay(DateTime.Today.Day); StoredModel.Enable.Should().BeTrue(); } [Test] public void inti_should_removed_jobs_that_no_longer_exist() { IEnumerable fakeJobs = new List { _fakeJob }; Mocker.SetConstant(fakeJobs); Subject.Init(); var deletedJob = Builder.CreateNew() .With(c => c.Id = 0) .Build(); Db.Insert(deletedJob); Subject.Init(); AllStoredModels.Should().HaveCount(1); AllStoredModels.Should().NotContain(c => c.Type == deletedJob.Type); } [Test] public void inti_should_removed_jobs_that_no_longer_exist_even_with_same_name() { IEnumerable fakeJobs = new List { _fakeJob }; Mocker.SetConstant(fakeJobs); Subject.Init(); var deletedJob = Builder.CreateNew() .With(c => c.Name = _fakeJob.Name) .With(c => c.Id = 0) .Build(); Db.Insert(deletedJob); Subject.Init(); AllStoredModels.Should().HaveCount(1); AllStoredModels.Should().NotContain(c => c.Type == deletedJob.Type); } [Test] public void init_should_update_existing_job() { var oldJob = Builder.CreateNew() .With(c => c.Id = 0) .With(c => c.Name = "OldName") .With(c => c.Type = typeof(FakeJob).ToString()) .With(c => c.Interval = 0) .With(c => c.Enable = true) .With(c => c.Success = true) .With(c => c.LastExecution = DateTime.Now.AddDays(-7).Date) .Build(); Storage.Insert(oldJob); var newJob = new FakeJob(); IEnumerable fakeJobs = new List { newJob }; Mocker.SetConstant(fakeJobs); Subject.Init(); AllStoredModels.Should().HaveCount(1); StoredModel.Type.Should().Be(newJob.GetType().FullName); StoredModel.Name.Should().Be(newJob.Name); StoredModel.Interval.Should().Be((int)newJob.DefaultInterval.TotalMinutes); StoredModel.Enable.Should().Be(true); StoredModel.Success.Should().Be(oldJob.Success); StoredModel.LastExecution.Should().Be(oldJob.LastExecution); } [Test] public void jobs_with_zero_interval_are_registered_as_disabled() { IEnumerable fakeJobs = new List { _disabledJob }; Mocker.SetConstant(fakeJobs); Subject.Init(); Storage.All().Should().HaveCount(1); Storage.All().First().Enable.Should().BeFalse(); } [Test] public void pending_job_should_get_jobs_that_have_matured() { var oldJob = Builder.CreateNew() .With(c => c.Id = 0) .With(c => c.Interval = 1) .With(c => c.Enable = true) .With(c => c.Success = true) .With(c => c.LastExecution = DateTime.Now.AddMinutes(-5)) .Build(); Storage.Insert(oldJob); Subject.GetPendingJobs().Should().HaveCount(1); } [Test] public void pending_job_should_not_get_jobs_that_havent_matured() { var recent = Builder.CreateNew() .With(c => c.Id = 0) .With(c => c.Interval = 60) .With(c => c.Enable = true) .With(c => c.Success = true) .With(c => c.LastExecution = DateTime.Now.AddMinutes(-5)) .Build(); Storage.Insert(recent); Subject.GetPendingJobs().Should().BeEmpty(); } /* [Test] public void disabled_jobs_arent_run_by_scheduler() { IEnumerable BaseFakeJobs = new List { disabledJob }; Mocker.SetConstant(BaseFakeJobs); var jobProvider = Mocker.Resolve(); jobProvider.QueueScheduled(); WaitForQueue(); disabledJob.ExecutionCount.Should().Be(0); }*/ } }