python – django UnitTest案例中的伪子函数(模拟)

发布时间：2020-12-20 13:37:17 所属栏目：Python 来源：网络整理

导读：def func_b(**kwargs): return somethingdef func_a(request,*args,**kwargs): //do something b = func_b(**kwargs) //do something return something 我正在为func_a编写UnitTest,但我想使用模拟或类似于django中的模拟来伪造func_b的输出. 意思是我只是想

def func_b(**kwargs):
    return something


def func_a(request,*args,**kwargs):
    //do something
    b = func_b(**kwargs)
    //do something
    return something

我正在为func_a编写UnitTest,但我想使用模拟或类似于django中的模拟来伪造func_b的输出.
意思是我只是想伪造func_b的输出,并且总是希望在测试用例中从func_b返回一些静态值.
有没有办法做到这一点？

解决方法

您应该使用mock.patch并指定return_value.这是一个示例,我们将func_b()的返回值修补为Fake值：

from mock import patch
import unittest


def func_b():
    return "Real value"


def func_a():
    return "The result of func_b is '%s'" % func_b()


class MyTestCase(unittest.TestCase):
    def test_fake_value(self):
        with patch('test.func_b',return_value="Fake value") as mock_function:
            self.assertEqual(func_a(),"The result of func_b is 'Fake value'")

UPD：

with patch.object(module_name,'func_b') as mock_function:
    mock_function.return_value = "Fake value"
    self.assertEqual(func_a(),"The result of func_b is 'Fake value'")

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!