d2-ui/src/formula-editor/__tests__/formula-parser.spec.js

d2-ui

import { formulaParser } from '../formula-parser';

describe('Formula parser', () => {
    it('should be a function', () => {
        expect(typeof formulaParser).toBe('function');
    });

    it('should return an array when an empty formula is given', () => {
        expect(formulaParser('')).toEqual([]);
    });

    it('should return an empty array when no formula is given', () => {
        expect(formulaParser()).toEqual([]);
    });

    it('should return the expected parts for the passed formula', () => {
        expect(formulaParser('#{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}')).toEqual([
            {
                displaySubstitute: 'A',
                entityType: 'dataElement',
                value: '#{cYeuwXTCPkU.pq2XI5kz2BY}',
                index: 0,
            },
            {
                entityType: 'operator',
                value: '+',
                index: 26,
            },
            {
                displaySubstitute: 'B',
                entityType: 'dataElement',
                value: '#{cYeuwXTCPkU.PT59n8BQbqM}',
                index: 27,
            },
        ]);
    });

    it('should return the expected part for a simple indicator expression', () => {
        expect(formulaParser('#{fbfJHSPpUQD}')).toEqual([{
            displaySubstitute: 'A',
            entityType: 'dataElement',
            value: '#{fbfJHSPpUQD}',
            index: 0,
        }]);
    });

    it('should go to AA style identifiers when the alphabet is used up', () => {
        const parsedFormula = formulaParser(`
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
            #{cYeuwXTCPkU.pq2XI5kz2BY}+#{cYeuwXTCPkU.PT59n8BQbqM}
        `);

        expect(parsedFormula[parsedFormula.length - 1]).toEqual({
            displaySubstitute: 'AD',
            entityType: 'dataElement',
            value: '#{cYeuwXTCPkU.PT59n8BQbqM}',
            index: 964,
        });
    });

    it('should include brackets when they are in the formula', () => {
        const parsedFormula = formulaParser('(#{cYeuwXTCPkU.pq2XI5kz2BY} + 2) - #{cYeuwXTCPkU.PT59n8BQbqM}');

        expect(parsedFormula[0]).toEqual({
            entityType: 'bracket',
            value: '(',
            index: 0,
        });
    });
});