如何在测试中Mock函数文件里导入的Config依赖？

我在测试安全模块中的get_user_profile_details函数时遇到了问题：导入该函数时会加载其依赖的config，而config初始化时创建ElasticsearchConfig实例会触发403权限错误，导致测试在执行函数前就失败。

相关代码与错误信息 #

安全模块代码：

import aiohttp
from fastapi import Header, HTTPException
from .util.config import config 

async def get_user_profile_details(
        user_profile_id: str, ..., ...
):
    user_profile_url = f"{config.entity.ENTITY_BASE_URL}/.../..."
    async with aiohttp.ClientSession() as session: 
      async with session.get(user_profile_url, headers=auth_header) as ...
    ....

测试代码（尝试过的写法）：

import unittest
from unittest.mock import patch
from aioresponses import aioresponses
from my_module.security import get_user_profile_details

    @patch('my_module.util.config.es', autospec=True)
    async def test_get_user_profile_details_success(self, mock_config):
        mock_config.entity.ENTITY_BASE_URL = "http://mocked-entity-url"

        user_profile_id = "123"
        ... 
        ...

错误信息：

es = ElasticsearchConfig()

E   google.api_core.exceptions.PermissionDenied: 403 Permission denied on resource project test.

Config定义：

class Config(BaseSettings):
   ... 
   ...   
   es = ElasticsearchConfig() # 需要Mock的对象
   entity = EntityAPIConfig() 
... 
config: Config = Config()

我已经尝试了这些方式但都无效：

• Mock my_module.util.config.es
• Mock my_module.util.config.config
• Mock my_module.util.Config
• 在asyncSetUp中初始化aioresponses并Mock请求

请问正确的Mock方式是什么？

正确的Mock方式 #

核心问题是导入my_module.security时，config已经被实例化完成，导致测试函数内的Patch生效太晚。必须在导入任何依赖config的模块之前，就阻止ElasticsearchConfig的真实初始化。

方法一：提前Patch，再导入被测试模块 #

在测试文件最开头先Patch掉ElasticsearchConfig，避免其真实初始化触发权限错误，之后再导入被测试的函数：

import unittest
from unittest.mock import patch, MagicMock
from aioresponses import aioresponses

# 第一步：先Mock掉ElasticsearchConfig，阻止真实初始化
patch('my_module.util.config.ElasticsearchConfig', autospec=True).start()

# 第二步：再导入被测试的模块
from my_module.security import get_user_profile_details

class TestSecurity(unittest.IsolatedAsyncioTestCase):
    def tearDown(self):
        # 测试结束后停止所有Patch，避免影响其他测试用例
        patch.stopall()

    async def test_get_user_profile_details_success(self):
        # Mock config中的entity属性，替换为测试需要的URL
        with patch('my_module.util.config.config') as mock_config:
            mock_config.entity.ENTITY_BASE_URL = "http://mocked-entity-url"
            
            user_profile_id = "123"
            # 使用aioresponses Mock接口请求
            with aioresponses() as m:
                # 构造Mock的请求URL并设置响应
                mocked_url = f"{mock_config.entity.ENTITY_BASE_URL}/.../..."
                m.get(mocked_url, status=200, json={"id": user_profile_id})
                
                # 调用被测试函数
                result = await get_user_profile_details(user_profile_id, ...)
                
                # 执行断言逻辑
                self.assertIsNotNone(result)
                m.assert_called_once_with(mocked_url, headers=...)

方法二：类级别Patch，确保导入前生效 #

如果习惯用装饰器，也可以在测试类上使用@patch，并确保在导入被测试模块前应用：

import unittest
from unittest.mock import patch
from aioresponses import aioresponses

# 提前Patch ElasticsearchConfig
@patch('my_module.util.config.ElasticsearchConfig', autospec=True)
class TestSecurity(unittest.IsolatedAsyncioTestCase):
    def setUp(self):
        # 导入被测试模块（此时ElasticsearchConfig已经被Mock）
        global get_user_profile_details
        from my_module.security import get_user_profile_details

    async def test_get_user_profile_details_success(self, mock_es_config):
        with patch('my_module.util.config.config') as mock_config:
            mock_config.entity.ENTITY_BASE_URL = "http://mocked-entity-url"
            
            user_profile_id = "123"
            with aioresponses() as m:
                mocked_url = f"{mock_config.entity.ENTITY_BASE_URL}/.../..."
                m.get(mocked_url, status=200, json={"id": user_profile_id})
                
                result = await get_user_profile_details(user_profile_id, ...)
                
                self.assertIsNotNone(result)
                m.assert_called_once()

关键说明 #

• 所有Patch必须在导入my_module.security之前完成，因为一旦导入该模块，from .util.config import config会立即执行config = Config()，触发ElasticsearchConfig的真实初始化。
• 如果需要Mockconfig的其他属性（比如entity.ENTITY_BASE_URL），可以在测试函数内再Patch整个config对象，覆盖需要的属性值。

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