diff options
Diffstat (limited to 'libpqpp/pq-mock.cpp')
| -rw-r--r-- | libpqpp/pq-mock.cpp | 170 |
1 files changed, 85 insertions, 85 deletions
diff --git a/libpqpp/pq-mock.cpp b/libpqpp/pq-mock.cpp index 018dad4..34d903e 100644 --- a/libpqpp/pq-mock.cpp +++ b/libpqpp/pq-mock.cpp @@ -1,46 +1,46 @@ #include "pq-mock.h" #include "pq-connection.h" +#include <boost/algorithm/string.hpp> #include <compileTimeFormatter.h> #include <modifycommand.h> #include <selectcommand.h> #include <selectcommandUtil.impl.h> -#include <boost/algorithm/string.hpp> NAMEDFACTORY("postgresql", PQ::Mock, DB::MockDatabaseFactory); namespace PQ { -Mock::Mock(const std::string & masterdb, const std::string & name, const std::vector<std::filesystem::path> & ss) : - MockServerDatabase(masterdb, name, "postgresql"), - tablespacePath(std::filesystem::temp_directory_path() / testDbName), - serverVersion(std::static_pointer_cast<Connection>(master)->serverVersion()) -{ - try { - CreateNewDatabase(); - PlaySchemaScripts(ss); - SetTablesToUnlogged(); - } - catch (...) { - DropDatabase(); - throw; + Mock::Mock(const std::string & masterdb, const std::string & name, const std::vector<std::filesystem::path> & ss) : + MockServerDatabase(masterdb, name, "postgresql"), + tablespacePath(std::filesystem::temp_directory_path() / testDbName), + serverVersion(std::static_pointer_cast<Connection>(master)->serverVersion()) + { + try { + CreateNewDatabase(); + PlaySchemaScripts(ss); + SetTablesToUnlogged(); + } + catch (...) { + DropDatabase(); + throw; + } } -} -AdHocFormatter(MockConnStr, "user=postgres dbname=%?"); -DB::ConnectionPtr -Mock::openConnection() const -{ - return std::make_shared<Connection>(MockConnStr::get(boost::algorithm::to_lower_copy(testDbName))); -} - -AdHocFormatter(MockSetUnlogged, "ALTER TABLE %?.%? SET UNLOGGED"); -void -Mock::SetTablesToUnlogged() const -{ - if (!hasUnloggedTables()) { - return; + AdHocFormatter(MockConnStr, "user=postgres dbname=%?"); + DB::ConnectionPtr + Mock::openConnection() const + { + return std::make_shared<Connection>(MockConnStr::get(boost::algorithm::to_lower_copy(testDbName))); } - auto s = master->select(R"SQL( + + AdHocFormatter(MockSetUnlogged, "ALTER TABLE %?.%? SET UNLOGGED"); + void + Mock::SetTablesToUnlogged() const + { + if (!hasUnloggedTables()) { + return; + } + auto s = master->select(R"SQL( SELECT n.nspname, c.relname FROM pg_class c, pg_namespace n WHERE c.relkind = 'r' @@ -55,69 +55,69 @@ AND NOT EXISTS ( AND ck.relpersistence = 'p') AND n.oid = c.relnamespace ORDER BY 1, 2)SQL"); - s->bindParamS(0, "pg_catalog"); - s->bindParamS(1, "information_schema"); - unsigned int n = 0; - do { - n = 0; - for (const auto [ nspname, relname ] : s->as<std::string, std::string>()) { - master->execute(MockSetUnlogged::get(nspname, relname)); - n += 1; - } - } while(n); -} + s->bindParamS(0, "pg_catalog"); + s->bindParamS(1, "information_schema"); + unsigned int n = 0; + do { + n = 0; + for (const auto [nspname, relname] : s->as<std::string, std::string>()) { + master->execute(MockSetUnlogged::get(nspname, relname)); + n += 1; + } + } while (n); + } -Mock::~Mock() -{ - Mock::DropDatabase(); -} + Mock::~Mock() + { + Mock::DropDatabase(); + } -bool -Mock::hasUnloggedTables() const -{ - // v9.5 server required for unlogged tables - return (serverVersion >= 90500); -} + bool + Mock::hasUnloggedTables() const + { + // v9.5 server required for unlogged tables + return (serverVersion >= 90500); + } -bool -Mock::hasCopyToProgram() const -{ - // v9.3 server required to use COPY ... TO PROGRAM ... - return (serverVersion >= 90300); -} + bool + Mock::hasCopyToProgram() const + { + // v9.3 server required to use COPY ... TO PROGRAM ... + return (serverVersion >= 90300); + } -AdHocFormatter(MockCreateTablespaceDir, "COPY (SELECT '%?') TO PROGRAM 'xargs mkdir -p'"); -AdHocFormatter(MockCreateTablespace, "CREATE TABLESPACE %? LOCATION '%?'"); -AdHocFormatter(MockCreateDatabase, "CREATE DATABASE %? TABLESPACE %?"); -AdHocFormatter(MockDropTablespace, "DROP TABLESPACE IF EXISTS %?"); -AdHocFormatter(MockDropTablespaceDir, "COPY (SELECT '%?') TO PROGRAM 'xargs rm -rf'"); + AdHocFormatter(MockCreateTablespaceDir, "COPY (SELECT '%?') TO PROGRAM 'xargs mkdir -p'"); + AdHocFormatter(MockCreateTablespace, "CREATE TABLESPACE %? LOCATION '%?'"); + AdHocFormatter(MockCreateDatabase, "CREATE DATABASE %? TABLESPACE %?"); + AdHocFormatter(MockDropTablespace, "DROP TABLESPACE IF EXISTS %?"); + AdHocFormatter(MockDropTablespaceDir, "COPY (SELECT '%?') TO PROGRAM 'xargs rm -rf'"); -void -Mock::CreateNewDatabase() const -{ - if (hasCopyToProgram()) { - DropDatabase(); - master->execute(MockCreateTablespaceDir::get(tablespacePath)); - master->execute(MockCreateTablespace::get(testDbName, tablespacePath.string())); - master->execute(MockCreateDatabase::get(testDbName, testDbName)); - } - else { - MockServerDatabase::CreateNewDatabase(); + void + Mock::CreateNewDatabase() const + { + if (hasCopyToProgram()) { + DropDatabase(); + master->execute(MockCreateTablespaceDir::get(tablespacePath)); + master->execute(MockCreateTablespace::get(testDbName, tablespacePath.string())); + master->execute(MockCreateDatabase::get(testDbName, testDbName)); + } + else { + MockServerDatabase::CreateNewDatabase(); + } } -} -void -Mock::DropDatabase() const -{ - auto t = master->modify("SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE LOWER(datname) = LOWER(?)"); - t->bindParamS(0, testDbName); - t->execute(); - MockServerDatabase::DropDatabase(); - if (hasCopyToProgram()) { - master->execute(MockDropTablespace::get(testDbName)); - master->execute(MockDropTablespaceDir::get(tablespacePath)); + void + Mock::DropDatabase() const + { + auto t = master->modify( + "SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE LOWER(datname) = LOWER(?)"); + t->bindParamS(0, testDbName); + t->execute(); + MockServerDatabase::DropDatabase(); + if (hasCopyToProgram()) { + master->execute(MockDropTablespace::get(testDbName)); + master->execute(MockDropTablespaceDir::get(tablespacePath)); + } } -} } - |