feat!: move snapshot implementation into @vitest/snapshot (#3032)

Author: sheremet-va

packages/browser/src/client/main.ts

@@ -3,11 +3,11 @@ import { createClient } from '@vitest/ws-client'
 import type { ResolvedConfig } from 'vitest'
 import type { VitestRunner } from '@vitest/runner'
 import { createBrowserRunner } from './runner'
-import { BrowserSnapshotEnvironment } from './snapshot'
 import { importId } from './utils'
 import { setupConsoleLogSpy } from './logger'
 import { createSafeRpc, rpc, rpcDone } from './rpc'
 import { setupDialogsSpy } from './dialog'
+import { BrowserSnapshotEnvironment } from './snapshot'
 
 // @ts-expect-error mocking some node apis
 globalThis.process = { env: {}, argv: [], cwd: () => '/', stdout: { write: () => {} }, nextTick: cb => cb() }
@@ -75,19 +75,17 @@ ws.addEventListener('open', async () => {
 
   await setupConsoleLogSpy()
   setupDialogsSpy()
-  await runTests(paths, config)
+  await runTests(paths, config!)
 })
 
-let hasSnapshot = false
-async function runTests(paths: string[], config: any) {
+async function runTests(paths: string[], config: ResolvedConfig) {
   // need to import it before any other import, otherwise Vite optimizer will hang
   const viteClientPath = '/@vite/client'
   await import(viteClientPath)
 
   const {
     startTests,
     setupCommonEnv,
-    setupSnapshotEnvironment,
     takeCoverageInsideWorker,
   } = await importId('vitest/browser') as typeof import('vitest/browser')
 
@@ -101,10 +99,8 @@ async function runTests(paths: string[], config: any) {
     runner = new BrowserRunner({ config, browserHashMap })
   }
 
-  if (!hasSnapshot) {
-    setupSnapshotEnvironment(new BrowserSnapshotEnvironment())
-    hasSnapshot = true
-  }
+  if (!config.snapshotOptions.snapshotEnvironment)
+    config.snapshotOptions.snapshotEnvironment = new BrowserSnapshotEnvironment()
 
   try {
     await setupCommonEnv(config)

packages/browser/src/client/snapshot.ts

@@ -2,6 +2,14 @@ import { rpc } from './rpc'
 import type { SnapshotEnvironment } from '#types'
 
 export class BrowserSnapshotEnvironment implements SnapshotEnvironment {
+  getVersion(): string {
+    return '1'
+  }
+
+  getHeader(): string {
+    return `// Vitest Snapshot v${this.getVersion()}, https://vitest.dev/guide/snapshot.html`
+  }
+
   readSnapshotFile(filepath: string): Promise<string | null> {
     return rpc().readFile(filepath)
   }