deflake 1.2.6 → 1.2.8

package/cli.js

@@ -960,6 +960,27 @@ async function analyzeFailures(artifacts, fullLog, client) {
     if (results.length > 0 && argv.fix && argv.report) {
         showFrameworkReport();
     }
+
+    return results.length;
+}
+
+/**
+ * Helper to run a shell command and capture output.
+ */
+async function runCommand(cmd, args) {
+    return new Promise((resolve) => {
+        console.log(`🚀 Running command: ${cmd} ${args.join(' ')}`);
+        const child = spawn(cmd, args, { shell: true, stdio: 'pipe' });
+        let stdout = '';
+        let stderr = '';
+
+        child.stdout.on('data', (data) => { process.stdout.write(data); stdout += data.toString(); });
+        child.stderr.on('data', (data) => { process.stderr.write(data); stderr += data.toString(); });
+
+        child.on('close', (code) => {
+            resolve({ code, output: stdout + "\n" + stderr });
+        });
+    });
 }
 
 async function main() {
@@ -976,26 +997,29 @@ async function main() {
     if (command.length > 0) {
         const cmd = command[0];
         const args = command.slice(1);
-        console.log(`🚀 Running command: ${cmd} ${args.join(' ')}`);
-
-        const child = spawn(cmd, args, { shell: true, stdio: 'pipe' });
-        let stdout = '';
-        let stderr = '';
-
-        child.stdout.on('data', (data) => { process.stdout.write(data); stdout += data.toString(); });
-        child.stderr.on('data', (data) => { process.stderr.write(data); stderr += data.toString(); });
-
-        child.on('close', async (code) => {
-            if (code !== 0) {
-                console.log(`\n🔴 Command failed with code ${code}. Activating DeFlake...`);
-                const artifacts = detectAllArtifacts(null, argv.html);
-                await analyzeFailures(artifacts, stdout + "\n" + stderr, client);
-                process.exit(code);
-            } else {
-                console.log("\n🟢 Command passed successfully.");
-                process.exit(0);
+        
+        let { code, output } = await runCommand(cmd, args);
+
+        if (code !== 0) {
+            console.log(`\n🔴 Command failed with code ${code}. Activating DeFlake...`);
+            const artifacts = detectAllArtifacts(null, argv.html);
+            const fixesApplied = await analyzeFailures(artifacts, output, client);
+
+            // AUTO-VERIFICATION
+            if (fixesApplied > 0 && argv.fix) {
+                console.log(`\n${C.BRIGHT}💉 Fixes applied. Re-running tests to verify...${C.RESET}`);
+                const secondRun = await runCommand(cmd, args);
+                if (secondRun.code === 0) {
+                    console.log(`\n${C.GREEN}${C.BRIGHT}✅ All tests passed after DeFlake healing!${C.RESET}`);
+                } else {
+                    console.log(`\n${C.YELLOW}⚠️  Some tests still failing after fixes. Check the report for details.${C.RESET}`);
+                }
             }
-        });
+            process.exit(code);
+        } else {
+            console.log("\n🟢 Command passed successfully.");
+            process.exit(0);
+        }
     } else {
         const artifacts = detectAllArtifacts(argv.log, argv.html);
         const fullLog = argv.log && fs.existsSync(argv.log) ? fs.readFileSync(argv.log, 'utf8') : null;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "deflake",
-  "version": "1.2.6",
+  "version": "1.2.8",
   "description": "AI-powered self-healing tool for Playwright, Cypress, and WebdriverIO tests.",
   "main": "client.js",
   "bin": {