From a89d8c3759ec11b76878ba5e28472d25483cebf6 Mon Sep 17 00:00:00 2001
From: Haitao Pan <manbuzhe2009@qq.com>
Date: Tue, 5 May 2026 14:48:46 +0800
Subject: [PATCH] test: cover artifact read limits

---
 src/exportArtifacts.test.ts | 45 +++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/src/exportArtifacts.test.ts b/src/exportArtifacts.test.ts
index cec328c..2d4fedb 100644
--- a/src/exportArtifacts.test.ts
+++ b/src/exportArtifacts.test.ts
@@ -176,6 +176,32 @@ describe("exportXWorkmateArtifacts", () => {
     });
   });
 
+  it("reads artifact metadata without inline content when the file exceeds the limit", async () => {
+    const root = await fs.mkdtemp(path.join(os.tmpdir(), "tmp-xworkmate-artifacts-"));
+    await fs.writeFile(path.join(root, "large.bin"), Buffer.from("large-content"));
+
+    const result = await readXWorkmateArtifact({
+      params: {
+        sessionKey: "thread-main",
+        runId: "run-1",
+        relativePath: "large.bin",
+        maxInlineBytes: 2,
+      },
+      pluginConfig: { workspaceDir: root },
+    });
+
+    expect(result.artifacts).toHaveLength(1);
+    expect(result.artifacts[0]).toMatchObject({
+      relativePath: "large.bin",
+      contentType: "application/octet-stream",
+      sizeBytes: Buffer.byteLength("large-content"),
+      sha256: createHash("sha256").update("large-content").digest("hex"),
+    });
+    expect(result.artifacts[0]?.encoding).toBeUndefined();
+    expect(result.artifacts[0]?.content).toBeUndefined();
+    expect(result.warnings).toContain("large.bin exceeds maxInlineBytes and was not inlined");
+  });
+
   it("rejects relative path traversal when reading artifacts", async () => {
     const root = await fs.mkdtemp(path.join(os.tmpdir(), "tmp-xworkmate-artifacts-"));
 
@@ -190,4 +216,23 @@ describe("exportXWorkmateArtifacts", () => {
       }),
     ).rejects.toThrow("relativePath must stay inside the workspace");
   });
+
+  it("rejects symlink escapes when reading artifacts", async () => {
+    const root = await fs.mkdtemp(path.join(os.tmpdir(), "tmp-xworkmate-artifacts-"));
+    const outsideRoot = await fs.mkdtemp(path.join(os.tmpdir(), "tmp-xworkmate-outside-"));
+    const outsideFile = path.join(outsideRoot, "secret.txt");
+    await fs.writeFile(outsideFile, "secret");
+    await fs.symlink(outsideFile, path.join(root, "linked-secret.txt"));
+
+    await expect(
+      readXWorkmateArtifact({
+        params: {
+          sessionKey: "thread-main",
+          runId: "run-1",
+          relativePath: "linked-secret.txt",
+        },
+        pluginConfig: { workspaceDir: root },
+      }),
+    ).rejects.toThrow("relativePath must stay inside the workspace");
+  });
 });