Improve testing

Author: kddejong

tst/unit/context/TransformContext.test.ts

@@ -7,6 +7,25 @@ describe('TransformContext', () => {
         // Mock syntax tree with SAM transform
         const mockNode = {
             childCount: 2,
+            descendantsOfType: () => [mockNode],
+            namedChildren: [
+                {
+                    type: 'block_mapping_pair',
+                    childForFieldName: (field: string) => {
+                        if (field === 'key') return { text: 'Transform' };
+                        if (field === 'value') return { text: 'AWS::Serverless-2016-10-31' };
+                        return null;
+                    },
+                },
+                {
+                    type: 'block_mapping_pair',
+                    childForFieldName: (field: string) => {
+                        if (field === 'key') return { text: 'Resources' };
+                        if (field === 'value') return { text: '{}' };
+                        return null;
+                    },
+                },
+            ],
             child: (index: number) => {
                 if (index === 0) {
                     return {
@@ -38,7 +57,12 @@ describe('TransformContext', () => {
     });
 
     it('should return false when no SAM transform present', () => {
-        const mockNode = { childCount: 0, child: () => null } as any;
+        const mockNode = {
+            childCount: 0,
+            child: () => null,
+            descendantsOfType: () => [],
+            namedChildren: [],
+        } as any;
         const context = new TransformContext(mockNode, DocumentType.YAML);
         expect(context.hasSamTransform()).toBe(false);
     });

tst/utils/MockContext.ts

@@ -34,12 +34,20 @@ const defaultContextParams: Partial<ContextParams> = {
 };
 
 function node(text: string, startPosition?: Point, endPosition?: Point, type?: string): SyntaxNode {
-    return {
+    const mockNode: Partial<SyntaxNode> = {
         text,
         startPosition,
         endPosition,
         type,
-    } as SyntaxNode;
+        parent: null,
+        childCount: 0,
+        child: () => null,
+        childForFieldName: () => null,
+        descendantsOfType: () => [],
+        namedChildren: [],
+        children: [],
+    };
+    return mockNode as SyntaxNode;
 }
 
 export function createMockContext(