测试时Mock PostgreSQL now()函数的解决方案咨询

这确实是个棘手的问题——Sinon能搞定Node/Express层的时间Mock，但PostgreSQL触发器里的now()属于数据库内部函数，完全不受Node层Mock的影响。好在我们可以在数据库层面做隔离，完全不需要改动业务代码，就能实现你要的时间快进功能。

核心思路：用自定义函数替换触发器中的now() #

我们可以创建一个自定义的时间获取函数，让触发器依赖这个函数而非直接调用now()。平时这个函数返回真实的当前时间，测试时我们可以临时替换它的行为，让它返回我们指定的时间，以此实现时间快进/回退。

步骤1：创建自定义时间函数 #

首先在数据库中创建一个默认返回now()的函数，这个函数会作为触发器的时间来源：

CREATE OR REPLACE FUNCTION get_current_timestamp()
RETURNS TIMESTAMPTZ AS $$
BEGIN
  -- 默认返回真实的当前时间，和业务逻辑完全兼容
  RETURN now();
END;
$$ LANGUAGE plpgsql STABLE;

标记为STABLE是为了保证在同一个事务中多次调用时返回相同值，和now()的行为保持一致。

步骤2：修改触发器函数使用自定义函数 #

更新你现有的触发器，把now()替换成我们刚创建的get_current_timestamp()：

CREATE OR REPLACE FUNCTION update_modified_column()
RETURNS TRIGGER AS $$
BEGIN
  -- 用自定义函数代替直接调用now()
  NEW.modified = get_current_timestamp();
  RETURN NEW;
END;
$$ LANGUAGE plpgsql;

这一步只是修改数据库中的触发器逻辑，完全不需要改动任何业务代码，业务服务依然像之前一样正常工作。

步骤3：测试时临时Mock时间 #

在测试会话中，我们可以通过会话级的设置来覆盖get_current_timestamp()的行为，实现时间控制：

1. 首先在测试数据库中创建一个用于存储Mock时间的配置项（只需执行一次）：

ALTER SYSTEM SET test.mock_timestamp TO '1970-01-01 00:00:00';
SELECT pg_reload_conf();

2. 在测试流程中，通过SET LOCAL设置当前会话的Mock时间：

-- 设置初始时间
SET LOCAL test.mock_timestamp = '2024-05-20 10:00:00';

-- 临时替换自定义函数，返回我们设置的Mock时间
CREATE OR REPLACE FUNCTION get_current_timestamp()
RETURNS TIMESTAMPTZ AS $$
BEGIN
  RETURN current_setting('test.mock_timestamp')::TIMESTAMPTZ;
END;
$$ LANGUAGE plpgsql STABLE;

测试流程示例（时间快进场景） #

以你提到的“修改内容→快进一天→再次修改→验证结果”为例，测试步骤如下：

1. 启动一个数据库事务（测试结束后回滚，避免影响其他测试）
2. 设置初始Mock时间并替换时间函数（如上一步）
3. 调用Web服务修改数据库内容，此时modified字段会被设为2024-05-20 10:00:00
4. 快进一天，更新Mock时间：

SET LOCAL test.mock_timestamp = '2024-05-21 10:00:00';

5. 再次调用Web服务修改数据库内容，此时modified字段会被设为2024-05-21 10:00:00
6. 调用Web服务查询“最近X秒内修改的记录”，验证结果符合预期
7. 回滚事务，数据库会自动恢复原来的get_current_timestamp()函数和时间设置

方案优势 #

• 完全不侵入业务代码：所有修改都在数据库层面，业务服务无需感知测试逻辑
• 灵活的时间控制：支持任意时间快进、回退，满足各种时间依赖性测试场景
• 会话级隔离：SET LOCAL只会影响当前测试会话，不会干扰其他并行测试
• 业务兼容性：生产环境下函数默认返回真实时间，和原有逻辑完全一致

备选方案（不推荐，仅作参考） #

如果暂时不想修改触发器，也可以在测试时手动更新modified字段，但这需要在测试代码中额外处理数据，不如上面的方案优雅：

-- 修改记录后手动更新modified字段到指定时间
UPDATE your_table SET modified = '2024-05-21 10:00:00' WHERE id = 123;

但这种方法需要在测试中硬编码数据操作，容易遗漏，且无法完全模拟触发器的自动行为。

内容的提问来源于stack exchange，提问作者Joris