Python trả về đối tượng MagicMock thay vì return_value

Tôi có một tệp python a.pychứa hai lớp Avà 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()

Tôi muốn nổi nhất method_btrong lớp Bbằng cách chế giễu A. Đây là nội dung của tệp testa.pycho mục đích này:

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()

Tôi mong đợi để có được Mocked Ađầu ra. Nhưng những gì tôi nhận được là:

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

Tôi đang làm sai ở đâu?

Khi bạn @mock.patch('a.A'), bạn đang thay thế lớp Atrong mã được kiểm tra bằng mock_a.

Trong B.method_bbạn sau đó thiết lập a = A(), mà bây giờ a = mock_a()- tức alà return_valuecủa mock_a. Vì bạn chưa chỉ định giá trị này, nó là một giá trị thường xuyên MagicMock; điều này cũng không được định cấu hình, vì vậy bạn sẽ nhận được phản hồi mặc định (nhưng khác MagicMock) khi gọi các phương thức trên đó.

Thay vào đó, bạn muốn cấu hình các return_valuecủamock_a để có phương pháp thích hợp, mà bạn có thể làm là một trong hai:

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

hoặc, có lẽ rõ ràng hơn:

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

Mã của bạn sẽ hoạt động trong trường hợp này a = A(gán lớp chứ không phải tạo một thể hiện), vì sau đó a.method_a()sẽ kích hoạt phương thức giả của bạn.