From 50ac0fcba86df4ae8811e6b3fe3f33a0a5a47342 Mon Sep 17 00:00:00 2001
From: Viktorplus <36795799+viktorplus@users.noreply.github.com>
Date: Sun, 7 Jun 2026 16:41:29 +0200
Subject: [PATCH] test: cover kimi agent staging cleanup tracking

---
 tests/runtime-artifact-layout.test.cjs | 37 ++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/tests/runtime-artifact-layout.test.cjs b/tests/runtime-artifact-layout.test.cjs
index bb1d5d541..5cc091006 100644
--- a/tests/runtime-artifact-layout.test.cjs
+++ b/tests/runtime-artifact-layout.test.cjs
@@ -23,6 +23,7 @@ const fs = require('fs');
 const path = require('path');
 
 const { resolveRuntimeArtifactLayout } = require('../gsd-core/bin/lib/runtime-artifact-layout.cjs');
+const installProfiles = require('../gsd-core/bin/lib/install-profiles.cjs');
 
 const FAKE_DIR = '/tmp/fake-config-dir';
 
@@ -444,6 +445,42 @@ describe('stage — skills kind (kimi global)', () => {
     assert.match(executorYaml, /kimi_cli\.tools\./);
     assert.doesNotMatch(executorYaml, /mcp__/);
   });
+
+  test('tracks Kimi agent staging dir before writing artifacts', () => {
+    const layout = resolveRuntimeArtifactLayout('kimi', FAKE_STAGE_DIR, 'global');
+    const agentsKind = layout.kinds.find(k => k.kind === 'kimi-agents');
+    assert.ok(agentsKind, 'should have a kimi-agents kind');
+
+    const originalWriteFileSync = fs.writeFileSync;
+    const before = new Set(installProfiles.STAGED_DIRS);
+    let added = [];
+
+    try {
+      fs.writeFileSync = function writeFileSyncWithInjectedFailure(file, ...args) {
+        const filePath = String(file);
+        if (filePath.includes('gsd-kimi-agents-') && path.basename(filePath) === 'gsd.yaml') {
+          throw new Error('forced Kimi stage write failure');
+        }
+        return originalWriteFileSync.call(this, file, ...args);
+      };
+
+      assert.throws(
+        () => agentsKind.stage(PROFILE_FULL),
+        /forced Kimi stage write failure/
+      );
+
+      added = [...installProfiles.STAGED_DIRS]
+        .filter(dir => !before.has(dir) && path.basename(dir).startsWith('gsd-kimi-agents-'));
+      assert.strictEqual(added.length, 1, 'partially written Kimi stage dir must be tracked for cleanup');
+      assert.ok(fs.existsSync(added[0]), 'tracked partial Kimi stage dir should exist');
+    } finally {
+      fs.writeFileSync = originalWriteFileSync;
+      for (const dir of added) {
+        fs.rmSync(dir, { recursive: true, force: true });
+        installProfiles.STAGED_DIRS.delete(dir);
+      }
+    }
+  });
 });
 
 describe('stage — opencode commands kind', () => {