python - 如何在 Python 中模拟具有嵌套属性的对象？

Question

考虑以下代码:

class Foo:
    @staticmethod
    def is_room_member(invitee, msg):
        return invitee in msg.frm.room.occupants

我想测试 is_room_member 方法，其中 invitee 是一个字符串并且occupants 是一个字符串列表。

如果 invitee = 'batman' 和 occupants = ['batman', 'superman'] 方法 is_room_member 返回 True 。

msg 是需要模拟的对象，以便我可以测试此方法。

我如何测试这个方法，因为它需要这个具有嵌套属性的 msg 对象？

我希望测试是这样的:

class Testing(unittest.TestCase):
    def test_is_room_member(self):
        occupants = ['batman', 'superman']
        # mocking 
        # msg = MagicMock()
        # msg.frm.room.occupants = occupants
        self.assertTrue(Foo.is_room_member('batman', msg))

Answer 1

您的问题已有答案: Mocking nested properties with mock

import unittest
import mock

class Foo:
    @staticmethod
    def is_room_member(invitee, msg):
        return invitee in msg.frm.room.occupants

class Testing(unittest.TestCase):
    def test_is_room_member(self):
        occupants = ['batman', 'superman']

        # mocking
        mocked_msg = mock.MagicMock()
        mocked_msg.frm.room.occupants = occupants

        self.assertTrue(Foo.is_room_member('batman', mocked_msg))

if __name__ == '__main__':
    unittest.main()