fixed test cases that were broken

Author: Louis Kuczykowski

media/RL(Final).png

@@ -44,7 +44,7 @@
         {
           "id": "react-labyrinth",
           "title": "React Labyrinth",
-          "icon": "/media/test1.svg"
+          "icon": "/media/RL(Final).png"
         }
       ]
     },

package.json

@@ -18,7 +18,7 @@ export function createPanel(context: vscode.ExtensionContext, data: Tree, column
     );
 
     // Set the icon logo of extension webview
-    panel.iconPath = vscode.Uri.joinPath(context.extensionUri, 'media', 'favicon.ico');
+    panel.iconPath = vscode.Uri.joinPath(context.extensionUri, 'media', 'RL(Final).png');
 
     // Set URI to be the path to bundle
     const bundlePath: vscode.Uri = vscode.Uri.joinPath(context.extensionUri, 'build', 'bundle.js');

src/panel.ts

@@ -57,9 +57,28 @@ export class Parser {
         };
         this.tree = root;
         this.parser(root);
+        // clean up nodes with error: 'File not found'
+        this.removeTreesWithError(this.tree);
         return this.tree;
     }
 
+    private removeTreesWithError(tree: Tree): void {
+        // base case
+        if(tree.children.length === 0) return;
+        // iterate over tree.children array to check for error. 
+        for(let i = 0; i < tree.children.length; i++){
+            // call removeTreesWithError on every tree in the children array
+            if(tree.children[i].children.length !== 0){
+                this.removeTreesWithError(tree.children[i]);
+            }
+            if(tree.children[i].error && (tree.children[i].error === 'File not found' || tree.children[i].error === 'Error while processing this file/node')){
+                // when an error is found, splice the tree out of the children array
+                tree.children.splice(i,1);
+                i--; // decrement to account for change in children array length
+            }
+        }
+    };
+
     public getTree(): Tree {
         return this.tree;
     }
@@ -278,7 +297,7 @@ export class Parser {
             }
         };
 
-        console.log('directive: ', directive);
+        // console.log('directive: ', directive);
         // Initial check for use of directives (ex: 'use client', 'use server', 'use strict')
         // Accounts for more than one directive 
         for (let i = 0; i < directive.length; i++) {

src/parser.ts

@@ -55,12 +55,10 @@ describe('Parser Test Suite', () => {
             file = path.join(__dirname, '../../../../src/test/test_cases/tc_6/component/App.jsx');
             parser = new Parser(file);
             tree = parser.parse();
-            console.log(tree)
         });
 
-        test("Child component that doesn't exist still shows up on the node tree", () => {
-            expect(tree.children[0].name).toBe('anotherApp');
-			expect(tree.children[0].error).not.toBe('');
+        test("Child component with bad file path does not show up on the node tree", () => {
+            expect(tree.children.length).toBe(0);
         })
     })
 
@@ -71,10 +69,9 @@ describe('Parser Test Suite', () => {
             parser = new Parser(file);
             tree = parser.parse();
         });
+        
         test("Parser stops parsing when there is a syntax error in a component", () => {
-            expect(tree.children[0].name).toBe('App');
-            expect(tree.children[0].error).not.toBeUndefined;
-            expect((tree.children[0].children).length).toBe(0);
+            expect(tree.children.length).toBe(0);
         });
     })