updated validaor and tests to use treemap

packages/parser/src/language/treemap/treemap-validator.ts

@@ -1,5 +1,5 @@
 import type { ValidationAcceptor, ValidationChecks } from 'langium';
-import type { MermaidAstType, TreemapDoc } from '../generated/ast.js';
+import type { MermaidAstType, Treemap } from '../generated/ast.js';
 import type { TreemapServices } from './module.js';
 
 /**
@@ -9,10 +9,10 @@ export function registerValidationChecks(services: TreemapServices) {
   const validator = services.validation.TreemapValidator;
   const registry = services.validation.ValidationRegistry;
   if (registry) {
-    // Use any to bypass type checking since we know TreemapDoc is part of the AST
+    // Use any to bypass type checking since we know Treemap is part of the AST
     // but the type system is having trouble with it
     const checks: ValidationChecks<MermaidAstType> = {
-      TreemapDoc: validator.checkSingleRoot.bind(validator),
+      Treemap: validator.checkSingleRoot.bind(validator),
       // Remove unused validation for TreemapRow
     };
     registry.register(checks, validator);
@@ -27,7 +27,7 @@ export class TreemapValidator {
    * Validates that a treemap has only one root node.
    * A root node is defined as a node that has no indentation.
    */
-  checkSingleRoot(doc: TreemapDoc, accept: ValidationAcceptor): void {
+  checkSingleRoot(doc: Treemap, accept: ValidationAcceptor): void {
     let rootNodeIndentation;
 
     for (const row of doc.TreemapRows) {

packages/parser/tests/treemap.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from 'vitest';
 import { expectNoErrorsOrAlternatives } from './test-util.js';
-import type { TreemapDoc, Section, Leaf, TreemapRow } from '../src/language/generated/ast.js';
+import type { Treemap, Section, Leaf, TreemapRow } from '../src/language/generated/ast.js';
 import type { LangiumParser } from 'langium';
 import { createTreemapServices } from '../src/language/treemap/module.js';
 
@@ -9,21 +9,21 @@ describe('Treemap Parser', () => {
   const parser: LangiumParser = services.parser.LangiumParser;
 
   const parse = (input: string) => {
-    return parser.parse<TreemapDoc>(input);
+    return parser.parse<Treemap>(input);
   };
 
   describe('Basic Parsing', () => {
     it('should parse empty treemap', () => {
       const result = parse('treemap');
       expectNoErrorsOrAlternatives(result);
-      expect(result.value.$type).toBe('TreemapDoc');
+      expect(result.value.$type).toBe('Treemap');
       expect(result.value.TreemapRows).toHaveLength(0);
     });
 
     it('should parse a section node', () => {
       const result = parse('treemap\n"Root"');
       expectNoErrorsOrAlternatives(result);
-      expect(result.value.$type).toBe('TreemapDoc');
+      expect(result.value.$type).toBe('Treemap');
       expect(result.value.TreemapRows).toHaveLength(1);
       if (result.value.TreemapRows[0].item) {
         expect(result.value.TreemapRows[0].item.$type).toBe('Section');
@@ -39,7 +39,7 @@ describe('Treemap Parser', () => {
   "Child2" : 200
 `);
       expectNoErrorsOrAlternatives(result);
-      expect(result.value.$type).toBe('TreemapDoc');
+      expect(result.value.$type).toBe('Treemap');
       expect(result.value.TreemapRows).toHaveLength(3);
 
       if (result.value.TreemapRows[0].item) {
@@ -95,7 +95,7 @@ describe('Treemap Parser', () => {
 
       // We're only checking that the multiple root nodes parse successfully
       // The validation errors would be reported by the validator during validation
-      expect(result.value.$type).toBe('TreemapDoc');
+      expect(result.value.$type).toBe('Treemap');
       expect(result.value.TreemapRows).toHaveLength(2);
     });
   });
@@ -104,7 +104,7 @@ describe('Treemap Parser', () => {
     it('should parse a treemap with title', () => {
       const result = parse('treemap\ntitle My Treemap Diagram\n"Root"\n  "Child": 100');
       expectNoErrorsOrAlternatives(result);
-      expect(result.value.$type).toBe('TreemapDoc');
+      expect(result.value.$type).toBe('Treemap');
       // We can't directly test the title property due to how Langium processes TitleAndAccessibilities
       // but we can verify the TreemapRows are parsed correctly
       expect(result.value.TreemapRows).toHaveLength(2);
@@ -113,7 +113,7 @@ describe('Treemap Parser', () => {
     it('should parse a treemap with accTitle', () => {
       const result = parse('treemap\naccTitle: Accessible Title\n"Root"\n  "Child": 100');
       expectNoErrorsOrAlternatives(result);
-      expect(result.value.$type).toBe('TreemapDoc');
+      expect(result.value.$type).toBe('Treemap');
       // We can't directly test the accTitle property due to how Langium processes TitleAndAccessibilities
       expect(result.value.TreemapRows).toHaveLength(2);
     });
@@ -123,7 +123,7 @@ describe('Treemap Parser', () => {
         'treemap\naccDescr: This is an accessible description\n"Root"\n  "Child": 100'
       );
       expectNoErrorsOrAlternatives(result);
-      expect(result.value.$type).toBe('TreemapDoc');
+      expect(result.value.$type).toBe('Treemap');
       // We can't directly test the accDescr property due to how Langium processes TitleAndAccessibilities
       expect(result.value.TreemapRows).toHaveLength(2);
     });
@@ -136,7 +136,7 @@ accDescr: This is an accessible description
 "Root"
   "Child": 100`);
       expectNoErrorsOrAlternatives(result);
-      expect(result.value.$type).toBe('TreemapDoc');
+      expect(result.value.$type).toBe('Treemap');
       // We can't directly test these properties due to how Langium processes TitleAndAccessibilities
       expect(result.value.TreemapRows).toHaveLength(2);
     });