2020-01-03 12:49:24 +00:00
|
|
|
using System.Collections.Generic;
|
|
|
|
using System.Linq;
|
2020-08-18 20:11:44 +00:00
|
|
|
using Dapper;
|
2017-09-04 02:20:56 +00:00
|
|
|
using FizzWare.NBuilder;
|
2014-02-14 23:15:45 +00:00
|
|
|
using NUnit.Framework;
|
|
|
|
using NzbDrone.Core.Datastore;
|
2020-01-03 12:49:24 +00:00
|
|
|
using NzbDrone.Core.MediaFiles;
|
2017-10-18 02:55:02 +00:00
|
|
|
using NzbDrone.Core.Music;
|
2020-01-03 12:49:24 +00:00
|
|
|
using NzbDrone.Core.Profiles.Qualities;
|
2014-02-14 23:15:45 +00:00
|
|
|
using NzbDrone.Core.Qualities;
|
2020-01-03 12:49:24 +00:00
|
|
|
using NzbDrone.Core.Test.Framework;
|
2014-02-14 23:15:45 +00:00
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.Datastore
|
|
|
|
{
|
|
|
|
[TestFixture]
|
2020-08-18 20:11:44 +00:00
|
|
|
public class LazyLoadingFixture : DbTest
|
2014-02-14 23:15:45 +00:00
|
|
|
{
|
|
|
|
[SetUp]
|
|
|
|
public void Setup()
|
|
|
|
{
|
2020-08-18 20:11:44 +00:00
|
|
|
SqlBuilderExtensions.LogSql = true;
|
|
|
|
|
2019-02-23 22:39:11 +00:00
|
|
|
var profile = new QualityProfile
|
2017-09-04 02:20:56 +00:00
|
|
|
{
|
|
|
|
Name = "Test",
|
2017-11-16 02:24:33 +00:00
|
|
|
Cutoff = Quality.MP3_320.Id,
|
2017-09-04 02:20:56 +00:00
|
|
|
Items = Qualities.QualityFixture.GetDefaultQualities()
|
|
|
|
};
|
|
|
|
|
2014-06-08 08:22:55 +00:00
|
|
|
profile = Db.Insert(profile);
|
2014-02-14 23:15:45 +00:00
|
|
|
|
2018-12-15 00:02:43 +00:00
|
|
|
var metadata = Builder<ArtistMetadata>.CreateNew()
|
|
|
|
.With(v => v.Id = 0)
|
|
|
|
.Build();
|
|
|
|
Db.Insert(metadata);
|
|
|
|
|
2017-10-18 02:55:02 +00:00
|
|
|
var artist = Builder<Artist>.CreateListOfSize(1)
|
2014-02-14 23:15:45 +00:00
|
|
|
.All()
|
2018-12-15 00:02:43 +00:00
|
|
|
.With(v => v.Id = 0)
|
2019-02-23 22:39:11 +00:00
|
|
|
.With(v => v.QualityProfileId = profile.Id)
|
2018-12-15 00:02:43 +00:00
|
|
|
.With(v => v.ArtistMetadataId = metadata.Id)
|
2014-02-14 23:15:45 +00:00
|
|
|
.BuildListOfNew();
|
|
|
|
|
2017-10-18 02:55:02 +00:00
|
|
|
Db.InsertMany(artist);
|
2014-02-14 23:15:45 +00:00
|
|
|
|
2017-11-25 05:16:46 +00:00
|
|
|
var albums = Builder<Album>.CreateListOfSize(3)
|
|
|
|
.All()
|
2018-12-15 00:02:43 +00:00
|
|
|
.With(v => v.Id = 0)
|
|
|
|
.With(v => v.ArtistMetadataId = metadata.Id)
|
2017-11-25 05:16:46 +00:00
|
|
|
.BuildListOfNew();
|
|
|
|
|
|
|
|
Db.InsertMany(albums);
|
|
|
|
|
2018-12-15 00:02:43 +00:00
|
|
|
var releases = new List<AlbumRelease>();
|
|
|
|
foreach (var album in albums)
|
|
|
|
{
|
|
|
|
releases.Add(
|
|
|
|
Builder<AlbumRelease>.CreateNew()
|
|
|
|
.With(v => v.Id = 0)
|
|
|
|
.With(v => v.AlbumId = album.Id)
|
|
|
|
.With(v => v.ForeignReleaseId = "test" + album.Id)
|
|
|
|
.Build());
|
|
|
|
}
|
2020-01-03 12:49:24 +00:00
|
|
|
|
2018-12-15 00:02:43 +00:00
|
|
|
Db.InsertMany(releases);
|
|
|
|
|
2017-10-18 02:55:02 +00:00
|
|
|
var trackFiles = Builder<TrackFile>.CreateListOfSize(1)
|
2014-02-14 23:15:45 +00:00
|
|
|
.All()
|
2018-12-15 00:02:43 +00:00
|
|
|
.With(v => v.Id = 0)
|
2019-02-16 14:49:24 +00:00
|
|
|
.With(v => v.AlbumId = albums[0].Id)
|
2014-07-28 11:16:40 +00:00
|
|
|
.With(v => v.Quality = new QualityModel())
|
2014-02-14 23:15:45 +00:00
|
|
|
.BuildListOfNew();
|
|
|
|
|
2017-10-18 02:55:02 +00:00
|
|
|
Db.InsertMany(trackFiles);
|
2014-02-14 23:15:45 +00:00
|
|
|
|
2017-10-18 02:55:02 +00:00
|
|
|
var tracks = Builder<Track>.CreateListOfSize(10)
|
2014-02-14 23:15:45 +00:00
|
|
|
.All()
|
2018-12-15 00:02:43 +00:00
|
|
|
.With(v => v.Id = 0)
|
2017-10-18 02:55:02 +00:00
|
|
|
.With(v => v.TrackFileId = trackFiles[0].Id)
|
2018-12-15 00:02:43 +00:00
|
|
|
.With(v => v.AlbumReleaseId = releases[0].Id)
|
2014-02-14 23:15:45 +00:00
|
|
|
.BuildListOfNew();
|
|
|
|
|
2017-10-18 02:55:02 +00:00
|
|
|
Db.InsertMany(tracks);
|
2014-02-14 23:15:45 +00:00
|
|
|
}
|
|
|
|
|
2019-02-16 14:49:24 +00:00
|
|
|
[Test]
|
|
|
|
public void should_lazy_load_artist_for_track()
|
|
|
|
{
|
2020-08-18 20:11:44 +00:00
|
|
|
var db = Mocker.Resolve<TrackRepository>();
|
2019-02-16 14:49:24 +00:00
|
|
|
|
2020-08-18 20:11:44 +00:00
|
|
|
var tracks = db.All();
|
2019-02-16 14:49:24 +00:00
|
|
|
|
|
|
|
Assert.IsNotEmpty(tracks);
|
|
|
|
foreach (var track in tracks)
|
|
|
|
{
|
|
|
|
Assert.IsFalse(track.Artist.IsLoaded);
|
|
|
|
Assert.IsNotNull(track.Artist.Value);
|
|
|
|
Assert.IsTrue(track.Artist.IsLoaded);
|
|
|
|
Assert.IsTrue(track.Artist.Value.Metadata.IsLoaded);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_lazy_load_artist_for_trackfile()
|
|
|
|
{
|
|
|
|
var db = Mocker.Resolve<IDatabase>();
|
2022-01-26 00:08:27 +00:00
|
|
|
var tracks = db.Query<TrackFile>(new SqlBuilder(db.DatabaseType)).ToList();
|
2019-02-16 14:49:24 +00:00
|
|
|
|
|
|
|
Assert.IsNotEmpty(tracks);
|
|
|
|
foreach (var track in tracks)
|
|
|
|
{
|
|
|
|
Assert.IsFalse(track.Artist.IsLoaded);
|
|
|
|
Assert.IsNotNull(track.Artist.Value);
|
|
|
|
Assert.IsTrue(track.Artist.IsLoaded);
|
|
|
|
Assert.IsTrue(track.Artist.Value.Metadata.IsLoaded);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_lazy_load_trackfile_if_not_joined()
|
|
|
|
{
|
|
|
|
var db = Mocker.Resolve<IDatabase>();
|
2022-01-26 00:08:27 +00:00
|
|
|
var tracks = db.Query<Track>(new SqlBuilder(db.DatabaseType)).ToList();
|
2019-02-16 14:49:24 +00:00
|
|
|
|
|
|
|
foreach (var track in tracks)
|
|
|
|
{
|
|
|
|
Assert.IsFalse(track.TrackFile.IsLoaded);
|
|
|
|
Assert.IsNotNull(track.TrackFile.Value);
|
|
|
|
Assert.IsTrue(track.TrackFile.IsLoaded);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_explicit_load_everything_if_joined()
|
|
|
|
{
|
|
|
|
var db = Mocker.Resolve<IDatabase>();
|
2020-08-18 20:11:44 +00:00
|
|
|
var files = MediaFileRepository.Query(db,
|
2022-01-26 00:08:27 +00:00
|
|
|
new SqlBuilder(db.DatabaseType)
|
2020-08-18 20:11:44 +00:00
|
|
|
.Join<TrackFile, Track>((f, t) => f.Id == t.TrackFileId)
|
|
|
|
.Join<TrackFile, Album>((t, a) => t.AlbumId == a.Id)
|
|
|
|
.Join<Album, Artist>((album, artist) => album.ArtistMetadataId == artist.ArtistMetadataId)
|
|
|
|
.Join<Artist, ArtistMetadata>((a, m) => a.ArtistMetadataId == m.Id));
|
2019-02-16 14:49:24 +00:00
|
|
|
|
|
|
|
Assert.IsNotEmpty(files);
|
|
|
|
foreach (var file in files)
|
|
|
|
{
|
|
|
|
Assert.IsTrue(file.Tracks.IsLoaded);
|
|
|
|
Assert.IsNotEmpty(file.Tracks.Value);
|
|
|
|
Assert.IsTrue(file.Album.IsLoaded);
|
|
|
|
Assert.IsTrue(file.Artist.IsLoaded);
|
|
|
|
Assert.IsTrue(file.Artist.Value.Metadata.IsLoaded);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_lazy_load_tracks_if_not_joined_to_trackfile()
|
|
|
|
{
|
|
|
|
var db = Mocker.Resolve<IDatabase>();
|
2020-08-18 20:11:44 +00:00
|
|
|
var files = db.QueryJoined<TrackFile, Album, Artist, ArtistMetadata>(
|
2022-01-26 00:08:27 +00:00
|
|
|
new SqlBuilder(db.DatabaseType)
|
2020-08-18 20:11:44 +00:00
|
|
|
.Join<TrackFile, Album>((t, a) => t.AlbumId == a.Id)
|
|
|
|
.Join<Album, Artist>((album, artist) => album.ArtistMetadataId == artist.ArtistMetadataId)
|
|
|
|
.Join<Artist, ArtistMetadata>((a, m) => a.ArtistMetadataId == m.Id),
|
|
|
|
(file, album, artist, metadata) =>
|
|
|
|
{
|
|
|
|
file.Album = album;
|
|
|
|
file.Artist = artist;
|
|
|
|
file.Artist.Value.Metadata = metadata;
|
|
|
|
return file;
|
|
|
|
});
|
2019-02-16 14:49:24 +00:00
|
|
|
|
|
|
|
Assert.IsNotEmpty(files);
|
|
|
|
foreach (var file in files)
|
|
|
|
{
|
|
|
|
Assert.IsFalse(file.Tracks.IsLoaded);
|
|
|
|
Assert.IsNotNull(file.Tracks.Value);
|
|
|
|
Assert.IsNotEmpty(file.Tracks.Value);
|
|
|
|
Assert.IsTrue(file.Tracks.IsLoaded);
|
|
|
|
Assert.IsTrue(file.Album.IsLoaded);
|
|
|
|
Assert.IsTrue(file.Artist.IsLoaded);
|
|
|
|
Assert.IsTrue(file.Artist.Value.Metadata.IsLoaded);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_lazy_load_tracks_if_not_joined()
|
|
|
|
{
|
|
|
|
var db = Mocker.Resolve<IDatabase>();
|
2022-01-26 00:08:27 +00:00
|
|
|
var release = db.Query<AlbumRelease>(new SqlBuilder(db.DatabaseType).Where<AlbumRelease>(x => x.Id == 1)).SingleOrDefault();
|
2019-02-16 14:49:24 +00:00
|
|
|
|
|
|
|
Assert.IsFalse(release.Tracks.IsLoaded);
|
|
|
|
Assert.IsNotNull(release.Tracks.Value);
|
|
|
|
Assert.IsNotEmpty(release.Tracks.Value);
|
|
|
|
Assert.IsTrue(release.Tracks.IsLoaded);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_lazy_load_track_if_not_joined()
|
|
|
|
{
|
|
|
|
var db = Mocker.Resolve<IDatabase>();
|
2022-01-26 00:08:27 +00:00
|
|
|
var tracks = db.Query<TrackFile>(new SqlBuilder(db.DatabaseType)).ToList();
|
2019-02-16 14:49:24 +00:00
|
|
|
|
|
|
|
foreach (var track in tracks)
|
|
|
|
{
|
|
|
|
Assert.IsFalse(track.Tracks.IsLoaded);
|
|
|
|
Assert.IsNotNull(track.Tracks.Value);
|
|
|
|
Assert.IsTrue(track.Tracks.IsLoaded);
|
|
|
|
}
|
|
|
|
}
|
2014-02-14 23:15:45 +00:00
|
|
|
}
|
2017-09-04 02:20:56 +00:00
|
|
|
}
|