如何模拟内部调用（xlwt）？

Question

我有一个这样的测试用例：

@mock.patch('xlwt.Workbook.Workbook.save')
    def test_Generic_send_test_email_address(self, workspace_mock):
        workspace_mock.return_value = None
        oi = OptinInvitesGeneric()
        oi.compute(...)
        self.assert ...

实际方法会进行一些处理，并将结果保存在excel电子表格中。

class OptinInvitesGeneric(OptinBase):
    def compute(...):
      ...
      wb = excel_tool.write_xls(...)
      wb.save('{0}.xls'.format(category))

看来我的模拟补丁没有接管workbook.save（）。 我想念什么？

Answer 1

我不知道您为什么要修补xlwt.Workbook.Workbook ，但是这两个对我xlwt.Workbook.Workbook ：

@patch.object(xlwt.Workbook, 'save', return_value=None)
def test_patch_object(mock):
    wb = xlwt.Workbook()
    assert wb.save() == None

@patch('xlwt.Workbook.save', return_value=None)
def test_patch(mock):
    wb = xlwt.Workbook()
    assert wb.save() == None