1 files changed, 159 insertions, 0 deletions
diff --git a/src/database/test/test_database.cpp b/src/database/test/test_database.cpp
new file mode 100644
index 00000000..d61421ee
--- /dev/null
+++ b/src/database/test/test_database.cpp
@@ -0,0 +1,159 @@
+#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 "song.hpp"
+#include "spi_fixture.hpp"
+#include "tag_parser.hpp"
+
+namespace database {
+
+class TestBackends : public IFileGatherer, public ITagParser {
+ public:
+  std::map<std::string, SongTags> songs;
+
+  auto MakeSong(const std::string& path, const std::string& title) -> void {
+    SongTags tags;
+    tags.encoding = Encoding::kMp3;
+    tags.title = title;
+    songs[path] = tags;
+  }
+
+  auto FindFiles(const std::string& root,
+                 std::function<void(const std::string&)> cb) -> void override {
+    for (auto keyval : songs) {
+      std::invoke(cb, keyval.first);
+    }
+  }
+
+  auto ReadAndParseTags(const std::string& path, SongTags* out)
+      -> bool override {
+    if (songs.contains(path)) {
+      *out = songs.at(path);
+      return true;
+    }
+    return false;
+  }
+};
+
+TEST_CASE("song database", "[integration]") {
+  I2CFixture i2c;
+  SpiFixture spi;
+  drivers::DriverCache drivers;
+  auto storage = drivers.AcquireStorage();
+
+  Database::Destroy();
+
+  TestBackends songs;
+  auto open_res = Database::Open(&songs, &songs);
+  REQUIRE(open_res.has_value());
+  std::unique_ptr<Database> db(open_res.value());
+
+  SECTION("empty database") {
+    std::unique_ptr<std::vector<Song>> res(db->GetSongs(10).get().values());
+    REQUIRE(res->size() == 0);
+  }
+
+  SECTION("add new songs") {
+    songs.MakeSong("song1.mp3", "Song 1");
+    songs.MakeSong("song2.wav", "Song 2");
+    songs.MakeSong("song3.exe", "Song 3");
+
+    db->Update();
+
+    std::unique_ptr<std::vector<Song>> res(db->GetSongs(10).get().values());
+    REQUIRE(res->size() == 3);
+    CHECK(*res->at(0).tags().title == "Song 1");
+    CHECK(res->at(0).data().id() == 1);
+    CHECK(*res->at(1).tags().title == "Song 2");
+    CHECK(res->at(1).data().id() == 2);
+    CHECK(*res->at(2).tags().title == "Song 3");
+    CHECK(res->at(2).data().id() == 3);
+
+    SECTION("update with no filesystem changes") {
+      db->Update();
+
+      std::unique_ptr<std::vector<Song>> new_res(
+          db->GetSongs(10).get().values());
+      REQUIRE(new_res->size() == 3);
+      CHECK(res->at(0) == new_res->at(0));
+      CHECK(res->at(1) == new_res->at(1));
+      CHECK(res->at(2) == new_res->at(2));
+    }
+
+    SECTION("update with all songs gone") {
+      songs.songs.clear();
+
+      db->Update();
+
+      std::unique_ptr<std::vector<Song>> new_res(
+          db->GetSongs(10).get().values());
+      CHECK(new_res->size() == 0);
+
+      SECTION("update with one song returned") {
+        songs.MakeSong("song2.wav", "Song 2");
+
+        db->Update();
+
+        std::unique_ptr<std::vector<Song>> new_res(
+            db->GetSongs(10).get().values());
+        REQUIRE(new_res->size() == 1);
+        CHECK(res->at(1) == new_res->at(0));
+      }
+    }
+
+    SECTION("update with one song gone") {
+      songs.songs.erase("song2.wav");
+
+      db->Update();
+
+      std::unique_ptr<std::vector<Song>> new_res(
+          db->GetSongs(10).get().values());
+      REQUIRE(new_res->size() == 2);
+      CHECK(res->at(0) == new_res->at(0));
+      CHECK(res->at(2) == new_res->at(1));
+    }
+
+    SECTION("update with tags changed") {
+      songs.MakeSong("song3.exe", "The Song 3");
+
+      db->Update();
+
+      std::unique_ptr<std::vector<Song>> new_res(
+          db->GetSongs(10).get().values());
+      REQUIRE(new_res->size() == 3);
+      CHECK(res->at(0) == new_res->at(0));
+      CHECK(res->at(1) == new_res->at(1));
+      CHECK(*new_res->at(2).tags().title == "The Song 3");
+      // The id should not have changed, since this was just a tag update.
+      CHECK(res->at(2).data().id() == new_res->at(2).data().id());
+    }
+
+    SECTION("update with one new song") {
+      songs.MakeSong("my song.midi", "Song 1 (nightcore remix)");
+
+      db->Update();
+
+      std::unique_ptr<std::vector<Song>> new_res(
+          db->GetSongs(10).get().values());
+      REQUIRE(new_res->size() == 4);
+      CHECK(res->at(0) == new_res->at(0));
+      CHECK(res->at(1) == new_res->at(1));
+      CHECK(res->at(2) == new_res->at(2));
+      CHECK(*new_res->at(3).tags().title == "Song 1 (nightcore remix)");
+      CHECK(new_res->at(3).data().id() == 4);
+    }
+  }
+}
+
+}  // namespace database