Add TrackFiles Housekeeper (#66)

Add TrackFiles Housekeeper

src/NzbDrone.Core.Test/Housekeeping/Housekeepers/CleanupOrphanedTrackFilesFixture.cs

@@ -1,49 +1,49 @@
-using System.Linq;
+using System.Linq;
 using FizzWare.NBuilder;
 using FluentAssertions;
 using NUnit.Framework;
 using NzbDrone.Core.Housekeeping.Housekeepers;
 using NzbDrone.Core.MediaFiles;
 using NzbDrone.Core.Test.Framework;
-using NzbDrone.Core.Tv;
+using NzbDrone.Core.Music;
 using NzbDrone.Core.Qualities;
 
 namespace NzbDrone.Core.Test.Housekeeping.Housekeepers
 {
     [TestFixture]
-    public class CleanupOrphanedEpisodeFilesFixture : DbTest<CleanupOrphanedEpisodeFiles, EpisodeFile>
+    public class CleanupOrphanedTrackFilesFixture : DbTest<CleanupOrphanedTrackFiles, TrackFile>
     {
         [Test]
-        public void should_delete_orphaned_episode_files()
+        public void should_delete_orphaned_track_files()
         {
-            var episodeFile = Builder<EpisodeFile>.CreateNew()
+            var trackFile = Builder<TrackFile>.CreateNew()
                                                   .With(h => h.Quality = new QualityModel())
                                                   .BuildNew();
 
-            Db.Insert(episodeFile);
+            Db.Insert(trackFile);
             Subject.Clean();
             AllStoredModels.Should().BeEmpty();
         }
 
         [Test]
-        public void should_not_delete_unorphaned_episode_files()
+        public void should_not_delete_unorphaned_track_files()
         {
-            var episodeFiles = Builder<EpisodeFile>.CreateListOfSize(2)
+            var trackFiles = Builder<TrackFile>.CreateListOfSize(2)
                                                    .All()
                                                    .With(h => h.Quality = new QualityModel())
                                                    .BuildListOfNew();
 
-            Db.InsertMany(episodeFiles);
+            Db.InsertMany(trackFiles);
 
-            var episode = Builder<Episode>.CreateNew()
+            var track = Builder<Track>.CreateNew()
-                                          .With(e => e.EpisodeFileId = episodeFiles.First().Id)
+                                          .With(e => e.TrackFileId = trackFiles.First().Id)
                                           .BuildNew();
 
-            Db.Insert(episode);
+            Db.Insert(track);
 
             Subject.Clean();
             AllStoredModels.Should().HaveCount(1);
-            Db.All<Episode>().Should().Contain(e => e.EpisodeFileId == AllStoredModels.First().Id);
+            Db.All<Track>().Should().Contain(e => e.TrackFileId == AllStoredModels.First().Id);
         }
     }
 }

src/NzbDrone.Core.Test/NzbDrone.Core.Test.csproj

@@ -236,7 +236,7 @@
     <Compile Include="Housekeeping\Housekeepers\CleanupAbsolutePathMetadataFilesFixture.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupDuplicateMetadataFilesFixture.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedBlacklistFixture.cs" />
-    <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedEpisodeFilesFixture.cs" />
+    <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedTrackFilesFixture.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedEpisodesFixture.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedIndexerStatusFixture.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedHistoryItemsFixture.cs" />

src/NzbDrone.Core/Housekeeping/Housekeepers/CleanupOrphanedEpisodeFiles.cs

@@ -1,26 +0,0 @@
-using NzbDrone.Core.Datastore;
-
-namespace NzbDrone.Core.Housekeeping.Housekeepers
-{
-    public class CleanupOrphanedEpisodeFiles : IHousekeepingTask
-    {
-        private readonly IMainDatabase _database;
-
-        public CleanupOrphanedEpisodeFiles(IMainDatabase database)
-        {
-            _database = database;
-        }
-
-        public void Clean()
-        {
-            var mapper = _database.GetDataMapper();
-
-            mapper.ExecuteNonQuery(@"DELETE FROM EpisodeFiles
-                                     WHERE Id IN (
-                                     SELECT EpisodeFiles.Id FROM EpisodeFiles
-                                     LEFT OUTER JOIN Episodes
-                                     ON EpisodeFiles.Id = Episodes.EpisodeFileId
-                                     WHERE Episodes.Id IS NULL)");
-        }
-    }
-}

src/NzbDrone.Core/Housekeeping/Housekeepers/CleanupOrphanedTrackFiles.cs

@@ -0,0 +1,26 @@
+using NzbDrone.Core.Datastore;
+
+namespace NzbDrone.Core.Housekeeping.Housekeepers
+{
+    public class CleanupOrphanedTrackFiles : IHousekeepingTask
+    {
+        private readonly IMainDatabase _database;
+
+        public CleanupOrphanedTrackFiles(IMainDatabase database)
+        {
+            _database = database;
+        }
+
+        public void Clean()
+        {
+            var mapper = _database.GetDataMapper();
+
+            mapper.ExecuteNonQuery(@"DELETE FROM TrackFiles
+                                     WHERE Id IN (
+                                     SELECT TrackFiles.Id FROM TrackFiles
+                                     LEFT OUTER JOIN Tracks
+                                     ON TrackFiles.Id = Tracks.TrackFileId
+                                     WHERE Tracks.Id IS NULL)");
+        }
+    }
+}

src/NzbDrone.Core/NzbDrone.Core.csproj

@@ -589,7 +589,7 @@
     <Compile Include="Housekeeping\Housekeepers\CleanupAbsolutePathMetadataFiles.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupDuplicateMetadataFiles.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedBlacklist.cs" />
-    <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedEpisodeFiles.cs" />
+    <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedTrackFiles.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedEpisodes.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedIndexerStatus.cs" />
     <Compile Include="Housekeeping\Housekeepers\CleanupOrphanedHistoryItems.cs" />