temp

Author: benfdking

vscode/extension/tests/bad_setup.spec.ts

@@ -75,19 +75,11 @@ test('missing LSP dependencies shows install prompt', async ({}, testInfo) => {
 })
 
 test('lineage, no sqlmesh found', async ({}, testInfo) => {
-  testInfo.setTimeout(120_000) // 2 minutes for venv creation and package installation
   const tempDir = await fs.mkdtemp(
     path.join(os.tmpdir(), 'vscode-test-tcloud-'),
   )
   const pythonEnvDir = path.join(tempDir, '.venv')
   const pythonDetails = await createVirtualEnvironment(pythonEnvDir)
-  const custom_materializations = path.join(
-    REPO_ROOT,
-    'examples',
-    'custom_materializations',
-  )
-  const sqlmeshWithExtras = `${REPO_ROOT}[bigquery]`
-  await pipInstall(pythonDetails, [sqlmeshWithExtras, custom_materializations])
 
   try {
     // Copy sushi project
@@ -110,16 +102,26 @@ test('lineage, no sqlmesh found', async ({}, testInfo) => {
     // Open lineage view
     await openLineageView(window)
 
-    await window.waitForTimeout(1_000)
+    // Assert shows that sqlmesh is not installed
+    await window.waitForSelector('text=SQLMesh language server not found')
 
-    // Assert cannot see the Loading models text
+    // Assert cannot see wrong notifications
+    await expect(window.locator('LSP dependencies missing')).not.toBeVisible()
+
+    // Cancel the notification
+    await window.getByRole('button', { name: 'Cancel' }).click()
+
+    // When cancel notification, the lineage view should show the error
     await expect(
-      window.locator('text=Loading models...'),
-    ).not.toBeVisible()
+      window.locator('text=SQLMesh language server not found'),
+    ).toBeVisible()
+
+    // Assert cannot see the Loading models text
+    await expect(window.locator('text=Loading models...')).not.toBeVisible()
 
     await close()
   } finally {
     // Clean up
     await fs.remove(tempDir)
   }
-})
+})