mermaid/packages/parser/tests/pie.test.ts

import { describe, expect, it } from 'vitest';

import { Pie } from '../src/language/index.js';
import { expectNoErrorsOrAlternatives, pieParse as parse } from './test-util.js';

describe('pie', () => {
  describe('should handle pie definition with or without showData', () => {
    it.each([
      `pie`,
      `  pie  `,
      `\tpie\t`,
      `
    \tpie
    `,
    ])('should handle regular pie', (context: string) => {
      const result = parse(context);
      expectNoErrorsOrAlternatives(result);
      expect(result.value.$type).toBe(Pie);
    });

    it.each([
      `pie showData`,
      `  pie  showData  `,
      `\tpie\tshowData\t`,
      `
    pie\tshowData
    `,
    ])('should handle regular showData', (context: string) => {
      const result = parse(context);
      expectNoErrorsOrAlternatives(result);
      expect(result.value.$type).toBe(Pie);

      const { showData } = result.value;
      expect(showData).toBeTruthy();
    });
  });
  describe('should handle TitleAndAccessibilities', () => {
    describe('should handle TitleAndAccessibilities without showData', () => {
      it.each([
        `pie title sample title`,
        `  pie  title sample title  `,
        `\tpie\ttitle sample title\t`,
        `pie
        \ttitle sample title
        `,
      ])('should handle regular pie + title in same line', (context: string) => {
        const result = parse(context);
        expectNoErrorsOrAlternatives(result);
        expect(result.value.$type).toBe(Pie);

        const { title } = result.value;
        expect(title).toBe('sample title');
      });

      it.each([
        `pie
        title sample title`,
        `pie
        title sample title
        `,
        `pie
        title sample title`,
        `pie
        title sample title
        `,
      ])('should handle regular pie + title in different line', (context: string) => {
        const result = parse(context);
        expectNoErrorsOrAlternatives(result);
        expect(result.value.$type).toBe(Pie);

        const { title } = result.value;
        expect(title).toBe('sample title');
      });
    });

    describe('should handle TitleAndAccessibilities with showData', () => {
      it.each([
        `pie showData title sample title`,
        `pie showData title sample title
        `,
      ])('should handle regular pie + showData + title', (context: string) => {
        const result = parse(context);
        expectNoErrorsOrAlternatives(result);
        expect(result.value.$type).toBe(Pie);

        const { showData, title } = result.value;
        expect(showData).toBeTruthy();
        expect(title).toBe('sample title');
      });

      it.each([
        `pie showData
        title sample title`,
        `pie showData
        title sample title
        `,
        `pie showData
        title sample title`,
        `pie showData
        title sample title
        `,
      ])('should handle regular showData + title in different line', (context: string) => {
        const result = parse(context);
        expectNoErrorsOrAlternatives(result);
        expect(result.value.$type).toBe(Pie);

        const { showData, title } = result.value;
        expect(showData).toBeTruthy();
        expect(title).toBe('sample title');
      });
    });
  });

  describe('should handle sections', () => {
    it.each([
      `pie
        "GitHub":100
        "GitLab":50`,
      `pie
        "GitHub"   :   100
        "GitLab"   :   50`,
      `pie
        "GitHub"\t:\t100
        "GitLab"\t:\t50`,
      `pie
        \t"GitHub" \t : \t 100
        \t"GitLab" \t : \t  50
        `,
    ])('should handle regular sections', (context: string) => {
      const result = parse(context);
      expectNoErrorsOrAlternatives(result);
      expect(result.value.$type).toBe(Pie);

      const { sections } = result.value;
      expect(sections[0].label).toBe('GitHub');
      expect(sections[0].value).toBe(100);

      expect(sections[1].label).toBe('GitLab');
      expect(sections[1].value).toBe(50);
    });

    it('should handle sections with showData', () => {
      const context = `pie showData
        "GitHub": 100
        "GitLab": 50`;
      const result = parse(context);
      expectNoErrorsOrAlternatives(result);
      expect(result.value.$type).toBe(Pie);

      const { showData, sections } = result.value;
      expect(showData).toBeTruthy();

      expect(sections[0].label).toBe('GitHub');
      expect(sections[0].value).toBe(100);

      expect(sections[1].label).toBe('GitLab');
      expect(sections[1].value).toBe(50);
    });

    it('should handle sections with title', () => {
      const context = `pie title sample wow
        "GitHub": 100
        "GitLab": 50`;
      const result = parse(context);
      expectNoErrorsOrAlternatives(result);
      expect(result.value.$type).toBe(Pie);

      const { title, sections } = result.value;
      expect(title).toBe('sample wow');

      expect(sections[0].label).toBe('GitHub');
      expect(sections[0].value).toBe(100);

      expect(sections[1].label).toBe('GitLab');
      expect(sections[1].value).toBe(50);
    });

    it('should handle value with positive decimal', () => {
      const context = `pie
        "ash": 60.67
        "bat": 40`;
      const result = parse(context);
      expectNoErrorsOrAlternatives(result);
      expect(result.value.$type).toBe(Pie);

      const { sections } = result.value;
      expect(sections[0].label).toBe('ash');
      expect(sections[0].value).toBe(60.67);

      expect(sections[1].label).toBe('bat');
      expect(sections[1].value).toBe(40);
    });

    it('should handle sections with accTitle', () => {
      const context = `pie accTitle: sample wow
        "GitHub": 100
        "GitLab": 50`;
      const result = parse(context);
      expectNoErrorsOrAlternatives(result);
      expect(result.value.$type).toBe(Pie);

      const { accTitle, sections } = result.value;
      expect(accTitle).toBe('sample wow');

      expect(sections[0].label).toBe('GitHub');
      expect(sections[0].value).toBe(100);

      expect(sections[1].label).toBe('GitLab');
      expect(sections[1].value).toBe(50);
    });

    it('should handle sections with single line accDescr', () => {
      const context = `pie accDescr: sample wow
        "GitHub": 100
        "GitLab": 50`;
      const result = parse(context);
      expectNoErrorsOrAlternatives(result);
      expect(result.value.$type).toBe(Pie);

      const { accDescr, sections } = result.value;
      expect(accDescr).toBe('sample wow');

      expect(sections[0].label).toBe('GitHub');
      expect(sections[0].value).toBe(100);

      expect(sections[1].label).toBe('GitLab');
      expect(sections[1].value).toBe(50);
    });

    it('should handle sections with multi line accDescr', () => {
      const context = `pie accDescr {
            sample wow
        }
        "GitHub": 100
        "GitLab": 50`;
      const result = parse(context);
      expectNoErrorsOrAlternatives(result);
      expect(result.value.$type).toBe(Pie);

      const { accDescr, sections } = result.value;
      expect(accDescr).toBe('sample wow');

      expect(sections[0].label).toBe('GitHub');
      expect(sections[0].value).toBe(100);

      expect(sections[1].label).toBe('GitLab');
      expect(sections[1].value).toBe(50);
    });
  });
});