lava-oushudb-dt-sql-parser/test/parser/flinksql/suggestion/tokenSuggestion.test.ts

import fs from 'fs';
import path from 'path';
import { CaretPosition } from '../../../../src/parser/common/basic-parser-types';
import FlinkSQL from '../../../../src/parser/flinksql'

const tokenSql = fs.readFileSync(path.join(__dirname, 'fixtures', 'tokenSuggestion.sql'), 'utf-8');

describe('Flink SQL Token Suggestion', () => {
    const parser = new FlinkSQL();

    test('Use Statement ', () => {
        const pos: CaretPosition = {
            lineNumber: 3,
            column: 5
        }
        const suggestion = parser.getSuggestionAtCaretPosition(tokenSql, pos)?.keywords;
   
        expect(suggestion)
            .toEqual([ 'MODULES', 'CATALOG' ])
    })

    test('Create Statement ', () => {
        const pos: CaretPosition = {
            lineNumber: 5,
            column: 8
        }
        const suggestion = parser.getSuggestionAtCaretPosition(tokenSql, pos)?.keywords;
   
        expect(suggestion)
            .toEqual([ 'CATALOG', 'FUNCTION', 'TEMPORARY', 'VIEW', 'DATABASE', 'TABLE' ])
    })

    test('Show Statement ', () => {
        const pos: CaretPosition = {
            lineNumber: 7,
            column: 6
        }
        const suggestion = parser.getSuggestionAtCaretPosition(tokenSql, pos)?.keywords;
   
        expect(suggestion)
            .toEqual([
                'MODULES',
                'FULL',
                'FUNCTIONS',
                'USER',
                'CREATE',
                'COLUMNS',
                'TABLES',
                'CURRENT',
                'CATALOGS',
                'DATABASES',
                'JARS',
                'VIEWS'
            ])
    })

})