Merge branch 'codex/file-tree-virtualization'

Author: yeahhe365

structure-insight/components/FileTree.test.tsx

@@ -0,0 +1,127 @@
+import React from 'react';
+import { cleanup, fireEvent, render, screen } from '@testing-library/react';
+import { afterEach, describe, expect, it, vi } from 'vitest';
+import FileTree from './FileTree';
+import type { FileNode } from '../types';
+
+vi.mock('react-virtuoso', () => ({
+  Virtuoso: ({ data, itemContent }: any) => (
+    <div data-testid="virtuoso" data-count={String(data.length)}>
+      {data.slice(0, 12).map((item: any, index: number) => (
+        <div key={item.path}>{itemContent(index, item)}</div>
+      ))}
+    </div>
+  ),
+}));
+
+const makeLargeTree = (): FileNode[] => [
+  {
+    name: 'src',
+    path: 'src',
+    isDirectory: true,
+    children: Array.from({ length: 200 }, (_, index) => ({
+      name: `file-${index}.ts`,
+      path: `src/file-${index}.ts`,
+      isDirectory: false,
+      children: [],
+      status: 'processed' as const,
+    })),
+  },
+];
+
+afterEach(() => {
+  cleanup();
+});
+
+describe('FileTree virtualization', () => {
+  it('passes the flattened visible rows into the virtual list instead of mounting the full tree', () => {
+    render(
+      <FileTree
+        nodes={makeLargeTree()}
+        onFileSelect={vi.fn()}
+        onDeleteFile={vi.fn()}
+        onCopyPath={vi.fn()}
+        onToggleExclude={vi.fn()}
+        selectedFilePath={null}
+        showCharCount={false}
+      />
+    );
+
+    expect(screen.getByTestId('virtuoso').getAttribute('data-count')).toBe('201');
+    expect(screen.getByText('file-0.ts')).not.toBeNull();
+    expect(screen.queryByText('file-199.ts')).toBeNull();
+  });
+
+  it('removes descendant rows when collapsing a directory', () => {
+    render(
+      <FileTree
+        nodes={[
+          {
+            name: 'src',
+            path: 'src',
+            isDirectory: true,
+            children: [
+              {
+                name: 'index.ts',
+                path: 'src/index.ts',
+                isDirectory: false,
+                children: [],
+                status: 'processed',
+              },
+            ],
+          },
+        ]}
+        onFileSelect={vi.fn()}
+        onDeleteFile={vi.fn()}
+        onCopyPath={vi.fn()}
+        onToggleExclude={vi.fn()}
+        selectedFilePath={null}
+        showCharCount={false}
+      />
+    );
+
+    expect(screen.getByText('index.ts')).not.toBeNull();
+    fireEvent.click(screen.getByText('src'));
+    expect(screen.queryByText('index.ts')).toBeNull();
+  });
+
+  it('navigates visible rows with the keyboard and opens processed files on Enter', () => {
+    const onFileSelect = vi.fn();
+
+    render(
+      <FileTree
+        nodes={[
+          {
+            name: 'src',
+            path: 'src',
+            isDirectory: true,
+            children: [
+              {
+                name: 'index.ts',
+                path: 'src/index.ts',
+                isDirectory: false,
+                children: [],
+                status: 'processed',
+              },
+            ],
+          },
+        ]}
+        onFileSelect={onFileSelect}
+        onDeleteFile={vi.fn()}
+        onCopyPath={vi.fn()}
+        onToggleExclude={vi.fn()}
+        selectedFilePath={null}
+        showCharCount={false}
+      />
+    );
+
+    const tree = screen.getByRole('tree', { name: '资源管理器' });
+    tree.focus();
+
+    fireEvent.keyDown(tree, { key: 'ArrowDown' });
+    fireEvent.keyDown(tree, { key: 'ArrowDown' });
+    fireEvent.keyDown(tree, { key: 'Enter' });
+
+    expect(onFileSelect).toHaveBeenCalledWith('src/index.ts');
+  });
+});