117 lines
3.6 KiB
C++
117 lines
3.6 KiB
C++
#include "../lmdb-boost-serialization.hh"
|
|
#include "../lmdb-typed.hh"
|
|
|
|
#define CATCH_CONFIG_MAIN // for catch2 < 3
|
|
#ifdef CATCH2_SPLIT_HEADERS
|
|
#include <catch2/catch_all.hpp>
|
|
#else
|
|
#include <catch2/catch.hpp>
|
|
#endif
|
|
|
|
#include <c++utilities/application/global.h>
|
|
|
|
#include <ctime>
|
|
|
|
#include <unistd.h>
|
|
|
|
using namespace std;
|
|
using namespace LMDBSafe;
|
|
|
|
struct Member {
|
|
std::string firstName;
|
|
std::string lastName;
|
|
std::time_t enrolled;
|
|
};
|
|
|
|
template <class Archive> void serialize(Archive &ar, Member &g, const unsigned int version)
|
|
{
|
|
CPP_UTILITIES_UNUSED(version)
|
|
ar &g.firstName &g.lastName &g.enrolled;
|
|
}
|
|
|
|
TEST_CASE("Basic typed tests", "[basictyped]")
|
|
{
|
|
unlink("./tests-typed");
|
|
using tmembers_t = TypedDBI<Member, index_on<Member, string, &Member::firstName>, index_on<Member, string, &Member::lastName>,
|
|
index_on<Member, time_t, &Member::enrolled>>;
|
|
auto tmembers = tmembers_t(getMDBEnv("./tests-typed.lmdb", MDB_CREATE | MDB_NOSUBDIR, 0600), "members");
|
|
|
|
// insert three rows
|
|
auto txn = tmembers.getRWTransaction();
|
|
auto m = Member{ "bert", "hubert", std::time_t() };
|
|
txn.put(m);
|
|
m.firstName = "bertus";
|
|
m.lastName = "testperson";
|
|
m.enrolled = time(0);
|
|
txn.put(m);
|
|
m.firstName = "other";
|
|
txn.put(m);
|
|
|
|
// find and read back inserted rows via ID
|
|
auto out = Member{};
|
|
REQUIRE(txn.get(1, out));
|
|
REQUIRE(out.firstName == "bert");
|
|
REQUIRE(out.lastName == "hubert");
|
|
REQUIRE(out.enrolled == std::time_t());
|
|
REQUIRE(txn.get(2, out));
|
|
REQUIRE(out.firstName == "bertus");
|
|
REQUIRE(out.lastName == "testperson");
|
|
REQUIRE(out.enrolled == m.enrolled);
|
|
REQUIRE(txn.get(3, out));
|
|
REQUIRE(out.firstName == "other");
|
|
REQUIRE(out.lastName == "testperson");
|
|
REQUIRE(out.enrolled == m.enrolled);
|
|
REQUIRE(!txn.get(4, out));
|
|
REQUIRE(txn.size<0>() == txn.size());
|
|
REQUIRE(txn.size<1>() == txn.size());
|
|
REQUIRE(txn.size<2>() == txn.size());
|
|
|
|
// find rows by prefix via index
|
|
auto range = txn.prefix_range<0>("bert");
|
|
auto names = std::vector<std::string>();
|
|
for (auto &iter = range.first; iter != range.second; ++iter) {
|
|
names.push_back(iter->firstName);
|
|
}
|
|
REQUIRE(names == std::vector<std::string>{ "bert", "bertus" });
|
|
auto range2 = txn.prefix_range<0>("nosuchperson");
|
|
REQUIRE(range2.first != range2.second);
|
|
|
|
// override existing row
|
|
m.firstName = "another";
|
|
m.enrolled += 1;
|
|
txn.put(m, 3);
|
|
REQUIRE(txn.get(3, out));
|
|
REQUIRE(out.firstName == "another");
|
|
REQUIRE(out.lastName == "testperson");
|
|
REQUIRE(out.enrolled == m.enrolled);
|
|
REQUIRE(txn.size<0>() == txn.size());
|
|
REQUIRE(txn.size<1>() == txn.size());
|
|
REQUIRE(txn.size<2>() == txn.size());
|
|
|
|
// row only retrievable via index with updated field value
|
|
REQUIRE(!txn.get<0>("other", out));
|
|
out.firstName.clear();
|
|
out.lastName.clear();
|
|
REQUIRE(txn.get<0>("another", out));
|
|
REQUIRE(out.firstName == "another");
|
|
REQUIRE(out.lastName == "testperson");
|
|
|
|
// modify existing row via modify function
|
|
txn.modify(3, [] (Member &member) { member.firstName = "yetanother"; });
|
|
REQUIRE(!txn.get<0>("another", out));
|
|
REQUIRE(txn.get<0>("yetanother", out));
|
|
REQUIRE(out.firstName == "yetanother");
|
|
REQUIRE(out.lastName == "testperson");
|
|
REQUIRE(out.enrolled == m.enrolled);
|
|
|
|
// rebuild the database
|
|
txn.rebuild([] (IDType, Member *member) {
|
|
return member->firstName != "bertus";
|
|
});
|
|
REQUIRE(txn.size() == 2);
|
|
REQUIRE(txn.size<0>() == txn.size());
|
|
REQUIRE(!txn.get<0>("bertus", out));
|
|
|
|
txn.abort();
|
|
}
|