lms/src/libs/database/test/Mood.cpp

/*
 * Copyright (C) 2025 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 "Common.hpp"

namespace lms::db::tests
{
    using ScopedMood = ScopedEntity<db::Mood>;

    TEST_F(DatabaseFixture, Mood_create)
    {
        {
            auto transaction{ session.createReadTransaction() };
            EXPECT_EQ(Mood::getCount(session), 0);
        }

        ScopedMood mood{ session, "Happy" };

        {
            auto transaction{ session.createReadTransaction() };
            EXPECT_EQ(Mood::getCount(session), 1);

            const Mood::pointer found{ Mood::find(session, mood.getId()) };
            ASSERT_TRUE(found);
            EXPECT_EQ(found->getName(), "Happy");
        }
    }

    TEST_F(DatabaseFixture, Mood_findByName)
    {
        ScopedMood mood{ session, "Chill" };

        {
            auto transaction{ session.createReadTransaction() };

            EXPECT_TRUE(Mood::find(session, "Chill"));
            EXPECT_FALSE(Mood::find(session, "chill"));
            EXPECT_FALSE(Mood::find(session, ""));
            EXPECT_FALSE(Mood::find(session, "Chil"));
        }
    }

    TEST_F(DatabaseFixture, Mood_orphan)
    {
        ScopedMood mood{ session, "Sad" };

        {
            auto transaction{ session.createReadTransaction() };
            const auto orphans{ Mood::findOrphanIds(session) };
            ASSERT_EQ(orphans.results.size(), 1);
            EXPECT_EQ(orphans.results.front(), mood.getId());
        }
    }

    TEST_F(DatabaseFixture, Mood_singleTrack)
    {
        ScopedTrack track{ session };
        ScopedMood mood1{ session, "Dark" };
        ScopedMood mood2{ session, "Upbeat" };

        {
            auto transaction{ session.createReadTransaction() };
            EXPECT_EQ(Mood::findOrphanIds(session).results.size(), 2);
            EXPECT_EQ(track->getMoods().size(), 0);
            EXPECT_EQ(track->getMoodIds().size(), 0);
        }

        {
            auto transaction{ session.createWriteTransaction() };
            track.get().modify()->setMoods(std::array{ mood1.get() });
        }

        {
            auto transaction{ session.createReadTransaction() };

            const auto moods{ Mood::findIds(session, Mood::FindParameters{}.setTrack(track.getId())) };
            ASSERT_EQ(moods.results.size(), 1);
            EXPECT_EQ(moods.results.front(), mood1.getId());

            const auto orphans{ Mood::findOrphanIds(session) };
            ASSERT_EQ(orphans.results.size(), 1);
            EXPECT_EQ(orphans.results.front(), mood2.getId());

            const auto trackMoods{ track->getMoods() };
            ASSERT_EQ(trackMoods.size(), 1);
            EXPECT_EQ(trackMoods.front()->getId(), mood1.getId());

            const auto trackMoodIds{ track->getMoodIds() };
            ASSERT_EQ(trackMoodIds.size(), 1);
            EXPECT_EQ(trackMoodIds.front(), mood1.getId());
        }

        {
            auto transaction{ session.createReadTransaction() };
            const auto tracks{ Track::findIds(session, Track::FindParameters{}.setFilters(Filters{}.setMood(mood1.getId()))) };
            ASSERT_EQ(tracks.results.size(), 1);
            EXPECT_EQ(tracks.results.front(), track.getId());

            const auto tracks2{ Track::findIds(session, Track::FindParameters{}.setFilters(Filters{}.setMood(mood2.getId()))) };
            EXPECT_EQ(tracks2.results.size(), 0);
        }
    }

    TEST_F(DatabaseFixture, Mood_multipleMoodsOnTrack)
    {
        ScopedTrack track{ session };
        ScopedMood mood1{ session, "Melancholic" };
        ScopedMood mood2{ session, "Romantic" };

        {
            auto transaction{ session.createWriteTransaction() };
            track.get().modify()->setMoods(std::array{ mood1.get(), mood2.get() });
        }

        {
            auto transaction{ session.createReadTransaction() };
            EXPECT_EQ(Mood::findOrphanIds(session).results.size(), 0);

            const auto trackMoods{ track->getMoods() };
            EXPECT_EQ(trackMoods.size(), 2);
        }

        {
            auto transaction{ session.createReadTransaction() };
            const auto tracks{ Track::findIds(session, Track::FindParameters{}.setFilters(Filters{}.setMood(mood1.getId()))) };
            ASSERT_EQ(tracks.results.size(), 1);
            EXPECT_EQ(tracks.results.front(), track.getId());

            const auto tracks2{ Track::findIds(session, Track::FindParameters{}.setFilters(Filters{}.setMood(mood2.getId()))) };
            ASSERT_EQ(tracks2.results.size(), 1);
            EXPECT_EQ(tracks2.results.front(), track.getId());
        }
    }

    TEST_F(DatabaseFixture, Mood_sortByName)
    {
        ScopedMood m1{ session, "Zzz" };
        ScopedMood m2{ session, "Aaa" };
        ScopedMood m3{ session, "Mmm" };

        {
            auto transaction{ session.createReadTransaction() };
            const auto moods{ Mood::findIds(session, Mood::FindParameters{}.setSortMethod(MoodSortMethod::Name)) };
            ASSERT_EQ(moods.results.size(), 3);
            EXPECT_EQ(moods.results[0], m2.getId());
            EXPECT_EQ(moods.results[1], m3.getId());
            EXPECT_EQ(moods.results[2], m1.getId());
        }
    }

    TEST_F(DatabaseFixture, Mood_sortByTrackCount)
    {
        ScopedMood m1{ session, "Energetic" };
        ScopedMood m2{ session, "Chill" };
        ScopedTrack track1{ session };
        ScopedTrack track2{ session };
        ScopedTrack track3{ session };

        {
            auto transaction{ session.createWriteTransaction() };
            track1.get().modify()->setMoods(std::array{ m1.get() });
            track2.get().modify()->setMoods(std::array{ m1.get() });
            track3.get().modify()->setMoods(std::array{ m2.get() });
        }

        {
            auto transaction{ session.createReadTransaction() };
            const auto moods{ Mood::findIds(session, Mood::FindParameters{}.setSortMethod(MoodSortMethod::TrackCountDesc)) };
            ASSERT_EQ(moods.results.size(), 2);
            EXPECT_EQ(moods.results[0], m1.getId());
            EXPECT_EQ(moods.results[1], m2.getId());
        }
    }
} // namespace lms::db::tests