restoreAllMocks doesn't restore mocks about vitest HOT 4 CLOSED

Inconveniently named restoreAllMocks is not related to vi.mock, so it doesn't mention the vi.mock behaviour in the documentation.

So is there no way to have different mocks for different test / suites?

There are several ways you can achieve it. vi.mock is called only once, but you can setup a hoisted spy that can be changed in different tests:

import { envSchema, fetchCdkConfig } from './cdk-config.js'

const getFromSsm = vi.hoisted(() => vi.fn())

vi.mock('./config-client.js', () => {
  const ConfigClient = vi.fn();
  ConfigClient.prototype.getFromSsm = getFromSsm
  return { ConfigClient };
})

test('returns a valid envSchema object', () => {
  getFromSssm.mockReturnValue(generateMock(envSchema))
  const actual = await fetchCdkConfig();
  expect(envSchema.safeParse(actual).success).toBe(true);
})

test('breaks if it gets an invalid envSchema object', async () => {
  getFromSssm.mockReturnValue({ name: 'home', catchphrase: 'doh' })
  const sut = async () => await fetchCdkConfig();
  await expect(sut).rejects.toThrow(); // make sure you await it
})

You can also just spy on methods without mocking the module:

import { envSchema, fetchCdkConfig } from './cdk-config.js'
import { ConfigClient } from './config-client.js'

const getFromSsm = vi.fn()
ConfigClient.prototype.getFromSsm = getFromSsm

// same code for tests

If you really want to mock several times, there is vi.doMock API: https://vitest.dev/api/vi.html#vi-domock
but make sure that you import from ./cdk-config.js inside of every test again as it only mocks the next import:

The next dynamic import of the module will be mocked.

from vitest.

Very helpful, thank you very much for the explanation!

from vitest.

vi.mock is hoisted, so your code is only actually called once. Please refer to the documentation: https://vitest.dev/api/vi.html#vi-mock

from vitest.

I see. I looked at the documentation for restAllMocks where nothing of the sort is mentioned

Will call .mockReset() on all spies. This will clear mock history and reset its implementation to an empty function (will return undefined).

So is there no way to have different mocks for different test / suites?

from vitest.