python教程—Python返回MagicMock对象,而不是return_value-Python实用宝典

python教程—Python返回MagicMock对象,而不是return_value

我有一个python文件a.py,它包含两个类a和B。我想在类B中通过mock a来unittest method_b,这是test .py文件的内容。

我有一个python文件a.py,它包含两个类a和B。

    class A(object): def method_a(self): return "Class A method a" class B(object): def method_b(self): a = A() print a.method_a()

我想在类B中通过mock a来unittest method_b,这是test .py文件的内容。

    import unittest import mock import a class TestB(unittest.TestCase): @mock.patch('a.A') def test_method_b(self, mock_a): mock_a.method_a.return_value = 'Mocked A' b = a.B() b.method_b() if __name__ == '__main__': unittest.main()

我期望在输出中模拟A。但我得到的是:

    <MagicMock name='A().method_a()' id='4326621392'>

我哪里做错了?

回答

当您@mock.patch('a.A')时,您正在用mock_a替换测试代码中的类A。

在B。然后设置a = a(),它现在是a = mock_a() -即a是mock_a的return_value。因为没有指定这个值,所以它是一个常规的MagicMock;这也没有配置,所以在调用它的方法时,您将得到默认响应(另一个MagicMock)。

相反,您希望配置 mock_a的return_value来获得适当的方法,您可以这样做:

    mock_a().method_a.return_value = 'Mocked A' # ^ note parentheses

或者,也许更明确地说:

    mock_a.return_value.method_a.return_value = 'Mocked A'

您的代码将在a = a的情况下工作(分配类,而不是创建实例),因为a.method_a()将触发模拟方法。

​Python实用宝典 (pythondict.com)
不只是一个宝典
欢迎关注公众号:Python实用宝典

本文由 Python实用宝典 作者:Python实用宝典 发表,其版权均为 Python实用宝典 所有,文章内容系作者个人观点,不代表 Python实用宝典 对观点赞同或支持。如需转载,请注明文章来源。
0

发表评论