模拟日期

有时你需要控制日期以确保测试时的一致性。Vitest 使用 @sinonjs/fake-timers 包来操作计时器和系统日期。你可以在这里找到有关特定 API 的详细信息。

示例

import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'

const businessHours = [9, 17]

function purchase() {
  const currentHour = new Date().getHours()
  const [open, close] = businessHours

  if (currentHour > open && currentHour < close) {
    return { message: 'Success' }
  }

  return { message: 'Error' }
}

describe('purchasing flow', () => {
  beforeEach(() => {
    // 告诉 vitest 我们使用模拟时间
    vi.useFakeTimers()
  })

  afterEach(() => {
    // 在每次测试运行后恢复日期
    vi.useRealTimers()
  })

  it('allows purchases within business hours', () => {
    // 设置在营业时间内的小时数
    const date = new Date(2000, 1, 1, 13)
    vi.setSystemTime(date)

    // 访问 `Date.now()` 将会返回上面设置的日期
    expect(purchase()).toEqual({ message: 'Success' })
  })

  it('disallows purchases outside of business hours', () => {
    // 设置在营业时间外的小时数
    const date = new Date(2000, 1, 1, 19)
    vi.setSystemTime(date)

    // 访问 `Date.now()` 将会返回上面设置的日期
    expect(purchase()).toEqual({ message: 'Error' })
  })
})