如何为单元测试创建PostgreSQL临时用户与数据库？

我来帮你梳理一套能解决问题且可移植的PostgreSQL单元测试自动化方案，刚好我之前做过类似的C++数据库测试流程搭建，咱们一步步来：

1. 无交互创建带密码的临时用户 #

你遇到的createuser参数无效、psql连接错误，本质是没利用好PostgreSQL本地认证机制：

• Ubuntu 16.04上的PostgreSQL默认给postgres超级用户开启了peer认证，本地用postgres用户执行psql无需密码，也不会有交互。
• 之前的psql: FATAL: database 'user' does not exist错误，是因为你直接用自己的Unix用户执行psql时，它默认会尝试连接和你用户名同名的数据库（这个数据库不存在），而postgres用户默认连接的是系统自带的postgres数据库（肯定存在）。

所以正确的无交互创建带密码用户的命令是：

sudo -u postgres psql -c "CREATE ROLE test_temp_user WITH LOGIN PASSWORD 'temp_pass_123';"

同理，后续创建/删除数据库、删除用户，都用sudo -u postgres前缀执行即可。

2. Google Test夹具实现完整测试流程 #

咱们可以用Google Test的测试套件级夹具（SetUpTestSuite/TearDownTestSuite）来实现全局的用户/数据库创建销毁，再用测试用例级的SetUp/TearDown处理数据库连接，这样既高效又能保证测试隔离。

示例代码片段 #

#include <gtest/gtest.h>
#include <pqxx/pqxx>
#include <cstdlib>
#include <memory>
#include <string>

// 定义测试夹具，整个测试套件共用一套临时用户和数据库
class DBIntegrationTest : public ::testing::Test {
protected:
    // 测试套件启动时执行：创建用户+数据库
    static void SetUpTestSuite() {
        // 创建临时用户
        int ret = system("sudo -u postgres psql -c \"CREATE ROLE test_temp_user WITH LOGIN PASSWORD 'temp_pass_123';\"");
        ASSERT_EQ(ret, 0) << "Failed to create temporary database user";

        // 创建测试数据库，并指定所有者为临时用户
        ret = system("sudo -u postgres createdb -O test_temp_user test_temp_db");
        ASSERT_EQ(ret, 0) << "Failed to create temporary database";
    }

    // 测试套件结束时执行：删除数据库+用户
    static void TearDownTestSuite() {
        // 先删数据库（用户有依赖时必须先删库）
        system("sudo -u postgres dropdb test_temp_db");
        // 再删用户
        system("sudo -u postgres psql -c \"DROP ROLE test_temp_user;\"");
    }

    // 每个测试用例启动时：建立数据库连接
    void SetUp() override {
        try {
            conn_ = std::make_unique<pqxx::connection>(
                "dbname=test_temp_db user=test_temp_user password=temp_pass_123 host=localhost"
            );
            ASSERT_TRUE(conn_->is_open()) << "Failed to connect to test database";
        } catch (const std::exception& e) {
            FAIL() << "Database connection error: " << e.what();
        }
    }

    // 每个测试用例结束时：关闭连接
    void TearDown() override {
        if (conn_ && conn_->is_open()) {
            conn_->close();
        }
    }

    std::unique_ptr<pqxx::connection> conn_;
};

// 编写测试用例示例
TEST_F(DBIntegrationTest, TestBasicQuery) {
    // 执行测试逻辑，比如创建表、插入数据、验证结果
    pqxx::work txn(*conn_);
    txn.exec("CREATE TABLE test_table (id INT PRIMARY KEY, name VARCHAR(255));");
    txn.exec("INSERT INTO test_table VALUES (1, 'test_data');");
    txn.commit();

    pqxx::result res = txn.exec("SELECT name FROM test_table WHERE id = 1;");
    ASSERT_EQ(res.size(), 1);
    ASSERT_EQ(res[0][0].as<std::string>(), "test_data");
}

3. 可移植性与优化建议 #

• 避免硬编码密码：可以把临时用户密码放到环境变量里（比如TEST_DB_PASSWORD），代码里读取环境变量拼接连接字符串，这样不同测试环境可以灵活配置。
• 替代system()调用：如果想更健壮地捕获命令执行的输出和错误，可以用C++的进程库（比如boost::process）或者封装自定义的进程执行函数，比system()更容易调试。
• 适配不同Linux发行版：大部分发行版的PostgreSQL超级用户都是postgres，如果遇到特殊情况，可以通过环境变量指定超级用户名称。
• 权限简化（可选）：如果不想每次测试都输入sudo密码，可以修改pg_hba.conf，允许你的Unix用户以trust方式连接postgres数据库（仅测试环境使用，生产环境绝对不能这么做）：
  编辑/etc/postgresql/9.5/main/pg_hba.conf（Ubuntu 16.04对应路径），添加一行：

  local   postgres   your_unix_username   trust
  

  然后重启PostgreSQL服务：sudo systemctl restart postgresql，之后就可以不用sudo直接执行psql命令了。

内容的提问来源于stack exchange，提问作者The Quantum Physicist