Python用多个返回值修补多个mock

Question

所以我尝试运行一个有两个补丁mock的单元，其中第一个补丁mock（insertIntoParent）被调用三次并返回[17,18,19]的值。这是我的测试：

@mock.patch('dao.AccessSQLInterface.AccessSQLInterface.insertIntoParent')
@mock.patch('dao.AccessSQLInterface.AccessSQLInterface.insertBulkData')
def test_hiertablecreatebulkinsert(self, insertBulkData, insertIntoParent):
  parse = ParseWorksheet()
  dao = AccessSQLInterface()

  dao.insertIntoParent = MagicMock(side_effect=[17,18,19])

  wb = openpyxl.Workbook()
  ws = wb.create_sheet('SheetData')

  dictHier = {}
  dictHier['C1'] = 4
  dictHier['C2'] = 2

  ws['A1'].value = 'Nameheader'
  ws['C1'].value = 'Parent1'
  ws['C2'].value = 'Parent2_1'
  ws['E2'].value = 'Parent2_2'
  ws['A4'].value = 'Cat1'
  ws['A5'].value = 'Cat2'
  ws['C4'].value = 'TxtVal1'
  ws['D5'].value = 4
  ws['E5'].value = 'TxtVal6'
  ws['F4'].value = 7

  genHeader = 'INSERT INTO dbo.Tbl1(Nameheader,Header1,Header2,ParentXID,StudyXID)'

  valueString1 = "('Cat1','TxtVal1',NULL,18,1),('Cat2',NULL,4,18,1)"
  valueString2 = "('Cat1',NULL,7,19,1),('Cat2','TxtVal6',NULL,19,1)"


  parse.createHierTableInserts(1, ws, 'Tbl1', genHeader, dictHier, 'A', 'False','C4','F5',dao)


  insertIntoParent.assert_called_once()

  insertBulkData.assert_any_call(genHeader, valueString1)
  insertBulkData.assert_any_call(genHeader, valueString2)

我知道这行：（dao.insertIntoParent = MagicMock(side_effect=[17,18,19])）正确地模拟了响应，因为副作用值显示在insertBulkData.assert_any_call(genHeader, valueString[])检查中。测试在insertIntoParent.assert_called_once()行失败，表示“insertIntoParent被调用了0次”，但是如果我注释掉它，测试就通过了。dao.insertIntoParent是否覆盖了我的insertIntoParent修补过的mock？我该怎么设置这个？在

Answer 1

所以实际上这是一个非常简单的修复，我只需要改变副作用线（正如我所想的，它覆盖了我的补丁mock）。这是我的新测试：

@mock.patch('dao.AccessSQLInterface.AccessSQLInterface.insertIntoParent')
@mock.patch('dao.AccessSQLInterface.AccessSQLInterface.insertBulkData')
def test_hiertablecreatebulkinsert(self, insertBulkData, insertIntoParent):
  parse = ParseWorksheet()
  dao = AccessSQLInterface()

  insertIntoParent.side_effect=[1,2,3]

  wb = openpyxl.Workbook()
  ws = wb.create_sheet('SheetData')

  dictHier = {}
  dictHier['C1'] = 4
  dictHier['C2'] = 2

  ws['A1'].value = 'Nameheader'
  ws['C1'].value = 'Parent1'
  ws['C2'].value = 'Parent2_1'
  ws['E2'].value = 'Parent2_2'
  ws['A4'].value = 'Cat1'
  ws['A5'].value = 'Cat2'
  ws['C4'].value = 'TxtVal1'
  ws['D5'].value = 4
  ws['E5'].value = 'TxtVal6'
  ws['F4'].value = 7

  genHeader = 'INSERT INTO dbo.Tbl1(Nameheader,Header1,Header2,ParentXID,StudyXID)'

  valueString1 = "('Cat1','TxtVal1',NULL,2,1),('Cat2',NULL,4,2,1)"
  valueString2 = "('Cat1',NULL,7,3,1),('Cat2','TxtVal6',NULL,3,1)"

  parse.createHierTableInserts(1, ws, 'Tbl1', genHeader, dictHier, 'A', 'False','C4','F5',dao)

  insertIntoParent.assert_any_call('Parent1',mock.ANY,'C1','Tbl1','NULL')
  insertIntoParent.assert_any_call('Parent2_1',mock.ANY,'C2','Tbl1',1)
  insertIntoParent.assert_any_call('Parent2_2',mock.ANY,'E2','Tbl1',1)

  insertBulkData.assert_any_call(genHeader, valueString1)
  insertBulkData.assert_any_call(genHeader, valueString2)