Vitest Zustand mock store

setupTests.tsx

import { mockSong } from '@/data/song';
import { useCadenceStore } from '@/store/cadence';
import { CadenceStore } from '@/types/cadence';
import '@testing-library/jest-dom';
import * as matchers from '@testing-library/jest-dom/matchers';
import { expect, vi } from 'vitest';

expect.extend(matchers);

vi.mock('@/store/cadence', () => ({
  useCadenceStore: vi.fn(),
}));

const useCadenceStoreMock = vi.mocked(useCadenceStore);

export const mockUseCadenceStore = (overrides: Partial<CadenceStore> = {}) => {
  useCadenceStoreMock.mockReturnValue({
    song: mockSong,
    setSong: vi.fn(),
    handleCellChange: vi.fn(),
    ...overrides,
  });
};

beforeEach(() => {
  mockUseCadenceStore();
});