chore(vscode): use uv to speed up e2e tests

Author: benfdking

.github/workflows/pr.yaml

@@ -42,6 +42,8 @@ jobs:
         uses: actions/setup-python@v5
         with:
           python-version: '3.12'
+      - name: Install uv
+        uses: astral-sh/setup-uv@v4
       - name: Cache pip dependencies
         uses: actions/cache@v4
         with:
@@ -62,7 +64,7 @@ jobs:
             ${{ runner.os }}-venv-
       - name: Install python dependencies
         run: |
-          python -m venv .venv
+          uv venv .venv
           source .venv/bin/activate
           make install-dev
       - name: Install code-server

vscode/extension/tests/utils.ts

@@ -1,7 +1,6 @@
 import path from 'path'
 import { Page } from '@playwright/test'
-import { exec } from 'child_process'
-import { promisify } from 'util'
+import { execAsync } from '../src/utilities/exec'
 
 // Where your extension lives on disk
 export const EXT_PATH = path.resolve(__dirname, '..')
@@ -34,25 +33,36 @@ export const clickExplorerTab = async (page: Page): Promise<void> => {
   }
 }
 
-const execAsync = promisify(exec)
-
 export interface PythonEnvironment {
   pythonPath: string
   pipPath: string
 }
 
 /**
- * Create a virtual environment in the given directory.
+ * Create a virtual environment in the given directory using uv.
  * @param venvDir The directory to create the virtual environment in.
  */
 export const createVirtualEnvironment = async (
   venvDir: string,
 ): Promise<PythonEnvironment> => {
-  const pythonCmd = process.platform === 'win32' ? 'python' : 'python3'
-  const { stderr } = await execAsync(`${pythonCmd} -m venv "${venvDir}"`)
-  if (stderr && !stderr.includes('WARNING')) {
-    throw new Error(`Failed to create venv: ${stderr}`)
+  // Try to use uv first, fallback to python -m venv
+  try {
+    const { exitCode, stderr } = await execAsync(`uv venv "${venvDir}"`)
+    if (exitCode !== 0) {
+      throw new Error(`Failed to create venv with uv: ${stderr}`)
+    }
+  } catch (error) {
+    // Fallback to python -m venv if uv is not available
+    console.warn('uv not available, falling back to python -m venv')
+    const pythonCmd = process.platform === 'win32' ? 'python' : 'python3'
+    const { stderr, exitCode } = await execAsync(
+      `${pythonCmd} -m venv "${venvDir}"`,
+    )
+    if (exitCode !== 0) {
+      throw new Error(`Failed to create venv: ${stderr}`)
+    }
   }
+
   // Get paths
   const isWindows = process.platform === 'win32'
   const binDir = path.join(venvDir, isWindows ? 'Scripts' : 'bin')
@@ -66,19 +76,33 @@ export const createVirtualEnvironment = async (
 }
 
 /**
- * Install packages in the given virtual environment.
+ * Install packages in the given virtual environment using uv (with pip fallback).
  * @param pythonDetails The Python environment to use.
  * @param packagePaths The paths to the packages to install (string[]).
  */
 export const pipInstall = async (
   pythonDetails: PythonEnvironment,
   packagePaths: string[],
 ): Promise<void> => {
-  const { pipPath } = pythonDetails
-  const execString = `"${pipPath}" install -e "${packagePaths.join('" -e "')}"`
-  const { stderr } = await execAsync(execString)
-  if (stderr && !stderr.includes('WARNING') && !stderr.includes('notice')) {
-    throw new Error(`Failed to install package: ${stderr}`)
+  // Try to use uv first, fallback to pip
+  try {
+    const packages = packagePaths.map(pkg => `-e "${pkg}"`).join(' ')
+    const execString = `uv pip install --python "${pythonDetails.pythonPath}" ${packages}`
+    const { stderr, exitCode } = await execAsync(execString)
+    if (exitCode !== 0) {
+      throw new Error(`Failed to install package with uv: ${stderr}`)
+    }
+  } catch (error) {
+    // Fallback to pip if uv is not available
+    console.warn(
+      `uv not available, falling back to pip, because of the error ${error}`,
+    )
+    const { pipPath } = pythonDetails
+    const execString = `"${pipPath}" install -e "${packagePaths.join('" -e "')}"`
+    const { stderr, exitCode } = await execAsync(execString)
+    if (exitCode !== 0) {
+      throw new Error(`Failed to install package: ${stderr}`)
+    }
   }
 }