sqlmesh/vscode/extension/tests/broken_project.spec.ts at 85c4d49c4a0cadbda6c2e8052f700b6a76924473 · SQLMesh/sqlmesh · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
import { test } from '@playwright/test'
import fs from 'fs-extra'
import os from 'os'
import path from 'path'
import { openLineageView, startVSCode, SUSHI_SOURCE_PATH } from './utils'

test('bad project, double model', async ({}) => {
  const tempDir = await fs.mkdtemp(
    path.join(os.tmpdir(), 'vscode-test-tcloud-'),
  )
  await fs.copy(SUSHI_SOURCE_PATH, tempDir)

  // Read the customers.sql file
  const customersSql = await fs.readFile(
    path.join(tempDir, 'models', 'customers.sql'),
    'utf8',
  )

  // Write the customers.sql file with a double model
  await fs.writeFile(
    path.join(tempDir, 'models', 'customers_duplicated.sql'),
    customersSql,
  )

  const { window, close } = await startVSCode(tempDir)
  try {
    await window.waitForSelector('text=models')

    await window
      .getByRole('treeitem', { name: 'models', exact: true })
      .locator('a')
      .click()

    await window
      .getByRole('treeitem', { name: 'customers.sql', exact: true })
      .locator('a')
      .click()

    await window.waitForSelector('text=Error creating context')

    await window.waitForTimeout(1000)
  } finally {
    await close()
    await fs.remove(tempDir)
  }
})

test('bad project, double model, then fixed', async ({}) => {
  const tempDir = await fs.mkdtemp(
    path.join(os.tmpdir(), 'vscode-test-tcloud-'),
  )
  await fs.copy(SUSHI_SOURCE_PATH, tempDir)

  // Read the customers.sql file
  const customersSql = await fs.readFile(
    path.join(tempDir, 'models', 'customers.sql'),
    'utf8',
  )

  // Write the customers.sql file with a double model
  await fs.writeFile(
    path.join(tempDir, 'models', 'customers_duplicated.sql'),
    customersSql,
  )

  const { window, close } = await startVSCode(tempDir)
  try {
    await window.waitForSelector('text=models')

    await window
      .getByRole('treeitem', { name: 'models', exact: true })
      .locator('a')
      .click()

    await window
      .getByRole('treeitem', { name: 'customers.sql', exact: true })
      .locator('a')
      .click()

    await window.waitForSelector('text=Error creating context')

    // Remove the duplicated model
    await fs.remove(path.join(tempDir, 'models', 'customers_duplicated.sql'))

    // Open the linage view
    await openLineageView(window)

    // Wait for the error to go away
    await window.waitForSelector('text=Loaded SQLMesh context')
  } finally {
    await close()
    await fs.remove(tempDir)
  }
})

test('bad project, double model, check lineage', async ({}) => {
  const tempDir = await fs.mkdtemp(
    path.join(os.tmpdir(), 'vscode-test-tcloud-'),
  )
  await fs.copy(SUSHI_SOURCE_PATH, tempDir)

  // Read the customers.sql file
  const customersSql = await fs.readFile(
    path.join(tempDir, 'models', 'customers.sql'),
    'utf8',
  )

  // Write the customers.sql file with a double model
  await fs.writeFile(
    path.join(tempDir, 'models', 'customers_duplicated.sql'),
    customersSql,
  )

  const { window, close } = await startVSCode(tempDir)
  try {
    await window.waitForSelector('text=models')

    // Open the lineage view
    await openLineageView(window)

    await window.waitForSelector('text=Error creating context')

    await window.waitForSelector('text=Error:')

    await window.waitForTimeout(1000)
  } finally {
    await close()
    await fs.remove(tempDir)
  }
})