7de192d486
* ci: add dependencies about lint tool * ci: replace eslint with prettier * ci: add husky, cz and commitlint * style: lint fix via prettier * ci: add prettier and check-types to github workflow '
33 lines
1.1 KiB
TypeScript
33 lines
1.1 KiB
TypeScript
import { AbstractParseTreeVisitor } from 'antlr4ts/tree/AbstractParseTreeVisitor';
|
|
import { SparkSqlParserVisitor } from '../../../src/lib/spark/SparkSqlParserVisitor';
|
|
import SparkSQL from '../../../src/parser/spark';
|
|
|
|
describe('Spark SQL Visitor Tests', () => {
|
|
const expectTableName = 'user1';
|
|
const sql = `select id,name,sex from ${expectTableName};`;
|
|
const parser = new SparkSQL();
|
|
|
|
const parserTree = parser.parse(sql, (error) => {
|
|
console.log('Parse error:', error);
|
|
});
|
|
|
|
test('Visitor visitRelationPrimary', () => {
|
|
class MyVisitor
|
|
extends AbstractParseTreeVisitor<any>
|
|
implements SparkSqlParserVisitor<any>
|
|
{
|
|
result: string = '';
|
|
protected defaultResult() {
|
|
return this.result;
|
|
}
|
|
visitRelationPrimary = (ctx): void => {
|
|
this.result = ctx.text.toLowerCase();
|
|
};
|
|
}
|
|
const visitor = new MyVisitor();
|
|
visitor.visit(parserTree);
|
|
|
|
expect(visitor.result).toBe(expectTableName);
|
|
});
|
|
});
|