无需安装unixODBC，如何Mock pyodbc解决测试导入报错问题？

完全可以在导入主文件前Mock pyodbc的导入，无需在构建服务器安装unixODBC。核心思路是利用Python的模块导入机制，在main.py真正导入pyodbc之前，用Mock对象替换掉系统中的pyodbc模块引用。

方法一：使用unittest.mock.patch（推荐） #

在test_main.py开头先执行Mock操作，再导入main模块：

from unittest.mock import patch, Mock

# 提前Mock pyodbc模块，替换掉真实的pyodbc
patch('pyodbc').start()

# 此时导入main.py不会触发真实pyodbc的导入
from main import function_to_test

def test_function_to_test():
    # 进一步Mock pyodbc的具体方法（比如connect）
    with patch('main.pyodbc.connect') as mock_connect:
        # 配置Mock的返回值和行为
        mock_cursor = Mock()
        mock_connect.return_value.cursor.return_value = mock_cursor
        mock_cursor.fetchall.return_value = [('sample_data',)]
        
        # 执行测试
        result = function_to_test()
        
        # 验证调用逻辑
        mock_connect.assert_called_once_with('你的数据库连接字符串')
        assert result == 预期结果

方法二：直接修改sys.modules #

手动创建Mock模块并注入到系统模块列表中：

import sys
from unittest.mock import Mock

# 创建Mock版的pyodbc模块
mock_pyodbc = Mock()
# 可以提前给Mock模块添加必要的属性，比如connect方法
mock_pyodbc.connect = Mock()

# 将Mock模块注入到sys.modules，后续import pyodbc都会拿到这个Mock
sys.modules['pyodbc'] = mock_pyodbc

# 安全导入main模块
from main import function_to_test

原理说明 #

Python在导入模块时会先检查sys.modules中是否已存在该模块的引用。通过上述两种方式，我们在main.py执行import pyodbc之前，就已经把sys.modules['pyodbc']替换成了Mock对象，从而完全绕过了对真实pyodbc库及其底层unixODBC依赖的加载。

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