From fb7fa980b8411e285c5e396a42868b2f72d68b92 Mon Sep 17 00:00:00 2001
From: Tianyi Cui <53024+tianyicui@users.noreply.github.com>
Date: Wed, 15 Jul 2026 22:58:44 +0800
Subject: [PATCH] test(examples): include task controls in CLI composition

---
 packages/examples/cli-demo/tests/cli-demo.spec.ts | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/packages/examples/cli-demo/tests/cli-demo.spec.ts b/packages/examples/cli-demo/tests/cli-demo.spec.ts
index 4cb7a243a5..1fb0e09999 100644
--- a/packages/examples/cli-demo/tests/cli-demo.spec.ts
+++ b/packages/examples/cli-demo/tests/cli-demo.spec.ts
@@ -89,7 +89,14 @@ describe('dsh-cli-demo app composition', () => {
       ctx.tools.register({ name, description: name, parameters: {}, execute: async () => [] })
     }
     expect(JSON.stringify(await composePrefix(ctx))).toContain('- `cli-skill`: CLI...')
-    expect((await ctx.systemPrompt.assemble()).tools.map(tool => tool.name)).toEqual(['zulu', 'alpha', 'skill'])
+    expect((await ctx.systemPrompt.assemble()).tools.map(tool => tool.name)).toEqual([
+      'zulu',
+      'alpha',
+      'skill',
+      'task_kill',
+      'task_list',
+      'task_output',
+    ])
   })
 
   it('exposes the Loader-safe namespace plugin shape and schema', () => {