From 897683e63f5ecedf66f5660eb70d0ec6613a039b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E5=BE=B7=E5=AE=87?= <kvtodev@outlook.com>
Date: Sat, 5 Sep 2026 13:17:51 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A1=A5=E5=85=85UI=E6=A0=91=E5=88=A0=E9=99=A4?=
 =?UTF-8?q?=E8=AE=A1=E6=95=B0=E6=B5=8B=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

覆盖嵌套节点定位与后代节点递归计数
---
 .../tests/uiTreeUtils.test.ts                 | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 apps/ai-game-creator-shell/tests/uiTreeUtils.test.ts

diff --git a/apps/ai-game-creator-shell/tests/uiTreeUtils.test.ts b/apps/ai-game-creator-shell/tests/uiTreeUtils.test.ts
new file mode 100644
index 000000000..196077400
--- /dev/null
+++ b/apps/ai-game-creator-shell/tests/uiTreeUtils.test.ts
@@ -0,0 +1,30 @@
+import { describe, expect, it } from 'vitest';
+
+import {
+  countUiNodeDescendants,
+  findUiNode,
+} from '../src/features/ui-editor/treeUtils';
+import type { Node } from '../src/features/ui-editor/types/Node';
+
+function node(id: string, children: Node[] = []): Node {
+  return {
+    id,
+    layout: {} as Node['layout'],
+    metadata: {} as Node['metadata'],
+    components: [],
+    children_display_mode: 'Stack',
+    children,
+  };
+}
+
+describe('ui tree utilities', () => {
+  it('finds a nested node and counts all descendants', () => {
+    const nested = node('nested', [node('leaf')]);
+    const root = node('root', [node('page'), nested]);
+
+    expect(findUiNode(root, 'nested')).toBe(nested);
+    expect(findUiNode(root, 'missing')).toBeNull();
+    expect(countUiNodeDescendants(nested)).toBe(1);
+    expect(countUiNodeDescendants(root)).toBe(3);
+  });
+});