feat: add ErrorStrategy(#230)
* refactor: rename errorHandler to errorListener * feat: add ErrorStrategy to mark context exceptions * test: errorStrategy unit tests
This commit is contained in:
		
							
								
								
									
										62
									
								
								test/parser/flinksql/errorStrategy.test.ts
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										62
									
								
								test/parser/flinksql/errorStrategy.test.ts
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,62 @@
 | 
			
		||||
import FlinkSQL from '../../../src/parser/flinksql';
 | 
			
		||||
import { FlinkSqlSplitListener } from '../../../src/parser/flinksql';
 | 
			
		||||
import { FlinkSqlParserListener } from '../../../src/lib/flinksql/FlinkSqlParserListener';
 | 
			
		||||
 | 
			
		||||
const validSQL1 = `INSERT INTO country_page_view
 | 
			
		||||
VALUES ('Chinese', 'mumiao', 18),
 | 
			
		||||
    ('Amercian', 'georage', 22);`;
 | 
			
		||||
const validSQL2 = 'SELECT * FROM tb;';
 | 
			
		||||
const inValidSQL = 'CREATE TABLE VALUES;';
 | 
			
		||||
 | 
			
		||||
describe('FlinkSQL ErrorStrategy test', () => {
 | 
			
		||||
    const flinkSQL = new FlinkSQL();
 | 
			
		||||
    test('begin inValid', () => {
 | 
			
		||||
        const sql = [inValidSQL, validSQL1, validSQL2].join('\n');
 | 
			
		||||
        // parse with empty errorListener
 | 
			
		||||
        const parseTree = flinkSQL.parse(sql, () => {});
 | 
			
		||||
        const splitListener = new FlinkSqlSplitListener();
 | 
			
		||||
        flinkSQL.listen(splitListener as FlinkSqlParserListener, parseTree);
 | 
			
		||||
 | 
			
		||||
        const statementCount = splitListener.statementsContext.length;
 | 
			
		||||
        splitListener.statementsContext.map((item, index) => {
 | 
			
		||||
            if (index !== statementCount - 1 && index !== statementCount - 2) {
 | 
			
		||||
                expect(item.exception).not.toBe(undefined);
 | 
			
		||||
            } else {
 | 
			
		||||
                expect(item.exception).toBe(undefined);
 | 
			
		||||
            }
 | 
			
		||||
        });
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    test('middle inValid', () => {
 | 
			
		||||
        const sql = [validSQL1, inValidSQL, validSQL2].join('\n');
 | 
			
		||||
        // parse with empty errorListener
 | 
			
		||||
        const parseTree = flinkSQL.parse(sql, () => {});
 | 
			
		||||
        const splitListener = new FlinkSqlSplitListener();
 | 
			
		||||
        flinkSQL.listen(splitListener as FlinkSqlParserListener, parseTree);
 | 
			
		||||
 | 
			
		||||
        const statementCount = splitListener.statementsContext.length;
 | 
			
		||||
        splitListener.statementsContext.map((item, index) => {
 | 
			
		||||
            if (index !== statementCount - 1 && index !== 0) {
 | 
			
		||||
                expect(item.exception).not.toBe(undefined);
 | 
			
		||||
            } else {
 | 
			
		||||
                expect(item.exception).toBe(undefined);
 | 
			
		||||
            }
 | 
			
		||||
        });
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    test('end inValid', () => {
 | 
			
		||||
        const sql = [validSQL1, validSQL2, inValidSQL].join('\n');
 | 
			
		||||
        // parse with empty errorListener
 | 
			
		||||
        const parseTree = flinkSQL.parse(sql, () => {});
 | 
			
		||||
        const splitListener = new FlinkSqlSplitListener();
 | 
			
		||||
        flinkSQL.listen(splitListener as FlinkSqlParserListener, parseTree);
 | 
			
		||||
 | 
			
		||||
        splitListener.statementsContext.map((item, index) => {
 | 
			
		||||
            if (index !== 0 && index !== 1) {
 | 
			
		||||
                expect(item.exception).not.toBe(undefined);
 | 
			
		||||
            } else {
 | 
			
		||||
                expect(item.exception).toBe(undefined);
 | 
			
		||||
            }
 | 
			
		||||
        });
 | 
			
		||||
    });
 | 
			
		||||
});
 | 
			
		||||
		Reference in New Issue
	
	Block a user