talon-agent 1.0.0 → 1.2.0

package/src/__tests__/fuzz.test.ts

@@ -51,6 +51,11 @@ const { handleSharedAction } = await import("../core/gateway-actions.js");
 const { resolveModelName } = await import("../storage/chat-settings.js");
 const { Cron } = await import("croner");
 
+// ── Configuration ───────────────────────────────────────────────────────────
+
+const NUM_RUNS = Number(process.env.FAST_CHECK_NUM_RUNS) || 100;
+const fcParams = { numRuns: NUM_RUNS };
+
 // ── Fuzz tests ───────────────────────────────────────────────────────────────
 
 describe("fuzz: classify()", () => {
@@ -63,6 +68,7 @@ describe("fuzz: classify()", () => {
         expect(typeof result.retryable).toBe("boolean");
         expect(typeof result.message).toBe("string");
       }),
+      fcParams,
     );
   });
 
@@ -73,6 +79,7 @@ describe("fuzz: classify()", () => {
         expect(result).toBeInstanceOf(TalonError);
         expect(result.reason).toBeTruthy();
       }),
+      fcParams,
     );
   });
 
@@ -93,6 +100,7 @@ describe("fuzz: classify()", () => {
           expect(result.reason).toBeTruthy();
         },
       ),
+      fcParams,
     );
   });
 
@@ -115,6 +123,7 @@ describe("fuzz: classify()", () => {
         const result = classify(input);
         expect(validReasons.has(result.reason)).toBe(true);
       }),
+      fcParams,
     );
   });
 
@@ -126,18 +135,25 @@ describe("fuzz: classify()", () => {
           expect(result.retryAfterMs).toBeGreaterThanOrEqual(0);
         }
       }),
+      fcParams,
     );
   });
 });
 
 // Real cron validator (the import is mocked, so we use Cron directly)
-function realValidateCron(expr: string, timezone?: string): { valid: boolean; error?: string } {
+function realValidateCron(
+  expr: string,
+  timezone?: string,
+): { valid: boolean; error?: string } {
   try {
     const cron = new Cron(expr, { timezone: timezone ?? undefined });
     cron.nextRun();
     return { valid: true };
   } catch (err) {
-    return { valid: false, error: err instanceof Error ? err.message : String(err) };
+    return {
+      valid: false,
+      error: err instanceof Error ? err.message : String(err),
+    };
   }
 }
 
@@ -151,6 +167,7 @@ describe("fuzz: validateCronExpression()", () => {
           expect(typeof result.error).toBe("string");
         }
       }),
+      fcParams,
     );
   });
 
@@ -160,6 +177,7 @@ describe("fuzz: validateCronExpression()", () => {
         const result = realValidateCron(expr, tz);
         expect(typeof result.valid).toBe("boolean");
       }),
+      fcParams,
     );
   });
 });
@@ -170,15 +188,24 @@ describe("fuzz: handleSharedAction() — unknown actions", () => {
       fc.asyncProperty(fc.string(), async (actionName) => {
         // Skip known action names to test only the default/unknown path
         const knownActions = new Set([
-          "read_history", "search_history", "get_user_messages",
-          "list_known_users", "list_media", "web_search", "fetch_url",
-          "create_cron_job", "list_cron_jobs", "edit_cron_job", "delete_cron_job",
+          "read_history",
+          "search_history",
+          "get_user_messages",
+          "list_known_users",
+          "list_media",
+          "web_search",
+          "fetch_url",
+          "create_cron_job",
+          "list_cron_jobs",
+          "edit_cron_job",
+          "delete_cron_job",
         ]);
         if (knownActions.has(actionName)) return;
 
         const result = await handleSharedAction({ action: actionName }, 123);
         expect(result).toBeNull();
       }),
+      fcParams,
     );
   });
 
@@ -192,9 +219,17 @@ describe("fuzz: handleSharedAction() — unknown actions", () => {
         }),
         async (body) => {
           const knownActions = new Set([
-            "read_history", "search_history", "get_user_messages",
-            "list_known_users", "list_media", "web_search", "fetch_url",
-            "create_cron_job", "list_cron_jobs", "edit_cron_job", "delete_cron_job",
+            "read_history",
+            "search_history",
+            "get_user_messages",
+            "list_known_users",
+            "list_media",
+            "web_search",
+            "fetch_url",
+            "create_cron_job",
+            "list_cron_jobs",
+            "edit_cron_job",
+            "delete_cron_job",
           ]);
           if (knownActions.has(body.action)) return;
 
@@ -202,6 +237,7 @@ describe("fuzz: handleSharedAction() — unknown actions", () => {
           expect(result).toBeNull();
         },
       ),
+      fcParams,
     );
   });
 });
@@ -216,7 +252,10 @@ describe("fuzz: fetch_url URL validation", () => {
 
     fc.assert(
       fc.asyncProperty(fc.string(), async (url) => {
-        const result = await handleSharedAction({ action: "fetch_url", url }, 123);
+        const result = await handleSharedAction(
+          { action: "fetch_url", url },
+          123,
+        );
         // Should either return an error result or handle gracefully
         expect(result).not.toBeUndefined();
         if (result !== null) {
@@ -226,6 +265,7 @@ describe("fuzz: fetch_url URL validation", () => {
           }
         }
       }),
+      fcParams,
     );
 
     globalThis.fetch = originalFetch;
@@ -234,11 +274,22 @@ describe("fuzz: fetch_url URL validation", () => {
   it("rejects all non-http/https protocols", () => {
     fc.assert(
       fc.asyncProperty(
-        fc.constantFrom("ftp", "file", "javascript", "data", "ws", "wss", "ssh"),
+        fc.constantFrom(
+          "ftp",
+          "file",
+          "javascript",
+          "data",
+          "ws",
+          "wss",
+          "ssh",
+        ),
         fc.webUrl(),
         async (protocol, path) => {
           const url = `${protocol}://${path.replace(/^https?:\/\//, "")}`;
-          const result = await handleSharedAction({ action: "fetch_url", url }, 123);
+          const result = await handleSharedAction(
+            { action: "fetch_url", url },
+            123,
+          );
           if (result !== null) {
             // Should not be ok for non-http protocols
             // (some invalid URLs will get "Invalid URL" error at parse time)
@@ -246,6 +297,7 @@ describe("fuzz: fetch_url URL validation", () => {
           }
         },
       ),
+      fcParams,
     );
   });
 });
@@ -257,6 +309,7 @@ describe("fuzz: resolveModelName()", () => {
         const result = resolveModelName(input);
         expect(typeof result).toBe("string");
       }),
+      fcParams,
     );
   });
 
@@ -266,11 +319,22 @@ describe("fuzz: resolveModelName()", () => {
         const result = resolveModelName(input);
         expect(result).toBe(result.trim());
       }),
+      fcParams,
     );
   });
 
   it("known aliases always resolve to claude model names", () => {
-    const aliases = ["sonnet", "opus", "haiku", "sonnet-4.6", "opus-4.6", "haiku-4.5", "sonnet-4-6", "opus-4-6", "haiku-4-5"];
+    const aliases = [
+      "sonnet",
+      "opus",
+      "haiku",
+      "sonnet-4.6",
+      "opus-4.6",
+      "haiku-4.5",
+      "sonnet-4-6",
+      "opus-4-6",
+      "haiku-4-5",
+    ];
     fc.assert(
       fc.property(
         fc.constantFrom(...aliases),
@@ -280,6 +344,7 @@ describe("fuzz: resolveModelName()", () => {
           expect(result).toMatch(/^claude-/);
         },
       ),
+      fcParams,
     );
   });
 
@@ -288,15 +353,22 @@ describe("fuzz: resolveModelName()", () => {
       fc.property(fc.string(), (input) => {
         const trimmed = input.trim().toLowerCase();
         const knownAliases = [
-          "sonnet", "opus", "haiku",
-          "sonnet-4.6", "opus-4.6", "haiku-4.5",
-          "sonnet-4-6", "opus-4-6", "haiku-4-5",
+          "sonnet",
+          "opus",
+          "haiku",
+          "sonnet-4.6",
+          "opus-4.6",
+          "haiku-4.5",
+          "sonnet-4-6",
+          "opus-4-6",
+          "haiku-4-5",
         ];
         if (knownAliases.includes(trimmed)) return;
 
         const result = resolveModelName(input);
         expect(result).toBe(input.trim());
       }),
+      fcParams,
     );
   });
 });