chore(vscode): add an integration test for gtd (#4608)

Author: benfdking

vscode/extension/tests/go_to_definition.spec.ts

@@ -11,10 +11,10 @@ test('Go to definition for macro', async () => {
     try {
       const { window, close } = await startVSCode(tempDir);
 
-      //   Wait for the models folder to be visible
+      // Wait for the models folder to be visible
       await window.waitForSelector('text=models');
 
-      // Click on the models folder, excluding external_models
+      // Click on the models folder
       await window.getByRole('treeitem', { name: 'models', exact: true }).locator('a').click();
 
       // Open the customer_revenue_lifetime model
@@ -24,7 +24,7 @@ test('Go to definition for macro', async () => {
       await window.waitForSelector('text=Loaded SQLMesh Context')
 
       // Render the model
-      window.locator("text=@MULTIPLY").click({
+      await window.locator("text=@MULTIPLY").click({
         modifiers: ["Meta"]
       })
 
@@ -35,4 +35,34 @@ test('Go to definition for macro', async () => {
     } finally {
       await fs.removeSync(tempDir);
     }
-  });
+  });
+
+test("Go to definition for model", async () => {
+    const tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'vscode-test-sushi-'));
+    await fs.copy(SUSHI_SOURCE_PATH, tempDir);
+
+    try {
+      const { window, close } = await startVSCode(tempDir);
+
+      // Wait for the models folder to be visible
+      await window.waitForSelector('text=models');
+
+      // Click on the models folder
+      await window.getByRole('treeitem', { name: 'models', exact: true }).locator('a').click();
+
+      // Open the top_waiters model
+      await window.getByRole('treeitem', { name: 'top_waiters.sql', exact: true }).locator('a').click();
+
+      await window.waitForSelector('text=grain');
+      await window.waitForSelector('text=Loaded SQLMesh Context')
+
+      // Go to definition for the model
+      await window.locator("text=sushi.waiter_revenue_by_day").first().click({
+        modifiers: ["Meta"]
+      })
+      await expect(window.locator('text=SUM(oi.quantity * i.price)::DOUBLE AS revenue')).toBeVisible();
+      await close();
+    } finally {
+      await fs.removeSync(tempDir);
+    }
+})