python-3.x - 在模块导入期间调用的 python 模拟函数调用

Question

我需要对在模块导入期间运行的 python 代码执行模拟

例如我有这样的代码

import configparser

config = configparser.ConfigParser()
config.read('test.ini')
a = float(config['config']['a'])
b = float(config['config']['b'])
c = float(config['config']['c'])

print(a)
print(b)
print(c)

我需要模拟“配置”进行测试

import pytest
import mock

import app

@mock.patch('app.configparser.ConfigParser')
def test_config_mock(config_mock):
     config_mock.return_value = {'config': { 'a' : 1 } }

然而，这个测试函数在实际导入后被调用，所以我的 mock 没有任何意义

做这种事情的正确方法是什么？

Answer 1

在这种情况下，您可以做的是修补 config使用 mock.patch.dict 的实例:

# test_coolio.py
import mock
from app.fun import coolio


@mock.patch.dict('app.configparser.config', values={'config': {'a': 15}})
def test_config_mock():
    assert coolio() == '15'


# app/fun.py
from app.configparser import config


def coolio():
    return config['config']['a']