How to spy on store functions during Pinia store test?

export const useTestStore = defineStore('test-store', () => {
  function test1 () {
    console.log('before')
    test2()
    console.log('after')
  }

  function test2 () {
    console.log('test 2 called')
  }

  return {
    test1,
    test2
  }
})

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

describe('useTestStore', () => {
  beforeEach(() => {
    const pinia = createPinia()
    setActivePinia(pinia)
  })

it('should call test2', () => {
    const testStore = useTestStore()
    const test2Spy = vi.spyOn(testStore, 'test2')
    testStore.test1()
    expect(test2Spy).toHaveBeenCalledOnce()
  })
})