flake-monster 0.3.1 → 0.3.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "flake-monster",
-  "version": "0.3.1",
+  "version": "0.3.3",
   "description": "Source-to-source test hardener that injects async delays to surface flaky tests",
   "type": "module",
   "bin": {

package/src/adapters/javascript/index.js

@@ -1,7 +1,7 @@
 import { fileURLToPath } from 'node:url';
 import { dirname, join, posix } from 'node:path';
 import { parseSource } from './parser.js';
-import { computeInjections, computeRuntimeImportInsertion, applyInsertions } from './injector.js';
+import { computeInjections, computeRuntimeImportInsertion, applyInsertions, MARKER_PREFIX } from './injector.js';
 import { recoverDelays, scanForRecovery } from './remover.js';
 
 const __dirname = dirname(fileURLToPath(import.meta.url));
@@ -43,6 +43,11 @@ export function createJavaScriptAdapter() {
     },
 
     inject(source, options) {
+      if (source.includes(MARKER_PREFIX)) {
+        console.warn(`  Skipping ${options.filePath}: already injected (restore first)`);
+        return { source, points: [], runtimeNeeded: false };
+      }
+
       try {
         const { ast } = parseSource(source);
         const { insertions, points } = computeInjections(ast, source, options);

package/src/adapters/javascript/injector.js

@@ -405,13 +405,20 @@ export function computeRuntimeImportInsertion(ast, source, runtimeImportPath) {
     }
   }
 
-  // Find the newline after the last import (or start of file)
-  let insertOffset = lastImportEnd;
-  while (insertOffset < source.length && source[insertOffset] !== '\n') {
-    insertOffset++;
-  }
-  if (insertOffset < source.length) {
-    insertOffset++; // past the \n
+  let insertOffset;
+  if (lastImportEnd === 0) {
+    // No existing imports — insert at the very start of the file
+    // (not after the first line, which may be inside a block comment)
+    insertOffset = 0;
+  } else {
+    // Find the newline after the last import
+    insertOffset = lastImportEnd;
+    while (insertOffset < source.length && source[insertOffset] !== '\n') {
+      insertOffset++;
+    }
+    if (insertOffset < source.length) {
+      insertOffset++; // past the \n
+    }
   }
 
   const text = `import { ${DELAY_OBJECT} } from '${runtimeImportPath}';\n`;

package/src/adapters/javascript/parser.js

@@ -1,5 +1,15 @@
 import * as acorn from 'acorn';
-import { attachComments } from 'astravel';
+import { defaultTraveler, attachComments } from 'astravel';
+
+// Patch astravel bug: PropertyDefinition reuses MethodDefinition handler which
+// calls this.go(node.value) without a null guard. Uninitialized class fields
+// (e.g. `bar;`) have value: null, crashing the traversal.
+defaultTraveler.PropertyDefinition = function (node, state) {
+  this.go(node.key, state);
+  if (node.value != null) {
+    this.go(node.value, state);
+  }
+};
 
 /**
  * Parse JS source to ESTree AST with comments attached to nodes.