我如何模拟当时在开玩笑之后被调用的方法？

Question

我有以下下载PDF的代码：

import html2pdf from 'html2pdf.js'
export const addFooter = (pdf) => {
  const totalPages = pdf.internal.getNumberOfPages()
  ...
}
export const downloadPDF = async (...) => {
  const pdfContent = document.getElementById("export")
  const opts = {...}
  await html2pdf().from(pdfContent).set(opts).toPdf().get('pdf').then(addFooter).save()
}

我希望能够使用开玩笑模拟保存方法。

到目前为止，我有：

jest.mock('html2pdf.js', () => () => ({
  from: () => ({
    set: () => ({
      toPdf: () => ({
        get: () => Promise.resolve({
          internal: {
            getNumberOfPages: () => {}
          }
        })
      })
    })
  })
}))

但是运行测试给我：

TypeError: (0 , _html2pdf.default)(...).from(...).set(...).toPdf(...).get(...).then(...).save is not a function

也许有人知道如何模拟保存方法？