error handling for streamed data

Author: LeonOstrez

src/commands/export.js

@@ -17,7 +17,7 @@ const {
     pleaseCaptureLoginTestLog,
     enterLoginRouteLog,
     testEligibleForExportLog,
-    jestAuthFileGenerationLog
+    testExportStartedLog
 } = require("../utils/cmdPrint");
 const {
     getJestAuthFunction,
@@ -102,6 +102,7 @@ function cleanupDataFolder() {
 }
 
 async function exportTest(originalTest, exportsMetadata) {
+    testExportStartedLog();
     let test = convertOldTestForGPT(originalTest);
     let jestTest = await getJestTest(test);
     let testName = await getJestTestName(jestTest, Object.values(exportsMetadata).map(obj => obj.testName));

src/helpers/api.js

@@ -1,7 +1,7 @@
 const https = require('https');
 const _ = require('lodash');
 const axios = require('axios');
-const {} = require('../utils/cmdPrint');
+const { jestAuthFileGenerationLog } = require('../utils/cmdPrint');
 const args = require('../utils/getArgs.js');
 
 function extractGPTMessageFromStreamData(input) {
@@ -38,12 +38,19 @@ function setOptions({protocol, hostname, port, path, method, headers}) {
 
 async function makeRequest(data, options) {
     let gptResponse = '';
+    let end = false;
 
     return new Promise((resolve, reject) => {
         const req = https.request(_.omit(options, ['protocol']), function(res){
             res.on('data', (chunk) => {
                 try {
                     let stringified = chunk.toString();
+                    if (stringified === 'pythagora_end') {
+                        gptResponse = '';
+                        end = true;
+                        return;
+                    }
+                    if (end) return gptResponse = stringified;
                     try {
                         let json = JSON.parse(stringified);
                         if (json.error) gptResponse = json.error;
@@ -60,8 +67,9 @@ async function makeRequest(data, options) {
 
                 } catch (e) {}
             });
-            res.on('end', async () => {
-                if (res.statusCode >= 400) throw new Error(`Response status code: ${res.statusCode}.`);
+            res.on('end', async (a,b,c) => {
+                process.stdout.write('\n');
+                if (res.statusCode >= 400) throw new Error(`Response status code: ${res.statusCode}. Error message: ${gptResponse}`);
                 if (gptResponse.message) throw new Error(`Error: ${gptResponse.message}. Code: ${gptResponse.code}`);
                 gptResponse = cleanupGPTResponse(gptResponse);
                 resolve(gptResponse);
@@ -102,7 +110,7 @@ async function isEligibleForExport(jestTest) {
         let options = setOptions({ path: '/check-if-eligible' });
 
         const response = await axios.post(
-            `${options.protocol}://${options.hostname}:${options.port}${options.path}`,
+            `${options.protocol}://${options.hostname}${options.port ? ':' + options.port : ''}${options.path}`,
             JSON.stringify({ jestTest }),
             { headers: options.headers }
         );

src/utils/cmdPrint.js

@@ -182,7 +182,7 @@ function pleaseCaptureLoginTestLog(loginEndpointPath) {
     console.log(`Login path is: ${green}${bold}${loginEndpointPath}${reset}`);
     console.log('Please run the Pythagora capture command, log into your app and try the export again.');
     console.log('To start Pythagora capture, run:');
-    console.log(`${blue}${bold}npx pythagora --init-command \"${args.init_command.join(' ')}\" --mode capture${reset}`);
+    console.log(`${blue}${bold}npx pythagora --init-command \"my start command\" --mode capture${reset}`);
 }
 
 function logLoginEndpointCaptured() {