add a unit test to Type Tree and Schema Tree

Author: Rheilly Aguilar

packages/idea-parser/tests/Schema.test.ts

@@ -4,6 +4,7 @@ import { expect, use } from 'chai';
 import deepEqualInAnyOrder from 'deep-equal-in-any-order';
 import SchemaTree from '../src/trees/SchemaTree';
 import Compiler from '../src/Compiler';
+import Exception from '../src/Exception';
 
 use(deepEqualInAnyOrder);
 const cleanAST = (node: any) => {
@@ -36,4 +37,12 @@ describe('Schema Tree', () => {
     const final = JSON.parse(fs.readFileSync(`${__dirname}/fixtures/final.json`, 'utf8'));
     expect(Compiler.final(actual)).to.deep.equalInAnyOrder(final);
   });
+
+
+  // Line 81 - 86 
+  it('Should throw an exception when there is an unexpected token at the end of the code', () => {
+  const codeWithUnexpectedToken = 'enum TestEnum { VALUE1, VALUE2, } unexpectedToken';
+  
+  expect(() => SchemaTree.parse(codeWithUnexpectedToken)).to.throw(Exception, /Unexpected token ,/);
+  });
 });

packages/idea-parser/tests/TypeTree.test.ts

@@ -37,4 +37,37 @@ describe('Type Tree', () => {
     //console.log(JSON.stringify(actual, null, 2));
     expect(actual).to.deep.equalInAnyOrder(expected);
   });
+
+
+  // Line 39 - 40
+  it('Should correctly identify "typeword" token at the start of the input string', () => {
+    const lexerMock = {
+      define: (name: string, callback: Function) => {
+        if (name === 'TypeWord') {
+          const result = callback('type Example', 0);
+          expect(result).to.deep.equal({
+            type: '_TypeWord',
+            start: 0,
+            end: 4,
+            value: 'type',
+            raw: 'type'
+          });
+        }
+      }
+    };
+    TypeTree.definitions(lexerMock as any);
+  });
+
+  // Line 55
+  it('Should throw an error when the input code is an empty string', () => {
+    expect(() => {
+      const lexerMock = {
+        expect: (tokenType: string) => { throw new Error('Unexpected end of input'); },
+        load: () => { }
+      };
+      const typeTree = new TypeTree();
+      (typeTree as any)._lexer = lexerMock;
+      typeTree.parse('');
+    }).to.throw(Error, 'Unexpected end of input');
+  });
 });