workspace-browser.spec.tsx 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785
  1. // @vitest-environment jsdom
  2. import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
  3. import { act, cleanup, createEvent, fireEvent, render, screen, waitFor } from '@testing-library/react'
  4. import { bindSnapshotSelector } from '@deepseek-ai/dsh-client-web-react'
  5. import type {
  6. SessionId, SessionListState, SessionSummary, WorkspaceId, WorkspaceListState, WorkspaceView,
  7. } from '@deepseek-ai/dsh-client-runtime/client'
  8. import { makeTranslate } from '@deepseek-ai/dsh-client-test-runtime'
  9. import { zh as commonZh } from '@deepseek-ai/dsh-client-locale/src/locales/zh.ts'
  10. import type { WorkspaceBrowserProps } from '../src/client/contract/slots.ts'
  11. import { createWorkspaceViewStore } from '../src/client/stores.ts'
  12. import { WorkspaceBrowser } from '../src/client/WorkspaceBrowser.tsx'
  13. import { zh } from '../src/client/locales.ts'
  14. afterEach(cleanup)
  15. beforeEach(() => { localStorage.clear() })
  16. // The seat's key domain is workspace ∪ common; the stub mirrors the real
  17. // lookup chain (namespace, then common vocabulary, then the key).
  18. const t: WorkspaceBrowserProps['t'] = makeTranslate(zh, commonZh)
  19. const sid = (id: string) => id as SessionId
  20. const wid = (id: string) => id as WorkspaceId
  21. const summary = (id: string, updatedAt: number, overrides: Partial<SessionSummary> = {}): SessionSummary => ({
  22. id: sid(id), displayTitle: id, running: false, blank: false, updatedAt, ...overrides,
  23. })
  24. const sessionState = (items: readonly SessionSummary[], overrides: Partial<SessionListState> = {}): SessionListState => ({
  25. ids: items.map(item => item.id),
  26. byId: Object.fromEntries(items.map(item => [item.id, item])),
  27. current: undefined,
  28. phase: 'ready',
  29. subagentsByParent: {},
  30. currentAddress: undefined,
  31. ...overrides,
  32. })
  33. const workspace = (id: string, sessionIds: string[], title = id): WorkspaceView => ({
  34. workspaceId: wid(id), path: `/projects/${id}`, title,
  35. sessionIds: sessionIds.map(sid), createdAt: '2026-01-01T00:00:00.000Z', updatedAt: '2026-01-01T00:00:00.000Z',
  36. })
  37. const workspaceState = (items: readonly WorkspaceView[], archivedSessionIds: readonly SessionId[] = []): WorkspaceListState => ({
  38. items, archivedSessionIds, state: 'idle', phase: 'ready', error: null, baselinesReady: true,
  39. recentWorkspaceId: items[0]?.workspaceId,
  40. })
  41. function hook<T>(snapshot: T) {
  42. return function select<S>(selector: (state: T) => S): S { return selector(snapshot) }
  43. }
  44. /** jsdom lacks DragEvent — the fireEvent fallback drops clientY, so pin it on the built event. */
  45. function fireDrag(row: HTMLElement, kind: 'dragOver' | 'drop', clientY: number): void {
  46. const event = kind === 'dragOver' ? createEvent.dragOver(row) : createEvent.drop(row)
  47. Object.defineProperty(event, 'clientY', { value: clientY })
  48. Object.defineProperty(event, 'dataTransfer', { value: { effectAllowed: '', dropEffect: '' } })
  49. fireEvent(row, event)
  50. }
  51. function mount(overrides: Partial<WorkspaceBrowserProps> = {}) {
  52. const store = createWorkspaceViewStore().create()
  53. const props: WorkspaceBrowserProps = {
  54. wide: true,
  55. expandSidebar: vi.fn(),
  56. useSessions: hook(sessionState([])),
  57. useWorkspaces: hook(workspaceState([])),
  58. useStore: bindSnapshotSelector(store),
  59. actions: store.actions,
  60. startSession: vi.fn(),
  61. open: vi.fn(),
  62. searchSessions: vi.fn(async () => ({ items: [], hasMore: false })),
  63. searchResultLimit: 20,
  64. renameSession: vi.fn(async () => {}),
  65. forkSession: vi.fn(),
  66. renameWorkspace: vi.fn(async () => {}),
  67. deleteWorkspace: vi.fn(async () => {}),
  68. archiveSession: vi.fn(async () => {}),
  69. insertSessionBefore: vi.fn(async () => {}),
  70. createWorkspace: vi.fn(async () => workspace('created', [])),
  71. useDirectoryFlow: bindSnapshotSelector({ getSnapshot: () => true, subscribe: () => () => {} }),
  72. renderSlot: ((_name: string, owner: { open: boolean }) => (owner.open ? <div data-testid="directory-flow" /> : null)) as never,
  73. t,
  74. ...overrides,
  75. }
  76. const view = render(<WorkspaceBrowser {...props} />)
  77. return { view, props, store }
  78. }
  79. /** Re-render with (possibly) changed props — WorkspaceBrowser has no side channel. */
  80. function rerender(b: ReturnType<typeof mount>, overrides: Partial<WorkspaceBrowserProps>) {
  81. Object.assign(b.props, overrides)
  82. b.view.rerender(<WorkspaceBrowser {...b.props} />)
  83. }
  84. describe('WorkspaceBrowser', () => {
  85. it('renders the grouped tree by default and switches to the flat list via Group by', () => {
  86. const sessions = sessionState([summary('alpha-s', 2), summary('beta-s', 1)])
  87. const b = mount({
  88. useSessions: hook(sessions),
  89. useWorkspaces: hook(workspaceState([workspace('alpha', ['alpha-s']), workspace('beta', ['beta-s'])])),
  90. })
  91. expect(screen.getByText('工作区')).toBeTruthy()
  92. expect(screen.getByText('alpha')).toBeTruthy()
  93. // Sessions hidden while their group is folded.
  94. expect(screen.queryByText('alpha-s')).toBeNull()
  95. fireEvent.click(screen.getByRole('button', { name: '分组方式' }))
  96. expect(screen.getByText('分组方式')).toBeTruthy() // the menu heading label
  97. fireEvent.click(screen.getByRole('menuitem', { name: '单列表' }))
  98. // Store-driven flip: title changes, rows flatten newest-first, headers gone.
  99. expect(b.store.getSnapshot().groupBy).toBe('flat')
  100. expect(screen.getByText('会话')).toBeTruthy()
  101. expect(screen.queryByText('alpha')).toBeNull()
  102. expect(screen.getByText('alpha-s')).toBeTruthy()
  103. expect(screen.getByText('beta-s')).toBeTruthy()
  104. // Back to workspace grouping through the same menu.
  105. fireEvent.click(screen.getByRole('button', { name: '分组方式' }))
  106. fireEvent.click(screen.getByRole('menuitem', { name: '按工作区' }))
  107. expect(b.store.getSnapshot().groupBy).toBe('workspace')
  108. expect(screen.getByText('工作区')).toBeTruthy()
  109. // Escape closes the menu without picking.
  110. fireEvent.click(screen.getByRole('button', { name: '分组方式' }))
  111. fireEvent.keyDown(document, { key: 'Escape' })
  112. expect(screen.queryByRole('menu')).toBeNull()
  113. expect(b.store.getSnapshot().groupBy).toBe('workspace')
  114. })
  115. it('expands a group on click and opens a session row', () => {
  116. const open = vi.fn()
  117. mount({
  118. useSessions: hook(sessionState([summary('alpha-s', 1)])),
  119. useWorkspaces: hook(workspaceState([workspace('alpha', ['alpha-s'])])),
  120. open,
  121. })
  122. fireEvent.click(screen.getByText('alpha'))
  123. fireEvent.click(screen.getByText('alpha-s'))
  124. expect(open).toHaveBeenCalledWith(sid('alpha-s'))
  125. // Collapse hides the row again.
  126. fireEvent.click(screen.getByText('alpha'))
  127. expect(screen.queryByText('alpha-s')).toBeNull()
  128. })
  129. it('archives a session from the row menu and hides archived rows in both modes', async () => {
  130. const archiveSession = vi.fn(async () => {})
  131. const b = mount({
  132. useSessions: hook(sessionState([summary('kept-s', 2), summary('gone-s', 1)])),
  133. useWorkspaces: hook(workspaceState([workspace('alpha', ['kept-s', 'gone-s'])])),
  134. archiveSession,
  135. })
  136. fireEvent.click(screen.getByText('alpha'))
  137. fireEvent.click(screen.getByRole('button', { name: '会话“gone-s”的操作' }))
  138. fireEvent.click(screen.getByRole('menuitem', { name: '归档会话' }))
  139. expect(archiveSession).toHaveBeenCalledWith(sid('gone-s'))
  140. // The archive-set echo hides the row in grouped mode (count included) and flat mode.
  141. rerender(b, { useWorkspaces: hook(workspaceState([workspace('alpha', ['kept-s', 'gone-s'])], [sid('gone-s')])) })
  142. expect(screen.queryByText('gone-s')).toBeNull()
  143. expect(screen.getByText('1 个会话')).toBeTruthy()
  144. fireEvent.click(screen.getByRole('button', { name: '分组方式' }))
  145. fireEvent.click(screen.getByRole('menuitem', { name: '单列表' }))
  146. expect(screen.getByText('kept-s')).toBeTruthy()
  147. expect(screen.queryByText('gone-s')).toBeNull()
  148. })
  149. it('logs and keeps the tree when the archive call rejects', async () => {
  150. const rejection = new Error('archive exploded')
  151. const archiveSession = vi.fn(async () => { throw rejection })
  152. const warn = vi.spyOn(console, 'warn').mockImplementation(() => {})
  153. try {
  154. mount({
  155. useSessions: hook(sessionState([summary('alpha-s', 1)])),
  156. useWorkspaces: hook(workspaceState([workspace('alpha', ['alpha-s'])])),
  157. archiveSession,
  158. })
  159. fireEvent.click(screen.getByText('alpha'))
  160. fireEvent.click(screen.getByRole('button', { name: '会话“alpha-s”的操作' }))
  161. fireEvent.click(screen.getByRole('menuitem', { name: '归档会话' }))
  162. await Promise.resolve()
  163. await Promise.resolve()
  164. expect(warn).toHaveBeenCalledWith('session archive rejected:', rejection)
  165. expect(screen.getByText('alpha-s')).toBeTruthy()
  166. } finally {
  167. warn.mockRestore()
  168. }
  169. })
  170. it('renders a fork child as a top-level row without a session twist', () => {
  171. const parent = summary('parent-s', 2)
  172. const child = { ...summary('child-s', 1), parentId: parent.id }
  173. mount({
  174. useSessions: hook(sessionState([parent, child])),
  175. useWorkspaces: hook(workspaceState([workspace('alpha', ['parent-s', 'child-s'])])),
  176. })
  177. fireEvent.click(screen.getByText('alpha'))
  178. expect(screen.getByText('child-s')).toBeTruthy()
  179. expect(screen.queryByRole('button', { name: /展开|收起/ })).toBeNull()
  180. expect(screen.getByText('child-s').closest('[role="treeitem"]')?.getAttribute('draggable')).toBe('true')
  181. })
  182. it('auto-expands the selected session group and starts a session from the group +', () => {
  183. const startSession = vi.fn()
  184. mount({
  185. useSessions: hook(sessionState([summary('alpha-s', 1)], { current: sid('alpha-s') })),
  186. useWorkspaces: hook(workspaceState([workspace('alpha', ['alpha-s'])])),
  187. startSession,
  188. })
  189. // The current-group effect expanded the owning group without a click.
  190. expect(screen.getByText('alpha-s')).toBeTruthy()
  191. fireEvent.click(screen.getByRole('button', { name: '在“alpha”中新建会话' }))
  192. expect(startSession).toHaveBeenCalledWith(wid('alpha'))
  193. })
  194. it('auto-expands the Ungrouped bucket for a loose current session; its header has no menu and its + is inert', () => {
  195. const startSession = vi.fn()
  196. mount({
  197. useSessions: hook(sessionState([summary('loose', 1)], { current: sid('loose') })),
  198. useWorkspaces: hook(workspaceState([workspace('alpha', [])])),
  199. startSession,
  200. })
  201. // The loose session's group is UNGROUPED_KEY: expanded by the effect.
  202. expect(screen.getByText('loose')).toBeTruthy()
  203. expect(screen.queryByRole('button', { name: '工作区“未分组”的操作' })).toBeNull()
  204. fireEvent.click(screen.getByRole('button', { name: '在“未分组”中新建会话' }))
  205. expect(startSession).not.toHaveBeenCalled()
  206. })
  207. it('keeps an already-expanded group when the selection moves within it', () => {
  208. const first = sessionState([summary('a', 2), summary('b', 1)], { current: sid('a') })
  209. const b = mount({
  210. useSessions: hook(first),
  211. useWorkspaces: hook(workspaceState([workspace('alpha', ['a', 'b'])])),
  212. })
  213. expect(screen.getByText('a')).toBeTruthy()
  214. // Selection hop inside the same group: the effect re-runs and leaves the
  215. // expansion list unchanged (no duplicate key, group still open).
  216. rerender(b, { useSessions: hook({ ...first, current: sid('b') }) })
  217. expect(screen.getByText('b')).toBeTruthy()
  218. fireEvent.click(screen.getByText('alpha'))
  219. expect(screen.queryByText('b')).toBeNull()
  220. })
  221. it('shows only the current blank session as the localized New Session, excluded from search', () => {
  222. const currentBlank = summary('alpha-blank', 9, { blank: true })
  223. const staleBlank = summary('beta-blank', 8, { blank: true })
  224. const sessions = sessionState(
  225. [currentBlank, staleBlank],
  226. { current: currentBlank.id },
  227. )
  228. const b = mount({
  229. useSessions: hook(sessions),
  230. useWorkspaces: hook(workspaceState([
  231. workspace('alpha', ['alpha-blank']), workspace('beta', ['beta-blank']),
  232. ])),
  233. })
  234. expect(screen.getByText('新会话')).toBeTruthy()
  235. expect(screen.queryByText('alpha-blank')).toBeNull()
  236. expect(screen.queryByText('beta-blank')).toBeNull()
  237. expect(screen.getByText('1 个会话')).toBeTruthy()
  238. rerender(b, { useSessions: hook({ ...sessions, current: staleBlank.id }) })
  239. expect(screen.getAllByText('新会话')).toHaveLength(1)
  240. b.store.actions.setGroupBy('flat')
  241. rerender(b, {})
  242. expect(screen.getAllByText('新会话')).toHaveLength(1)
  243. // Search excludes blank rows entirely — neither the canonical stored
  244. // title nor the localized display label participates in matching.
  245. fireEvent.change(screen.getByPlaceholderText('搜索名称、关键词…'), { target: { value: 'new session' } })
  246. expect(screen.queryByText('新会话')).toBeNull()
  247. fireEvent.change(screen.getByPlaceholderText('搜索名称、关键词…'), { target: { value: '新会话' } })
  248. expect(screen.queryByText('新会话')).toBeNull()
  249. })
  250. it('shows local metadata matches immediately, then clears back to the grouped tree', async () => {
  251. vi.useFakeTimers()
  252. try {
  253. const sessions = sessionState([
  254. summary('needle-row', 2, { displayTitle: 'Needle row' }),
  255. summary('other-row', 1, { displayTitle: 'Other row' }),
  256. ])
  257. mount({
  258. useSessions: hook(sessions),
  259. useWorkspaces: hook(workspaceState([workspace('alpha', ['needle-row', 'other-row'])])),
  260. })
  261. const input = screen.getByPlaceholderText<HTMLInputElement>('搜索名称、关键词…')
  262. fireEvent.change(input, { target: { value: 'needle' } })
  263. const resultTree = screen.getByRole('tree', { name: '搜索结果' })
  264. expect(screen.getByText('Needle row')).toBeTruthy()
  265. expect(screen.queryByText('Other row')).toBeNull()
  266. const status = screen.getByRole('status')
  267. expect(status.textContent).toBe('正在搜索会话历史…')
  268. expect(resultTree.contains(status)).toBe(false)
  269. fireEvent.change(input, { target: { value: 'zzz' } })
  270. await act(async () => { await vi.advanceTimersByTimeAsync(250) })
  271. expect(screen.getByText('无匹配会话')).toBeTruthy()
  272. fireEvent.click(screen.getByRole('button', { name: '清除搜索' }))
  273. expect(input.value).toBe('')
  274. expect(screen.getByRole('tree', { name: '会话' })).toBeTruthy()
  275. // Clicking the field row focuses the input (wide mode).
  276. fireEvent.click(input.parentElement as HTMLElement)
  277. expect(document.activeElement).toBe(input)
  278. } finally {
  279. vi.useRealTimers()
  280. }
  281. })
  282. it('adds Host content hits with context, shows the result bound, and opens without clearing the query', async () => {
  283. vi.useFakeTimers()
  284. try {
  285. const open = vi.fn()
  286. const searchSessions = vi.fn(async () => ({
  287. items: [{ sessionId: sid('body-hit'), snippet: '…the waterfall token appears here…' }],
  288. hasMore: true,
  289. }))
  290. mount({
  291. useSessions: hook(sessionState([
  292. summary('body-hit', 1, { displayTitle: 'Research notes' }),
  293. ])),
  294. useWorkspaces: hook(workspaceState([
  295. workspace('research', ['body-hit'], 'Research Workspace'),
  296. ])),
  297. open,
  298. searchSessions,
  299. })
  300. const input = screen.getByPlaceholderText<HTMLInputElement>('搜索名称、关键词…')
  301. fireEvent.change(input, { target: { value: 'waterfall token' } })
  302. expect(screen.getByText('正在搜索会话历史…')).toBeTruthy()
  303. expect(screen.queryByText('Research notes')).toBeNull()
  304. await act(async () => { await vi.advanceTimersByTimeAsync(250) })
  305. expect(searchSessions).toHaveBeenCalledWith('waterfall token', expect.any(AbortSignal))
  306. expect(screen.getByText('Research notes')).toBeTruthy()
  307. expect(screen.getByText('Research Workspace')).toBeTruthy()
  308. expect(screen.getByText('…the waterfall token appears here…')).toBeTruthy()
  309. expect(screen.getByText('仅显示前 20 条结果,请缩小搜索范围。')).toBeTruthy()
  310. fireEvent.click(screen.getByRole('treeitem'))
  311. expect(open).toHaveBeenCalledWith(sid('body-hit'))
  312. expect(input.value).toBe('waterfall token')
  313. } finally {
  314. vi.useRealTimers()
  315. }
  316. })
  317. it('bounds programmatic search input to a schema-valid request without splitting an astral character', async () => {
  318. vi.useFakeTimers()
  319. try {
  320. const searchSessions = vi.fn(async () => ({ items: [], hasMore: false }))
  321. mount({ searchSessions })
  322. const input = screen.getByPlaceholderText<HTMLInputElement>('搜索名称、关键词…')
  323. expect(input.maxLength).toBe(500)
  324. fireEvent.change(input, { target: { value: 'y'.repeat(501) } })
  325. expect(input.value).toBe('y'.repeat(500))
  326. const expected = `prefix${'x'.repeat(493)}`
  327. fireEvent.change(input, {
  328. target: { value: `prefix\0${'x'.repeat(493)}😀tail` },
  329. })
  330. expect(input.value).toBe(expected)
  331. expect(input.value.length).toBe(499)
  332. expect(input.value).not.toContain('\0')
  333. await act(async () => { await vi.advanceTimersByTimeAsync(250) })
  334. expect(searchSessions).toHaveBeenCalledOnce()
  335. expect(searchSessions).toHaveBeenCalledWith(expected, expect.any(AbortSignal))
  336. } finally {
  337. vi.useRealTimers()
  338. }
  339. })
  340. it('keeps local matches and shows a lightweight warning when Host search fails', async () => {
  341. vi.useFakeTimers()
  342. try {
  343. const searchSessions = vi.fn(async () => { throw new Error('index unavailable') })
  344. mount({
  345. useSessions: hook(sessionState([
  346. summary('local-hit', 1, { displayTitle: 'Needle title' }),
  347. ])),
  348. useWorkspaces: hook(workspaceState([workspace('alpha', ['local-hit'])])),
  349. searchSessions,
  350. })
  351. fireEvent.change(screen.getByPlaceholderText('搜索名称、关键词…'), {
  352. target: { value: 'needle' },
  353. })
  354. expect(screen.getByText('Needle title')).toBeTruthy()
  355. await act(async () => { await vi.advanceTimersByTimeAsync(250) })
  356. expect(screen.getByText('Needle title')).toBeTruthy()
  357. expect(screen.getByText('内容搜索暂不可用,仅显示名称匹配。')).toBeTruthy()
  358. expect(screen.queryByText('无匹配会话')).toBeNull()
  359. } finally {
  360. vi.useRealTimers()
  361. }
  362. })
  363. it('aborts a superseded request and ignores its stale result', async () => {
  364. vi.useFakeTimers()
  365. try {
  366. let resolveFirst!: (value: {
  367. items: { sessionId: SessionId; snippet: string }[]
  368. hasMore: boolean
  369. }) => void
  370. const first = new Promise<{
  371. items: { sessionId: SessionId; snippet: string }[]
  372. hasMore: boolean
  373. }>((resolve) => { resolveFirst = resolve })
  374. const searchSessions = vi.fn((query: string, _signal: AbortSignal) => query === 'first'
  375. ? first
  376. : Promise.resolve({
  377. items: [{ sessionId: sid('second-hit'), snippet: 'second excerpt' }],
  378. hasMore: false,
  379. }))
  380. mount({
  381. useSessions: hook(sessionState([
  382. summary('first-hit', 2, { displayTitle: 'Old result' }),
  383. summary('second-hit', 1, { displayTitle: 'Fresh result' }),
  384. ])),
  385. searchSessions,
  386. })
  387. const input = screen.getByPlaceholderText('搜索名称、关键词…')
  388. fireEvent.change(input, { target: { value: 'first' } })
  389. await act(async () => { await vi.advanceTimersByTimeAsync(250) })
  390. const firstSignal = searchSessions.mock.calls[0]?.[1] as AbortSignal
  391. expect(firstSignal.aborted).toBe(false)
  392. fireEvent.change(input, { target: { value: 'second' } })
  393. expect(firstSignal.aborted).toBe(true)
  394. await act(async () => { await vi.advanceTimersByTimeAsync(250) })
  395. expect(screen.getByText('Fresh result')).toBeTruthy()
  396. await act(async () => {
  397. resolveFirst({
  398. items: [{ sessionId: sid('first-hit'), snippet: 'stale excerpt' }],
  399. hasMore: false,
  400. })
  401. await Promise.resolve()
  402. })
  403. expect(screen.queryByText('Old result')).toBeNull()
  404. expect(screen.getByText('Fresh result')).toBeTruthy()
  405. } finally {
  406. vi.useRealTimers()
  407. }
  408. })
  409. it('ignores a rejected request after it has been superseded', async () => {
  410. vi.useFakeTimers()
  411. try {
  412. let rejectFirst!: (reason: Error) => void
  413. const first = new Promise<never>((_resolve, reject) => { rejectFirst = reject })
  414. const searchSessions = vi.fn((query: string) => query === 'first'
  415. ? first
  416. : Promise.resolve({ items: [], hasMore: false }))
  417. mount({ searchSessions })
  418. const input = screen.getByPlaceholderText('搜索名称、关键词…')
  419. fireEvent.change(input, { target: { value: 'first' } })
  420. await act(async () => { await vi.advanceTimersByTimeAsync(250) })
  421. fireEvent.change(input, { target: { value: 'second' } })
  422. await act(async () => {
  423. rejectFirst(new Error('stale failure'))
  424. await Promise.resolve()
  425. })
  426. expect(screen.queryByText('内容搜索暂不可用,仅显示名称匹配。')).toBeNull()
  427. } finally {
  428. vi.useRealTimers()
  429. }
  430. })
  431. it('shows the no-sessions empty state in both modes and resolves an empty search', async () => {
  432. vi.useFakeTimers()
  433. try {
  434. const b = mount()
  435. expect(screen.getByText('暂无会话')).toBeTruthy()
  436. b.store.actions.setGroupBy('flat')
  437. rerender(b, {})
  438. expect(screen.getByText('暂无会话')).toBeTruthy()
  439. fireEvent.change(screen.getByPlaceholderText('搜索名称、关键词…'), { target: { value: 'x' } })
  440. expect(screen.getByText('正在搜索会话历史…')).toBeTruthy()
  441. await act(async () => { await vi.advanceTimersByTimeAsync(250) })
  442. expect(screen.getByText('无匹配会话')).toBeTruthy()
  443. } finally {
  444. vi.useRealTimers()
  445. }
  446. })
  447. it('rail state renders icon controls that request expansion', () => {
  448. vi.useFakeTimers()
  449. try {
  450. const expandSidebar = vi.fn()
  451. const b = mount({ wide: false, expandSidebar })
  452. // No wide chrome in rail state.
  453. expect(screen.queryByText('工作区')).toBeNull()
  454. expect(screen.queryByPlaceholderText('搜索名称、关键词…')).toBeNull()
  455. fireEvent.click(screen.getByRole('button', { name: '搜索会话' }))
  456. expect(expandSidebar).toHaveBeenCalledTimes(1)
  457. // The wide flip mounts the input and focuses it after the slide.
  458. rerender(b, { wide: true })
  459. const input = screen.getByPlaceholderText('搜索名称、关键词…')
  460. act(() => { vi.advanceTimersByTime(300) })
  461. expect(document.activeElement).toBe(input)
  462. // Wide search button is decorative (tabIndex -1, no expand call).
  463. fireEvent.click(screen.getByRole('button', { name: '搜索会话' }))
  464. expect(expandSidebar).toHaveBeenCalledTimes(1)
  465. } finally {
  466. vi.useRealTimers()
  467. }
  468. })
  469. it('rail add-workspace raises the directory flow in place, with no menu and no expansion', () => {
  470. const expandSidebar = vi.fn()
  471. mount({ wide: false, expandSidebar, useWorkspaces: hook(workspaceState([workspace('alpha', [])])) })
  472. fireEvent.click(screen.getByRole('button', { name: '添加工作区' }))
  473. expect(expandSidebar).not.toHaveBeenCalled()
  474. // Adding is the header's only action, so the gesture IS that action: no
  475. // one-row popover, and existing workspaces stay in the tree below.
  476. expect(screen.queryByRole('menu')).toBeNull()
  477. expect(screen.queryByRole('menuitem', { name: 'alpha' })).toBeNull()
  478. expect(screen.getByTestId('directory-flow')).toBeTruthy()
  479. })
  480. it('hides the add button when no directory-flow occupant is composed', () => {
  481. mount({
  482. useWorkspaces: hook(workspaceState([workspace('alpha', [])])),
  483. useDirectoryFlow: bindSnapshotSelector({ getSnapshot: () => false, subscribe: () => () => {} }),
  484. })
  485. // Nothing to add with, so the header offers no dead button.
  486. expect(screen.queryByRole('button', { name: '添加工作区' })).toBeNull()
  487. expect(screen.getByText('alpha')).toBeTruthy()
  488. })
  489. it('drag reorder reports the anchor to insertSessionBefore and skips no-op drops', () => {
  490. const insertSessionBefore = vi.fn(async () => {})
  491. const sessions = sessionState([summary('one', 3), summary('two', 2), summary('three', 1)])
  492. mount({
  493. useSessions: hook(sessions),
  494. useWorkspaces: hook(workspaceState([workspace('alpha', ['one', 'two', 'three'])])),
  495. insertSessionBefore,
  496. })
  497. fireEvent.click(screen.getByText('alpha'))
  498. const rows = screen.getAllByRole('treeitem').slice(1) // drop the group header
  499. const [one, , three] = rows as [HTMLElement, HTMLElement, HTMLElement]
  500. three.getBoundingClientRect = () => ({
  501. top: 200, bottom: 234, left: 0, right: 200, width: 200, height: 34, x: 0, y: 200, toJSON: () => ({}),
  502. })
  503. const dataTransfer = { effectAllowed: '', dropEffect: '' }
  504. fireEvent.dragStart(one, { dataTransfer })
  505. // Drop on the top half of "three": insert one before three.
  506. fireDrag(three, 'dragOver', 205)
  507. fireDrag(three, 'drop', 205)
  508. expect(insertSessionBefore).toHaveBeenCalledWith(wid('alpha'), sid('one'), sid('three'))
  509. // Dropping right back onto its own position is a no-op — top half
  510. // (anchor = itself) and bottom half (anchor = the next root) alike.
  511. fireEvent.dragStart(one, { dataTransfer })
  512. one.getBoundingClientRect = () => ({
  513. top: 100, bottom: 134, left: 0, right: 200, width: 200, height: 34, x: 0, y: 100, toJSON: () => ({}),
  514. })
  515. fireDrag(one, 'dragOver', 105)
  516. fireDrag(one, 'drop', 105)
  517. expect(insertSessionBefore).toHaveBeenCalledTimes(1)
  518. fireEvent.dragStart(one, { dataTransfer })
  519. fireDrag(one, 'drop', 130)
  520. expect(insertSessionBefore).toHaveBeenCalledTimes(1)
  521. })
  522. it('still sends the reorder when the dragged row left the group mid-drag', () => {
  523. const insertSessionBefore = vi.fn(async () => {})
  524. const sessions = sessionState([summary('one', 2), summary('two', 1)])
  525. const b = mount({
  526. useSessions: hook(sessions),
  527. useWorkspaces: hook(workspaceState([workspace('alpha', ['one', 'two'])])),
  528. insertSessionBefore,
  529. })
  530. fireEvent.click(screen.getByText('alpha'))
  531. const one = screen.getByText('one').closest('[role="treeitem"]') as HTMLElement
  532. fireEvent.dragStart(one, { dataTransfer: { effectAllowed: '', dropEffect: '' } })
  533. // The host dropped "one" from the workspace account while the drag is in
  534. // flight: the source index is gone but the drop still resolves its anchor.
  535. rerender(b, { useWorkspaces: hook(workspaceState([workspace('alpha', ['two'])])) })
  536. const two = screen.getByText('two').closest('[role="treeitem"]') as HTMLElement
  537. two.getBoundingClientRect = () => ({
  538. top: 150, bottom: 184, left: 0, right: 200, width: 200, height: 34, x: 0, y: 150, toJSON: () => ({}),
  539. })
  540. fireDrag(two, 'drop', 155)
  541. expect(insertSessionBefore).toHaveBeenCalledWith(wid('alpha'), sid('one'), sid('two'))
  542. })
  543. it('drag end without a drop clears markers; bottom-half drop appends past the last row', () => {
  544. const insertSessionBefore = vi.fn(async () => {})
  545. const sessions = sessionState([summary('one', 2), summary('two', 1)])
  546. mount({
  547. useSessions: hook(sessions),
  548. useWorkspaces: hook(workspaceState([workspace('alpha', ['one', 'two'])])),
  549. insertSessionBefore,
  550. })
  551. fireEvent.click(screen.getByText('alpha'))
  552. const [one, two] = screen.getAllByRole('treeitem').slice(1) as [HTMLElement, HTMLElement]
  553. two.getBoundingClientRect = () => ({
  554. top: 150, bottom: 184, left: 0, right: 200, width: 200, height: 34, x: 0, y: 150, toJSON: () => ({}),
  555. })
  556. const dataTransfer = { effectAllowed: '', dropEffect: '' }
  557. fireEvent.dragStart(one, { dataTransfer })
  558. fireEvent.dragEnd(one)
  559. // The drag ended: rows no longer accept drops.
  560. fireDrag(two, 'drop', 180)
  561. expect(insertSessionBefore).not.toHaveBeenCalled()
  562. // Bottom half of the last row: append (anchor omitted).
  563. fireEvent.dragStart(one, { dataTransfer })
  564. fireDrag(two, 'dragOver', 180)
  565. fireDrag(two, 'drop', 180)
  566. expect(insertSessionBefore).toHaveBeenCalledWith(wid('alpha'), sid('one'), undefined)
  567. })
  568. it('logs and keeps the order when the reorder call rejects', async () => {
  569. const warn = vi.spyOn(console, 'warn').mockImplementation(() => {})
  570. try {
  571. const insertSessionBefore = vi.fn(async () => { throw new Error('stale anchor') })
  572. const sessions = sessionState([summary('one', 2), summary('two', 1)])
  573. mount({
  574. useSessions: hook(sessions),
  575. useWorkspaces: hook(workspaceState([workspace('alpha', ['one', 'two'])])),
  576. insertSessionBefore,
  577. })
  578. fireEvent.click(screen.getByText('alpha'))
  579. const [one, two] = screen.getAllByRole('treeitem').slice(1) as [HTMLElement, HTMLElement]
  580. two.getBoundingClientRect = () => ({
  581. top: 150, bottom: 184, left: 0, right: 200, width: 200, height: 34, x: 0, y: 150, toJSON: () => ({}),
  582. })
  583. const dataTransfer = { effectAllowed: '', dropEffect: '' }
  584. fireEvent.dragStart(one, { dataTransfer })
  585. fireDrag(two, 'drop', 180)
  586. await waitFor(() => { expect(warn).toHaveBeenCalledWith('session reorder rejected:', expect.any(Error)) })
  587. } finally {
  588. warn.mockRestore()
  589. }
  590. })
  591. it('renames a workspace through the row menu dialog', async () => {
  592. let resolveRename!: () => void
  593. const renameWorkspace = vi.fn(() => new Promise<void>((resolve) => { resolveRename = resolve }))
  594. mount({
  595. useWorkspaces: hook(workspaceState([workspace('alpha', [], 'Alpha'), workspace('beta', [], 'Beta')])),
  596. renameWorkspace,
  597. })
  598. fireEvent.click(screen.getByRole('button', { name: '工作区“Alpha”的操作' }))
  599. fireEvent.click(screen.getByRole('menuitem', { name: '重命名' }))
  600. const input = screen.getByLabelText<HTMLInputElement>('工作区名称')
  601. expect(input.value).toBe('Alpha')
  602. // Unchanged and blank names stay blocked.
  603. expect(screen.getByRole<HTMLButtonElement>('button', { name: '重命名' }).disabled).toBe(true)
  604. fireEvent.change(input, { target: { value: ' ' } })
  605. expect(screen.getByRole<HTMLButtonElement>('button', { name: '重命名' }).disabled).toBe(true)
  606. // A duplicate of another workspace's title shows the inline conflict.
  607. fireEvent.change(input, { target: { value: ' Beta ' } })
  608. expect(screen.getByRole('alert').textContent).toBe('已存在名为“Beta”的工作区。')
  609. expect(screen.getByRole<HTMLButtonElement>('button', { name: '重命名' }).disabled).toBe(true)
  610. fireEvent.change(input, { target: { value: 'Gamma' } })
  611. fireEvent.click(screen.getByRole('button', { name: '重命名' }))
  612. expect(renameWorkspace).toHaveBeenCalledWith(wid('alpha'), 'Gamma')
  613. // While renaming: input disabled, close blocked, Enter ignored.
  614. expect(input.disabled).toBe(true)
  615. fireEvent.keyDown(document, { key: 'Escape' })
  616. expect(screen.getByRole('dialog')).toBeTruthy()
  617. await act(async () => { resolveRename() })
  618. expect(screen.queryByRole('dialog')).toBeNull()
  619. })
  620. it('rename via Enter, failure surfaces the error, Cancel closes', async () => {
  621. const renameWorkspace = vi.fn(async () => { throw new Error('rename conflict') })
  622. mount({
  623. useWorkspaces: hook(workspaceState([workspace('alpha', [], 'Alpha')])),
  624. renameWorkspace,
  625. })
  626. fireEvent.click(screen.getByRole('button', { name: '工作区“Alpha”的操作' }))
  627. fireEvent.click(screen.getByRole('menuitem', { name: '重命名' }))
  628. const input = screen.getByLabelText<HTMLInputElement>('工作区名称')
  629. // Enter with a blocked draft (unchanged) does nothing.
  630. fireEvent.keyDown(input, { key: 'Enter' })
  631. expect(renameWorkspace).not.toHaveBeenCalled()
  632. fireEvent.change(input, { target: { value: 'Renamed' } })
  633. fireEvent.keyDown(input, { key: 'a' })
  634. fireEvent.keyDown(input, { key: 'Enter' })
  635. expect(renameWorkspace).toHaveBeenCalledWith(wid('alpha'), 'Renamed')
  636. await waitFor(() => { expect(screen.getByRole('alert').textContent).toBe('rename conflict') })
  637. // The dialog stays for retry; typing clears the error; Cancel closes.
  638. fireEvent.change(input, { target: { value: 'Renamed2' } })
  639. expect(screen.queryByRole('alert')).toBeNull()
  640. fireEvent.click(screen.getByRole('button', { name: '取消' }))
  641. expect(screen.queryByRole('dialog')).toBeNull()
  642. })
  643. it('reports non-Error rename failures as text', async () => {
  644. const renameWorkspace = vi.fn(async () => { throw 'denied' })
  645. mount({
  646. useWorkspaces: hook(workspaceState([workspace('alpha', [], 'Alpha')])),
  647. renameWorkspace,
  648. })
  649. fireEvent.click(screen.getByRole('button', { name: '工作区“Alpha”的操作' }))
  650. fireEvent.click(screen.getByRole('menuitem', { name: '重命名' }))
  651. fireEvent.change(screen.getByLabelText('工作区名称'), { target: { value: 'Other' } })
  652. fireEvent.click(screen.getByRole('button', { name: '重命名' }))
  653. await waitFor(() => { expect(screen.getByRole('alert').textContent).toBe('denied') })
  654. })
  655. it('confirms Workspace deletion, explains retention, and blocks duplicate submission', async () => {
  656. let resolveDelete!: () => void
  657. const deleteWorkspace = vi.fn(() => new Promise<void>((resolve) => { resolveDelete = resolve }))
  658. const browser = mount({
  659. useWorkspaces: hook(workspaceState([workspace('alpha', ['session'], 'Alpha')])),
  660. deleteWorkspace,
  661. })
  662. fireEvent.click(screen.getByRole('button', { name: '工作区“Alpha”的操作' }))
  663. fireEvent.click(screen.getByRole('menuitem', { name: '删除工作区' }))
  664. const dialog = screen.getByRole('dialog', { name: '删除工作区' })
  665. expect(dialog.textContent).toContain('将把“Alpha”从工作区列表中移除')
  666. expect(dialog.textContent).toContain('文件夹与会话记录会保留')
  667. expect(dialog.textContent).toContain('其会话将显示在“未分组”下')
  668. const confirm = screen.getByRole<HTMLButtonElement>('button', { name: '删除工作区' })
  669. fireEvent.click(confirm)
  670. fireEvent.click(confirm)
  671. expect(deleteWorkspace).toHaveBeenCalledOnce()
  672. expect(deleteWorkspace).toHaveBeenCalledWith(wid('alpha'))
  673. expect(confirm.disabled).toBe(true)
  674. expect(screen.getByRole<HTMLButtonElement>('button', { name: '取消' }).disabled).toBe(true)
  675. expect(screen.getByRole('status').textContent).toBe('正在删除工作区…')
  676. fireEvent.keyDown(document, { key: 'Escape' })
  677. fireEvent.click(screen.getByRole('button', { name: '关闭' }))
  678. expect(screen.getByRole('dialog', { name: '删除工作区' })).toBeTruthy()
  679. await act(async () => { resolveDelete() })
  680. // RPC success alone does not close: the component waits until its
  681. // useWorkspaces projection has committed the removal, preventing a stale
  682. // Workspace frame from leaking into the next gesture.
  683. expect(screen.getByRole('dialog', { name: '删除工作区' })).toBeTruthy()
  684. rerender(browser, { useWorkspaces: hook(workspaceState([])) })
  685. expect(screen.queryByRole('dialog', { name: '删除工作区' })).toBeNull()
  686. })
  687. it('keeps the delete dialog open on failure and allows retry or cancellation', async () => {
  688. const deleteWorkspace = vi.fn()
  689. .mockRejectedValueOnce(new Error('storage unavailable'))
  690. .mockRejectedValueOnce('denied')
  691. mount({
  692. useWorkspaces: hook(workspaceState([workspace('alpha', [], 'Alpha')])),
  693. deleteWorkspace,
  694. })
  695. fireEvent.click(screen.getByRole('button', { name: '工作区“Alpha”的操作' }))
  696. fireEvent.click(screen.getByRole('menuitem', { name: '删除工作区' }))
  697. fireEvent.click(screen.getByRole('button', { name: '删除工作区' }))
  698. await waitFor(() => { expect(screen.getByRole('alert').textContent).toBe('storage unavailable') })
  699. expect(screen.getByRole('dialog', { name: '删除工作区' })).toBeTruthy()
  700. fireEvent.click(screen.getByRole('button', { name: '删除工作区' }))
  701. await waitFor(() => { expect(screen.getByRole('alert').textContent).toBe('denied') })
  702. fireEvent.click(screen.getByRole('button', { name: '取消' }))
  703. expect(screen.queryByRole('dialog', { name: '删除工作区' })).toBeNull()
  704. })
  705. it('Cancel, Escape, and Close dismiss deletion without calling the action', () => {
  706. const deleteWorkspace = vi.fn(async () => {})
  707. mount({
  708. useWorkspaces: hook(workspaceState([workspace('alpha', [], 'Alpha')])),
  709. deleteWorkspace,
  710. })
  711. const open = () => {
  712. fireEvent.click(screen.getByRole('button', { name: '工作区“Alpha”的操作' }))
  713. fireEvent.click(screen.getByRole('menuitem', { name: '删除工作区' }))
  714. }
  715. open()
  716. fireEvent.click(screen.getByRole('button', { name: '取消' }))
  717. open()
  718. fireEvent.keyDown(document, { key: 'Escape' })
  719. open()
  720. fireEvent.click(screen.getByRole('button', { name: '关闭' }))
  721. expect(deleteWorkspace).not.toHaveBeenCalled()
  722. expect(screen.queryByRole('dialog', { name: '删除工作区' })).toBeNull()
  723. })
  724. it('search hides drag affordances (rows are not draggable during search)', () => {
  725. const sessions = sessionState([summary('needle-a', 2, { displayTitle: 'Needle A' })])
  726. mount({
  727. useSessions: hook(sessions),
  728. useWorkspaces: hook(workspaceState([workspace('alpha', ['needle-a'])])),
  729. })
  730. fireEvent.change(screen.getByPlaceholderText('搜索名称、关键词…'), { target: { value: 'needle' } })
  731. const row = screen.getByText('Needle A').closest('[role="treeitem"]') as HTMLElement
  732. expect(row.hasAttribute('draggable')).toBe(false)
  733. })
  734. })