Merge branch 'main' into feat/expand-unit-tests

Author: LeonOstrez

package.json

@@ -1,6 +1,6 @@
 {
   "name": "pythagora",
-  "version": "0.0.76",
+  "version": "0.0.78",
   "author": {
     "name": "Zvonimir Sabljic",
     "email": "hi@pythagora.ai"

src/bin/run.bash

@@ -60,7 +60,7 @@ do
     exit 0
   elif [[ "${args[$i]}" == "--unit-tests" ]]
   then
-    echo "${green}${bold}Generating unit tests...${reset}"
+    echo "${green}${bold}Starting generation of unit tests...${reset}"
     PYTHAGORA_CONFIG="$@" node "${pythagora_dir}/src/scripts/unit.js"
     exit 0
 

src/helpers/unitTests.js

@@ -69,7 +69,7 @@ async function processFile(filePath, filesToProcess) {
                 if (importedFile && !filesToProcess.includes(importedFile)) {
                     filesToProcess.push(importedFile);
                 }
-            } else if (node.type === "VariableDeclaration" && node.declarations[0].init.type === "CallExpression" && node.declarations[0].init.callee.name === "require") {
+            } else if (node.type === "VariableDeclaration" && node.declarations.length > 0 && node.declarations[0].init && node.declarations[0].init.type === "CallExpression" && node.declarations[0].init.callee.name === "require") {
                 let importedFile = path.resolve(path.dirname(filePath), node.declarations[0].init.arguments[0].value);
                 importedFile = resolveFilePath(importedFile, extension);
                 if (importedFile && !filesToProcess.includes(importedFile)) {
@@ -395,12 +395,14 @@ async function generateTestsForDirectory(args, processingFunction = 'getUnitTest
     API.checkForAPIKey();
     queriedPath = path.resolve(pathToProcess);
     rootPath = args.pythagora_root;
-    ({ screen, spinner, scrollableContent } = initScreenForUnitTests());
+    console.log('Processing folder structure...');
 
     await traverseDirectory(queriedPath, true, funcName, processingFunction);
     processedFiles = [];
     await traverseDirectory(queriedPath, true, funcName, processingFunction);
     processedFiles = [];
+    console.log('Generating tests...');
+    ({ screen, spinner, scrollableContent } = initScreenForUnitTests());
     await traverseDirectory(queriedPath, false, funcName, processingFunction);
 
     screen.destroy();