From 3880805080e463feb346d1620f5c99fcd8bbb00c Mon Sep 17 00:00:00 2001 From: liuxy0551 Date: Tue, 15 Oct 2024 22:13:53 +0800 Subject: [PATCH] test: suggestion wordRanges with range --- .../suggestion/multipleStatement.test.ts | 20 ++++++++++++- .../flink/suggestion/syntaxSuggestion.test.ts | 12 +++++++- .../hive/suggestion/multipleStatement.test.ts | 20 ++++++++++++- .../hive/suggestion/syntaxSuggestion.test.ts | 28 ++++++++++++++++++- .../suggestion/multipleStatement.test.ts | 20 ++++++++++++- .../suggestion/syntaxSuggestion.test.ts | 28 ++++++++++++++++++- .../suggestion/multipleStatement.test.ts | 20 ++++++++++++- .../mysql/suggestion/syntaxSuggestion.test.ts | 28 ++++++++++++++++++- .../suggestion/multipleStatement.test.ts | 20 ++++++++++++- .../suggestion/suggestionWithEntity.test.ts | 12 +++++++- .../suggestion/syntaxSuggestion.test.ts | 28 ++++++++++++++++++- .../suggestion/multipleStatement.test.ts | 20 ++++++++++++- .../spark/suggestion/syntaxSuggestion.test.ts | 28 ++++++++++++++++++- .../suggestion/multipleStatement.test.ts | 20 ++++++++++++- .../trino/suggestion/syntaxSuggestion.test.ts | 28 ++++++++++++++++++- 15 files changed, 317 insertions(+), 15 deletions(-) diff --git a/test/parser/flink/suggestion/multipleStatement.test.ts b/test/parser/flink/suggestion/multipleStatement.test.ts index e89e4bdf..e0ecd308 100644 --- a/test/parser/flink/suggestion/multipleStatement.test.ts +++ b/test/parser/flink/suggestion/multipleStatement.test.ts @@ -36,7 +36,25 @@ describe('FlinkSQL Multiple Statements Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.']); + expect(suggestion?.wordRanges?.length).toBe(2); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 16, + startIndex: 306, + endIndex: 307, + startColumn: 14, + endColumn: 16, + }, + { + text: '.', + line: 16, + startIndex: 308, + endIndex: 308, + startColumn: 16, + endColumn: 17, + }, + ]); }); test('Insert into table ', () => { diff --git a/test/parser/flink/suggestion/syntaxSuggestion.test.ts b/test/parser/flink/suggestion/syntaxSuggestion.test.ts index a017c641..4192a5ac 100644 --- a/test/parser/flink/suggestion/syntaxSuggestion.test.ts +++ b/test/parser/flink/suggestion/syntaxSuggestion.test.ts @@ -32,7 +32,17 @@ describe('Flink SQL Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['cat']); + expect(suggestion?.wordRanges?.length).toBe(1); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'cat', + line: 1, + startIndex: 13, + endIndex: 15, + startColumn: 14, + endColumn: 17, + }, + ]); }); test('Select table', () => { diff --git a/test/parser/hive/suggestion/multipleStatement.test.ts b/test/parser/hive/suggestion/multipleStatement.test.ts index 0c1d36fe..223ab846 100644 --- a/test/parser/hive/suggestion/multipleStatement.test.ts +++ b/test/parser/hive/suggestion/multipleStatement.test.ts @@ -36,7 +36,25 @@ describe('HiveSQL Multiple Statements Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.']); + expect(suggestion?.wordRanges?.length).toBe(2); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 9, + startIndex: 272, + endIndex: 273, + startColumn: 14, + endColumn: 16, + }, + { + text: '.', + line: 9, + startIndex: 274, + endIndex: 274, + startColumn: 16, + endColumn: 17, + }, + ]); }); test('Insert into table ', () => { diff --git a/test/parser/hive/suggestion/syntaxSuggestion.test.ts b/test/parser/hive/suggestion/syntaxSuggestion.test.ts index f2066a8f..541464aa 100644 --- a/test/parser/hive/suggestion/syntaxSuggestion.test.ts +++ b/test/parser/hive/suggestion/syntaxSuggestion.test.ts @@ -32,7 +32,33 @@ describe('Hive SQL Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.', 'tb']); + expect(suggestion?.wordRanges?.length).toBe(3); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 1, + startIndex: 12, + endIndex: 13, + startColumn: 13, + endColumn: 15, + }, + { + text: '.', + line: 1, + startIndex: 14, + endIndex: 14, + startColumn: 15, + endColumn: 16, + }, + { + text: 'tb', + line: 1, + startIndex: 15, + endIndex: 16, + startColumn: 16, + endColumn: 18, + }, + ]); }); test('Select table ', () => { diff --git a/test/parser/impala/suggestion/multipleStatement.test.ts b/test/parser/impala/suggestion/multipleStatement.test.ts index 115d2012..d9a00b02 100644 --- a/test/parser/impala/suggestion/multipleStatement.test.ts +++ b/test/parser/impala/suggestion/multipleStatement.test.ts @@ -36,7 +36,25 @@ describe('ImpalaSQL Multiple Statements Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.']); + expect(suggestion?.wordRanges?.length).toBe(2); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 9, + startIndex: 203, + endIndex: 204, + startColumn: 14, + endColumn: 16, + }, + { + text: '.', + line: 9, + startIndex: 205, + endIndex: 205, + startColumn: 16, + endColumn: 17, + }, + ]); }); test('Insert into table ', () => { diff --git a/test/parser/impala/suggestion/syntaxSuggestion.test.ts b/test/parser/impala/suggestion/syntaxSuggestion.test.ts index 08629187..02cf8d4d 100644 --- a/test/parser/impala/suggestion/syntaxSuggestion.test.ts +++ b/test/parser/impala/suggestion/syntaxSuggestion.test.ts @@ -26,7 +26,33 @@ describe('Impala SQL Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['cat', '.', 'a']); + expect(suggestion?.wordRanges?.length).toBe(3); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'cat', + line: 1, + startIndex: 14, + endIndex: 16, + startColumn: 15, + endColumn: 18, + }, + { + text: '.', + line: 1, + startIndex: 17, + endIndex: 17, + startColumn: 18, + endColumn: 19, + }, + { + text: 'a', + line: 1, + startIndex: 18, + endIndex: 18, + startColumn: 19, + endColumn: 20, + }, + ]); }); test('Function call', () => { diff --git a/test/parser/mysql/suggestion/multipleStatement.test.ts b/test/parser/mysql/suggestion/multipleStatement.test.ts index 48efa713..d08099b8 100644 --- a/test/parser/mysql/suggestion/multipleStatement.test.ts +++ b/test/parser/mysql/suggestion/multipleStatement.test.ts @@ -36,7 +36,25 @@ describe('MySQL Multiple Statements Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.']); + expect(suggestion?.wordRanges?.length).toBe(2); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 9, + startIndex: 306, + endIndex: 307, + startColumn: 14, + endColumn: 16, + }, + { + text: '.', + line: 9, + startIndex: 308, + endIndex: 308, + startColumn: 16, + endColumn: 17, + }, + ]); }); test('Insert into table ', () => { diff --git a/test/parser/mysql/suggestion/syntaxSuggestion.test.ts b/test/parser/mysql/suggestion/syntaxSuggestion.test.ts index 03ac43a9..a8977e61 100644 --- a/test/parser/mysql/suggestion/syntaxSuggestion.test.ts +++ b/test/parser/mysql/suggestion/syntaxSuggestion.test.ts @@ -30,7 +30,33 @@ describe('MySQL Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.', 'tb']); + expect(suggestion?.wordRanges?.length).toBe(3); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 1, + startIndex: 12, + endIndex: 13, + startColumn: 13, + endColumn: 15, + }, + { + text: '.', + line: 1, + startIndex: 14, + endIndex: 14, + startColumn: 15, + endColumn: 16, + }, + { + text: 'tb', + line: 1, + startIndex: 15, + endIndex: 16, + startColumn: 16, + endColumn: 18, + }, + ]); }); test('Select table ', () => { diff --git a/test/parser/postgresql/suggestion/multipleStatement.test.ts b/test/parser/postgresql/suggestion/multipleStatement.test.ts index 48e199a8..a688db41 100644 --- a/test/parser/postgresql/suggestion/multipleStatement.test.ts +++ b/test/parser/postgresql/suggestion/multipleStatement.test.ts @@ -69,6 +69,24 @@ describe('PgSQL Multiple Statements Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.']); + expect(suggestion?.wordRanges?.length).toBe(2); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 21, + startIndex: 682, + endIndex: 683, + startColumn: 62, + endColumn: 64, + }, + { + text: '.', + line: 21, + startIndex: 684, + endIndex: 684, + startColumn: 64, + endColumn: 65, + }, + ]); }); }); diff --git a/test/parser/postgresql/suggestion/suggestionWithEntity.test.ts b/test/parser/postgresql/suggestion/suggestionWithEntity.test.ts index 89ae26a6..a80adff3 100644 --- a/test/parser/postgresql/suggestion/suggestionWithEntity.test.ts +++ b/test/parser/postgresql/suggestion/suggestionWithEntity.test.ts @@ -166,7 +166,17 @@ describe('PostgreSql Syntax Suggestion with collect entity', () => { (syn) => syn.syntaxContextType === EntityContextType.COLUMN ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['a_column']); + expect(suggestion?.wordRanges?.length).toBe(1); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'a_column', + line: 13, + startIndex: 399, + endIndex: 406, + startColumn: 27, + endColumn: 35, + }, + ]); const entities = postgre.getAllEntities(sql, pos); expect(entities.length).toBe(1); diff --git a/test/parser/postgresql/suggestion/syntaxSuggestion.test.ts b/test/parser/postgresql/suggestion/syntaxSuggestion.test.ts index 664b2b52..3a7f2266 100644 --- a/test/parser/postgresql/suggestion/syntaxSuggestion.test.ts +++ b/test/parser/postgresql/suggestion/syntaxSuggestion.test.ts @@ -32,7 +32,33 @@ describe('Postgre SQL Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.', 'tb']); + expect(suggestion?.wordRanges?.length).toBe(3); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 3, + startIndex: 88, + endIndex: 89, + startColumn: 13, + endColumn: 15, + }, + { + text: '.', + line: 3, + startIndex: 90, + endIndex: 90, + startColumn: 15, + endColumn: 16, + }, + { + text: 'tb', + line: 3, + startIndex: 91, + endIndex: 92, + startColumn: 16, + endColumn: 18, + }, + ]); }); test('Alter table ', () => { diff --git a/test/parser/spark/suggestion/multipleStatement.test.ts b/test/parser/spark/suggestion/multipleStatement.test.ts index a65b020e..04ee292f 100644 --- a/test/parser/spark/suggestion/multipleStatement.test.ts +++ b/test/parser/spark/suggestion/multipleStatement.test.ts @@ -36,7 +36,25 @@ describe('SparkSQL Multiple Statements Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.']); + expect(suggestion?.wordRanges?.length).toBe(2); + expect(suggestion?.wordRanges).toEqual([ + { + endColumn: 17, + endIndex: 258, + line: 9, + startColumn: 15, + startIndex: 257, + text: 'db', + }, + { + endColumn: 18, + endIndex: 259, + line: 9, + startColumn: 17, + startIndex: 259, + text: '.', + }, + ]); }); test('Insert into table ', () => { diff --git a/test/parser/spark/suggestion/syntaxSuggestion.test.ts b/test/parser/spark/suggestion/syntaxSuggestion.test.ts index 2ed6c655..e6a2e85d 100644 --- a/test/parser/spark/suggestion/syntaxSuggestion.test.ts +++ b/test/parser/spark/suggestion/syntaxSuggestion.test.ts @@ -32,7 +32,33 @@ describe('Spark SQL Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.', 'tb']); + expect(suggestion?.wordRanges?.length).toBe(3); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 1, + startIndex: 12, + endIndex: 13, + startColumn: 13, + endColumn: 15, + }, + { + text: '.', + line: 1, + startIndex: 14, + endIndex: 14, + startColumn: 15, + endColumn: 16, + }, + { + text: 'tb', + line: 1, + startIndex: 15, + endIndex: 16, + startColumn: 16, + endColumn: 18, + }, + ]); }); test('Select table ', () => { diff --git a/test/parser/trino/suggestion/multipleStatement.test.ts b/test/parser/trino/suggestion/multipleStatement.test.ts index a224030d..1d8c1fb9 100644 --- a/test/parser/trino/suggestion/multipleStatement.test.ts +++ b/test/parser/trino/suggestion/multipleStatement.test.ts @@ -36,7 +36,25 @@ describe('TrinoSQL Multiple Statements Syntax Suggestion', () => { ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.']); + expect(suggestion?.wordRanges?.length).toBe(2); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 9, + startIndex: 137, + endIndex: 138, + startColumn: 17, + endColumn: 19, + }, + { + text: '.', + line: 9, + startIndex: 139, + endIndex: 139, + startColumn: 19, + endColumn: 20, + }, + ]); }); test('Insert into table ', () => { diff --git a/test/parser/trino/suggestion/syntaxSuggestion.test.ts b/test/parser/trino/suggestion/syntaxSuggestion.test.ts index 74bbd084..2d51fe0e 100644 --- a/test/parser/trino/suggestion/syntaxSuggestion.test.ts +++ b/test/parser/trino/suggestion/syntaxSuggestion.test.ts @@ -32,7 +32,33 @@ describe('Trino SQL Syntax Suggestion', () => { (syn) => syn.syntaxContextType === EntityContextType.TABLE ); expect(suggestion).not.toBeUndefined(); - expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.', 'tb']); + expect(suggestion?.wordRanges?.length).toBe(3); + expect(suggestion?.wordRanges).toEqual([ + { + text: 'db', + line: 1, + startIndex: 12, + endIndex: 13, + startColumn: 13, + endColumn: 15, + }, + { + text: '.', + line: 1, + startIndex: 14, + endIndex: 14, + startColumn: 15, + endColumn: 16, + }, + { + text: 'tb', + line: 1, + startIndex: 15, + endIndex: 16, + startColumn: 16, + endColumn: 18, + }, + ]); }); test('Select table ', () => {