amis/packages/amis-formula/__tests__/evalute.test.ts

import {evaluate, parse} from '../src';

test('evalute:simple', () => {
  expect(
    evaluate('a is ${a}', {
      a: 123
    })
  ).toBe('a is 123');
});

test('evalute:filter', () => {
  expect(
    evaluate(
      'a is ${a | abc}',
      {
        a: 123
      },
      {
        filters: {
          abc(input: any) {
            return `${input}456`;
          }
        }
      }
    )
  ).toBe('a is 123456');

  expect(
    evaluate(
      'a is ${a | concat:233}',
      {
        a: 123
      },
      {
        filters: {
          concat(input: any, arg: string) {
            return `${input}${arg}`;
          }
        }
      }
    )
  ).toBe('a is 123233');

  expect(
    evaluate(
      'a is ${concat(a, a)}',
      {
        a: 123
      },
      {
        filters: {
          concat(input: any, arg: string) {
            return `${input}${arg}`;
          }
        }
      }
    )
  ).toBe('a is 123123');
});

test('evalute:filter2', () => {
  expect(
    evaluate(
      'a is ${[1, 2, 3] | concat:4 | join}',
      {},
      {
        filters: {
          concat(input: any, ...args: Array<any>) {
            return input.concat.apply(input, args);
          },
          join(input: any) {
            return input.join(',');
          }
        }
      }
    )
  ).toBe('a is 1,2,3,4');
});

test('evalute:filter3', () => {
  expect(
    evaluate(
      'a is ${[1, 2, 3] | concat:"4" | join}',
      {},
      {
        filters: {
          concat(input: any, ...args: Array<any>) {
            return input.concat.apply(input, args);
          },
          join(input: any) {
            return input.join(',');
          }
        }
      }
    )
  ).toBe('a is 1,2,3,4');
});

test('evalute:filter4', () => {
  expect(
    evaluate(
      'a is ${[1, 2, 3] | concat:${a + 3} | join}',
      {
        a: 4
      },
      {
        filters: {
          concat(input: any, ...args: Array<any>) {
            return input.concat.apply(input, args);
          },
          join(input: any) {
            return input.join(',');
          }
        }
      }
    )
  ).toBe('a is 1,2,3,7');
});

test('evalute:keywords', () => {
  expect(evaluate('a is ${JOIN([1, 2, 3], "|")}', {})).toBe('a is 1|2|3');
});

test('evalute:oldVariable', () => {
  expect(
    evaluate('a is $a', {
      a: 4
    })
  ).toBe('a is 4');

  expect(
    evaluate('b is $b', {
      a: 4
    })
  ).toBe('b is ');

  expect(
    evaluate('a.b is $a.b', {
      a: {
        b: 233
      }
    })
  ).toBe('a.b is 233');
});

test('evalute:ariable2', () => {
  expect(
    evaluate('a is $$', {
      a: 4
    })
  ).toBe('a is [object Object]');
});

test('evalute:ariable3', () => {
  expect(
    evaluate(
      '$$',
      {
        a: 4
      },
      {
        defaultFilter: 'raw'
      }
    )
  ).toMatchObject({
    a: 4
  });
});

test('evalute:object-variable', () => {
  const data = {
    key: 'x',
    obj: {
      x: 1,
      y: 2
    }
  };

  expect(evaluate('a is ${obj.x}', data)).toBe('a is 1');
  expect(evaluate('a is ${obj[x]}', data)).toBe('a is 1');
  expect(evaluate('a is ${obj[`x`]}', data)).toBe('a is 1');
  expect(evaluate('a is ${obj["x"]}', data)).toBe('a is 1');
  expect(evaluate('a is ${obj[key]}', data)).toBe('a is 1');
  expect(evaluate('a is ${obj[`${key}`]}', data)).toBe('a is 1');
  expect(evaluate('a is ${obj[${key}]}', data)).toBe('a is 1');
});

test('evalute:literal-variable', () => {
  const data = {
    key: 'x',
    index: 0,
    obj: {
      x: 1,
      y: 2
    }
  };

  expect(evaluate('a is ${({x: 1})["x"]}', data)).toBe('a is 1');
  expect(evaluate('a is ${({x: 1}).x}', data)).toBe('a is 1');
  expect(evaluate('a is ${(["a", "b"])[index]}', data)).toBe('a is a');
  expect(evaluate('a is ${(["a", "b"])[1]}', data)).toBe('a is b');
  expect(evaluate('a is ${(["a", "b"]).0}', data)).toBe('a is a');
});

test('evalute:tempalte', () => {
  const data = {
    key: 'x'
  };

  expect(evaluate('abc${`11${3}22`}xyz', data)).toBe('abc11322xyz');
  expect(evaluate('abc${`${3}22`}xyz', data)).toBe('abc322xyz');
  expect(evaluate('abc${`11${3}`}xyz', data)).toBe('abc113xyz');
  expect(evaluate('abc${`${3}`}xyz', data)).toBe('abc3xyz');
  expect(evaluate('abc${`${key}`}xyz', data)).toBe('abcxxyz');
});

test('evalute:literal', () => {
  const data = {
    dynamicKey: 'alpha'
  };

  expect(
    evaluate('${{a: 1, 0: 2, "3": 3}}', data, {
      defaultFilter: 'raw'
    })
  ).toMatchObject({
    a: 1,
    0: 2,
    3: 3
  });

  expect(
    evaluate('${{a: 1, 0: 2, "3": 3, [`4`]: 4}}', data, {
      defaultFilter: 'raw'
    })
  ).toMatchObject({
    a: 1,
    0: 2,
    3: 3,
    4: 4
  });

  expect(
    evaluate('${{a: 1, 0: 2, "3": 3, [`${dynamicKey}233`]: 4}}', data, {
      defaultFilter: 'raw'
    })
  ).toMatchObject({
    a: 1,
    0: 2,
    3: 3,
    alpha233: 4
  });

  expect(
    evaluate('${[1, 2, `2${dynamicKey}2`, {a: 1, 0: 2, [`2`]: "3"}]}', data, {
      defaultFilter: 'raw'
    })
  ).toMatchObject([1, 2, `2alpha2`, {a: 1, 0: 2, [`2`]: '3'}]);
});

test('evalute:variableName', () => {
  const data = {
    'a_b': 'c',
    '222': 10222,
    '222_221': 233,
    '222_abcde': 'abcde',
    '222-221': 333
  };

  expect(evaluate('${a_b}', data)).toBe('c');
  expect(evaluate('${222}', data)).toBe(222);
  expect(evaluate('${222_221}', data)).toBe('233');
  expect(evaluate('${222-221}', data)).toBe(1);
  expect(evaluate('${222_abcde}', data)).toBe('abcde');
  expect(
    evaluate('${&["222-221"]}', data, {
      defaultFilter: 'raw'
    })
  ).toBe(333);
  expect(
    evaluate('222', data, {
      variableMode: true
    })
  ).toBe(10222);
});

test('evalute:3-1', () => {
  const data = {};

  expect(evaluate('${3-1}', data)).toBe(2);
  expect(evaluate('${-1 + 2.5 + 3}', data)).toBe(4.5);
  expect(evaluate('${-1 + -1}', data)).toBe(-2);
  expect(evaluate('${3 * -1}', data)).toBe(-3);

  expect(evaluate('${3 + +1}', data)).toBe(4);
});

test('evalate:0.1+0.2', () => {
  expect(evaluate('${0.1 + 0.2}', {})).toBe(0.3);
});

test('evalute:variable:com.xxx.xx', () => {
  const data = {
    'com.xxx.xx': 'abc',
    'com xxx%xx': 'cde',
    'com[xxx]': 'eee'
  };

  expect(evaluate('${com\\.xxx\\.xx}', data)).toBe('abc');
  expect(evaluate('${com\\ xxx\\%xx}', data)).toBe('cde');
  expect(evaluate('${com\\[xxx\\]}', data)).toBe('eee');
});

test('evalute:anonymous:function', () => {
  const data = {
    arr: [1, 2, 3],
    arr2: [
      {
        a: 1
      },
      {
        a: 2
      },
      {
        a: 3
      }
    ],
    outter: 4
  };

  expect(evaluate('${() => 233}', data)).toMatchObject({
    args: [],
    return: {type: 'literal', value: 233},
    type: 'anonymous_function'
  });

  expect(evaluate('${ARRAYMAP(arr, () => 1)}', data)).toMatchObject([1, 1, 1]);
  expect(evaluate('${ARRAYMAP(arr, item => item)}', data)).toMatchObject([
    1, 2, 3
  ]);
  expect(evaluate('${ARRAYMAP(arr, item => item * 2)}', data)).toMatchObject([
    2, 4, 6
  ]);
  expect(
    evaluate('${ARRAYMAP(arr2, (item, index) => `a${item.a}${index}`)}', data)
  ).toMatchObject(['a10', 'a21', 'a32']);
  expect(
    evaluate(
      '${ARRAYMAP(arr2, (item, index) => `a${item.a}${index}${outter}`)}',
      data
    )
  ).toMatchObject(['a104', 'a214', 'a324']);
  expect(
    evaluate(
      '${ARRAYMAP(arr2, (item, index) => {x: item.a, index: index})}',
      data
    )
  ).toMatchObject([
    {
      x: 1,
      index: 0
    },
    {
      x: 2,
      index: 1
    },
    {
      x: 3,
      index: 2
    }
  ]);
});

test('evalute:anonymous:function2', () => {
  const data = {
    arr: [1, 2, 3],
    arr2: [
      {
        x: 1,
        y: [
          {
            z: 1
          },
          {
            z: 1
          }
        ]
      },
      {
        x: 2,
        y: [
          {
            z: 2
          },
          {
            z: 2
          }
        ]
      }
    ]
  };

  expect(
    evaluate(
      '${ARRAYMAP(ARRAYMAP(arr, item => item * 2), item => item + 2)}',
      data
    )
  ).toMatchObject([4, 6, 8]);

  expect(
    evaluate('${ARRAYMAP(arr2, item => ARRAYMAP(item.y, i => i.z))}', data)
  ).toMatchObject([
    [1, 1],
    [2, 2]
  ]);
});

test('evalute:array:func', () => {
  const data = {
    arr1: [0, 1, false, 2, '', 3],
    arr2: ['a', 'b', 'c'],
    arr3: [1, 2, 3],
    arr4: [2, 4, 6],
    arr5: [
      {
        id: 1.1
      },
      {
        id: 2.2
      },
      {
        id: 1.1,
        name: 1.3
      }
    ],
    obj1: {
      p1: 'name',
      p2: 'age',
      p3: 'obj',
      p4: [
        {
          p41: 'Tom',
          p42: 'Jerry'
        },
        {
          p41: 'baidu',
          p42: 'amis'
        }
      ]
    }
  };

  expect(evaluate('${COMPACT(arr1)}', data)).toMatchObject([1, 2, 3]);

  expect(evaluate("${COMPACT([0, 1, false, 2, '', 3])}", data)).toMatchObject([
    1, 2, 3
  ]);

  expect(evaluate('${JOIN(arr2, "~")}', data)).toMatch('a~b~c');

  expect(evaluate('${SUM(arr3)}', data)).toBe(6);

  expect(evaluate('${AVG(arr4)}', data)).toBe(4);

  expect(evaluate('${MIN(arr4)}', data)).toBe(2);

  expect(evaluate('${MAX(arr4)}', data)).toBe(6);

  expect(evaluate('${CONCAT(arr3, arr4)}', data)).toMatchObject([
    1, 2, 3, 2, 4, 6
  ]);

  expect(evaluate('${CONCAT(arr, arr4)}', data)).toMatchObject([2, 4, 6]);

  expect(evaluate('${UNIQ(arr5)}', data)).toMatchObject(data.arr5);

  expect(evaluate('${UNIQ(arr5, "id")}', data)).toMatchObject([
    {id: 1.1},
    {id: 2.2}
  ]);

  expect(evaluate('${ARRAYFILTER(arr1, item => item)}', data)).toMatchObject([
    1, 2, 3
  ]);
  expect(
    evaluate('${ARRAYFILTER(arr1, item => item && item >=2)}', data)
  ).toMatchObject([2, 3]);

  expect(evaluate('${ARRAYFINDINDEX(arr3, item => item === 2)}', data)).toBe(1);

  expect(
    evaluate('${ARRAYFIND(arr5, item => item.name === 1.3)}', data)
  ).toMatchObject({
    id: 1.1,
    name: 1.3
  });

  expect(evaluate('${ARRAYSOME(arr5, item => item.name === 1.3)}', data)).toBe(
    true
  );

  expect(evaluate('${ARRAYEVERY(arr5, item => item.name === 1.3)}', data)).toBe(
    false
  );

  expect(evaluate('${ARRAYINCLUDES(arr1, false)}', data)).toBe(true);

  expect(evaluate('${GET(arr1, 2)}', data)).toBe(false);
  expect(evaluate('${GET(arr1, 6, "not-found")}', data)).toBe('not-found');
  expect(evaluate('${GET(arr5, "[2].name")}', data)).toBe(1.3);
  expect(evaluate('${GET(arr5, "2.name")}', data)).toBe(1.3);
  expect(evaluate('${GET(obj1, "p2")}', data)).toBe('age');
  expect(evaluate('${GET(obj1, "p4.1.p42")}', data)).toBe('amis');
  expect(evaluate('${GET(obj1, "p4[1].p42")}', data)).toBe('amis');

  expect(evaluate('${ENCODEJSON(obj1)}', data)).toBe(JSON.stringify(data.obj1));
  expect(
    evaluate('${DECODEJSON("{\\"name\\":\\"amis\\"}")}', data)
  ).toMatchObject(JSON.parse('{"name":"amis"}'));
});

test('evalute:ISTYPE', () => {
  const data = {
    a: 1,
    b: 'string',
    c: null,
    d: undefined,
    e: [1, 2],
    f: {a: 1, b: 2},
    g: new Date()
  };
  expect(evaluate('${ISTYPE(a, "number")}', data)).toBe(true);
  expect(evaluate('${ISTYPE(b, "number")}', data)).toBe(false);
  expect(evaluate('${ISTYPE(b, "string")}', data)).toBe(true);
  expect(evaluate('${ISTYPE(c, "nil")}', data)).toBe(true);
  expect(evaluate('${ISTYPE(d, "nil")}', data)).toBe(true);
  expect(evaluate('${ISTYPE(e, "array")}', data)).toBe(true);
  expect(evaluate('${ISTYPE(f, "array")}', data)).toBe(false);
  expect(evaluate('${ISTYPE(f, "plain-object")}', data)).toBe(true);
  expect(evaluate('${ISTYPE(g, "date")}', data)).toBe(true);
});

test('evalute:Math', () => {
  const data = {
    float: 0.5,
    integer1: 2,
    integer2: 4,
    negativeInteger: -2,
    array: [1, 2, 3],
    infinity: Infinity
  };

  expect(evaluate('${POW(integer1, integer2)}', data)).toBe(16);
  expect(evaluate('${POW(integer2, 0.5)}', data)).toBe(2);
  expect(evaluate('${POW(integer1, -2)}', data)).toBe(0.25);
  /** 计算结果不合法，则返回NaN */
  expect(evaluate('${POW(negativeInteger, 0.5)}', data)).toBe(NaN);
  /** 参数值不合法，则返回基数本身*/
  expect(evaluate('${POW(array, 2)}', data)).toBe(data.array);
  /** 测试Infinity */
  expect(evaluate('${POW(infinity, 2)}', data)).toBe(data.infinity);
  expect(evaluate('${POW(1, infinity)}', data)).toBe(NaN);
  expect(evaluate('${POW(2, infinity)}', data)).toBe(data.infinity);
});

test('evalute:namespace', () => {
  localStorage.setItem('a', '1');
  localStorage.setItem('b', '2');
  localStorage.setItem('c', '{"a": 1, "b": 2, "c": {"d": 4}}');
  localStorage.setItem('key', 'c');
  localStorage.setItem('spec-var-name', 'you are right');

  expect(evaluate('${ls: a}', {})).toBe(1);
  expect(evaluate('${ls: b}', {})).toBe(2);
  expect(evaluate('${ls: c}', {})).toMatchObject({a: 1, b: 2, c: {d: 4}});
  // 被认为是减操作
  expect(evaluate('${ls: spec-var-name}', {})).toBe(0);
  expect(evaluate('${ls: spec\\-var\\-name}', {})).toBe('you are right');
  expect(evaluate('${ls: &["spec-var-name"]}', {})).toBe('you are right');
  expect(evaluate('${ls: &["c"]["c"]}', {})).toMatchObject({d: 4});
  expect(evaluate('${ls: &["c"][key]}', {})).toMatchObject({d: 4});
});