python - 运行 python mox 测试时从未调用的预期方法

Question

我正在尝试编写一个 mox 测试来读取电子表格(4 列)、获取提要并将特定列(2 列)写入 CSV 文件。我试图通过获取列表提要的第一步，我的代码如下:

class SpreadsheetReader(mox.MoxTestBase):

  def setUp(self):
    mox.MoxTestBase.setUp(self)
    self.mock_gclient = self.mox.CreateMock(
                                            gdata.spreadsheet.service.SpreadsheetsService)
    self.mock_spreadsheet_key = 'fake_spreadsheet_key'
    self.mock_worksheet_id = 'default'
    self.test_data = [{'str_col':'col1', 'str_col':'col2', 'str_col':'col13'}]


  def testGetFeed(self):

    self.mock_gclient.GetListFeed(self.mock_spreadsheet_key,
                                  self.mock_worksheet_id).AndReturn(self.test_data)

    self.mox.ReplayAll()
    self.mox.Verify()


  def tearDown(self):
    mox.MoxTestBase.tearDown(self)

当我运行它时，出现以下错误:

ExpectedMethodCallsError: Verify: Expected methods never called:
  0.  SpreadsheetsService.GetListFeed('fake_spreadsheet_key', 'default') -> [{'str_col': 'col13'}]

知道如何解决这个错误吗？

Answer 1

您需要实际触发调用 GetListFeed 的函数。在调用 self.mox.ReplayAll() 之前，您只是在“记录”mox 进入重放模式后应该看到的内容。将 mox 置于重播模式后，您需要实际调用任何将调用 GetListFeed 的函数。在您的情况下，这似乎是 testGetFeed 或其父函数是什么。

此外，因为您在类定义中子类化了 mox.MoxTestBase()，所以您不需要在最后调用 self.mox.Verify() —根据 docs ,

you can make your test case be a subclass of mox.MoxTestBase; this will automatically create a mock object factory in self.mox, and will automatically verify all mock objects and unset stubs at the end of each test.