eelke/pgLab
1
0
Fork 0
Code Issues 29 Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add tests for loading the catalog

Browse source
This commit is contained in:
eelke 2026-10-18 12:32:16 +02:00
parent 591573b6b8
commit a1ac545086
13 changed files with 203 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/integration/CMakeLists.txt
View file

@ -1,10 +1,13 @@
add_executable(IntegrationTests
main.cpp
tst_Smoke.cpp
tst_CatalogLoad.cpp
tst_DatabaseSizes.cpp
)
target_link_libraries(IntegrationTests PRIVATE
pgintegration
pglablib
GTest::gtest
Qt6::Core
)

36
tests/integration/tst_CatalogLoad.cpp Normal file
View file

@ -0,0 +1,36 @@
#include "DbTest.h"
#include "PgTestEnvironment.h"
#include "catalog/PgClassContainer.h"
#include "catalog/PgDatabaseCatalog.h"
#include "Pgsql_Result.h"
using namespace pgtest;
class CatalogLoad : public DbTest {};
TEST_F(CatalogLoad, SuperuserLoadsCatalog)
{
PgDatabaseCatalog catalog;
catalog.loadAll(conn(), {});
EXPECT_EQ(catalog.serverVersion(), serverVersionNum());
EXPECT_EQ(catalog.getDBName(), dbName());
ASSERT_NE(catalog.classes(), nullptr);
EXPECT_NE(catalog.classes()->getByObjectNsAndName("public", "fixture_item"), nullptr);
}
TEST_F(CatalogLoad, PlainRoleLoadsCatalog)
{
auto plain = plainConn();
Pgsql::Result r = plain->query("SELECT rolsuper FROM pg_roles WHERE rolname = current_user");
ASSERT_EQ(r.rows(), 1);
EXPECT_EQ(QString::fromUtf8(r.val(0, 0)), QString("f"));
PgDatabaseCatalog catalog;
catalog.loadAll(*plain, {});
EXPECT_EQ(catalog.serverVersion(), serverVersionNum());
EXPECT_EQ(catalog.getDBName(), dbName());
ASSERT_NE(catalog.classes(), nullptr);
EXPECT_NE(catalog.classes()->getByObjectNsAndName("public", "fixture_item"), nullptr);
}

68
tests/integration/tst_DatabaseSizes.cpp Normal file
View file

@ -0,0 +1,68 @@
#include "DbTest.h"
#include "PgTestEnvironment.h"
#include "catalog/PgDatabaseSizes.h"
#include "Pgsql_Result.h"
using namespace pgtest;
namespace {
Oid oidOfDatabase(Pgsql::Connection &c, const QString &db)
{
Pgsql::Result r = c.query(QString("SELECT oid FROM pg_database WHERE datname = '%1'").arg(db));
if (!r || r.rows() != 1)
return InvalidOid;
return QString::fromUtf8(r.val(0, 0)).toUInt();
}
const PgDatabaseSize *findSize(const PgDatabaseSizes &sizes, Oid oid)
{
for (const auto &s : sizes) {
if (s.oid == oid)
return &s;
}
return nullptr;
}
} // namespace
class DatabaseSizes : public DbTest {};
TEST_F(DatabaseSizes, ConnectableDatabaseReturnsSize)
{
auto plain = plainConn();
Oid oid = oidOfDatabase(*plain, dbName());
ASSERT_NE(oid, InvalidOid);
PgDatabaseSizes sizes = loadDatabaseSizes(*plain);
const PgDatabaseSize *s = findSize(sizes, oid);
ASSERT_NE(s, nullptr);
EXPECT_GE(s->totalBytes, 0);
}
TEST_F(DatabaseSizes, RevokedConnectReturnsMinusOne)
{
Oid oid = oidOfDatabase(conn(), dbName());
ASSERT_NE(oid, InvalidOid);
conn().query(QString("REVOKE CONNECT ON DATABASE %1 FROM PUBLIC").arg(quoteIdent(dbName())));
auto plain = std::make_unique<Pgsql::Connection>();
plain->connect(urlForDbAsPlainRole("postgres"));
PgDatabaseSizes sizes = loadDatabaseSizes(*plain);
const PgDatabaseSize *s = findSize(sizes, oid);
ASSERT_NE(s, nullptr);
EXPECT_EQ(s->totalBytes, -1);
}
TEST_F(DatabaseSizes, SuperuserSeesAllDatabases)
{
PgDatabaseSizes sizes = loadDatabaseSizes(conn());
Pgsql::Result r = conn().query("SELECT count(*) FROM pg_database");
ASSERT_EQ(r.rows(), 1);
EXPECT_EQ(static_cast<size_t>(QString::fromUtf8(r.val(0, 0)).toULongLong()), sizes.size());
for (const auto &s : sizes)
EXPECT_GE(s.totalBytes, 0) << "database oid " << s.oid;
}

7
tests/pgintegration/DbTest.cpp
View file

@ -15,6 +15,13 @@ QString DbTest::dbUrl() const
return urlForDb(dbName_);
}
std::unique_ptr<Pgsql::Connection> DbTest::plainConn() const
{
auto c = std::make_unique<Pgsql::Connection>();
c->connect(urlForDbAsPlainRole(dbName_));
return c;
}
void DbTest::SetUp()
{
if (!available())

2
tests/pgintegration/DbTest.h
View file

@ -17,6 +17,8 @@ protected:
Pgsql::Connection& conn() { return *conn_; }
QString dbName() const { return dbName_; }
QString dbUrl() const;
/// Opens a new connection to this test's database as the plain (non-superuser) role.
std::unique_ptr<Pgsql::Connection> plainConn() const;
private:
QString dbName_;

26
tests/pgintegration/PgTestEnvironment.cpp
View file

@ -16,6 +16,8 @@ namespace {
bool g_available = false;
int g_versionNum = 0;
const char *kPlainRolePassword = "pglab_plain_test";
} // namespace
std::optional<QString> baseUrl()
@ -56,6 +58,23 @@ QString templateDbName()
return QString("tmpl_%1").arg(processId());
}
QString plainRoleName()
{
return QString("tst_%1").arg(processId());
}
QString urlForDbAsPlainRole(const QString &db)
{
auto base = baseUrl();
if (!base)
throw std::runtime_error("TEST_DATABASE_URL is not set");
QUrl url(*base);
url.setPath("/" + db);
url.setUserName(plainRoleName());
url.setPassword(QString::fromUtf8(kPlainRolePassword));
return url.toString();
}
QString quoteIdent(const QString &name)
{
QString r = name;
@ -82,6 +101,9 @@ void PgEnv::SetUp()
g_versionNum = QString::fromUtf8(r.val(0, 0)).toInt();
admin.query(QString("DROP DATABASE IF EXISTS %1 WITH (FORCE)").arg(quoteIdent(tmpl)));
admin.query(QString("CREATE DATABASE %1").arg(quoteIdent(tmpl)));
admin.query(QString("DROP ROLE IF EXISTS %1").arg(quoteIdent(plainRoleName())));
admin.query(QString("CREATE ROLE %1 LOGIN NOSUPERUSER NOCREATEDB NOCREATEROLE PASSWORD '%2'")
.arg(quoteIdent(plainRoleName()), QString::fromUtf8(kPlainRolePassword)));
}
{
QFile f(fixtureDir_ + "/schema.sql");
@ -101,6 +123,7 @@ void PgEnv::SetUp()
Pgsql::Connection admin;
admin.connect(*base);
admin.query(QString("DROP DATABASE IF EXISTS %1 WITH (FORCE)").arg(quoteIdent(tmpl)));
admin.query(QString("DROP ROLE IF EXISTS %1").arg(quoteIdent(plainRoleName())));
}
catch (const std::exception &) {
}
@ -115,9 +138,10 @@ void PgEnv::TearDown()
Pgsql::Connection admin;
admin.connect(*baseUrl());
admin.query(QString("DROP DATABASE IF EXISTS %1 WITH (FORCE)").arg(quoteIdent(templateDbName())));
admin.query(QString("DROP ROLE IF EXISTS %1").arg(quoteIdent(plainRoleName())));
}
catch (const std::exception &e) {
ADD_FAILURE() << "Dropping template database failed: " << e.what();
ADD_FAILURE() << "Dropping template database or plain role failed: " << e.what();
}
g_available = false;
}

6
tests/pgintegration/PgTestEnvironment.h
View file

@ -24,6 +24,12 @@ qint64 processId();
/// Name of the template database for this process.
QString templateDbName();
/// Name of the login role without superuser rights, unique per process.
QString plainRoleName();
/// Same as urlForDb() but with the plain role as user.
QString urlForDbAsPlainRole(const QString &db);
/// Quotes an identifier for use in SQL.
QString quoteIdent(const QString &name);