@godzillaba/mutest 1.3.5 → 1.4.1

package/README.md

@@ -10,7 +10,7 @@ npx @godzillaba/mutest src/Counter.sol
 
 Pass one or more Solidity files. Mutest will:
 
-1. Create 10 parallel copies of your project
+1. Create parallel copies of your project (8 by default)
 2. Generate mutants with Gambit (e.g. `++` -> `--`, assignments replaced)
 3. Run `forge test` against each mutant across the worker copies
 4. Report which mutants were killed (tests caught them) or survived (coverage gap)
@@ -27,6 +27,14 @@ npx @godzillaba/mutest
 
 This reads `gambit_out/survivors.json` (or falls back to `gambit_out/gambit_results.json`) and re-runs the test suite against those mutants. Useful after improving your tests to check if previously surviving mutants are now caught.
 
+### Options
+
+`--workers <n>` / `-w <n>` — number of parallel workers (default: 8).
+
+```sh
+npx @godzillaba/mutest --workers 4 src/Counter.sol
+```
+
 ## Requirements
 
 - [Foundry](https://getfoundry.sh/) (`forge`)

package/index.ts

@@ -1,7 +1,8 @@
 #!/usr/bin/env -S npx tsx
 import { execFile as execFileCb } from "child_process";
-import { readFile, writeFile, cp, rm } from "fs/promises";
+import { readFile, writeFile, cp, rm, readdir } from "fs/promises";
 import { promisify } from "util";
+import { join } from "path";
 
 const execFile = promisify(execFileCb);
 
@@ -28,7 +29,7 @@ async function setupWorkers(workerCount: number): Promise<string> {
   return root;
 }
 
-async function runGambit(solFiles: string[]): Promise<Mutant[]> {
+async function runGambit(solFiles: string[], concurrency: number): Promise<Mutant[]> {
   await rm("gambit_out", { recursive: true, force: true });
   const { stdout: remappingsRaw } = await execFile("forge", ["remappings"]);
   const remappings = remappingsRaw.trim().replaceAll('/=', '=').split("\n").filter(Boolean);
@@ -36,7 +37,6 @@ async function runGambit(solFiles: string[]): Promise<Mutant[]> {
 
   const results: Mutant[] = [];
   const pending = [...solFiles];
-  const concurrency = 10;
 
   async function worker() {
     while (pending.length > 0) {
@@ -47,7 +47,6 @@ async function runGambit(solFiles: string[]): Promise<Mutant[]> {
       ]);
       const raw = await readFile(`${outdir}/gambit_results.json`, "utf-8");
       const mutants: Mutant[] = JSON.parse(raw);
-      for (const m of mutants) m.name = `${file}/${m.name}`;
       results.push(...mutants);
     }
   }
@@ -74,7 +73,7 @@ async function processMutants(
       const dest = `${workerDir}/${mutant.original}`;
       const backup = `${dest}.orig`;
       await cp(dest, backup);
-      await cp(`gambit_out/${mutant.name}`, dest);
+      await cp(`gambit_out/${mutant.original}/${mutant.name}`, dest);
       try {
         await execFile("forge", ["test", "--optimize", "false", "--root", workerDir]);
         survivors.push(mutant);
@@ -92,6 +91,16 @@ async function processMutants(
   console.log(`Wrote survivors.json`);
 }
 
+async function findJsonFiles(dir: string, name: string): Promise<string[]> {
+  const results: string[] = [];
+  for (const entry of await readdir(dir, { withFileTypes: true })) {
+    const full = join(dir, entry.name);
+    if (entry.isDirectory()) results.push(...await findJsonFiles(full, name));
+    else if (entry.name === name) results.push(full);
+  }
+  return results;
+}
+
 async function loadExistingMutants(): Promise<Mutant[]> {
   try {
     const raw = await readFile("gambit_out/survivors.json", "utf-8");
@@ -101,13 +110,31 @@ async function loadExistingMutants(): Promise<Mutant[]> {
       return survivors;
     }
   } catch {}
-  const raw = await readFile("gambit_out/gambit_results.json", "utf-8");
-  return JSON.parse(raw);
+  const files = await findJsonFiles("gambit_out", "gambit_results.json");
+  const all: Mutant[] = [];
+  for (const f of files) {
+    const raw = await readFile(f, "utf-8");
+    all.push(...JSON.parse(raw));
+  }
+  return all;
+}
+
+function parseArgs() {
+  const args = process.argv.slice(2);
+  let workers = 8;
+  const solFiles: string[] = [];
+  for (let i = 0; i < args.length; i++) {
+    if (args[i] === "--workers" || args[i] === "-w") {
+      workers = parseInt(args[++i], 10);
+    } else {
+      solFiles.push(args[i]);
+    }
+  }
+  return { solFiles, workers };
 }
 
 async function main() {
-  const solFiles = process.argv.slice(2);
-  const workerCount = 10;
+  const { solFiles, workers: workerCount } = parseArgs();
   console.log(`Setting up ${workerCount} workers...`);
   const tempDir = await setupWorkers(workerCount);
 
@@ -115,7 +142,7 @@ async function main() {
     let mutants: Mutant[];
     if (solFiles.length > 0) {
       console.log(`Running gambit on ${solFiles.join(", ")}...`);
-      mutants = await runGambit(solFiles);
+      mutants = await runGambit(solFiles, workerCount);
     } else {
       console.log("No files specified, using existing gambit_out/...");
       mutants = await loadExistingMutants();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@godzillaba/mutest",
-  "version": "1.3.5",
+  "version": "1.4.1",
   "bin": {
     "mutest": "./index.ts"
   },