Merge branch 'feat/expand-unit-tests' of https://github.com/Pythagora-io/pythagora into feat/expand-unit-tests

Author: pavel-pythagora

README.md

@@ -26,7 +26,15 @@ npx pythagora --unit-tests --func <FUNCTION_NAME>
 
 Where `<FUNCTION_NAME>` is the name of the function you want to generate unit tests for. Just make sure that your function is exported from a file. You can see other options like generating tests for multiple files or folders [below in the Options section](#-options).
 
-<br>
+<br><br>
+If you wish to expand your current test suite with more tests to get better code coverage you can run:
+
+```bash
+npx pythagora --expand-unit-tests --path <PATH_TO_YOUR_TEST_SUITE>
+```
+for more details on expanding existing tests see [below in the Expanding existing tests section](#-expand-existing-tests).
+
+<br><br>
 
 **NOTE:** on Windows make sure to run all commands using `Git Bash` and not `Power Shell` or anything similiar
 
@@ -77,6 +85,16 @@ When Pythagora generates unit tests, it uses the following approach:
 
 <br>
 
+# 📈 Expand existing tests
+If you already have generated tests for your codebase but you just want to increase your code coverage or cover more edge cases, simply run:
+
+```bash
+npx pythagora --expand-unit-tests --path <PATH_TO_YOUR_TEST_SUITE>
+```
+When running command `PATH_TO_YOUR_TEST_SUITE` can be path to a single test file or to a folder and all test files inside of that folder will be processed and expanded.
+
+That's all, enjoy your new code coverage!
+
 # 📖 Options
 - To generate unit tests for **one single function**, run:
 

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)) {
@@ -374,12 +374,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();