fix test cases

joe-re · Aug 13, 2023 · 7be556e · 7be556e
1 parent 15fc402
commit 7be556e
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/packages/server/test/complete.test.ts b/packages/server/test/complete.test.ts
@@ -124,17 +124,18 @@ describe('keyword completion', () => {
   })
 
   test("complete 'DELETE' keyword", () => {
-    const sql = 'D'
+    const sql = 'DE'
     const result = complete(sql, { line: 0, column: sql.length })
-    expect(result.candidates.length).toEqual(2) // includes drop table
+    expect(result.candidates.length).toEqual(1)
     expect(result.candidates[0].label).toEqual('DELETE')
   })
 
-  test("complete 'DROP TABLE' keyword", () => {
-    const sql = 'D'
+  test("complete 'DROP'/'DROP TABLE' keyword", () => {
+    const sql = 'DR'
     const result = complete(sql, { line: 0, column: sql.length })
-    expect(result.candidates.length).toEqual(2) // includes delete
-    expect(result.candidates[1].label).toEqual('DROP TABLE')
+    expect(result.candidates.length).toEqual(2)
+    expect(result.candidates.map((v) => v.label)).toContain('DROP TABLE')
+    expect(result.candidates.map((v) => v.label)).toContain('DROP')
   })
 })