lava-oushudb-dt-sql-parser/test/parser/spark/lexer.test.ts

import SparkSQL from '../../../src/parser/spark';

describe('SparkSQL Lexer tests', () => {
    const parser = new SparkSQL();

    test('select id,name from user1;', () => {
        const sql = `select id,name from user1;`;
        const tokens = parser.getAllTokens(sql);
        expect(tokens.length).toBe(10);
    });

    test('SELECT * FROM t WHERE x = 1 AND y = 2;', () => {
        const sql = `SELECT * FROM t WHERE x = 1 AND y = 2;`;
        const tokens = parser.getAllTokens(sql);
        expect(tokens.length).toBe(24);
    });
});