From 29769ccfd29ec10a04cd7209a96742c7fd86bf06 Mon Sep 17 00:00:00 2001
From: Ben King <9087625+benfdking@users.noreply.github.com>
Date: Wed, 2 Jul 2025 14:34:33 +0100
Subject: [PATCH] chore: make vscode tests less flaky

---
 vscode/extension/tests/python_env.spec.ts | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/vscode/extension/tests/python_env.spec.ts b/vscode/extension/tests/python_env.spec.ts
index bbf0bafb36..95a726c25b 100644
--- a/vscode/extension/tests/python_env.spec.ts
+++ b/vscode/extension/tests/python_env.spec.ts
@@ -84,7 +84,9 @@ test.describe('python environment variable injection on sqlmesh_lsp', () => {
     }
   })
 
-  test('normal setup - set', async ({ page }) => {
+  test('normal setup - set', async ({ page }, testInfo) => {
+    testInfo.setTimeout(120_000)
+
     const [tempDir, _] = await setupEnvironment()
     writeEnvironmentConfig(tempDir)
     const env_file = path.join(tempDir, '.env')
@@ -126,7 +128,9 @@ async function setupTcloudProject(
 }
 
 test.describe('tcloud version', () => {
-  test('normal setup - error ', async ({ page }) => {
+  test('normal setup - error ', async ({ page }, testInfo) => {
+    testInfo.setTimeout(120_000)
+
     const [tempDir, pythonDetails] = await setupEnvironment()
     await setupTcloudProject(tempDir, pythonDetails)
     writeEnvironmentConfig(tempDir)
@@ -141,7 +145,9 @@ test.describe('tcloud version', () => {
     }
   })
 
-  test('normal setup - set', async ({ page }) => {
+  test('normal setup - set', async ({ page }, testInfo) => {
+    testInfo.setTimeout(120_000)
+
     const [tempDir, pythonDetails] = await setupEnvironment()
     await setupTcloudProject(tempDir, pythonDetails)
     writeEnvironmentConfig(tempDir)