Python補丁：在類中調用嵌套方法不起作用

Question

這是我的以下模板代碼：

import mock
import unittest
class ClassToPatch(object):
    def __init__(self, *args):
        pass
    def some_func(self):
        data = self._get_data()
        return data

    def _get_data(self):
        return 'class_data'

class TestCase(unittest.TestCase):
    @mock.patch('__main__.ClassToPatch', autospec = True)
    def test_1(self, mock1):
        #mock1.data = "mocked data"
        m = mock.Mock()
        m._get_data.return_value = 'mocked data'
        mock1.return_value = m
        u = ClassToPatch()
        self.assertEqual(u.some_func(), 'mocked data')

unittest.main()

但是，這會在斷言上引發錯誤。 當我將其更改為：

self.assertEqual(u._get_data(), 'mocked data')

它工作正常。 有人可以告訴我我在做什么錯嗎？

Answer 1

為了測試，僅修補_get_data ，而不修補整個類。

@mock.patch.object(ClassToPatch, '_get_data')
def test_1(self, mock):
    mock.return_value = 'mocked data'
    u = ClassToPatch()
    self.assertEqual(u.some_func(), 'mocked data')

要么

@mock.patch.object(ClassToPatch, '_get_data', lambda self: 'mocked data')
def test_1(self):
    u = ClassToPatch()
    self.assertEqual(u.some_func(), 'mocked data')

要么

@mock.patch('__main__.ClassToPatch._get_data', return_value='mocked data')
def test_1(self, mock1):
    u = ClassToPatch()
    self.assertEqual(u.some_func(), 'mocked data')