test_registry.cpp 1.1 KB

12345678910111213141516171819202122232425262728
  1. #include "db_test_util.hpp"
  2. #include "collection_registry.hpp"
  3. using namespace svapi;
  4. TEST(CollectionMeta, JsonRoundTrip) {
  5. CollectionMeta m{"docs", "vector", 1536, "text-embedding-3-small", 123};
  6. CollectionMeta r = CollectionMeta::fromJson(m.toJson());
  7. EXPECT_EQ(r.name, "docs"); EXPECT_EQ(r.kind, "vector");
  8. EXPECT_EQ(r.vectorDimension, 1536u); EXPECT_EQ(r.embeddingModel, "text-embedding-3-small");
  9. }
  10. TEST(CollectionRegistry, IsolatedPerProject) {
  11. SVAPI_REQUIRE_DB(db);
  12. const std::string pa = testutil::tmpName("rega"), pb = testutil::tmpName("regb");
  13. db->ensureProject(pa); db->ensureProject(pb);
  14. CollectionRegistry reg(*db);
  15. reg.ensure(pa); reg.ensure(pb);
  16. EXPECT_TRUE(reg.add(pa, {"users", "json", 0, "", 0}));
  17. EXPECT_TRUE(reg.get(pa, "users").has_value());
  18. EXPECT_FALSE(reg.get(pb, "users").has_value()); // isolation across projects
  19. EXPECT_EQ(reg.list(pa).size(), 1u);
  20. EXPECT_EQ(reg.list(pb).size(), 0u);
  21. EXPECT_TRUE(reg.remove(pa, "users"));
  22. EXPECT_FALSE(reg.get(pa, "users").has_value());
  23. db->dropProject(pa); db->dropProject(pb);
  24. }