vitest/test/benchmark/test/reporter.test.ts

import * as pathe from 'pathe'
import { assert, expect, it } from 'vitest'
import { runVitest } from '../../test-utils'

it('summary', async () => {
  const root = pathe.join(import.meta.dirname, '../fixtures/reporter')
  const result = await runVitest({ root }, ['summary.bench.ts'], 'benchmark')
  expect(result.stderr).toBe('')
  expect(result.stdout).not.toContain('NaNx')
  expect(result.stdout.split('BENCH  Summary')[1].replaceAll(/[0-9.]+x/g, '(?)')).toMatchSnapshot()
})

it('non-tty', async () => {
  const root = pathe.join(import.meta.dirname, '../fixtures/basic')
  const result = await runVitest({ root }, ['base.bench.ts'], 'benchmark')
  const lines = result.stdout.split('\n').slice(4).slice(0, 11)
  const expected = `\
 ✓ base.bench.ts > sort
     name
   · normal
   · reverse

 ✓ base.bench.ts > timeout
     name
   · timeout100
   · timeout75
   · timeout50
   · timeout25
`

  for (const [index, line] of expected.trim().split('\n').entries()) {
    expect(lines[index]).toMatch(line)
  }
})

it.for([true, false])('includeSamples %s', async (includeSamples) => {
  const result = await runVitest(
    {
      root: pathe.join(import.meta.dirname, '../fixtures/reporter'),
      benchmark: { includeSamples },
    },
    ['summary.bench.ts'],
    'benchmark',
  )
  assert(result.ctx)
  const allSamples = [...result.ctx.state.idMap.values()]
    .filter(t => t.meta.benchmark)
    .map(t => t.result?.benchmark?.samples)
  if (includeSamples) {
    expect(allSamples[0]).not.toEqual([])
  }
  else {
    expect(allSamples[0]).toEqual([])
  }
})