amis2/packages/amis-formula/__tests__/lexer.test.ts

import {lexer as createLexer, parse} from '../src';

function getTokens(input: string, options?: any) {
  const lexer = createLexer(input, options);
  const tokens: Array<any> = [];

  while (true) {
    const token = lexer.next();
    if (token) {
      tokens.push(token);

      if (token.type === 'EOF') {
        break;
      }
    } else {
      break;
    }
  }

  return tokens.map(token => `<${token.type}> ${token.value}`);
}

test('lexer:simple', () => {
  expect(
    getTokens('expression result is ${a + b}', {
      evalMode: false
    })
  ).toMatchSnapshot();
});

test('lexer:filter', () => {
  expect(
    getTokens('\\$abc is ${abc | date: YYYY-MM-DD HH\\:mm\\:ss}', {
      evalMode: false
    })
  ).toMatchSnapshot();

  expect(
    getTokens('\\$abc is ${abc | isTrue : trueValue : falseValue}', {
      evalMode: false
    })
  ).toMatchSnapshot();
});

// test('lexer:test', () => {
//   console.log(getTokens("{a: 1, 'b': 2, `c`: 3, d: {}}", {evalMode: true}));
// });

test('lexer:exception', () => {
  expect(() =>
    getTokens('\\aabc is ', {
      evalMode: false
    })
  ).toThrow('Unexpected token a in 1:3');

  expect(() =>
    getTokens('${a | filter: \\x2}', {
      evalMode: false
    })
  ).toThrow('Unexpected token x in 1:17');
});