测试类中mock builtins.open导致pytz调用异常的问题排查

方案一：精准Mock被测模块内的open（推荐） #

不要直接mock全局的builtins.open，而是mock被测类所在模块的open。Python中每个模块维护独立的命名空间，模块内的open是对builtins.open的本地引用，mock这个本地引用只会影响当前模块的代码，不会干扰pytz等其他模块的open调用。

假设被测代码在my_module.py中：

# my_module.py
import pytz

class Foo:
    def read_file(self, path):
        with open(path, 'r') as f:
            return f.read()
    
    def get_timezone(self):
        return pytz.timezone('GMT')

对应的测试代码写法：

# test_my_module.py
from unittest.mock import mock_open, patch
from my_module import Foo

def test_foo_read_and_timezone():
    # 仅mock my_module模块内的open
    with patch('my_module.open', mock_open(read_data='测试内容')):
        foo = Foo()
        # 被测类的read_file会使用mock的open
        result = foo.read_file('任意路径')
        assert result == '测试内容'
        # pytz内部的open调用不受影响，正常获取时区
        gmt_zone = foo.get_timezone()
        assert gmt_zone.zone == 'GMT'

方案二：限制Mock仅作用于指定次数的调用 #

如果必须mock全局builtins.open，可以通过side_effect控制调用行为：第一次调用返回mock对象，后续调用使用真实的open。这种方式依赖调用顺序，仅适用于确定调用顺序的场景。

示例测试代码：

# test_my_module.py
from unittest.mock import mock_open, patch
import builtins
from my_module import Foo

def test_foo_limited_mock():
    original_open = builtins.open
    mock_file = mock_open(read_data='测试内容')
    call_count = 0

    def controlled_open(*args, **kwargs):
        nonlocal call_count
        if call_count == 0:
            call_count += 1
            return mock_file(*args, **kwargs)
        return original_open(*args, **kwargs)

    with patch('builtins.open', side_effect=controlled_open):
        foo = Foo()
        # 第一次open调用（Foo.read_file）用mock
        result = foo.read_file('任意路径')
        assert result == '测试内容'
        # 后续pytz的open调用使用真实实现，不会报错
        gmt_zone = foo.get_timezone()
        assert gmt_zone.zone == 'GMT'

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