fix logging of folder structure and final log when generating of single unit test suite is done

Author: LeonOstrez

src/helpers/api.js

@@ -58,6 +58,7 @@ async function makeRequest(data, options, customLogFunction) {
                     } catch (e) {}
 
                     gptResponse += stringified;
+                    if (gptResponse.indexOf('pythagora_end:') > -1) return;
                     if (customLogFunction) customLogFunction(gptResponse);
                     else process.stdout.write(stringified);
                 } catch (e) {}

src/helpers/unitTests.js

@@ -154,7 +154,6 @@ async function createTests(filePath, funcToTest, processingFunction = 'getUnitTe
     try {
         let extension = path.extname(filePath);
         let ast = await getAstFromFilePath(filePath);
-        const fileIndex = folderStructureTree.findIndex(item => item.absolutePath === filePath);
 
         const foundFunctions = [];
 
@@ -184,6 +183,7 @@ async function createTests(filePath, funcToTest, processingFunction = 'getUnitTe
         );
 
         sortFolderTree();
+        const fileIndex = folderStructureTree.findIndex(item => item.absolutePath === filePath);
 
         for (const [i, funcData] of uniqueFoundFunctions.entries()) {
             let indexToPush = fileIndex + 1 + i;