Python ส่งคืนวัตถุ MagicMock แทน return_value

ฉันมีไฟล์ 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()

ผมอยากจะ UnitTest method_bในชั้นเรียนโดยการเยาะเย้ยB Aนี่คือเนื้อหาของไฟล์testa.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()

ฉันคาดว่าจะได้รับMocked Aในผลลัพธ์ แต่สิ่งที่ฉันได้รับคือ:

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

ฉันทำผิดตรงไหน?

เมื่อคุณ@mock.patch('a.A')คุณจะเปลี่ยนชั้นในรหัสภายใต้การทดสอบด้วยAmock_a

ในB.method_bที่คุณตั้งไว้แล้วa = A()ซึ่งขณะนี้a = mock_a()- คือaเป็นของreturn_value mock_aในขณะที่คุณยังไม่ได้ระบุค่านี้ก็เป็นปกติMagicMock; สิ่งนี้ไม่ได้กำหนดค่าไว้ดังนั้นคุณจะได้รับการตอบสนองเริ่มต้น (อีกครั้งMagicMock) เมื่อเรียกวิธีการ

แต่คุณต้องการกำหนดค่าของที่จะมีวิธีการที่เหมาะสมที่คุณสามารถทำตามอย่างใดอย่างหนึ่งreturn_valuemock_a

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()จะทำให้เกิดวิธีการจำลองของคุณ