python - 如何在类的方法中模拟外部函数

Question

我需要一些关于模拟的帮助。

我在 mymodule.py 中有以下代码：

from someModule import external_function

class Class1(SomeBaseClass):
    def method1(self, arg1, arg2):
        external_function(param)

现在我有测试代码：

import mock
from django.test import TestCase

from mymodule import class1
class Class1Test(TestCase) 
    def test_method1:
        '''how can I mock external_function here?'''

Answer 1

你会写：

class Class1Test(TestCase):

    @mock.patch('mymodule.external_function')
    def test_method1(self, mock_external_function):
        pass

看mymodule函数external_function是直接导入的。因此，您需要模拟，因为这是执行mymodule.external_function时将调用的函数。method1