summaryrefslogtreecommitdiff
path: root/src/tangara/database/test/test_database.cpp
diff options
context:
space:
mode:
authorjacqueline <me@jacqueline.id.au>2024-05-02 19:12:26 +1000
committerjacqueline <me@jacqueline.id.au>2024-05-02 19:12:26 +1000
commit1573a8c4cde1cd9528b422b2dcc598e37ffe94a7 (patch)
treed162822b8fd7054f81bace0c7a65ab4d5e6f93ef /src/tangara/database/test/test_database.cpp
parenta231fd1c8afedbeb14b0bc77d76bad61db986059 (diff)
downloadtangara-fw-1573a8c4cde1cd9528b422b2dcc598e37ffe94a7.tar.gz
WIP merge cyclically dependent components into one big component
Diffstat (limited to 'src/tangara/database/test/test_database.cpp')
-rw-r--r--src/tangara/database/test/test_database.cpp210
1 files changed, 210 insertions, 0 deletions
diff --git a/src/tangara/database/test/test_database.cpp b/src/tangara/database/test/test_database.cpp
new file mode 100644
index 00000000..6aec9bfb
--- /dev/null
+++ b/src/tangara/database/test/test_database.cpp
@@ -0,0 +1,210 @@
+/*
+ * Copyright 2023 jacqueline <me@jacqueline.id.au>
+ *
+ * SPDX-License-Identifier: GPL-3.0-only
+ */
+
+#include "database.hpp"
+
+#include <stdint.h>
+#include <iomanip>
+#include <map>
+#include <memory>
+#include <string>
+
+#include "catch2/catch.hpp"
+#include "driver_cache.hpp"
+#include "esp_log.h"
+#include "file_gatherer.hpp"
+#include "i2c_fixture.hpp"
+#include "leveldb/db.h"
+#include "spi_fixture.hpp"
+#include "tag_parser.hpp"
+#include "track.hpp"
+
+namespace database {
+
+class TestBackends : public IFileGatherer, public ITagParser {
+ public:
+ std::map<std::pmr::string, TrackTags> tracks;
+
+ auto MakeTrack(const std::pmr::string& path, const std::pmr::string& title)
+ -> void {
+ TrackTags tags;
+ tags.encoding = Encoding::kMp3;
+ tags.title = title;
+ tracks[path] = tags;
+ }
+
+ auto FindFiles(const std::pmr::string& root,
+ std::function<void(const std::pmr::string&)> cb)
+ -> void override {
+ for (auto keyval : tracks) {
+ std::invoke(cb, keyval.first);
+ }
+ }
+
+ auto ReadAndParseTags(const std::pmr::string& path, TrackTags* out)
+ -> bool override {
+ if (tracks.contains(path)) {
+ *out = tracks.at(path);
+ return true;
+ }
+ return false;
+ }
+};
+
+TEST_CASE("track database", "[integration]") {
+ I2CFixture i2c;
+ SpiFixture spi;
+ drivers::DriverCache drivers;
+ auto storage = drivers.AcquireStorage();
+
+ Database::Destroy();
+
+ TestBackends tracks;
+ auto open_res = Database::Open(&tracks, &tracks);
+ REQUIRE(open_res.has_value());
+ std::unique_ptr<Database> db(open_res.value());
+
+ SECTION("empty database") {
+ std::unique_ptr<Result<Track>> res(db->GetTracks(10).get());
+ REQUIRE(res->values().size() == 0);
+ }
+
+ SECTION("add new tracks") {
+ tracks.MakeTrack("track1.mp3", "Track 1");
+ tracks.MakeTrack("track2.wav", "Track 2");
+ tracks.MakeTrack("track3.exe", "Track 3");
+
+ db->Update();
+
+ std::unique_ptr<Result<Track>> res(db->GetTracks(10).get());
+ REQUIRE(res->values().size() == 3);
+ CHECK(*res->values().at(0).tags().title == "Track 1");
+ CHECK(res->values().at(0).data().id() == 1);
+ CHECK(*res->values().at(1).tags().title == "Track 2");
+ CHECK(res->values().at(1).data().id() == 2);
+ CHECK(*res->values().at(2).tags().title == "Track 3");
+ CHECK(res->values().at(2).data().id() == 3);
+
+ SECTION("update with no filesystem changes") {
+ db->Update();
+
+ std::unique_ptr<Result<Track>> new_res(db->GetTracks(10).get());
+ REQUIRE(new_res->values().size() == 3);
+ CHECK(res->values().at(0) == new_res->values().at(0));
+ CHECK(res->values().at(1) == new_res->values().at(1));
+ CHECK(res->values().at(2) == new_res->values().at(2));
+ }
+
+ SECTION("update with all tracks gone") {
+ tracks.tracks.clear();
+
+ db->Update();
+
+ std::unique_ptr<Result<Track>> new_res(db->GetTracks(10).get());
+ CHECK(new_res->values().size() == 0);
+
+ SECTION("update with one track returned") {
+ tracks.MakeTrack("track2.wav", "Track 2");
+
+ db->Update();
+
+ std::unique_ptr<Result<Track>> new_res(db->GetTracks(10).get());
+ REQUIRE(new_res->values().size() == 1);
+ CHECK(res->values().at(1) == new_res->values().at(0));
+ }
+ }
+
+ SECTION("update with one track gone") {
+ tracks.tracks.erase("track2.wav");
+
+ db->Update();
+
+ std::unique_ptr<Result<Track>> new_res(db->GetTracks(10).get());
+ REQUIRE(new_res->values().size() == 2);
+ CHECK(res->values().at(0) == new_res->values().at(0));
+ CHECK(res->values().at(2) == new_res->values().at(1));
+ }
+
+ SECTION("update with tags changed") {
+ tracks.MakeTrack("track3.exe", "The Track 3");
+
+ db->Update();
+
+ std::unique_ptr<Result<Track>> new_res(db->GetTracks(10).get());
+ REQUIRE(new_res->values().size() == 3);
+ CHECK(res->values().at(0) == new_res->values().at(0));
+ CHECK(res->values().at(1) == new_res->values().at(1));
+ CHECK(*new_res->values().at(2).tags().title == "The Track 3");
+ // The id should not have changed, since this was just a tag update.
+ CHECK(res->values().at(2).data().id() ==
+ new_res->values().at(2).data().id());
+ }
+
+ SECTION("update with one new track") {
+ tracks.MakeTrack("my track.midi", "Track 1 (nightcore remix)");
+
+ db->Update();
+
+ std::unique_ptr<Result<Track>> new_res(db->GetTracks(10).get());
+ REQUIRE(new_res->values().size() == 4);
+ CHECK(res->values().at(0) == new_res->values().at(0));
+ CHECK(res->values().at(1) == new_res->values().at(1));
+ CHECK(res->values().at(2) == new_res->values().at(2));
+ CHECK(*new_res->values().at(3).tags().title ==
+ "Track 1 (nightcore remix)");
+ CHECK(new_res->values().at(3).data().id() == 4);
+ }
+
+ SECTION("get tracks with pagination") {
+ std::unique_ptr<Result<Track>> res(db->GetTracks(1).get());
+
+ REQUIRE(res->values().size() == 1);
+ CHECK(res->values().at(0).data().id() == 1);
+ REQUIRE(res->next_page());
+
+ res.reset(db->GetPage(&res->next_page().value()).get());
+
+ REQUIRE(res->values().size() == 1);
+ CHECK(res->values().at(0).data().id() == 2);
+ REQUIRE(res->next_page());
+
+ res.reset(db->GetPage(&res->next_page().value()).get());
+
+ REQUIRE(res->values().size() == 1);
+ CHECK(res->values().at(0).data().id() == 3);
+ REQUIRE(!res->next_page());
+
+ SECTION("page backwards") {
+ REQUIRE(res->prev_page());
+
+ res.reset(db->GetPage(&res->prev_page().value()).get());
+
+ REQUIRE(res->values().size() == 1);
+ CHECK(res->values().at(0).data().id() == 2);
+ REQUIRE(res->prev_page());
+
+ res.reset(db->GetPage(&res->prev_page().value()).get());
+
+ REQUIRE(res->values().size() == 1);
+ CHECK(res->values().at(0).data().id() == 1);
+ REQUIRE(!res->prev_page());
+
+ SECTION("page forwards again") {
+ REQUIRE(res->next_page());
+
+ res.reset(db->GetPage(&res->next_page().value()).get());
+
+ REQUIRE(res->values().size() == 1);
+ CHECK(res->values().at(0).data().id() == 2);
+ CHECK(res->next_page());
+ CHECK(res->prev_page());
+ }
+ }
+ }
+ }
+}
+
+} // namespace database