using System; using System.IO; using Moq; using NUnit.Framework; using NzbDrone.Common; using NzbDrone.Core.Configuration; using NzbDrone.Core.MediaFiles; using NzbDrone.Core.Test.Framework; namespace NzbDrone.Core.Test.ProviderTests.RecycleBinProviderTests { [TestFixture] public class CleanupFixture : CoreTest { private const string RecycleBin = @"C:\Test\RecycleBin"; private void WithExpired() { Mocker.GetMock().Setup(s => s.GetLastFolderWrite(It.IsAny())) .Returns(DateTime.UtcNow.AddDays(-10)); Mocker.GetMock().Setup(s => s.GetLastFileWrite(It.IsAny())) .Returns(DateTime.UtcNow.AddDays(-10)); } private void WithNonExpired() { Mocker.GetMock().Setup(s => s.GetLastFolderWrite(It.IsAny())) .Returns(DateTime.UtcNow.AddDays(-3)); Mocker.GetMock().Setup(s => s.GetLastFileWrite(It.IsAny())) .Returns(DateTime.UtcNow.AddDays(-3)); } [SetUp] public void Setup() { Mocker.GetMock().SetupGet(s => s.RecycleBin).Returns(RecycleBin); Mocker.GetMock().Setup(s => s.GetDirectories(RecycleBin)) .Returns(new [] { @"C:\Test\RecycleBin\Folder1", @"C:\Test\RecycleBin\Folder2", @"C:\Test\RecycleBin\Folder3" }); Mocker.GetMock().Setup(s => s.GetFiles(RecycleBin, SearchOption.TopDirectoryOnly)) .Returns(new [] { @"C:\Test\RecycleBin\File1.avi", @"C:\Test\RecycleBin\File2.mkv" }); } [Test] public void should_return_if_recycleBin_not_configured() { Mocker.GetMock().SetupGet(s => s.RecycleBin).Returns(String.Empty); Mocker.Resolve().Cleanup(); Mocker.GetMock().Verify(v => v.GetDirectories(It.IsAny()), Times.Never()); } [Test] public void should_delete_all_expired_folders() { WithExpired(); Mocker.Resolve().Cleanup(); Mocker.GetMock().Verify(v => v.DeleteFolder(It.IsAny(), true), Times.Exactly(3)); } [Test] public void should_delete_all_expired_files() { WithExpired(); Mocker.Resolve().Cleanup(); Mocker.GetMock().Verify(v => v.DeleteFile(It.IsAny()), Times.Exactly(2)); } [Test] public void should_not_delete_all_non_expired_folders() { WithNonExpired(); Mocker.Resolve().Cleanup(); Mocker.GetMock().Verify(v => v.DeleteFolder(It.IsAny(), true), Times.Never()); } [Test] public void should_not_delete_all_non_expired_files() { WithNonExpired(); Mocker.Resolve().Cleanup(); Mocker.GetMock().Verify(v => v.DeleteFile(It.IsAny()), Times.Never()); } } }