test: deflake agent image root checks

RomneyDa · RomneyDa · commit 144fb1e30769 · 2026-05-27T21:16:34.000-07:00
diff --git a/src/agents/tools/image-tool.test.ts b/src/agents/tools/image-tool.test.ts
@@ -648,12 +648,14 @@ function installFastLocalImageProviderStubs(...providers: MediaUnderstandingProv
           fileName: path.basename(mediaUrl),
         };
       },
-      optimizeImageBufferForWebMedia: async ({ buffer, contentType, fileName }) => ({
-        buffer,
-        contentType: contentType ?? "image/png",
-        kind: "image",
-        fileName,
-      }),
+      optimizeImageBufferForWebMedia: async ({ buffer, contentType, fileName }) => {
+        return {
+          buffer,
+          contentType: contentType ?? "image/png",
+          kind: "image",
+          fileName,
+        };
+      },
     }),
   });
 }
@@ -1793,7 +1795,7 @@ describe("image tool implicit imageModel config", () => {
         await fs.rm(attachmentRoot, { recursive: true, force: true });
       }
     });
-  }, 240_000);
+  });
 
   it("allows image paths from current iMessage wildcard attachment roots", async () => {
     await withTempAgentDir(async (agentDir) => {
@@ -1852,7 +1854,7 @@ describe("image tool implicit imageModel config", () => {
         await fs.rm(attachmentRootParent, { recursive: true, force: true });
       }
     });
-  }, 240_000);
+  });
 
   it("allows workspace images via createOpenClawCodingTools when workspace root is explicit", async () => {
     await withTempWorkspacePng(async ({ workspaceDir, imagePath }) => {
