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

using System.Linq;
using FizzWare.NBuilder;
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Core.Housekeeping.Housekeepers;
using NzbDrone.Core.MediaFiles;
using NzbDrone.Core.Music;
using NzbDrone.Core.Qualities;
using NzbDrone.Core.Test.Framework;

namespace NzbDrone.Core.Test.Housekeeping.Housekeepers
{
    [TestFixture]
    public class CleanupOrphanedTrackFilesFixture : DbTest<CleanupOrphanedTrackFiles, TrackFile>
    {
        [Test]
        public void should_unlink_orphaned_track_files()
        {
            var trackFile = Builder<TrackFile>.CreateNew()
                .With(h => h.Quality = new QualityModel())
                .With(h => h.AlbumId = 1)
                .BuildNew();

            Db.Insert(trackFile);
            Subject.Clean();
            AllStoredModels[0].AlbumId.Should().Be(0);
        }

        [Test]
        public void should_not_unlink_unorphaned_track_files()
        {
            var trackFiles = Builder<TrackFile>.CreateListOfSize(2)
                .All()
                .With(h => h.Quality = new QualityModel())
                .With(h => h.AlbumId = 1)
                .BuildListOfNew();

            Db.InsertMany(trackFiles);

            var track = Builder<Track>.CreateNew()
                                          .With(e => e.TrackFileId = trackFiles.First().Id)
                                          .BuildNew();

            Db.Insert(track);

            Subject.Clean();
            AllStoredModels.Where(x => x.AlbumId == 1).Should().HaveCount(1);

            Db.All<Track>().Should().Contain(e => e.TrackFileId == AllStoredModels.First().Id);
        }
    }
}