fix(dev): restore local global snap test shims

Author: Salnika

packages/tools/src/local-cli.ts

@@ -277,7 +277,15 @@ export function runLocalGlobalSnapTest(args: string[]) {
 
   const result = spawnSync(
     process.execPath,
-    [toolBinPath, 'snap-test', '--dir', 'snap-tests-global', '--bin-dir', localVpBinDir, ...args],
+    [
+      toolBinPath,
+      'snap-test',
+      '--dir',
+      'snap-tests-global',
+      '--local-vp-bin-dir',
+      localVpBinDir,
+      ...args,
+    ],
     {
       cwd: process.cwd(),
       env: localCliEnv(),

packages/tools/src/snap-test.ts

@@ -1,3 +1,4 @@
+import { spawnSync } from 'node:child_process';
 import { randomUUID } from 'node:crypto';
 import fs, { readFileSync } from 'node:fs';
 import fsPromises from 'node:fs/promises';
@@ -91,6 +92,48 @@ function selectShard<T>(items: T[], index: number, total: number): T[] {
 }
 
 const NPM_GLOBAL_PREFIX_DIR = 'npm-global-lib-for-snap-tests';
+const vpBinaryName = process.platform === 'win32' ? 'vp.exe' : 'vp';
+
+function setupLocalGlobalShims(vitePlusHome: string, sourceBinDir: string): string {
+  const resolvedSourceBinDir = path.resolve(expandHome(sourceBinDir));
+  const sourceVpPath = path.join(resolvedSourceBinDir, vpBinaryName);
+  if (!fs.existsSync(sourceVpPath)) {
+    throw new Error(`Missing local vp binary for snap tests: ${sourceVpPath}`);
+  }
+
+  const currentBinDir = path.join(vitePlusHome, 'current', 'bin');
+  fs.mkdirSync(currentBinDir, { recursive: true });
+
+  const localVpPath = path.join(currentBinDir, vpBinaryName);
+  if (process.platform === 'win32') {
+    fs.copyFileSync(sourceVpPath, localVpPath);
+
+    const sourceShimPath = path.join(resolvedSourceBinDir, 'vp-shim.exe');
+    if (!fs.existsSync(sourceShimPath)) {
+      throw new Error(`Missing local vp trampoline for snap tests: ${sourceShimPath}`);
+    }
+    fs.copyFileSync(sourceShimPath, path.join(currentBinDir, 'vp-shim.exe'));
+  } else {
+    fs.symlinkSync(sourceVpPath, localVpPath);
+  }
+
+  const result = spawnSync(localVpPath, ['env', 'setup', '--refresh'], {
+    env: {
+      ...process.env,
+      VP_HOME: vitePlusHome,
+    },
+    encoding: 'utf-8',
+  });
+
+  if (!result.error && result.status === 0) {
+    return path.join(vitePlusHome, 'bin');
+  }
+
+  const output = [result.stdout, result.stderr].filter(Boolean).join('\n').trim();
+  throw new Error(
+    `Failed to prepare local global shims for snap tests.${output ? `\n${output}` : ''}`,
+  );
+}
 
 export async function snapTest() {
   const { positionals, values } = parseArgs({
@@ -99,6 +142,7 @@ export async function snapTest() {
     options: {
       dir: { type: 'string' },
       'bin-dir': { type: 'string' },
+      'local-vp-bin-dir': { type: 'string' },
       shard: { type: 'string' },
     },
   });
@@ -129,6 +173,9 @@ export async function snapTest() {
 
   const vitePlusHome = path.join(tempTmpDir, 'vite-plus-home');
   fs.mkdirSync(vitePlusHome, { recursive: true });
+  const effectiveBinDir = values['local-vp-bin-dir']
+    ? setupLocalGlobalShims(vitePlusHome, values['local-vp-bin-dir'])
+    : values['bin-dir'];
 
   // Remove .previous-version so command-upgrade-rollback snap test is stable
   const previousVersionPath = path.join(vitePlusHome, '.previous-version');
@@ -210,8 +257,7 @@ export async function snapTest() {
   for (const caseName of selectedCases) {
     const stepsPath = path.join(casesDir, caseName, 'steps.json');
     const steps: Steps = JSON.parse(readFileSync(stepsPath, 'utf-8'));
-    const task = () =>
-      runTestCase(caseName, tempTmpDir, casesDir, vitePlusHome, values['bin-dir']);
+    const task = () => runTestCase(caseName, tempTmpDir, casesDir, vitePlusHome, effectiveBinDir);
     if (steps.serial) {
       serialTasks.push(task);
     } else {