bb0fad1dbe
* refactor: rename flinksql to flink * refactor: rename pgsql to postgresql * refactor: rename trinosql to trino * refactor: replace all default exports with named export * refactor: rename basicParser to basicSQL * refactor: rename basic-parser-types to types * refactor: replace arrow func with plain func
75 lines
2.6 KiB
TypeScript
75 lines
2.6 KiB
TypeScript
import fs from 'fs';
|
|
import path from 'path';
|
|
import { PostgreSQL } from 'src/parser/postgresql';
|
|
import { CaretPosition, EntityContextType } from 'src/parser/common/types';
|
|
|
|
const syntaxSql = fs.readFileSync(
|
|
path.join(__dirname, 'fixtures', 'multipleStatement.sql'),
|
|
'utf-8'
|
|
);
|
|
|
|
describe('PgSQL Multiple Statements Syntax Suggestion', () => {
|
|
const postgresql = new PostgreSQL();
|
|
|
|
test('Create table ', () => {
|
|
const pos: CaretPosition = {
|
|
lineNumber: 1,
|
|
column: 14,
|
|
};
|
|
const syntaxes = postgresql.getSuggestionAtCaretPosition(syntaxSql, pos)?.syntax;
|
|
const suggestion = syntaxes?.find(
|
|
(syn) => syn.syntaxContextType === EntityContextType.TABLE_CREATE
|
|
);
|
|
|
|
expect(suggestion).not.toBeUndefined();
|
|
expect(suggestion?.wordRanges.map((token) => token.text)).toEqual([]);
|
|
});
|
|
|
|
test('Select from table or view', () => {
|
|
const pos: CaretPosition = {
|
|
lineNumber: 9,
|
|
column: 18,
|
|
};
|
|
const syntaxes = postgresql.getSuggestionAtCaretPosition(syntaxSql, pos)?.syntax;
|
|
const suggestion = syntaxes?.find(
|
|
(syn) => syn.syntaxContextType === EntityContextType.TABLE
|
|
);
|
|
const suggestionVw = syntaxes?.find(
|
|
(syn) => syn.syntaxContextType === EntityContextType.VIEW
|
|
);
|
|
|
|
expect(suggestion).not.toBeUndefined();
|
|
expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.']);
|
|
expect(suggestionVw).not.toBeUndefined();
|
|
expect(suggestionVw?.wordRanges.map((token) => token.text)).toEqual(['db', '.']);
|
|
});
|
|
|
|
test('Insert into table ', () => {
|
|
const pos: CaretPosition = {
|
|
lineNumber: 15,
|
|
column: 13,
|
|
};
|
|
const syntaxes = postgresql.getSuggestionAtCaretPosition(syntaxSql, pos)?.syntax;
|
|
const suggestion = syntaxes?.find(
|
|
(syn) => syn.syntaxContextType === EntityContextType.TABLE
|
|
);
|
|
|
|
expect(suggestion).not.toBeUndefined();
|
|
expect(suggestion?.wordRanges.map((token) => token.text)).toEqual([]);
|
|
});
|
|
|
|
test('Insert into select from table ', () => {
|
|
const pos: CaretPosition = {
|
|
lineNumber: 21,
|
|
column: 65,
|
|
};
|
|
const syntaxes = postgresql.getSuggestionAtCaretPosition(syntaxSql, pos)?.syntax;
|
|
const suggestion = syntaxes?.find(
|
|
(syn) => syn.syntaxContextType === EntityContextType.TABLE
|
|
);
|
|
|
|
expect(suggestion).not.toBeUndefined();
|
|
expect(suggestion?.wordRanges.map((token) => token.text)).toEqual(['db', '.']);
|
|
});
|
|
});
|