fix: change the tests

src/index.ts

@@ -15,25 +15,46 @@ const auditorResult = await auditorAgent.invoke({ solidityFile: coderResult.cont
 console.log("\n======= Auditor =======");
 // console.log(auditorResult.vulnerabilities);
 
-function mapFindingToReport(finding: Finding, sourceCode: string): VulnerabilityReport {
-  // Extract contract name from path (e.g., "contracts/CafeToken.sol" -> "CafeToken")
-  const nameMatch = finding.path.match(/([^\/]+)\.sol$/);
+function mapFindingToReport(finding: Partial<Finding> & { type?: string; severity?: string }, sourceCode: string): VulnerabilityReport {
+  const title = typeof finding.title === "string" && finding.title.trim().length > 0
+    ? finding.title
+    : typeof finding.type === "string" && finding.type.trim().length > 0
+      ? finding.type
+      : "Unknown vulnerability";
+
+  const description = typeof finding.description === "string" && finding.description.trim().length > 0
+    ? finding.description
+    : "No description provided by auditor.";
+
+  const nameMatch = finding.path?.match(/([^\/]+)\.sol$/);
   const contractName = nameMatch ? nameMatch[1] : "TargetContract";
 
+  const exploitablePaths = Array.isArray(finding.judgeReview?.exploitablePaths)
+    ? finding.judgeReview.exploitablePaths
+    : [];
+
+  const severity = finding.severity === "high" || finding.severity === "medium" || finding.severity === "low"
+    ? finding.severity
+    : "low";
+
+  if (!finding.path || !finding.judgeReview) {
+    console.warn("Auditor returned incomplete finding; using fallbacks for PoC generation.");
+  }
+
   return {
-    id: finding.title.toLowerCase().replace(/[^a-z0-9]+/g, "-").slice(0, 50),
-    severity: finding.severity === "high" ? "high" : finding.severity === "medium" ? "medium" : "low",
-    type: "custom",
-    title: finding.title,
-    description: finding.description,
+    id: title.toLowerCase().replace(/[^a-z0-9]+/g, "-").slice(0, 50),
+    severity,
+    type: typeof finding.type === "string" && finding.type.trim().length > 0 ? finding.type : "custom",
+    title,
+    description,
     affectedContract: {
       name: contractName,
-      sourceCode: sourceCode,
+      sourceCode,
     },
-    attackVector: finding.judgeReview.exploitablePaths[0] || "Unknown vector",
-    exploitablePaths: finding.judgeReview.exploitablePaths,
-    codeSnippet: finding.codeSnippet,
-    location: finding.location
+    attackVector: exploitablePaths[0] ?? "Unknown vector",
+    exploitablePaths,
+    codeSnippet: typeof finding.codeSnippet === "string" ? finding.codeSnippet : undefined,
+    location: typeof finding.location === "string" ? finding.location : undefined,
   };
 }
 

tests/e2e/poc-generator.test.ts

@@ -1,3 +1,5 @@
+import { describe, expect, it } from "vitest";
+
 import { runPoCGenerator } from "../../src/agents/tester/index.js";
 import { VulnerabilityReport } from "../../src/agents/tester/types.js";
 
@@ -26,31 +28,12 @@ const mockReport: VulnerabilityReport = {
   suggestedCheatcodes: ["vm.deal", "vm.startPrank", "vm.stopPrank"],
 };
 
-async function runE2ETest() {
-  console.log("Iniciando smoke test end-to-end (Reentrancy)...");
-  
-  // Garantir que o sandbox está limpo
-  // No mundo real, scripts/setup-sandbox.sh deve ser rodado uma vez no setup do sistema
-  
-  try {
+describe("PoC generator (e2e)", () => {
+  it("generates a PoC from a vulnerability report", async () => {
     const result = await runPoCGenerator(mockReport);
 
-    console.log("\n======= E2E RESULT =======");
-    console.log(`Status: ${result.status}`);
-    console.log(`Iterações: ${result.iterations}`);
-    console.log(`Logs: ${result.executionLogs.length} entrada(s)`);
-
-    console.assert(result.status === "success", `FALHOU: status esperado 'success', recebido '${result.status}'`);
-    console.assert(result.solidityCode.includes("test_Exploit"), "FALHOU: código não contém test_Exploit");
-
-    if (result.status === "success") {
-        console.log("\nSmoke test PASSOU: Vulnerabilidade confirmada via PoC!");
-    } else {
-        console.error("\nSmoke test FALHOU: Agente não conseguiu gerar PoC válido.");
-    }
-  } catch (error) {
-    console.error("Erro fatal no teste E2E:", error);
-  }
-}
-
-runE2ETest().catch(console.error);
+    expect(result.status).toBe("success");
+    expect(result.solidityCode).toContain("test_Exploit");
+    expect(result.executionLogs.length).toBeGreaterThan(0);
+  }, 120000);
+});

tests/scaffold.test.ts

@@ -1,15 +1,19 @@
+import { describe, expect, it } from "vitest";
+
 import { generateLocalScaffold } from "../src/agents/tester/tools/scaffoldGenerator.js";
 
-const mockReport = {
-  id: "t1", 
-  severity: "high" as const, 
-  type: "reentrancy", 
-  title: "Reentrancy in withdraw()",
-  description: "withdraw() sends ETH before zeroing balance",
-  attackVector: "Malicious callback",
-  affectedContract: { 
-    name: "VulnerableBank", 
-    sourceCode: `
+describe("generateLocalScaffold", () => {
+  it("creates a basic exploit scaffold", () => {
+    const mockReport = {
+      id: "t1",
+      severity: "high" as const,
+      type: "reentrancy",
+      title: "Reentrancy in withdraw()",
+      description: "withdraw() sends ETH before zeroing balance",
+      attackVector: "Malicious callback",
+      affectedContract: {
+        name: "VulnerableBank",
+        sourceCode: `
 pragma solidity ^0.8.20;
 contract VulnerableBank {
   mapping(address=>uint) public balances;
@@ -18,18 +22,15 @@ contract VulnerableBank {
     (bool ok,) = msg.sender.call{value:a}("");
     require(ok); balances[msg.sender] = 0;
   }
-}`
-  }
-};
-
-const scaffold = generateLocalScaffold(mockReport);
-console.log("--- Scaffold Output ---");
-console.log(scaffold);
-console.log("--- End Scaffold ---");
+}`,
+      },
+    };
 
-console.assert(scaffold.includes("contract ExploitTest is Test"), "Scaffold missing ExploitTest");
-console.assert(scaffold.includes("VulnerableBank target"), "Scaffold missing target declaration");
-console.assert(scaffold.includes("function setUp()"), "Scaffold missing setUp");
-console.assert(scaffold.includes("function test_Exploit()"), "Scaffold missing test_Exploit");
+    const scaffold = generateLocalScaffold(mockReport);
 
-console.log("Scaffold generator test passed");
+    expect(scaffold).toContain("contract ExploitTest is Test");
+    expect(scaffold).toContain("VulnerableBank target");
+    expect(scaffold).toContain("function setUp()");
+    expect(scaffold).toContain("function test_Exploit()");
+  });
+});

tests/state.test.ts

@@ -1,23 +1,16 @@
-import { PoCStateAnnotation } from "../src/agents/tester/state.js";
+import { describe, expect, it } from "vitest";
 
-const s = (PoCStateAnnotation as any).spec;
-if (s.iterations !== undefined) {
-    console.log("Estado OK: iterations existe");
-} else {
-    console.error("Erro: iterations não existe no spec");
-    process.exit(1);
-}
+import { PoCStateAnnotation } from "../src/agents/tester/state.js";
 
-// Teste de reducer aditivo para iterations
-const iterationsSpec = s.iterations;
-const initial = 0;
-const update = 1;
-const result = (iterationsSpec.reducer || ((x: number, y: number) => x + y))(initial, update);
-if (result === 1) {
-    console.log("Reducer aditivo OK: 0 + 1 = 1");
-} else {
-    console.error(`Erro no reducer aditivo: esperado 1, recebido ${result}`);
-    process.exit(1);
-}
+describe("PoCStateAnnotation", () => {
+  it("defines the iterations field", () => {
+    const spec = (PoCStateAnnotation as any).spec;
+    expect(spec.iterations).toBeDefined();
+  });
 
-console.log("Teste de estado concluído com sucesso");
+  it("uses additive reducer for iterations", () => {
+    const spec = (PoCStateAnnotation as any).spec;
+    const reducer = spec.iterations.reducer ?? ((x: number, y: number) => x + y);
+    expect(reducer(0, 1)).toBe(1);
+  });
+});