Python单元测试：从另一个模块模拟有问题的模块/功能

问题描述

B/moduleB.py@H_404_2@定义为：

def text_function():
    raise KeyError

text_function()
@H_404_2@

ModuleA.py@H_404_2@定义为：

from B.moduleB import text_function

class a():
    def __init__(self):
        self.text = 'abc'

    def mul(self,a,b):
        print(text_function() + str(a*b))

if __name__ == "__main__":
    rom = a()
    rom.mul(2,3)
@H_404_2@

这是单元测试test_module.py@H_404_2@：

from unittest.mock import Magicmock,patch
from moduleA import a

class TestmoduleA(unittest.TestCase):
    def setUp(self):
        pass

    def test_mul(self):
        print(a.mul)
        self.assertTrue(True)

@H_404_2@

现在，有没有一种方法可以在从moduleA导入时模拟B / moduleB.py中的text_function，使我无法得到KeyError@H_404_2@？

基本上，在测试模块A时，我不希望模块B运行任何东西

预先感谢：）

解决方法

您的主要问题是在导入模块时已经调用了有问题的函数，因此您无法修补该函数（因为修补会导入原始模块，然后调用该函数）。在这种情况下，您必须 在导入测试模块之前对导入本身进行模拟：

import unittest
from unittest import mock
import sys

sys.modules['B.moduleB'] = mock.Mock()

from moduleA import a

class TestmoduleA(unittest.TestCase):
    def test_mul(self):
        print(a.mul) 
        self.assertTrue(True)

另请参见how to mock an import。