moved media file service

2025-11-06 09:19:38 +02:00 · 2013-02-28 23:03:41 -08:00
parent 8900bbb3a1
commit de5d5b76e8
46 changed files with 512 additions and 488 deletions
--- a/NzbDrone.Core.Test/MediaFileTests/CleanUpDatabaseFixture.cs
+++ b/NzbDrone.Core.Test/MediaFileTests/CleanUpDatabaseFixture.cs
@@ -0,0 +1,132 @@
+using System;
+using System.Linq;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using NUnit.Framework;
+using NzbDrone.Core.Configuration;
+using NzbDrone.Core.MediaFiles;
+using NzbDrone.Core.Qualities;
+using NzbDrone.Core.Tv;
+using NzbDrone.Core.Model;
+using NzbDrone.Core.Test.Framework;
+
+namespace NzbDrone.Core.Test.MediaFileTests
+{
+    [TestFixture]
+    public class CleanUpDatabaseFixture : SqlCeTest
+    {
+
+        [SetUp]
+        public void Setup()
+        {
+            WithRealDb();
+        }
+
+        private void WithAutoIgnore(bool autoIgnore)
+        {
+
+            Mocker.GetMock<IConfigService>()
+                    .SetupGet(c => c.AutoIgnorePreviouslyDownloadedEpisodes).Returns(autoIgnore);
+        }
+
+
+
+        [Test]
+        public void CleanUpDatabse_should_detach_none_existing_file_from_episodes_with_auto_ignore()
+        {
+            WithAutoIgnore(true);
+
+            var episodes = Builder<Episode>.CreateListOfSize(3)
+                .All().With(c => c.GrabDate = DateTime.Now)
+                      .And(c => c.Ignored = false)
+                      .And(c => c.PostDownloadStatus = PostDownloadStatusType.NoError)
+                .Build();
+
+
+            Db.InsertMany(episodes);
+
+            //Act
+            var result = Db.Fetch<Episode>();
+
+            //Assert
+            result.Should().HaveSameCount(episodes);
+            result.Should().OnlyContain(e => e.EpisodeFileId == 0);
+            result.Should().OnlyContain(e => e.PostDownloadStatus == PostDownloadStatusType.Unknown);
+            result.Should().OnlyContain(e => e.Ignored);
+            result.Should().OnlyContain(e => e.GrabDate == null);
+        }
+
+        [Test]
+        public void CleanUpDatabse_should_detach_none_existing_file_from_episodes_with_no_auto_ignore()
+        {
+            WithAutoIgnore(false);
+
+            var episodes = Builder<Episode>.CreateListOfSize(3)
+                .All().With(c => c.GrabDate = DateTime.Now)
+                      .And(c => c.PostDownloadStatus = PostDownloadStatusType.NoError)
+                .TheFirst(2).With(c => c.Ignored = true)
+                .TheLast(1).With(c => c.Ignored = false)
+                .Build();
+
+
+            Db.InsertMany(episodes);
+
+            //Act
+            var result = Db.Fetch<Episode>();
+
+            //Assert
+            result.Should().HaveSameCount(episodes);
+            result.Should().OnlyContain(e => e.EpisodeFileId == 0);
+            result.Should().OnlyContain(e => e.PostDownloadStatus == PostDownloadStatusType.Unknown);
+            result.Should().OnlyContain(e => e.GrabDate == null);
+            result.Should().Contain(c => c.Ignored == true);
+            result.Should().Contain(c => c.Ignored == false);
+        }
+
+        [Test]
+        public void CleanUpDatabse_should_not_change_episodes_with_no_file_id()
+        {
+            //Setup
+            var episodes = Builder<Episode>.CreateListOfSize(3)
+                .All().With(c => c.GrabDate = DateTime.Now)
+                      .And(c => c.Ignored = false)
+                      .And(c => c.PostDownloadStatus = PostDownloadStatusType.NoError)
+                .Build();
+
+            Db.InsertMany(episodes);
+
+            //Act
+            var result = Db.Fetch<Episode>();
+
+            //Assert
+            result.Should().HaveSameCount(episodes);
+            result.Should().OnlyContain(e => e.EpisodeFileId == 0);
+            result.Should().NotContain(e => e.PostDownloadStatus == PostDownloadStatusType.Unknown);
+            result.Should().NotContain(e => e.Ignored);
+            result.Should().NotContain(e => e.GrabDate == null);
+        }
+
+
+        [Test]
+        public void DeleteOrphanedEpisodeFiles()
+        {
+            //Setup
+            var episodeFiles = Builder<EpisodeFile>
+                .CreateListOfSize(10)
+                .All()
+                .With(e => e.Quality = Quality.DVD)
+                .Build();
+            var episodes = Builder<Episode>.CreateListOfSize(5).Build();
+
+            Db.InsertMany(episodes);
+            Db.InsertMany(episodeFiles);
+
+            //Act
+            var result = Db.Fetch<EpisodeFile>();
+
+            //Assert
+            result.Should().HaveCount(5);
+            result.Should().OnlyContain(e => e.Id > 0);
+        }
+    }
+}