Jest 如何在测试中 mock 一个 Date 对象？

在前端开发中，我们经常需要对日期进行处理。而在测试中，我们也需要对日期进行模拟来确保代码的正确性。Jest 是一个流行的 JavaScript 测试框架，它提供了一些方法来 mock 日期对象，以便在测试中使用。

Date 对象的 mock

在 Jest 中，我们可以使用 jest.spyOn 方法来 mock 一个 Date 对象。这个方法接受两个参数：要 mock 的对象和要 mock 的方法名。在这里，我们要 mock 的是全局的 Date 对象，所以第一个参数应该是 global，第二个参数是 Date。

test('mock Date', () => {
  const dateNowSpy = jest.spyOn(global.Date, 'now')
  const now = new Date('2021-01-01').getTime()
  dateNowSpy.mockImplementation(() => now)
  expect(Date.now()).toEqual(now)
  dateNowSpy.mockRestore()
})

上面的代码中，我们首先使用 jest.spyOn 方法来创建一个 dateNowSpy 对象，然后创建一个新的 Date 对象，并获取它的时间戳。接着，我们使用 mockImplementation 方法来为 dateNowSpy 对象设置一个新的实现，这个实现会返回我们刚才创建的时间戳。最后，我们使用 expect 来验证 Date.now() 的返回值是否等于我们刚才创建的时间戳。最后，我们使用 mockRestore 方法来恢复原始的实现。

Date 对象的构造函数

除了 mock Date.now 方法之外，我们还可以 mock Date 构造函数。在 Jest 中，我们可以使用 jest.spyOn 方法来 mock Date 构造函数，然后使用 mockReturnValue 方法来设置返回值。

// javascriptcn.com 代码示例
test('mock Date constructor', () => {
  const dateConstructorSpy = jest.spyOn(global, 'Date')
  const now = new Date('2021-01-01').getTime()
  dateConstructorSpy.mockImplementation(() => {
    return {
      getTime: () => now
    }
  })
  expect(new Date().getTime()).toEqual(now)
  dateConstructorSpy.mockRestore()
})

上面的代码中，我们首先使用 jest.spyOn 方法来创建一个 dateConstructorSpy 对象，然后创建一个新的 Date 对象，并获取它的时间戳。接着，我们使用 mockImplementation 方法来为 dateConstructorSpy 对象设置一个新的实现，这个实现会返回一个包含 getTime 方法的对象，这个方法会返回我们刚才创建的时间戳。最后，我们使用 expect 来验证 new Date().getTime() 的返回值是否等于我们刚才创建的时间戳。最后，我们使用 mockRestore 方法来恢复原始的实现。

总结

在 Jest 中 mock 一个 Date 对象非常简单。我们可以使用 jest.spyOn 方法来创建一个 mock 对象，然后使用 mockImplementation 或 mockReturnValue 方法来设置返回值。这样，我们就可以在测试中模拟日期对象了。

示例代码

// javascriptcn.com 代码示例
test('mock Date', () => {
  const dateNowSpy = jest.spyOn(global.Date, 'now')
  const now = new Date('2021-01-01').getTime()
  dateNowSpy.mockImplementation(() => now)
  expect(Date.now()).toEqual(now)
  dateNowSpy.mockRestore()
})

test('mock Date constructor', () => {
  const dateConstructorSpy = jest.spyOn(global, 'Date')
  const now = new Date('2021-01-01').getTime()
  dateConstructorSpy.mockImplementation(() => {
    return {
      getTime: () => now
    }
  })
  expect(new Date().getTime()).toEqual(now)
  dateConstructorSpy.mockRestore()
})

来源：JavaScript中文网 ，转载请注明来源 本文地址：https://www.javascriptcn.com/post/6562dc92d2f5e1655dca03ea