using System; using System.Collections.Generic; using FizzWare.NBuilder; using Moq; using NUnit.Framework; using NzbDrone.Common; using NzbDrone.Core.MediaFiles; using NzbDrone.Core.MediaFiles.Commands; using NzbDrone.Core.Test.Framework; using NzbDrone.Core.Tv; using System.Linq; namespace NzbDrone.Core.Test.MediaFileTests { public class MediaFileTableCleanupServiceFixture : CoreTest { private const string DeletedPath = "ANY FILE WITH THIS PATH IS CONSIDERED DELETED!"; [SetUp] public void SetUp() { Mocker.GetMock() .Setup(s => s.GetSeries(It.IsAny())) .Returns(Builder.CreateNew().Build()); Mocker.GetMock() .Setup(e => e.FileExists(It.Is(c => c != DeletedPath))) .Returns(true); Mocker.GetMock() .Setup(c => c.GetEpisodesByFileId(It.IsAny())) .Returns(new List {new Episode()}); Mocker.GetMock() .Setup(s => s.IsParent(It.IsAny(), It.IsAny())) .Returns(true); } private void GivenEpisodeFiles(IEnumerable episodeFiles) { Mocker.GetMock() .Setup(c => c.GetFilesBySeries(It.IsAny())) .Returns(episodeFiles.ToList()); } private void GivenFilesAreNotAttachedToEpisode() { Mocker.GetMock() .Setup(c => c.GetEpisodesByFileId(It.IsAny())) .Returns(new List()); } private void GivenFileIsNotInSeriesFolder() { Mocker.GetMock() .Setup(s => s.IsParent(It.IsAny(), It.IsAny())) .Returns(false); } [Test] public void should_skip_files_that_exist_in_disk() { var episodeFiles = Builder.CreateListOfSize(10) .Build(); GivenEpisodeFiles(episodeFiles); Subject.Execute(new CleanMediaFileDb(0)); Mocker.GetMock().Verify(c => c.UpdateEpisode(It.IsAny()), Times.Never()); } [Test] public void should_delete_none_existing_files() { var episodeFiles = Builder.CreateListOfSize(10) .Random(2) .With(c => c.Path = DeletedPath) .Build(); GivenEpisodeFiles(episodeFiles); Subject.Execute(new CleanMediaFileDb(0)); Mocker.GetMock().Verify(c => c.Delete(It.Is(e => e.Path == DeletedPath), false), Times.Exactly(2)); } [Test] public void should_delete_files_that_dont_belong_to_any_episodes() { var episodeFiles = Builder.CreateListOfSize(10) .Random(10) .With(c => c.Path = "ExistingPath") .Build(); GivenEpisodeFiles(episodeFiles); GivenFilesAreNotAttachedToEpisode(); Subject.Execute(new CleanMediaFileDb(0)); Mocker.GetMock().Verify(c => c.Delete(It.IsAny(), false), Times.Exactly(10)); } [Test] public void should_delete_files_that_do_not_belong_to_the_series_path() { var episodeFiles = Builder.CreateListOfSize(10) .Random(10) .With(c => c.Path = "ExistingPath") .Build(); GivenEpisodeFiles(episodeFiles); GivenFileIsNotInSeriesFolder(); Subject.Execute(new CleanMediaFileDb(0)); Mocker.GetMock().Verify(c => c.Delete(It.IsAny(), false), Times.Exactly(10)); } } }