Create Orphaned Albums Housekeeper (#82)
parent
43677e8481
commit
0b22d7fce3
@ -0,0 +1,43 @@
|
||||
using FizzWare.NBuilder;
|
||||
using FluentAssertions;
|
||||
using NUnit.Framework;
|
||||
using NzbDrone.Core.Housekeeping.Housekeepers;
|
||||
using NzbDrone.Core.Test.Framework;
|
||||
using NzbDrone.Core.Music;
|
||||
|
||||
namespace NzbDrone.Core.Test.Housekeeping.Housekeepers
|
||||
{
|
||||
[TestFixture]
|
||||
public class CleanupOrphanedAlbumsFixture : DbTest<CleanupOrphanedAlbums, Album>
|
||||
{
|
||||
[Test]
|
||||
public void should_delete_orphaned_albums()
|
||||
{
|
||||
var album = Builder<Album>.CreateNew()
|
||||
.BuildNew();
|
||||
|
||||
Db.Insert(album);
|
||||
Subject.Clean();
|
||||
AllStoredModels.Should().BeEmpty();
|
||||
}
|
||||
|
||||
[Test]
|
||||
public void should_not_delete_unorphaned_albums()
|
||||
{
|
||||
var artist = Builder<Artist>.CreateNew()
|
||||
.BuildNew();
|
||||
|
||||
Db.Insert(artist);
|
||||
|
||||
var albums = Builder<Album>.CreateListOfSize(2)
|
||||
.TheFirst(1)
|
||||
.With(e => e.ArtistId = artist.Id)
|
||||
.BuildListOfNew();
|
||||
|
||||
Db.InsertMany(albums);
|
||||
Subject.Clean();
|
||||
AllStoredModels.Should().HaveCount(1);
|
||||
AllStoredModels.Should().Contain(e => e.ArtistId == artist.Id);
|
||||
}
|
||||
}
|
||||
}
|
@ -0,0 +1,26 @@
|
||||
using NzbDrone.Core.Datastore;
|
||||
|
||||
namespace NzbDrone.Core.Housekeeping.Housekeepers
|
||||
{
|
||||
public class CleanupOrphanedAlbums : IHousekeepingTask
|
||||
{
|
||||
private readonly IMainDatabase _database;
|
||||
|
||||
public CleanupOrphanedAlbums(IMainDatabase database)
|
||||
{
|
||||
_database = database;
|
||||
}
|
||||
|
||||
public void Clean()
|
||||
{
|
||||
var mapper = _database.GetDataMapper();
|
||||
|
||||
mapper.ExecuteNonQuery(@"DELETE FROM Albums
|
||||
WHERE Id IN (
|
||||
SELECT Albums.Id FROM Albums
|
||||
LEFT OUTER JOIN Artists
|
||||
ON Albums.ArtistId = Artists.Id
|
||||
WHERE Artists.Id IS NULL)");
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in new issue