数据库单元测试：如何在TSQL存储过程中模拟(mock)时间

方案1：会话级上下文变量传递自定义时间（最优推荐） #

该方案仅需对存储过程做一次无侵入性修改，完全不影响上游调用逻辑，也不需要额外新增入参：

• 操作步骤：
  1. 将存储过程内所有GETDATE()/GETUTCDATE()调用替换为如下逻辑：

  COALESCE(CAST(SESSION_CONTEXT(N'Test_CurrentDateTime') AS DATETIME2), GETDATE())
  

  2. 生产环境下未设置Test_CurrentDateTime上下文变量，COALESCE会自动回退到原生GETDATE()，上游调用无需做任何调整，无额外传参风险。
  3. Python单元测试时，建立数据库连接后先执行一次会话级配置：

  cursor.execute("EXEC sp_set_session_context @key=N'Test_CurrentDateTime', @value='2024-01-01 05:30:00'")
  

  后续同一会话内调用存储过程时，会自动使用指定的测试时间，测试结束断开连接即可自动销毁上下文变量，不会影响其他会话。
• 优势：无业务侵入、不需要修改上游调用、不依赖特定账号权限、性能损耗可忽略、存储过程可读性几乎不受影响。

方案2：封装时间获取逻辑为独立标量函数 #

如果不想在存储过程中添加上下文判断逻辑，可以用函数封装的方式实现时间的统一替换：

• 操作步骤：
  1. 新增一个独立的标量函数用于获取当前时间，生产环境下函数逻辑为返回原生时间：

  CREATE OR ALTER FUNCTION dbo.GetCurrentSysTime()
  RETURNS DATETIME2
  AS
  BEGIN
    RETURN GETDATE()
  END
  

  2. 将存储过程内所有GETDATE()替换为dbo.GetCurrentSysTime()，上游调用完全无感知。
  3. 单元测试时，在测试数据库中临时替换该函数的定义，固定返回需要的测试时间，测试完成后还原函数即可，存储过程本身始终和生产版本完全一致。
• 优势：存储过程逻辑更简洁，时间逻辑统一管理，后续修改时间规则不需要逐个调整存储过程。

方案3：零修改生产代码的临时替换方案 #

如果完全不允许修改现有存储过程的任何代码，可以用自动化临时替换的方式实现测试：

• 操作步骤：
  1. 单元测试执行前，通过pyodbc查询生产存储过程的原始定义代码：

  SELECT definition FROM sys.sql_modules WHERE object_id = OBJECT_ID(N'[你的存储过程名]')
  

  2. 用Python字符串替换功能，将代码中所有GETDATE()/GETUTCDATE()批量替换为需要的测试时间常量。
  3. 将替换后的存储过程部署到隔离的测试数据库中执行测试，测试完成后删除测试环境的存储过程即可。
• 优势：完全不需要修改生产环境的任何代码，不存在版本不一致的问题，每次测试都是基于最新的生产存储过程代码做的临时替换。

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