import { MySQL, MySqlParserVisitor, AbstractParseTreeVisitor } from '../../filters'; describe('MySQL Visitor Tests', () => { const expectTableName = 'user1'; const sql = `select id,name,sex from ${expectTableName};`; const parser = new MySQL(); const parseTree = parser.parse(sql, (error) => { console.log('Parse error:', error); }); test('Visitor visitTableName', () => { let result = ''; class MyVisitor extends AbstractParseTreeVisitor implements MySqlParserVisitor { protected defaultResult() { return result; } visitTableName = (ctx): void => { result = ctx.text.toLowerCase(); }; } const visitor = new MyVisitor(); visitor.visit(parseTree); expect(result).toBe(expectTableName); }); });