Cleanup Disk Scan Service (#69)

* Cleanup Disk Scan Service * Fix Encoding, Change mp4 to flac
2026-04-20 21:54:25 -04:00 · 2017-08-27 16:32:41 -04:00
parent 9fba2458d8
commit 19414912ad
10 changed files with 101 additions and 131 deletions
@@ -0,0 +1,112 @@
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Common.Disk;
+using NzbDrone.Core.MediaFiles;
+using NzbDrone.Core.Test.Framework;
+using NzbDrone.Core.Music;
+
+namespace NzbDrone.Core.Test.ProviderTests.DiskScanProviderTests
+{
+
+    public class GetAudioFilesFixture : CoreTest<DiskScanService>
+    {
+        private string[] _fileNames;
+
+        [SetUp]
+        public void Setup()
+        {
+            _fileNames = new[]
+                        {
+                            @"30 Rock1.mp3",
+                            @"30 Rock2.flac",
+                            @"30 Rock3.ogg",
+                            @"30 Rock4.m4a",
+                            @"30 Rock.avi",
+                            @"movie.exe",
+                            @"movie"
+                        };
+        }
+
+        private IEnumerable<string> GetFiles(string folder, string subFolder = "")
+        {
+            return _fileNames.Select(f => Path.Combine(folder, subFolder, f));
+        }
+
+        private void GivenFiles(IEnumerable<string> files)
+        {
+            var filesToReturn = files.ToArray();
+            Mocker.GetMock<IDiskProvider>()
+                  .Setup(s => s.GetFiles(It.IsAny<string>(), SearchOption.AllDirectories))
+                  .Returns(filesToReturn);
+        }
+
+        [Test]
+        public void should_check_all_directories()
+        {
+            var path = @"C:\Test\";
+
+            Subject.GetAudioFiles(path);
+
+            Mocker.GetMock<IDiskProvider>().Verify(s => s.GetFiles(path, SearchOption.AllDirectories), Times.Once());
+            Mocker.GetMock<IDiskProvider>().Verify(s => s.GetFiles(path, SearchOption.TopDirectoryOnly), Times.Never());
+        }
+
+        [Test]
+        public void should_check_all_directories_when_allDirectories_is_true()
+        {
+            var path = @"C:\Test\";
+
+            Subject.GetAudioFiles(path, true);
+
+            Mocker.GetMock<IDiskProvider>().Verify(s => s.GetFiles(path, SearchOption.AllDirectories), Times.Once());
+            Mocker.GetMock<IDiskProvider>().Verify(s => s.GetFiles(path, SearchOption.TopDirectoryOnly), Times.Never());
+        }
+
+        [Test]
+        public void should_check_top_level_directory_only_when_allDirectories_is_false()
+        {
+            var path = @"C:\Test\";
+
+            Subject.GetAudioFiles(path, false);
+
+            Mocker.GetMock<IDiskProvider>().Verify(s => s.GetFiles(path, SearchOption.AllDirectories), Times.Never());
+            Mocker.GetMock<IDiskProvider>().Verify(s => s.GetFiles(path, SearchOption.TopDirectoryOnly), Times.Once());
+        }
+
+        [Test]
+        public void should_return_audio_files_only()
+        {
+            var path = @"C:\Test\";
+            GivenFiles(GetFiles(path));
+
+            Subject.GetAudioFiles(path).Should().HaveCount(4);
+        }
+
+        [TestCase("Extras")]
+        [TestCase("@eadir")]
+        [TestCase("extrafanart")]
+        [TestCase("Plex Versions")]
+        [TestCase(".secret")]
+        [TestCase(".hidden")]
+        public void should_filter_certain_sub_folders(string subFolder)
+        {
+            var path = @"C:\Test\";
+            var files = GetFiles(path).ToList();
+            var specialFiles = GetFiles(path, subFolder).ToList();
+            var allFiles = files.Concat(specialFiles);
+
+            var artist = Builder<Artist>.CreateNew()
+                                        .With(s => s.Path = path)
+                                        .Build();
+
+            var filteredFiles = Subject.FilterFiles(artist, allFiles);
+            filteredFiles.Should().NotContain(specialFiles);
+            filteredFiles.Count.Should().BeGreaterThan(0);
+        }
+    }
+}