Add hydration tests for the ODB ORM. Apparently it works.

tests/odb-orm/orm_test_common.h

@@ -0,0 +1,159 @@
+#pragma once
+
+#include <cstdlib>
+#include <filesystem>
+#include <fstream>
+#include <sstream>
+#include <stdexcept>
+#include <string>
+#include <vector>
+
+#include <gtest/gtest.h>
+
+#include <cppbessot/model/Agent.h>
+
+inline std::string cppbessot_required_env(const char* name)
+{
+  const char* value = std::getenv(name);
+  if(value == nullptr || value[0] == '\0')
+  {
+    throw std::runtime_error(std::string("missing required environment variable: ") + name);
+  }
+  return value;
+}
+
+inline std::string cppbessot_env_or_default(const char* name, const char* fallback)
+{
+  const char* value = std::getenv(name);
+  if(value != nullptr && value[0] != '\0')
+  {
+    return value;
+  }
+
+  if(fallback != nullptr && fallback[0] != '\0')
+  {
+    return fallback;
+  }
+
+  throw std::runtime_error(std::string("missing required environment variable: ") + name);
+}
+
+inline std::vector<std::filesystem::path> cppbessot_sql_files()
+{
+  const std::filesystem::path sql_dir(CPPBESSOT_ODB_TEST_SQL_DIR);
+  if(!std::filesystem::is_directory(sql_dir))
+  {
+    throw std::runtime_error("SQL DDL directory does not exist: " + sql_dir.string());
+  }
+
+  std::vector<std::filesystem::path> files;
+  for(const auto& entry : std::filesystem::directory_iterator(sql_dir))
+  {
+    if(entry.is_regular_file() && entry.path().extension() == ".sql")
+    {
+      files.push_back(entry.path());
+    }
+  }
+
+  std::sort(files.begin(), files.end());
+  if(files.empty())
+  {
+    throw std::runtime_error("No SQL DDL files found in " + sql_dir.string());
+  }
+  return files;
+}
+
+inline std::string cppbessot_read_text_file(const std::filesystem::path& path)
+{
+  std::ifstream stream(path);
+  if(!stream)
+  {
+    throw std::runtime_error("Failed to open file: " + path.string());
+  }
+
+  std::ostringstream buffer;
+  buffer << stream.rdbuf();
+  return buffer.str();
+}
+
+template <typename Database, typename Transaction>
+void cppbessot_run_agent_orm_roundtrip(Database& db)
+{
+  {
+    Transaction t(db.begin());
+    models::Agent first{};
+    first.id = "agent-orm-1";
+    first.role = "REQUESTER";
+    first.persistent = true;
+    first.displayName = "ODB Agent";
+    db.persist(first);
+
+    models::Agent second{};
+    second.id = "agent-orm-2";
+    second.role = "PROVIDER";
+    second.persistent = false;
+    second.displayName = "ODB Agent Secondary";
+    db.persist(second);
+    t.commit();
+  }
+
+  {
+    Transaction t(db.begin());
+    models::Agent loaded{};
+    db.template load<models::Agent>("agent-orm-1", loaded);
+    EXPECT_EQ(loaded.id, "agent-orm-1");
+    EXPECT_EQ(loaded.role, "REQUESTER");
+    EXPECT_TRUE(loaded.persistent);
+    EXPECT_EQ(loaded.displayName, "ODB Agent");
+
+    loaded.displayName = "ODB Agent Updated";
+    db.update(loaded);
+    t.commit();
+  }
+
+  {
+    Transaction t(db.begin());
+    using query = odb::query<models::Agent>;
+    odb::result<models::Agent> results(
+      db.template query<models::Agent>(query::persistent == true || query::role == "PROVIDER"));
+    bool saw_updated_agent = false;
+    bool saw_secondary_agent = false;
+    std::size_t count = 0;
+    for(const models::Agent& row : results)
+    {
+      ++count;
+      if(row.id == "agent-orm-1")
+      {
+        saw_updated_agent = true;
+        EXPECT_EQ(row.role, "REQUESTER");
+        EXPECT_TRUE(row.persistent);
+        EXPECT_EQ(row.displayName, "ODB Agent Updated");
+      }
+      else if(row.id == "agent-orm-2")
+      {
+        saw_secondary_agent = true;
+        EXPECT_EQ(row.role, "PROVIDER");
+        EXPECT_FALSE(row.persistent);
+        EXPECT_EQ(row.displayName, "ODB Agent Secondary");
+      }
+      else
+      {
+        ADD_FAILURE() << "unexpected hydrated Agent row id: " << row.id;
+      }
+    }
+    EXPECT_EQ(count, 2U);
+    EXPECT_TRUE(saw_updated_agent);
+    EXPECT_TRUE(saw_secondary_agent);
+
+    db.template erase<models::Agent>("agent-orm-1");
+    db.template erase<models::Agent>("agent-orm-2");
+    t.commit();
+  }
+
+  {
+    Transaction t(db.begin());
+    EXPECT_FALSE(db.template find<models::Agent>("agent-orm-1"));
+    EXPECT_FALSE(db.template find<models::Agent>("agent-orm-2"));
+    t.commit();
+  }
+}