test: add filter keywords test case

Author: JackWang032

src/parser/postgresql/index.ts

@@ -33,7 +33,13 @@ export class PostgreSQL extends BasicSQL<PostgreSqlLexer, ProgramContext, Postgr
     /**
      * The rules that keywords you don't want to be suggested.
      */
-    protected excludeKeywordRules = new Set([PostgreSqlParser.RULE_nonReservedWord]);
+    protected excludeKeywordRules = new Set([
+        PostgreSqlParser.RULE_nonReservedWord,
+        PostgreSqlParser.RULE_identifier,
+        PostgreSqlParser.RULE_reservedKeyword,
+        PostgreSqlParser.RULE_typeFuncNameKeyword,
+        PostgreSqlParser.RULE_colNameKeyword,
+    ]);
 
     protected preferredRules: Set<number> = new Set([
         PostgreSqlParser.RULE_tableNameCreate, // table name

test/parser/flink/suggestion/fixtures/tokenSuggestion.sql

@@ -12,3 +12,5 @@ CREATE TABLE IF NOT EXISTS
 CREATE TABLE tb (id 
 
 );
+
+SELECT id FROM   ;

test/parser/flink/suggestion/tokenSuggestion.test.ts

@@ -105,4 +105,16 @@ describe('Flink SQL Token Suggestion', () => {
         )?.keywords;
         expect(suggestion.length).not.toBe(0);
     });
+
+    test('filter unreserved keywords', () => {
+        const pos: CaretPosition = {
+            lineNumber: 16,
+            column: 17,
+        };
+        const suggestion = flink.getSuggestionAtCaretPosition(
+            commentOtherLine(tokenSql, pos.lineNumber),
+            pos
+        )?.keywords;
+        expect(suggestion).toMatchUnorderedArray(['TABLE', 'LATERAL', 'UNNEST']);
+    });
 });

test/parser/hive/suggestion/fixtures/tokenSuggestion.sql

@@ -24,3 +24,5 @@ CREATE TABLE IF NOT EXISTS
 CREATE TABLE tb (id 
 
 );
+
+SELECT id FROM  ;

test/parser/hive/suggestion/tokenSuggestion.test.ts

@@ -272,4 +272,16 @@ describe('Hive SQL Token Suggestion', () => {
         )?.keywords;
         expect(suggestion.length).not.toBe(0);
     });
+
+    test('filter unreserved keywords', () => {
+        const pos: CaretPosition = {
+            lineNumber: 28,
+            column: 16,
+        };
+        const suggestion = hive.getSuggestionAtCaretPosition(
+            commentOtherLine(tokenSql, pos.lineNumber),
+            pos
+        )?.keywords;
+        expect(suggestion).toMatchUnorderedArray(['TABLE', 'UNIQUEJOIN']);
+    });
 });

test/parser/impala/suggestion/fixtures/tokenSuggestion.sql

@@ -15,3 +15,5 @@ CREATE TABLE IF NOT EXISTS;
 CREATE TABLE tb (id 
 
 );
+
+SELECT id FROM   ;

test/parser/impala/suggestion/tokenSuggestion.test.ts

@@ -185,4 +185,16 @@ describe('Impala SQL Token Suggestion', () => {
         )?.keywords;
         expect(suggestion.length).not.toBe(0);
     });
+
+    test('filter unreserved keywords', () => {
+        const pos: CaretPosition = {
+            lineNumber: 19,
+            column: 17,
+        };
+        const suggestion = impala.getSuggestionAtCaretPosition(
+            commentOtherLine(tokenSql, pos.lineNumber),
+            pos
+        )?.keywords;
+        expect(suggestion).toMatchUnorderedArray(['LATERAL', 'UNNEST']);
+    });
 });

test/parser/mysql/suggestion/fixtures/tokenSuggestion.sql

@@ -20,3 +20,5 @@ CREATE TABLE IF NOT EXISTS
 CREATE TABLE tb (id 
 
 );
+
+SELECT id FROM   ;

test/parser/mysql/suggestion/tokenSuggestion.test.ts

@@ -297,4 +297,16 @@ describe('MySQL Token Suggestion', () => {
         )?.keywords;
         expect(suggestion.length).not.toBe(0);
     });
+
+    test('filter unreserved keywords', () => {
+        const pos: CaretPosition = {
+            lineNumber: 24,
+            column: 17,
+        };
+        const suggestion = mysql.getSuggestionAtCaretPosition(
+            commentOtherLine(tokenSql, pos.lineNumber),
+            pos
+        )?.keywords;
+        expect(suggestion).toMatchUnorderedArray(['JSON_TABLE', 'LATERAL', 'SELECT']);
+    });
 });

test/parser/postgresql/suggestion/fixtures/tokenSuggestion.sql

@@ -13,3 +13,5 @@ CREATE TABLE IF NOT EXISTS;
 CREATE TABLE tb (id 
 
 );
+
+SELECT id as  FROM a1