lava-oushudb-dt-sql-parser/test/parser/pgsql/errorStrategy.test.ts

import PgSQL from '../../../src/parser/pgsql';
import { PgSqlSplitListener } from '../../../src/parser/pgsql';
import { PostgreSQLParserListener } from '../../../src/lib/pgsql/PostgreSQLParserListener';

const validSQL1 = `INSERT INTO country_page_view
VALUES ('Chinese', 'mumiao', 18),
    ('Amercian', 'georage', 22);`;
const validSQL2 = 'SELECT * FROM tb;';
const inValidSQL = 'CREATE TABLE';

describe('PgSQL ErrorStrategy test', () => {
    const pgSQL = new PgSQL();

    // TODO: handle unexpected case
    // test('begin inValid', () => {
    //     const sql = [inValidSQL, validSQL1, validSQL2].join('\n');
    //     // parse with empty errorListener
    //     const parseTree = pgSQL.parse(sql, () => {});
    //     const splitListener = new PgSqlSplitListener();
    //     pgSQL.listen(splitListener as PostgreSQLParserListener, 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 = pgSQL.parse(sql, () => {});
        const splitListener = new PgSqlSplitListener();
        pgSQL.listen(splitListener as PostgreSQLParserListener, 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 = pgSQL.parse(sql, () => {});
        const splitListener = new PgSqlSplitListener();
        pgSQL.listen(splitListener as PostgreSQLParserListener, parseTree);

        splitListener.statementsContext.map((item, index) => {
            if (index !== 0 && index !== 1) {
                expect(item.exception).not.toBe(undefined);
            } else {
                expect(item.exception).toBe(undefined);
            }
        });
    });
});