@tng-sh/js 0.0.6 → 0.0.8

package/bin/tng.js

@@ -10,8 +10,8 @@ const { ping, getUserStats } = require('../index');
 
 program
     .name('tng')
-    .description('TNG - Automated Test Generation for JavaScript')
-    .version('0.0.6');
+    .description('TNG - Automated Test Generation, and audit generation for JavaScript')
+    .version('0.0.8');
 
 /**
  * @command init

package/lib/generateTestsUi.js

@@ -174,7 +174,7 @@ class GenerateTestsUI {
         const displayName = method.class_name ? `${method.class_name}#${method.name}` : `${fileName}#${method.name}`;
         const actionName = isAudit ? 'Auditing' : 'Generating test for';
 
-        const progressHandler = (progress) => {
+        const progressHandler = async (progress) => {
             progress.update('Preparing request...');
 
             try {
@@ -240,19 +240,22 @@ class GenerateTestsUI {
                 );
 
                 if (isAudit) {
-                    // Auto-exit the progress UI so we can immediately show the audit results
-                    progress.complete('Audit complete!', { auto_exit: true });
+                    progress.complete('Audit ready!', { auto_exit: false });
                     return {
-                        message: 'Audit complete!',
+                        message: 'Audit ready!',
                         resultJson: resultJson
                     };
                 }
 
                 progress.update('Tests generated successfully!');
 
+                // Save the file inside the progress handler to match Python flow
+                const fileInfo = await saveTestFile(resultJson);
+
                 return {
                     message: 'Tests generated successfully!',
-                    resultJson
+                    resultJson,
+                    file_path: fileInfo?.file_path
                 };
             } catch (e) {
                 progress.error(`Failed to generate: ${e.message}`);
@@ -269,11 +272,7 @@ class GenerateTestsUI {
         }
 
         if (uiResult && uiResult.resultJson) {
-            const fileInfo = await saveTestFile(uiResult.resultJson);
-            return {
-                ...uiResult,
-                file_path: fileInfo?.file_path
-            };
+            return uiResult;
         }
 
         return null;
@@ -287,12 +286,71 @@ class GenerateTestsUI {
             const choice = this.goUiSession.showPostGenerationMenu(filePath, runCommand);
             if (choice === 'back' || !choice) break;
 
-            if (choice === 'copy_command') {
+            if (choice === 'run_tests') {
+                this._runAndShowTestResults(runCommand);
+            } else if (choice === 'copy_command') {
                 this._copyToClipboard(runCommand);
             }
         }
     }
 
+    _runAndShowTestResults(command) {
+        const { execSync } = require('child_process');
+
+        const spinnerHandler = () => {
+            try {
+                const output = execSync(command, { encoding: 'utf8', stdio: 'pipe' });
+                return {
+                    success: true,
+                    message: "Tests completed",
+                    output: output,
+                    exit_code: 0
+                };
+            } catch (error) {
+                return {
+                    success: true, // We still want to show the results even if tests failed
+                    message: "Tests completed",
+                    output: error.stdout + error.stderr,
+                    exit_code: error.status
+                };
+            }
+        };
+
+        const testOutput = this.goUiSession.showSpinner("Running tests...", spinnerHandler);
+
+        // Note: For simplicity in the first pass, we provide counts.
+        // We can enhance _parseTestOutput if needed.
+        const { passed, failed, errors, total } = this._parseTestOutput(testOutput.output || "", testOutput.exit_code || 0);
+
+        this.goUiSession.showTestResults("Test Results", passed, failed, errors, total, []);
+    }
+
+    _parseTestOutput(output, exitCode) {
+        let passed = 0, failed = 0, errors = 0;
+
+        const passedMatch = output.match(/(\d+) passed/);
+        const failedMatch = output.match(/(\d+) failed/);
+        const errorMatch = output.match(/(\d+) error/);
+
+        if (passedMatch) passed = parseInt(passedMatch[1]);
+        if (failedMatch) failed = parseInt(failedMatch[1]);
+        if (errorMatch) errors = parseInt(errorMatch[1]);
+
+        let total = passed + failed + errors;
+
+        if (total === 0) {
+            if (exitCode === 0) {
+                passed = 1;
+                total = 1;
+            } else {
+                failed = 1;
+                total = 1;
+            }
+        }
+
+        return { passed, failed, errors, total };
+    }
+
     _copyToClipboard(text) {
         const { execSync } = require('child_process');
         try {

package/lib/goUiSession.js

@@ -106,9 +106,11 @@ class GoUISession {
 
             fs.writeFileSync(controlFile, JSON.stringify(status));
 
+            // Wait for process to exit
             // Wait for process to exit
             // In Node, we can't easily wait sync for background spawned process without busy loop or async
             // But for this simple implementation, we'll wait for the process to exit
+            return result;
         } catch (error) {
             const status = { status: 'error', message: error.message };
             fs.writeFileSync(controlFile, JSON.stringify(status));

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@tng-sh/js",
-    "version": "0.0.6",
+    "version": "0.0.8",
     "description": "TNG JavaScript CLI",
     "repository": {
         "type": "git",