From 9835608ff0d4f0f85e7e8238a3c0d3d7d60c82e8 Mon Sep 17 00:00:00 2001
From: Keivan Beigi <kay.one@gmail.com>
Date: Tue, 16 Jul 2013 10:55:11 -0700
Subject: [PATCH] skip episode folder if drop folder doesn't exist.

---
 .../DropFolderImportServiceFixture.cs         | 32 +++++++++++++++----
 .../DownloadedEpisodesImportService.cs        | 25 ++++++++-------
 2 files changed, 40 insertions(+), 17 deletions(-)

diff --git a/NzbDrone.Core.Test/MediaFileTests/DropFolderImportServiceFixture.cs b/NzbDrone.Core.Test/MediaFileTests/DropFolderImportServiceFixture.cs
index e96ddfd78..5e3bd33b7 100644
--- a/NzbDrone.Core.Test/MediaFileTests/DropFolderImportServiceFixture.cs
+++ b/NzbDrone.Core.Test/MediaFileTests/DropFolderImportServiceFixture.cs
@@ -1,4 +1,3 @@
-using System;
 using System.Collections.Generic;
 using System.IO;
 using FizzWare.NBuilder;
@@ -7,11 +6,11 @@
 using NzbDrone.Common;
 using NzbDrone.Core.Configuration;
 using NzbDrone.Core.MediaFiles;
+using NzbDrone.Core.MediaFiles.Commands;
 using NzbDrone.Core.MediaFiles.EpisodeImport;
-using NzbDrone.Core.MediaFiles.Events;
 using NzbDrone.Core.Parser;
-using NzbDrone.Core.Tv;
 using NzbDrone.Core.Test.Framework;
+using NzbDrone.Test.Common;
 
 namespace NzbDrone.Core.Test.MediaFileTests
 {
@@ -34,14 +33,20 @@ public void Setup()
             Mocker.GetMock<IDiskProvider>().Setup(c => c.GetDirectories(It.IsAny<string>()))
                   .Returns(_subFolders);
 
+
+            Mocker.GetMock<IDiskProvider>().Setup(c => c.FolderExists(It.IsAny<string>()))
+                  .Returns(true);
+
             Mocker.GetMock<IConfigService>().SetupGet(c => c.DownloadedEpisodesFolder)
                   .Returns("c:\\drop\\");
+
+
         }
 
         [Test]
         public void should_import_file()
         {
-            Subject.ProcessDownloadedEpisodesFolder();
+            Subject.Execute(new DownloadedEpisodesScanCommand());
 
             VerifyImport();
         }
@@ -49,18 +54,33 @@ public void should_import_file()
         [Test]
         public void should_search_for_series_using_folder_name()
         {
-            Subject.ProcessDownloadedEpisodesFolder();
+            Subject.Execute(new DownloadedEpisodesScanCommand());
+
 
             Mocker.GetMock<IParsingService>().Verify(c => c.GetSeries("foldername"), Times.Once());
         }
 
+        [Test]
+        public void should_skip_import_if_dropfolder_doesnt_exist()
+        {
+            Mocker.GetMock<IDiskProvider>().Setup(c => c.FolderExists(It.IsAny<string>())).Returns(false);
+            
+            Subject.Execute(new DownloadedEpisodesScanCommand());
+
+            Mocker.GetMock<IDiskProvider>().Verify(c => c.GetDirectories(It.IsAny<string>()), Times.Never());
+            Mocker.GetMock<IDiskProvider>().Verify(c => c.GetFiles(It.IsAny<string>(), It.IsAny<SearchOption>()), Times.Never());
+
+            ExceptionVerification.ExpectedWarns(1);
+        }
+
         [Test]
         public void should_skip_if_file_is_in_use_by_another_process()
         {
             Mocker.GetMock<IDiskProvider>().Setup(c => c.IsFileLocked(It.IsAny<FileInfo>()))
                   .Returns(true);
 
-            Subject.ProcessDownloadedEpisodesFolder();
+            Subject.Execute(new DownloadedEpisodesScanCommand());
+
             VerifyNoImport();
         }
 
diff --git a/NzbDrone.Core/MediaFiles/DownloadedEpisodesImportService.cs b/NzbDrone.Core/MediaFiles/DownloadedEpisodesImportService.cs
index f91beaf08..dd6bb58f4 100644
--- a/NzbDrone.Core/MediaFiles/DownloadedEpisodesImportService.cs
+++ b/NzbDrone.Core/MediaFiles/DownloadedEpisodesImportService.cs
@@ -18,7 +18,6 @@ public class DownloadedEpisodesImportService : IExecute<DownloadedEpisodesScanCo
         private readonly IDiskProvider _diskProvider;
         private readonly IDiskScanService _diskScanService;
         private readonly ISeriesService _seriesService;
-        private readonly IMoveEpisodeFiles _episodeFileMover;
         private readonly IParsingService _parsingService;
         private readonly IConfigService _configService;
         private readonly IMakeImportDecision _importDecisionMaker;
@@ -28,7 +27,6 @@ public class DownloadedEpisodesImportService : IExecute<DownloadedEpisodesScanCo
         public DownloadedEpisodesImportService(IDiskProvider diskProvider,
             IDiskScanService diskScanService,
             ISeriesService seriesService,
-            IMoveEpisodeFiles episodeFileMover,
             IParsingService parsingService,
             IConfigService configService,
             IMakeImportDecision importDecisionMaker,
@@ -38,7 +36,6 @@ public DownloadedEpisodesImportService(IDiskProvider diskProvider,
             _diskProvider = diskProvider;
             _diskScanService = diskScanService;
             _seriesService = seriesService;
-            _episodeFileMover = episodeFileMover;
             _parsingService = parsingService;
             _configService = configService;
             _importDecisionMaker = importDecisionMaker;
@@ -46,7 +43,7 @@ public DownloadedEpisodesImportService(IDiskProvider diskProvider,
             _logger = logger;
         }
 
-        public void ProcessDownloadedEpisodesFolder()
+        private void ProcessDownloadedEpisodesFolder()
         {
             //TODO: We should also process the download client's category folder
             var downloadedEpisodesFolder = _configService.DownloadedEpisodesFolder;
@@ -57,24 +54,30 @@ public void ProcessDownloadedEpisodesFolder()
                 return;
             }
 
-            foreach (var subfolder in _diskProvider.GetDirectories(downloadedEpisodesFolder))
+            if (!_diskProvider.FolderExists(downloadedEpisodesFolder))
+            {
+                _logger.Warn("Downloaded Episodes Folder [{0}] doesn't exist.", downloadedEpisodesFolder);
+                return;
+            }
+
+            foreach (var subFolder in _diskProvider.GetDirectories(downloadedEpisodesFolder))
             {
                 try
                 {
-                    if (!_seriesService.SeriesPathExists(subfolder))
+                    if (!_seriesService.SeriesPathExists(subFolder))
                     {
-                        ProcessSubFolder(new DirectoryInfo(subfolder));
+                        ProcessSubFolder(new DirectoryInfo(subFolder));
 
                         //Todo: We should make sure the file(s) are actually imported
-                        if (_diskProvider.GetFolderSize(subfolder) < NotSampleSpecification.SampleSizeLimit)
+                        if (_diskProvider.GetFolderSize(subFolder) < NotSampleSpecification.SampleSizeLimit)
                         {
-                            _diskProvider.DeleteFolder(subfolder, true);
+                            _diskProvider.DeleteFolder(subFolder, true);
                         }
                     }
                 }
                 catch (Exception e)
                 {
-                    _logger.ErrorException("An error has occurred while importing folder: " + subfolder, e);
+                    _logger.ErrorException("An error has occurred while importing folder: " + subFolder, e);
                 }
             }
 
@@ -121,7 +124,7 @@ private void ProcessVideoFile(string videoFile, Series series)
                 return;
             }
 
-            ProcessFiles(new [] { videoFile }, series);
+            ProcessFiles(new[] { videoFile }, series);
         }
 
         private void ProcessFiles(IEnumerable<string> videoFiles, Series series)