Added Album Artist support
This commit is contained in:
emeric
+23
-5
@@ -1,14 +1,32 @@
|
||||
|
||||
TESTS = som-test
|
||||
TESTS = som database
|
||||
|
||||
check_PROGRAMS = som-test
|
||||
check_PROGRAMS = som database
|
||||
|
||||
som_test_SOURCES = \
|
||||
$(srcdir)/som-test/SomTest.cpp \
|
||||
som_SOURCES = \
|
||||
$(srcdir)/som/SomTest.cpp \
|
||||
$(top_srcdir)/src/similarity/features/som/DataNormalizer.cpp \
|
||||
$(top_srcdir)/src/similarity/features/som/Network.cpp
|
||||
|
||||
som_test_CXXFLAGS=-std=c++14 -Wall -I${top_srcdir}/src/ -I${top_srcdir}/src/similarity/features/som/
|
||||
som_CXXFLAGS=-std=c++14 -Wall -I${top_srcdir}/src/ -I${top_srcdir}/src/similarity/features/som/
|
||||
|
||||
|
||||
database_SOURCES = \
|
||||
$(srcdir)/database/DatabaseTest.cpp \
|
||||
$(top_srcdir)/src/database/Artist.cpp \
|
||||
$(top_srcdir)/src/database/Cluster.cpp \
|
||||
$(top_srcdir)/src/database/DatabaseHandler.cpp \
|
||||
$(top_srcdir)/src/database/TrackArtistLink.cpp \
|
||||
$(top_srcdir)/src/database/TrackFeatures.cpp \
|
||||
$(top_srcdir)/src/database/TrackList.cpp \
|
||||
$(top_srcdir)/src/database/Release.cpp \
|
||||
$(top_srcdir)/src/database/ScanSettings.cpp \
|
||||
$(top_srcdir)/src/database/SimilaritySettings.cpp \
|
||||
$(top_srcdir)/src/database/SqlQuery.cpp \
|
||||
$(top_srcdir)/src/database/Track.cpp \
|
||||
$(top_srcdir)/src/database/User.cpp \
|
||||
$(top_srcdir)/src/utils/Logger.cpp \
|
||||
$(top_srcdir)/src/utils/Utils.cpp
|
||||
|
||||
database_CXXFLAGS=-std=c++14 -Wall -I${top_srcdir}/src/
|
||||
|
||||
|
||||
@@ -0,0 +1,948 @@
|
||||
/*
|
||||
* Copyright (C) 2019 Emeric Poupon
|
||||
*
|
||||
* This file is part of LMS.
|
||||
*
|
||||
* LMS is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* LMS is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with LMS. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <cstdlib>
|
||||
|
||||
#include <boost/filesystem.hpp>
|
||||
|
||||
#include "database/Artist.hpp"
|
||||
#include "database/Cluster.hpp"
|
||||
#include "database/DatabaseHandler.hpp"
|
||||
#include "database/TrackList.hpp"
|
||||
#include "database/Release.hpp"
|
||||
#include "database/Track.hpp"
|
||||
|
||||
using namespace Database;
|
||||
|
||||
class ScopedFileDeleter final
|
||||
{
|
||||
public:
|
||||
ScopedFileDeleter(const boost::filesystem::path& path) : _path {path} {}
|
||||
~ScopedFileDeleter() { boost::filesystem::remove(_path); }
|
||||
private:
|
||||
boost::filesystem::path _path;
|
||||
};
|
||||
|
||||
/*static void check(bool cond)
|
||||
{
|
||||
if (!cond)
|
||||
}*/
|
||||
|
||||
#define CHECK(PRED) \
|
||||
{ \
|
||||
if (!(PRED)) \
|
||||
{ \
|
||||
std::string msg {"Predicate '" + std::string {#PRED} + "' at " + __FUNCTION__ + "@l." + std::to_string(__LINE__)}; \
|
||||
throw std::runtime_error(msg.c_str()); \
|
||||
} \
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrack(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "MyTrackFile")};
|
||||
CHECK(track);
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
CHECK(track);
|
||||
CHECK(Track::getAll(session).size() == 1);
|
||||
|
||||
track.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleArtist(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType artistId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist {Artist::create(session, "MyArtist")};
|
||||
CHECK(artist);
|
||||
session.flush();
|
||||
artistId = artist.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
CHECK(Artist::getAll(session).size() == 1);
|
||||
|
||||
auto artists {Artist::getAllOrphans(session)};
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
CHECK(artist);
|
||||
artist.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleRelease(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType releaseId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto release {Release::create(session, "MyRelease")};
|
||||
CHECK(release);
|
||||
session.flush();
|
||||
releaseId = release.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto releases {Release::getAllOrphans(session)};
|
||||
CHECK(releases.size() == 1);
|
||||
CHECK(releases.front().id() == releaseId);
|
||||
|
||||
releases = Release::getAll(session);
|
||||
CHECK(releases.size() == 1);
|
||||
CHECK(releases.front().id() == releaseId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto release {Release::getById(session, releaseId)};
|
||||
CHECK(release);
|
||||
release.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleCluster(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType clusterTypeId {};
|
||||
IdType clusterId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto clusterType {ClusterType::create(session, "MyType")};
|
||||
CHECK(clusterType);
|
||||
auto cluster {Cluster::create(session, clusterType, "MyCluster")};
|
||||
CHECK(cluster);
|
||||
session.flush();
|
||||
clusterTypeId = clusterType.id();
|
||||
clusterId = cluster.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto clusters {Cluster::getAll(session)};
|
||||
CHECK(clusters.size() == 1);
|
||||
CHECK(clusters.front().id() == clusterId);
|
||||
CHECK(clusters.front()->getType().id() == clusterTypeId);
|
||||
|
||||
auto clusterTypes {ClusterType::getAll(session)};
|
||||
CHECK(clusterTypes.size() == 1);
|
||||
CHECK(clusterTypes.front().id() == clusterTypeId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto cluster {Cluster::getById(session, clusterId)};
|
||||
CHECK(cluster);
|
||||
CHECK(cluster->getType().id() == clusterTypeId);
|
||||
auto clusterType {ClusterType::getById(session, clusterTypeId)};
|
||||
CHECK(clusterType);
|
||||
|
||||
cluster.remove();
|
||||
clusterType.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleArtist(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType artistId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "dummy")};
|
||||
CHECK(track);
|
||||
auto artist {Artist::create(session, "dummy")};
|
||||
CHECK(artist);
|
||||
|
||||
auto trackArtistLink {TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::Artist)};
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
artistId = artist.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
CHECK(track);
|
||||
|
||||
auto artists {track->getArtists()};
|
||||
CHECK(artists.size() == 1);
|
||||
auto artist {artists.front()};
|
||||
CHECK(artist.id() == artistId);
|
||||
|
||||
CHECK(track->getArtistLinks().size() == 1);
|
||||
auto artistLink {track->getArtistLinks().front()};
|
||||
CHECK(artistLink->getTrack().id() == trackId);
|
||||
CHECK(artistLink->getArtist().id() == artistId);
|
||||
|
||||
CHECK(track->getArtists(TrackArtistLink::Type::Artist).size() == 1);
|
||||
CHECK(track->getArtists(TrackArtistLink::Type::ReleaseArtist).size() == 0);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
CHECK(artist);
|
||||
auto tracks {artist->getTracks()};
|
||||
CHECK(tracks.size() == 1);
|
||||
auto track {tracks.front()};
|
||||
CHECK(track.id() == trackId);
|
||||
|
||||
CHECK(artist->getTracks(TrackArtistLink::Type::ReleaseArtist).size() == 0);
|
||||
CHECK(artist->getTracks(TrackArtistLink::Type::Artist).size() == 1);
|
||||
|
||||
track.remove();
|
||||
artist.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleArtistMultiRoles(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType artistId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "MyTrack")};
|
||||
auto artist {Artist::create(session, "MyArtist")};
|
||||
|
||||
TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::Artist);
|
||||
TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::ReleaseArtist);
|
||||
TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::Writer);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
artistId = artist.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
CHECK(track);
|
||||
|
||||
auto artists {track->getArtists(TrackArtistLink::Type::Artist)};
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
|
||||
artists = track->getArtists(TrackArtistLink::Type::ReleaseArtist);
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
|
||||
CHECK(track->getArtistLinks().size() == 3);
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
CHECK(artist);
|
||||
CHECK(artist->getTracks().size() == 1);
|
||||
CHECK(artist->getTracks(TrackArtistLink::Type::ReleaseArtist).size() == 1);
|
||||
CHECK(artist->getTracks(TrackArtistLink::Type::Artist).size() == 1);
|
||||
CHECK(artist->getTracks(TrackArtistLink::Type::Writer).size() == 1);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
auto track {Track::getById(session, trackId)};
|
||||
|
||||
track.remove();
|
||||
artist.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackMultiArtists(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType artist1Id {};
|
||||
IdType artist2Id {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "dummy")};
|
||||
auto artist1 {Artist::create(session, "artist1")};
|
||||
auto artist2 {Artist::create(session, "artist2")};
|
||||
|
||||
TrackArtistLink::create(session, track, artist1, TrackArtistLink::Type::Artist);
|
||||
TrackArtistLink::create(session, track, artist2, TrackArtistLink::Type::Artist);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
artist1Id = artist1.id();
|
||||
artist2Id = artist2.id();
|
||||
|
||||
CHECK(artist1Id != artist2Id);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
CHECK(track);
|
||||
|
||||
auto artists {track->getArtists()};
|
||||
CHECK(artists.size() == 2);
|
||||
CHECK((artists[0].id() == artist1Id && artists[1].id() == artist2Id)
|
||||
|| (artists[0].id() == artist2Id && artists[1].id() == artist1Id));
|
||||
|
||||
CHECK(track->getArtists(TrackArtistLink::Type::Artist).size() == 2);
|
||||
CHECK(track->getArtists(TrackArtistLink::Type::ReleaseArtist).size() == 0);
|
||||
CHECK(Artist::getAll(session).size() == 2);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist1 {Artist::getById(session, artist1Id)};
|
||||
CHECK(artist1);
|
||||
auto tracks {artist1->getTracks()};
|
||||
CHECK(tracks.size() == 1);
|
||||
auto track {tracks.front()};
|
||||
CHECK(track.id() == trackId);
|
||||
auto artist2 {Artist::getById(session, artist2Id)};
|
||||
CHECK(artist2);
|
||||
CHECK(artist2->getTracks().front() == track);
|
||||
|
||||
|
||||
CHECK(artist1->getTracks(TrackArtistLink::Type::ReleaseArtist).size() == 0);
|
||||
CHECK(artist1->getTracks(TrackArtistLink::Type::Artist).size() == 1);
|
||||
CHECK(artist2->getTracks(TrackArtistLink::Type::ReleaseArtist).size() == 0);
|
||||
CHECK(artist2->getTracks(TrackArtistLink::Type::Artist).size() == 1);
|
||||
|
||||
track.remove();
|
||||
artist1.remove();
|
||||
artist2.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleRelease(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType releaseId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "dummy")};
|
||||
CHECK(track);
|
||||
auto release {Release::create(session, "dummy")};
|
||||
CHECK(release);
|
||||
|
||||
track.modify()->setRelease(release);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
releaseId = release.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
CHECK(track);
|
||||
CHECK(track->getRelease());
|
||||
CHECK(track->getRelease().id() == releaseId);
|
||||
|
||||
auto release {Release::getById(session, releaseId)};
|
||||
CHECK(release);
|
||||
CHECK(release->getTracks().size() == 1);
|
||||
CHECK(release->getTracks().front().id() == trackId);
|
||||
|
||||
track.remove();
|
||||
release.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleCluster(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType clusterId {};
|
||||
IdType clusterTypeId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "dummy")};
|
||||
auto clusterType {ClusterType::create(session, "MyType")};
|
||||
auto cluster {Cluster::create(session, clusterType, "MyCluster")};
|
||||
|
||||
cluster.modify()->addTrack(track);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
clusterTypeId = clusterType.id();
|
||||
clusterId = cluster.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto tracks {Track::getByFilter(session, {clusterId})};
|
||||
CHECK(tracks.size() == 1);
|
||||
CHECK(tracks.front().id() == trackId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
auto cluster {Cluster::getById(session, clusterId)};
|
||||
auto clusterType {ClusterType::getById(session, clusterTypeId)};
|
||||
|
||||
track.remove();
|
||||
cluster.remove();
|
||||
clusterType.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleReleaseSingleCluster(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType releaseId {};
|
||||
IdType clusterId {};
|
||||
IdType clusterTypeId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "MyTrackFile")};
|
||||
auto release {Release::create(session, "MyRelease")};
|
||||
auto clusterType {ClusterType::create(session, "MyType")};
|
||||
auto cluster {Cluster::create(session, clusterType, "MyCluster")};
|
||||
|
||||
track.modify()->setRelease(release);
|
||||
cluster.modify()->addTrack(track);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
releaseId = release.id();
|
||||
clusterTypeId = clusterType.id();
|
||||
clusterId = cluster.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto releases {Release::getByFilter(session, {clusterId})};
|
||||
CHECK(releases.size() == 1);
|
||||
CHECK(releases.front().id() == releaseId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
auto release {Release::getById(session, releaseId)};
|
||||
auto cluster {Cluster::getById(session, clusterId)};
|
||||
auto clusterType {ClusterType::getById(session, clusterTypeId)};
|
||||
|
||||
track.remove();
|
||||
release.remove();
|
||||
cluster.remove();
|
||||
clusterType.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleArtistMultiClusters(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType artistId {};
|
||||
IdType cluster1Id {};
|
||||
IdType cluster2Id {};
|
||||
IdType clusterTypeId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "MyTrackFile")};
|
||||
auto artist {Artist::create(session, "MyArtist")};
|
||||
auto clusterType {ClusterType::create(session, "MyType")};
|
||||
auto cluster1 {Cluster::create(session, clusterType, "MyCluster1")};
|
||||
auto cluster2 {Cluster::create(session, clusterType, "MyCluster2")};
|
||||
|
||||
auto trackArtistLink {TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::Artist)};
|
||||
cluster1.modify()->addTrack(track);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
artistId = artist.id();
|
||||
clusterTypeId = clusterType.id();
|
||||
cluster1Id = cluster1.id();
|
||||
cluster2Id = cluster2.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artists {Artist::getByFilter(session, {cluster1Id})};
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
|
||||
artists = Artist::getByFilter(session, {cluster2Id});
|
||||
CHECK(artists.size() == 0);
|
||||
|
||||
auto cluster2 {Cluster::getById(session, cluster2Id)};
|
||||
auto track {Track::getById(session, trackId)};
|
||||
cluster2.modify()->addTrack(track);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artists {Artist::getByFilter(session, {cluster1Id})};
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
|
||||
artists = Artist::getByFilter(session, {cluster2Id});
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
|
||||
artists = Artist::getByFilter(session, {cluster1Id, cluster2Id});
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
auto cluster1 {Cluster::getById(session, cluster1Id)};
|
||||
auto cluster2 {Cluster::getById(session, cluster2Id)};
|
||||
auto clusterType {ClusterType::getById(session, clusterTypeId)};
|
||||
|
||||
track.remove();
|
||||
artist.remove();
|
||||
cluster1.remove();
|
||||
cluster2.remove();
|
||||
clusterType.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleArtistMultiRolesMultiClusters(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType artistId {};
|
||||
IdType clusterId {};
|
||||
IdType clusterTypeId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "MyTrackFile")};
|
||||
auto artist {Artist::create(session, "MyArtist")};
|
||||
auto clusterType {ClusterType::create(session, "MyType")};
|
||||
auto cluster {Cluster::create(session, clusterType, "MyCluster")};
|
||||
|
||||
TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::Artist);
|
||||
TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::ReleaseArtist);
|
||||
cluster.modify()->addTrack(track);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
artistId = artist.id();
|
||||
clusterTypeId = clusterType.id();
|
||||
clusterId = cluster.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artists {Artist::getByFilter(session, {clusterId})};
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
auto cluster {Cluster::getById(session, clusterId)};
|
||||
auto clusterType {ClusterType::getById(session, clusterTypeId)};
|
||||
|
||||
track.remove();
|
||||
artist.remove();
|
||||
cluster.remove();
|
||||
clusterType.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testMultiTracksSingleArtistMultiClusters(Wt::Dbo::Session& session)
|
||||
{
|
||||
const std::size_t nbTracks {10};
|
||||
const std::size_t nbClusters {5};
|
||||
IdType artistId {};
|
||||
IdType clusterTypeId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist {Artist::create(session, "MyArtist")};
|
||||
|
||||
auto clusterType {ClusterType::create(session, "MyType")};
|
||||
std::vector<Cluster::pointer> clusters;
|
||||
|
||||
for (std::size_t i {}; i < nbClusters; ++i)
|
||||
clusters.push_back(Cluster::create(session, clusterType, "MyCluster" + std::to_string(i)));
|
||||
|
||||
for (std::size_t i {}; i < nbTracks; ++i)
|
||||
{
|
||||
auto track {Track::create(session, "MyTrackFile")};
|
||||
TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::Artist);
|
||||
|
||||
for (const auto& cluster : clusters)
|
||||
cluster.modify()->addTrack(track);
|
||||
}
|
||||
|
||||
session.flush();
|
||||
artistId = artist.id();
|
||||
clusterTypeId = clusterType.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
std::vector<Cluster::pointer> clusters {Cluster::getAll(session)};
|
||||
CHECK(clusters.size() == nbClusters);
|
||||
|
||||
std::set<IdType> clusterIds;
|
||||
std::transform(std::cbegin(clusters), std::cend(clusters), std::inserter(clusterIds, std::begin(clusterIds)), [](const Cluster::pointer& cluster) { return cluster.id(); });
|
||||
|
||||
auto artists {Artist::getByFilter(session, clusterIds)};
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
std::vector<Cluster::pointer> clusters {Cluster::getAll(session)};
|
||||
for (auto& cluster : clusters)
|
||||
cluster.remove();
|
||||
|
||||
std::vector<Track::pointer> tracks {Track::getAll(session)};
|
||||
for (auto& track : tracks)
|
||||
track.remove();
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
auto clusterType {ClusterType::getById(session, clusterTypeId)};
|
||||
|
||||
artist.remove();
|
||||
clusterType.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleReleaseSingleArtist(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType releaseId {};
|
||||
IdType artistId {};
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "dummy")};
|
||||
auto release {Release::create(session, "dummy")};
|
||||
auto artist {Artist::create(session, "dummy")};
|
||||
|
||||
auto trackArtistLink {TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::Artist)};
|
||||
track.modify()->setRelease(release);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
releaseId = release.id();
|
||||
artistId = artist.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
CHECK(artist);
|
||||
auto releases {artist->getReleases()};
|
||||
CHECK(releases.size() == 1);
|
||||
CHECK(releases.front().id() == releaseId);
|
||||
|
||||
auto release {Release::getById(session, releaseId)};
|
||||
CHECK(release);
|
||||
auto artists {release->getArtists()};
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
|
||||
auto track {Track::getById(session, trackId)};
|
||||
|
||||
track.remove();
|
||||
artist.remove();
|
||||
release.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleReleaseSingleArtistSingleCluster(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType releaseId {};
|
||||
IdType artistId {};
|
||||
IdType clusterId {};
|
||||
IdType clusterTypeId {};
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "MyTrackFile")};
|
||||
auto release {Release::create(session, "MyRelease")};
|
||||
auto artist {Artist::create(session, "MyArtist")};
|
||||
|
||||
auto clusterType {ClusterType::create(session, "MyType")};
|
||||
auto cluster {Cluster::create(session, clusterType, "MyCluster")};
|
||||
|
||||
auto trackArtistLink {TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::Artist)};
|
||||
track.modify()->setRelease(release);
|
||||
cluster.modify()->addTrack(track);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
releaseId = release.id();
|
||||
artistId = artist.id();
|
||||
clusterId = cluster.id();
|
||||
clusterTypeId = clusterType.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artists {Artist::getByFilter(session, {clusterId})};
|
||||
CHECK(artists.size() == 1);
|
||||
CHECK(artists.front().id() == artistId);
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
auto releases {artist->getReleases()};
|
||||
CHECK(releases.size() == 1);
|
||||
CHECK(releases.front().id() == releaseId);
|
||||
|
||||
releases = artist->getReleases({clusterId});
|
||||
CHECK(releases.size() == 1);
|
||||
CHECK(releases.front().id() == releaseId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
auto release {Release::getById(session, releaseId)};
|
||||
auto track {Track::getById(session, trackId)};
|
||||
auto cluster {Cluster::getById(session, clusterId)};
|
||||
auto clusterType {ClusterType::getById(session, clusterTypeId)};
|
||||
|
||||
track.remove();
|
||||
artist.remove();
|
||||
release.remove();
|
||||
cluster.remove();
|
||||
clusterType.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testSingleTrackSingleReleaseSingleArtistMultiClusters(Wt::Dbo::Session& session)
|
||||
{
|
||||
IdType trackId {};
|
||||
IdType releaseId {};
|
||||
IdType artistId {};
|
||||
IdType cluster1Id {};
|
||||
IdType cluster2Id {};
|
||||
IdType clusterTypeId {};
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto track {Track::create(session, "MyTrackFile")};
|
||||
auto release {Release::create(session, "MyRelease")};
|
||||
auto artist {Artist::create(session, "MyArtist")};
|
||||
|
||||
auto clusterType {ClusterType::create(session, "MyType")};
|
||||
auto cluster1 {Cluster::create(session, clusterType, "MyCluster1")};
|
||||
auto cluster2 {Cluster::create(session, clusterType, "MyCluster2")};
|
||||
|
||||
auto trackArtistLink {TrackArtistLink::create(session, track, artist, TrackArtistLink::Type::Artist)};
|
||||
track.modify()->setRelease(release);
|
||||
cluster1.modify()->addTrack(track);
|
||||
cluster2.modify()->addTrack(track);
|
||||
|
||||
session.flush();
|
||||
trackId = track.id();
|
||||
releaseId = release.id();
|
||||
artistId = artist.id();
|
||||
cluster1Id = cluster1.id();
|
||||
cluster2Id = cluster2.id();
|
||||
clusterTypeId = clusterType.id();
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
auto releases {artist->getReleases()};
|
||||
CHECK(releases.size() == 1);
|
||||
CHECK(releases.front().id() == releaseId);
|
||||
|
||||
releases = artist->getReleases({cluster1Id, cluster2Id});
|
||||
CHECK(releases.size() == 1);
|
||||
CHECK(releases.front().id() == releaseId);
|
||||
}
|
||||
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
auto artist {Artist::getById(session, artistId)};
|
||||
auto release {Release::getById(session, releaseId)};
|
||||
auto track {Track::getById(session, trackId)};
|
||||
auto cluster1 {Cluster::getById(session, cluster1Id)};
|
||||
auto cluster2 {Cluster::getById(session, cluster1Id)};
|
||||
auto clusterType {ClusterType::getById(session, clusterTypeId)};
|
||||
|
||||
track.remove();
|
||||
artist.remove();
|
||||
release.remove();
|
||||
cluster1.remove();
|
||||
cluster2.remove();
|
||||
clusterType.remove();
|
||||
}
|
||||
}
|
||||
|
||||
static
|
||||
void
|
||||
testDatabaseEmpty(Wt::Dbo::Session& session)
|
||||
{
|
||||
Wt::Dbo::Transaction transaction {session};
|
||||
|
||||
CHECK(Artist::getAll(session).size() == 0);
|
||||
CHECK(Cluster::getAll(session).size() == 0);
|
||||
CHECK(ClusterType::getAll(session).size() == 0);
|
||||
CHECK(Release::getAll(session).size() == 0);
|
||||
CHECK(Track::getAll(session).size() == 0);
|
||||
}
|
||||
|
||||
int main(int argc, char* argv[])
|
||||
{
|
||||
|
||||
try
|
||||
{
|
||||
boost::filesystem::path tmpFile {boost::filesystem::temp_directory_path() / boost::filesystem::unique_path()};
|
||||
ScopedFileDeleter tmpFileDeleter {tmpFile};
|
||||
|
||||
std::cout << "Database test file: '" << tmpFile.string() << "'" << std::endl;
|
||||
|
||||
std::unique_ptr<Wt::Dbo::SqlConnectionPool> connectionPool{ Handler::createConnectionPool(tmpFile) };
|
||||
|
||||
Handler db {*connectionPool};
|
||||
Wt::Dbo::Session& session {db.getSession()};
|
||||
|
||||
auto runTest = [&session](const std::string& name, std::function<void(Wt::Dbo::Session&)> testFunc)
|
||||
{
|
||||
std::cout << "Running test '" << name << "'..." << std::endl;
|
||||
testFunc(session);
|
||||
testDatabaseEmpty(session);
|
||||
std::cout << "Running test '" << name << "': SUCCESS" << std::endl;
|
||||
};
|
||||
|
||||
#define RUN_TEST(test) runTest(#test, test)
|
||||
|
||||
RUN_TEST(testSingleTrack);
|
||||
RUN_TEST(testSingleArtist);
|
||||
RUN_TEST(testSingleRelease);
|
||||
RUN_TEST(testSingleCluster);
|
||||
|
||||
RUN_TEST(testSingleTrackSingleArtist);
|
||||
RUN_TEST(testSingleTrackSingleArtistMultiRoles);
|
||||
RUN_TEST(testSingleTrackMultiArtists);
|
||||
|
||||
RUN_TEST(testSingleTrackSingleRelease);
|
||||
|
||||
RUN_TEST(testSingleTrackSingleCluster);
|
||||
|
||||
RUN_TEST(testSingleTrackSingleReleaseSingleCluster);
|
||||
RUN_TEST(testSingleTrackSingleArtistMultiClusters);
|
||||
RUN_TEST(testSingleTrackSingleArtistMultiRolesMultiClusters);
|
||||
RUN_TEST(testMultiTracksSingleArtistMultiClusters);
|
||||
|
||||
RUN_TEST(testSingleTrackSingleReleaseSingleArtist);
|
||||
|
||||
RUN_TEST(testSingleTrackSingleReleaseSingleArtistSingleCluster);
|
||||
RUN_TEST(testSingleTrackSingleReleaseSingleArtistMultiClusters);
|
||||
|
||||
}
|
||||
catch (std::exception& e)
|
||||
{
|
||||
std::cerr << "Caught exception: " << e.what() << std::endl;
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
@@ -1,12 +1,29 @@
|
||||
/*
|
||||
* Copyright (C) 2019 Emeric Poupon
|
||||
*
|
||||
* This file is part of LMS.
|
||||
*
|
||||
* LMS is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* LMS is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with LMS. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <sstream>
|
||||
#include <cassert>
|
||||
#include <assert.h>
|
||||
#include <iostream>
|
||||
|
||||
#include "DataNormalizer.hpp"
|
||||
#include "Network.hpp"
|
||||
|
||||
#define CHECK(pred) assert((pred))
|
||||
|
||||
using namespace SOM;
|
||||
|
||||
int main(int argc, char* argv[])
|
||||
Reference in New Issue
Block a user