Python测试中如何Mock被调用函数以跳过其原有执行逻辑？

你遇到的「Mock之后被替换函数的原执行逻辑仍在运行」，不是mock功能本身的限制，是使用方式存在偏差。
按照mock.patch的原生设计：只要替换逻辑生效，被patch的对象会从上下文进入块的瞬间被完全替换成无实体的Mock实例，原函数的所有代码根本不会被触发执行。
可以用一段验证代码直观证明：

def f1():
    # 在f1第一行加打印语句，只要被执行就会输出控制台
    print("==== 我是原f1的执行逻辑 ====")
    a = 'do not make this when f1() is mocked'
    return 10, True

def f2():
    num, stat = f1()
    return 2*num, stat

from unittest import mock

# 上下文外调用：会打印原f1的语句，输出(20, True)
print(f2())

with mock.patch('__main__.f1') as MockClass:
    MockClass.return_value = (30, True)
    # 这里调用f2，控制台根本不会打印原f1的字符串
    # 因为此时__main__命名空间下的f1已经被Mock对象完全顶替
    print(f2())

为什么会出现「原逻辑照常运行」的错觉？ #

大概率是踩到了两个常见坑：

• patch锚点写错：如果f2是从其他模块导入的f1（比如f2的文件里写了from utils import f1），此时如果去patch utils.f1，是无法替换f2内部实际持有的f1引用的。f2拿着的还是原始的f1对象，自然会照常执行原逻辑。正确的patch锚点应该写被调用方所在的命名空间位置，也就是你当前模块下的f1引用路径。
• 误用了wraps参数：如果给patch加了wraps=原函数参数，这个参数的作用就是让Mock对象收到调用时，自动转发执行原函数的全部逻辑。如果你是想完全屏蔽原函数，去掉这个参数即可。

测试单个函数时，完全隔离其内部依赖的其他函数、屏蔽被依赖函数的内部执行逻辑，是单元测试的核心原则——我们测f2的时候，只需要验证f2拿到f1的返回值后的处理逻辑是否正确，完全不需要关心f1内部的数据库查询、接口请求等逻辑有没有正常运行。

可行的实现方案 #

• 方案1：正确配置的标准mock.patch
  只要选对patch锚点、去掉不必要的wraps参数，mock.patch天然就能实现完全屏蔽原函数的效果，也是最常用的方案。你可以做个快速验证：在原f1的第一行加一句raise RuntimeError("原f1被执行了")，如果mock生效，调用f2时根本不会触发这个异常——因为f1压根没被运行。
• 方案2：手动临时替换函数
  如果不想依赖mock库，也可以直接做手动的临时重载，逻辑更直白：

  # 先存好原函数的引用，方便测试后还原
  original_f1 = f1
  
  # 完全自定义重载的f1，不会执行原函数的任何一行代码
  def mocked_f1():
      return (30, True)
  
  # 临时替换成mock版本
  f1 = mocked_f1
  try:
      print(f2()) # 输出(60, True)
  finally:
      # 测试结束必须还原原函数，避免污染其他测试用例
      f1 = original_f1
  

• 方案3：精细化控制Mock行为
  如果需要模拟更复杂的场景，比如根据入参动态返回不同值、统计函数调用次数、验证传入参数是否正确，可以给Mock实例配置side_effect规则，完全替代原函数的行为：

  with mock.patch('__main__.f1') as mock_f1:
      # 自定义动态返回规则，完全不触发原函数
      mock_f1.side_effect = lambda req_val: (req_val*2, req_val>10) if req_val <100 else (0, False)
      # 测试时可以直接断言f1的调用是否符合预期
      mock_f1.assert_called_once_with(5)
  

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