python - 如何模拟内部调用（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

我不知道您为什么要尝试 patch 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