mirror of
https://github.com/getml/sqlgen.git
synced 2026-01-01 15:09:46 -06:00
109 lines
4.1 KiB
C++
109 lines
4.1 KiB
C++
#ifndef SQLGEN_BUILD_DRY_TESTS_ONLY
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <rfl.hpp>
|
|
#include <rfl/json.hpp>
|
|
#include <sqlgen.hpp>
|
|
#include <sqlgen/postgres.hpp>
|
|
#include <vector>
|
|
|
|
namespace test_enum_cross_table {
|
|
enum class AccessRestriction { PUBLIC = 1, INTERNAL = 2, CONFIDENTIAL = 3 };
|
|
struct Employee {
|
|
sqlgen::PrimaryKey<uint32_t, sqlgen::auto_incr> id;
|
|
std::string first_name;
|
|
std::string last_name;
|
|
AccessRestriction access_level;
|
|
};
|
|
|
|
struct Document {
|
|
sqlgen::PrimaryKey<uint32_t, sqlgen::auto_incr> id;
|
|
AccessRestriction min_access_level;
|
|
std::string name;
|
|
std::string path;
|
|
};
|
|
|
|
TEST(postgres, test_enum_cross_table) {
|
|
using namespace sqlgen;
|
|
using namespace sqlgen::literals;
|
|
|
|
auto employees = std::vector<Employee>({
|
|
Employee{.first_name = "Homer",
|
|
.last_name = "Simpson",
|
|
.access_level = AccessRestriction::PUBLIC},
|
|
Employee{.first_name = "Waylon",
|
|
.last_name = "Smithers",
|
|
.access_level = AccessRestriction::INTERNAL},
|
|
Employee{.first_name = "Montgomery",
|
|
.last_name = "Burns",
|
|
.access_level = AccessRestriction::CONFIDENTIAL},
|
|
});
|
|
auto documents = std::vector<Document>({
|
|
Document{.min_access_level = AccessRestriction::PUBLIC,
|
|
.name = "Power Plant Safety Manual",
|
|
.path = "/documents/powerplant/safety_manual.txt"},
|
|
Document{.min_access_level = AccessRestriction::INTERNAL,
|
|
.name = "Staff Memo",
|
|
.path = "/documents/powerplant/staff_memo.txt"},
|
|
Document{.min_access_level = AccessRestriction::CONFIDENTIAL,
|
|
.name = "Operations Report",
|
|
.path = "/documents/powerplant/operations_report.pdf"},
|
|
Document{.min_access_level = AccessRestriction::PUBLIC,
|
|
.name = "Project Plan",
|
|
.path = "/documents/powerplant/project_plan.md"},
|
|
Document{.min_access_level = AccessRestriction::INTERNAL,
|
|
.name = "Budget Q1",
|
|
.path = "/documents/powerplant/budget_q1.pdf"},
|
|
Document{.min_access_level = AccessRestriction::CONFIDENTIAL,
|
|
.name = "HR Policies",
|
|
.path = "/documents/powerplant/hr_policies.pdf"},
|
|
Document{.min_access_level = AccessRestriction::PUBLIC,
|
|
.name = "Team Photo",
|
|
.path = "/documents/powerplant/team.jpg"},
|
|
Document{.min_access_level = AccessRestriction::CONFIDENTIAL,
|
|
.name = "Executive Summary",
|
|
.path = "/documents/powerplant/executive_summary.docx"},
|
|
Document{.min_access_level = AccessRestriction::INTERNAL,
|
|
.name = "Release Notes",
|
|
.path = "/documents/powerplant/release_notes.txt"},
|
|
});
|
|
|
|
const auto credentials = sqlgen::postgres::Credentials{.user = "postgres",
|
|
.password = "password",
|
|
.host = "localhost",
|
|
.dbname = "postgres"};
|
|
const auto conn = postgres::connect(credentials);
|
|
conn.and_then(drop<Employee> | if_exists)
|
|
.and_then(drop<Document> | if_exists);
|
|
|
|
write(conn, employees);
|
|
write(conn, documents);
|
|
|
|
const auto smithers = conn.and_then(sqlgen::read<Employee> |
|
|
where("last_name"_c == "Smithers" and
|
|
"first_name"_c == "Waylon"))
|
|
.value();
|
|
// EXPECT_FALSE(smithers.empty());
|
|
|
|
const auto smithers_level = smithers.access_level;
|
|
const auto smithers_documents =
|
|
conn.and_then(sqlgen::read<std::vector<Document>> |
|
|
where("min_access_level"_c == smithers_level ||
|
|
"min_access_level"_c == AccessRestriction::PUBLIC) |
|
|
order_by("name"_c))
|
|
.value();
|
|
|
|
const auto expected_ids = std::set<uint32_t>{1, 2, 4, 5, 7, 9};
|
|
std::set<uint32_t> actual_ids;
|
|
for (const auto &d : smithers_documents) {
|
|
actual_ids.emplace(d.id());
|
|
}
|
|
|
|
EXPECT_EQ(expected_ids, actual_ids);
|
|
}
|
|
|
|
} // namespace test_enum_cross_table
|
|
|
|
#endif
|