a99721162b
* feat: add text and word utils * feat: add entity collector class * refactor: rename SyntaxContextType to EntityContextType * refactor: improve EntityCollector * feat: improve mysql parser grammar * feat: add mysql entity collector * test: mysql entity collector tests * feat: remove useless method * feat: improve spark grammar file * feat: add spark entity collector * test: spark entity collector unit tests * feat: remove useless code * feat: add queryStatement label * feat: add crateDatabaseStmt * feat: add trino entity collector * feat: rename trinosql to trino * test: trino collect entity unit tests * test: fix spark test * feat(impala): support impale entity collector (#256) * Feat/collect entity hive (#263) * feat(hive): support hive collect entity * feat(hive): update tableAllColumns * feat: replace antlr4ts with antlr4ng * feat(pgsql): pgsql collect entity (#268) * feat(pgsql): pgsql collect entity * feat(pgsql): optimize some name --------- Co-authored-by: zhaoge <> * feat: get word text by token.text * feat: supprt collect db/function and add splitListener (#270) * feat: supprt collect db/function and add splitListner * feat: remove SplitListener interface in baseParser to use SplitListener in root * fix(mysql): fix show create xxx not celloct as createXXXEntity type * test: fix pgsql unit tests * Feat/error recover predicate (#274) * feat: optimize pgsql grammar * feat: add sql parser base * feat: apply SQLParserBase * feat: add geAllEntities method * test: test collect table when missing column * feat: compose collect and suggestion (#276) * feat: mark stmt which contain caret * test: correct name of getAllEntities * test: remove misscolumn unit tests * test: add suggestionWithEntity tests * feat: flink collect entity (#277) * feat: improve flink sql parser * feat: support flink entity collector * test: flink entity collect unit test * feat: move combine entities to parent class --------- Co-authored-by: 霜序 <976060700@qq.com> Co-authored-by: XCynthia <942884029@qq.com>
83 lines
2.8 KiB
TypeScript
83 lines
2.8 KiB
TypeScript
import TrinoSQL from 'src/parser/trino';
|
|
import { readSQL } from 'test/helper';
|
|
|
|
const features = {
|
|
select: readSQL(__dirname, 'select.sql'),
|
|
selectWithClause: readSQL(__dirname, 'select_with_clause.sql'),
|
|
selectWithSetOperations: readSQL(__dirname, 'select_with_set_operations.sql'),
|
|
selectWithSubQueries: readSQL(__dirname, 'select_with_sub_queries.sql'),
|
|
selectWithTableSample: readSQL(__dirname, 'select_with_table_sample.sql'),
|
|
selectWithRowType: readSQL(__dirname, 'select_with_row_type.sql'),
|
|
selectWithOffset: readSQL(__dirname, 'select_with_offset.sql'),
|
|
selectWithJoin: readSQL(__dirname, 'select_with_join.sql'),
|
|
selectWithFetch: readSQL(__dirname, 'select_with_fetch.sql'),
|
|
selectWithUNNEST: readSQL(__dirname, 'select_with_ unnest.sql'),
|
|
selectWithExists: readSQL(__dirname, 'select_with_exists.sql'),
|
|
selectWithUnion: readSQL(__dirname, 'select_with_union.sql'),
|
|
};
|
|
|
|
describe('TrinoSQL Select Statements Syntax Tests', () => {
|
|
const parser = new TrinoSQL();
|
|
features.select.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithClause.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithSetOperations.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
|
|
features.selectWithSubQueries.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithTableSample.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithRowType.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithOffset.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithJoin.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithFetch.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithUNNEST.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithExists.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
features.selectWithUnion.forEach((sql) => {
|
|
it(sql, () => {
|
|
expect(parser.validate(sql).length).toBe(0);
|
|
});
|
|
});
|
|
});
|