@alumbwe/anvil 1.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (529) hide show
  1. package/README.md +84 -0
  2. package/bin/anvil +2 -0
  3. package/bin/tree-sitter.wasm +0 -0
  4. package/package.json +67 -0
  5. package/src/__tests__/README.md +256 -0
  6. package/src/__tests__/bash-mode.test.ts +442 -0
  7. package/src/__tests__/cli-args.test.ts +181 -0
  8. package/src/__tests__/cli-harness.test.ts +52 -0
  9. package/src/__tests__/e2e/first-time-login.test.ts +173 -0
  10. package/src/__tests__/e2e/logout-relogin-flow.test.ts +110 -0
  11. package/src/__tests__/e2e/returning-user-auth.test.ts +113 -0
  12. package/src/__tests__/e2e-cli.test.ts +183 -0
  13. package/src/__tests__/helpers/launcher-disconnect-fixture.cjs +176 -0
  14. package/src/__tests__/helpers/mock-api-client.ts +64 -0
  15. package/src/__tests__/helpers/renderer-cleanup-fixture.tsx +89 -0
  16. package/src/__tests__/helpers/terminal-watchdog-fixture.ts +75 -0
  17. package/src/__tests__/home-directory-detection.test.ts +213 -0
  18. package/src/__tests__/integration/api-integration.test.ts +288 -0
  19. package/src/__tests__/integration/credentials-storage.test.ts +486 -0
  20. package/src/__tests__/integration/local-agents.test.ts +1187 -0
  21. package/src/__tests__/integration/login-flow-instrumentation.test.ts +134 -0
  22. package/src/__tests__/integration/login-polling-working.test.ts +285 -0
  23. package/src/__tests__/integration/queue-editing.test.tsx +261 -0
  24. package/src/__tests__/integration/usage-refresh-on-completion.test.ts +168 -0
  25. package/src/__tests__/integration-tmux.test.ts +188 -0
  26. package/src/__tests__/launcher-avx2-fallback.test.ts +630 -0
  27. package/src/__tests__/launcher-disconnect.test.ts +104 -0
  28. package/src/__tests__/mocks/hover-toggle-controller.ts +71 -0
  29. package/src/__tests__/open-tui-native-bundle.test.ts +378 -0
  30. package/src/__tests__/path-completion.test.ts +243 -0
  31. package/src/__tests__/project-files-chat-id.test.ts +34 -0
  32. package/src/__tests__/release/proxy-http-get.test.ts +667 -0
  33. package/src/__tests__/release/wrapper-safety.test.ts +412 -0
  34. package/src/__tests__/renderer-cleanup.test.ts +174 -0
  35. package/src/__tests__/rerender-perf.integration.test.ts +283 -0
  36. package/src/__tests__/terminal-reset-sequences.test.ts +36 -0
  37. package/src/__tests__/terminal-watchdog.test.ts +388 -0
  38. package/src/__tests__/test-utils.ts +271 -0
  39. package/src/__tests__/tmux-poc.ts +150 -0
  40. package/src/__tests__/unit/agent-mode-toggle.test.ts +137 -0
  41. package/src/__tests__/unit/copy-button.test.ts +157 -0
  42. package/src/__tests__/unit/create-run-config.test.ts +66 -0
  43. package/src/__tests__/unit/publish-confirmation.test.ts +70 -0
  44. package/src/__tests__/unit/segmented-control.test.ts +152 -0
  45. package/src/__tests__/utils/env.test.ts +169 -0
  46. package/src/__tests__/utils/project-picker.test.ts +39 -0
  47. package/src/agents/bundled-agents.generated.d.ts +14 -0
  48. package/src/agents/bundled-agents.generated.ts +3692 -0
  49. package/src/app.tsx +399 -0
  50. package/src/chat.tsx +1698 -0
  51. package/src/cli-args.ts +134 -0
  52. package/src/commands/__tests__/bash-command.test.ts +445 -0
  53. package/src/commands/__tests__/command-args.test.ts +349 -0
  54. package/src/commands/__tests__/copy-conversation-clipboard.test.ts +108 -0
  55. package/src/commands/__tests__/copy-conversation.test.ts +226 -0
  56. package/src/commands/__tests__/diagnostics-command.test.ts +35 -0
  57. package/src/commands/__tests__/freebuff-command-aliases.test.ts +56 -0
  58. package/src/commands/__tests__/image.test.ts +99 -0
  59. package/src/commands/__tests__/init.test.ts +456 -0
  60. package/src/commands/__tests__/new-command-rotates-chat.test.ts +91 -0
  61. package/src/commands/__tests__/process-diagnostics.test.ts +54 -0
  62. package/src/commands/__tests__/prompt-builders.test.ts +28 -0
  63. package/src/commands/__tests__/router-input.test.ts +298 -0
  64. package/src/commands/ads.ts +44 -0
  65. package/src/commands/command-registry.ts +726 -0
  66. package/src/commands/copy-conversation.ts +392 -0
  67. package/src/commands/help.ts +13 -0
  68. package/src/commands/image.ts +20 -0
  69. package/src/commands/init.ts +121 -0
  70. package/src/commands/process-diagnostics.ts +120 -0
  71. package/src/commands/prompt-builders.ts +100 -0
  72. package/src/commands/publish.ts +203 -0
  73. package/src/commands/router-utils.ts +80 -0
  74. package/src/commands/router.ts +487 -0
  75. package/src/commands/usage.ts +26 -0
  76. package/src/components/__tests__/ad-banner.test.tsx +83 -0
  77. package/src/components/__tests__/chat-input-bar.test.tsx +130 -0
  78. package/src/components/__tests__/freebuff-landing-heading-row.test.tsx +91 -0
  79. package/src/components/__tests__/freebuff-model-selector.test.tsx +519 -0
  80. package/src/components/__tests__/freebuff-offer-invariants.test.ts +54 -0
  81. package/src/components/__tests__/freebuff-takeover-prompt.test.tsx +212 -0
  82. package/src/components/__tests__/grid-layout.integration.test.tsx +304 -0
  83. package/src/components/__tests__/grid-layout.test.tsx +1051 -0
  84. package/src/components/__tests__/message-block.completion.test.tsx +104 -0
  85. package/src/components/__tests__/message-block.streaming.test.tsx +78 -0
  86. package/src/components/__tests__/message-with-agents.test.tsx +579 -0
  87. package/src/components/__tests__/multiline-input.test.tsx +1132 -0
  88. package/src/components/__tests__/queue-panel.test.tsx +398 -0
  89. package/src/components/__tests__/selectable-list.test.ts +232 -0
  90. package/src/components/__tests__/status-bar.test.tsx +44 -0
  91. package/src/components/__tests__/status-indicator.test.tsx +204 -0
  92. package/src/components/__tests__/terminal-thinking-orb.test.tsx +111 -0
  93. package/src/components/__tests__/user-error-banner.test.tsx +102 -0
  94. package/src/components/ad-banner.tsx +323 -0
  95. package/src/components/agent-checklist.tsx +350 -0
  96. package/src/components/agent-mode-toggle.tsx +241 -0
  97. package/src/components/anvil-mark.tsx +52 -0
  98. package/src/components/ask-user/__tests__/multiple-choice-form.test.ts +432 -0
  99. package/src/components/ask-user/__tests__/validation.test.ts +213 -0
  100. package/src/components/ask-user/components/accordion-question.tsx +153 -0
  101. package/src/components/ask-user/components/custom-answer-input.tsx +66 -0
  102. package/src/components/ask-user/components/options-list.tsx +133 -0
  103. package/src/components/ask-user/components/question-header.tsx +80 -0
  104. package/src/components/ask-user/components/question-option.tsx +89 -0
  105. package/src/components/ask-user/constants.ts +38 -0
  106. package/src/components/ask-user/index.tsx +631 -0
  107. package/src/components/ask-user/utils/validation.ts +85 -0
  108. package/src/components/attachment-card.tsx +65 -0
  109. package/src/components/blocks/agent-block-grid.tsx +58 -0
  110. package/src/components/blocks/agent-branch-item.tsx +311 -0
  111. package/src/components/blocks/agent-branch-wrapper.tsx +493 -0
  112. package/src/components/blocks/agent-list-branch.tsx +77 -0
  113. package/src/components/blocks/ask-user-branch.tsx +81 -0
  114. package/src/components/blocks/block-helpers.ts +11 -0
  115. package/src/components/blocks/blocks-renderer.tsx +289 -0
  116. package/src/components/blocks/content-with-markdown.tsx +33 -0
  117. package/src/components/blocks/image-block.tsx +128 -0
  118. package/src/components/blocks/implementor-row.tsx +429 -0
  119. package/src/components/blocks/single-block.tsx +201 -0
  120. package/src/components/blocks/thinking-block.tsx +67 -0
  121. package/src/components/blocks/tool-block-group.tsx +57 -0
  122. package/src/components/blocks/tool-branch.tsx +206 -0
  123. package/src/components/blocks/user-content-copy.tsx +157 -0
  124. package/src/components/bottom-banner.tsx +136 -0
  125. package/src/components/build-mode-buttons.tsx +110 -0
  126. package/src/components/button.tsx +66 -0
  127. package/src/components/chat-header.tsx +77 -0
  128. package/src/components/chat-history-screen.tsx +413 -0
  129. package/src/components/chat-input-bar.tsx +523 -0
  130. package/src/components/clickable-title-box.tsx +66 -0
  131. package/src/components/clickable.tsx +119 -0
  132. package/src/components/collapse-button.tsx +34 -0
  133. package/src/components/copy-button.tsx +203 -0
  134. package/src/components/elapsed-timer.tsx +55 -0
  135. package/src/components/error-boundary.tsx +55 -0
  136. package/src/components/feedback-container.tsx +181 -0
  137. package/src/components/feedback-icon-button.tsx +89 -0
  138. package/src/components/feedback-input-mode.tsx +344 -0
  139. package/src/components/file-attachment-card.tsx +98 -0
  140. package/src/components/freebuff-active-session-summary.tsx +63 -0
  141. package/src/components/freebuff-landing-screen.tsx +865 -0
  142. package/src/components/freebuff-model-selector.tsx +1087 -0
  143. package/src/components/freebuff-referral-banner.tsx +604 -0
  144. package/src/components/freebuff-superseded-screen.tsx +62 -0
  145. package/src/components/grid-layout.tsx +80 -0
  146. package/src/components/help-banner.tsx +127 -0
  147. package/src/components/highlighted-text.tsx +52 -0
  148. package/src/components/image-card.tsx +167 -0
  149. package/src/components/image-thumbnail.tsx +114 -0
  150. package/src/components/input-cursor.tsx +77 -0
  151. package/src/components/input-mode-banner.tsx +57 -0
  152. package/src/components/load-previous-button.tsx +48 -0
  153. package/src/components/login-modal.tsx +475 -0
  154. package/src/components/message-block.tsx +346 -0
  155. package/src/components/message-footer.tsx +258 -0
  156. package/src/components/message-with-agents.tsx +506 -0
  157. package/src/components/mode-divider.tsx +43 -0
  158. package/src/components/multiline-input.tsx +1241 -0
  159. package/src/components/out-of-credits-banner.tsx +160 -0
  160. package/src/components/pending-attachments-banner.tsx +116 -0
  161. package/src/components/pending-bash-message.tsx +59 -0
  162. package/src/components/progress-bar.tsx +81 -0
  163. package/src/components/project-picker-screen.tsx +485 -0
  164. package/src/components/publish-confirmation.tsx +493 -0
  165. package/src/components/publish-container.tsx +687 -0
  166. package/src/components/queue-panel.tsx +310 -0
  167. package/src/components/raised-pill.tsx +95 -0
  168. package/src/components/renderers/plan-box.tsx +59 -0
  169. package/src/components/review-screen.tsx +112 -0
  170. package/src/components/scroll-to-bottom-button.tsx +34 -0
  171. package/src/components/segmented-control.tsx +187 -0
  172. package/src/components/selectable-list.tsx +248 -0
  173. package/src/components/selected-chips.tsx +80 -0
  174. package/src/components/separator.tsx +40 -0
  175. package/src/components/session-ended-banner.tsx +226 -0
  176. package/src/components/shimmer-text.tsx +248 -0
  177. package/src/components/status-bar.tsx +313 -0
  178. package/src/components/subscription-limit-banner.tsx +206 -0
  179. package/src/components/suggested-prompts.tsx +184 -0
  180. package/src/components/suggestion-menu.tsx +215 -0
  181. package/src/components/terminal-command-display.tsx +151 -0
  182. package/src/components/terminal-link.tsx +97 -0
  183. package/src/components/terminal-thinking-orb.tsx +95 -0
  184. package/src/components/text-attachment-card.tsx +77 -0
  185. package/src/components/thinking.tsx +109 -0
  186. package/src/components/tools/__tests__/apply-patch.test.tsx +84 -0
  187. package/src/components/tools/__tests__/code-search.test.tsx +45 -0
  188. package/src/components/tools/__tests__/gravity-index.test.ts +76 -0
  189. package/src/components/tools/__tests__/render-ui.test.tsx +92 -0
  190. package/src/components/tools/__tests__/run-terminal-command.test.ts +233 -0
  191. package/src/components/tools/apply-patch.tsx +95 -0
  192. package/src/components/tools/code-search.tsx +45 -0
  193. package/src/components/tools/composio.tsx +152 -0
  194. package/src/components/tools/diff-viewer.tsx +76 -0
  195. package/src/components/tools/glob.tsx +61 -0
  196. package/src/components/tools/gravity-index.tsx +77 -0
  197. package/src/components/tools/list-directory.tsx +61 -0
  198. package/src/components/tools/read-docs.tsx +37 -0
  199. package/src/components/tools/read-files.tsx +91 -0
  200. package/src/components/tools/read-subtree.tsx +44 -0
  201. package/src/components/tools/read-url.tsx +33 -0
  202. package/src/components/tools/registry.ts +126 -0
  203. package/src/components/tools/render-ui.tsx +122 -0
  204. package/src/components/tools/run-terminal-command.tsx +79 -0
  205. package/src/components/tools/skill.tsx +29 -0
  206. package/src/components/tools/str-replace.tsx +77 -0
  207. package/src/components/tools/suggest-followups.tsx +369 -0
  208. package/src/components/tools/task-completed.tsx +16 -0
  209. package/src/components/tools/tool-call-item.tsx +273 -0
  210. package/src/components/tools/types.ts +55 -0
  211. package/src/components/tools/web-search.tsx +33 -0
  212. package/src/components/tools/write-file.tsx +16 -0
  213. package/src/components/tools/write-todos.tsx +95 -0
  214. package/src/components/top-banner.tsx +178 -0
  215. package/src/components/usage-banner.tsx +244 -0
  216. package/src/components/user-error-banner.tsx +56 -0
  217. package/src/components/validation-error-popover.tsx +192 -0
  218. package/src/contexts/__tests__/chat-runtime-context.test.tsx +149 -0
  219. package/src/contexts/chat-runtime-context.tsx +240 -0
  220. package/src/data/slash-commands.test.ts +24 -0
  221. package/src/data/slash-commands.ts +249 -0
  222. package/src/entry.ts +12 -0
  223. package/src/hooks/__tests__/holds-live-freebuff-slot.test.ts +49 -0
  224. package/src/hooks/__tests__/session-fetch-signal.test.ts +63 -0
  225. package/src/hooks/__tests__/use-activity-query.test.ts +1000 -0
  226. package/src/hooks/__tests__/use-ask-user-bridge.test.ts +176 -0
  227. package/src/hooks/__tests__/use-auth-query.test.ts +21 -0
  228. package/src/hooks/__tests__/use-chat-input-remount.test.tsx +66 -0
  229. package/src/hooks/__tests__/use-clipboard.test.tsx +73 -0
  230. package/src/hooks/__tests__/use-connection-status.test.ts +112 -0
  231. package/src/hooks/__tests__/use-directory-browser.test.ts +203 -0
  232. package/src/hooks/__tests__/use-gravity-ad.test.ts +73 -0
  233. package/src/hooks/__tests__/use-grid-layout.test.ts +349 -0
  234. package/src/hooks/__tests__/use-input-history.test.ts +699 -0
  235. package/src/hooks/__tests__/use-message-queue-editing.test.tsx +209 -0
  236. package/src/hooks/__tests__/use-message-queue.test.tsx +71 -0
  237. package/src/hooks/__tests__/use-path-tab-completion.test.ts +315 -0
  238. package/src/hooks/__tests__/use-queue-controls.test.ts +68 -0
  239. package/src/hooks/__tests__/use-queue-ui.test.tsx +83 -0
  240. package/src/hooks/__tests__/use-searchable-list.test.ts +359 -0
  241. package/src/hooks/__tests__/use-send-message-timer.test.ts +146 -0
  242. package/src/hooks/__tests__/use-suggestion-engine-mention.test.ts +361 -0
  243. package/src/hooks/__tests__/use-suggestion-engine.test.ts +485 -0
  244. package/src/hooks/__tests__/use-terminal-focus.test.ts +66 -0
  245. package/src/hooks/__tests__/use-terminal-layout.test.ts +675 -0
  246. package/src/hooks/__tests__/use-timeout.test.ts +330 -0
  247. package/src/hooks/__tests__/use-usage-query.test.ts +502 -0
  248. package/src/hooks/__tests__/use-user-details-query.test.ts +195 -0
  249. package/src/hooks/helpers/__tests__/send-message.test.ts +1918 -0
  250. package/src/hooks/helpers/send-message.ts +654 -0
  251. package/src/hooks/stream-state.ts +83 -0
  252. package/src/hooks/use-activity-query.ts +668 -0
  253. package/src/hooks/use-agent-validation.ts +74 -0
  254. package/src/hooks/use-ask-user-bridge.ts +92 -0
  255. package/src/hooks/use-auth-query.ts +253 -0
  256. package/src/hooks/use-auth-state.ts +152 -0
  257. package/src/hooks/use-chat-input.ts +105 -0
  258. package/src/hooks/use-chat-keyboard.ts +321 -0
  259. package/src/hooks/use-chat-messages.ts +254 -0
  260. package/src/hooks/use-chat-state.ts +165 -0
  261. package/src/hooks/use-chat-streaming.ts +180 -0
  262. package/src/hooks/use-chat-ui.ts +131 -0
  263. package/src/hooks/use-clipboard.ts +151 -0
  264. package/src/hooks/use-connection-status.ts +142 -0
  265. package/src/hooks/use-directory-browser.ts +88 -0
  266. package/src/hooks/use-elapsed-time.ts +116 -0
  267. package/src/hooks/use-event.ts +34 -0
  268. package/src/hooks/use-exit-handler.ts +66 -0
  269. package/src/hooks/use-fetch-login-url.ts +65 -0
  270. package/src/hooks/use-fingerprint.ts +61 -0
  271. package/src/hooks/use-freebuff-ctrl-c-exit.ts +23 -0
  272. package/src/hooks/use-freebuff-session-progress.ts +34 -0
  273. package/src/hooks/use-freebuff-session.ts +806 -0
  274. package/src/hooks/use-freebuff-streak-query.ts +61 -0
  275. package/src/hooks/use-gravity-ad.ts +656 -0
  276. package/src/hooks/use-grid-layout.ts +73 -0
  277. package/src/hooks/use-input-history.ts +173 -0
  278. package/src/hooks/use-login-keyboard-handlers.ts +75 -0
  279. package/src/hooks/use-login-polling.ts +114 -0
  280. package/src/hooks/use-logo.tsx +161 -0
  281. package/src/hooks/use-message-queue.ts +387 -0
  282. package/src/hooks/use-now.ts +20 -0
  283. package/src/hooks/use-path-tab-completion.ts +88 -0
  284. package/src/hooks/use-publish-mutation.ts +60 -0
  285. package/src/hooks/use-queue-controls.ts +68 -0
  286. package/src/hooks/use-queue-ui.ts +74 -0
  287. package/src/hooks/use-scroll-management.ts +184 -0
  288. package/src/hooks/use-searchable-list.ts +98 -0
  289. package/src/hooks/use-send-message.ts +773 -0
  290. package/src/hooks/use-sheen-animation.tsx +88 -0
  291. package/src/hooks/use-subscription-query.ts +71 -0
  292. package/src/hooks/use-suggestion-engine.ts +783 -0
  293. package/src/hooks/use-terminal-breakpoints.ts +77 -0
  294. package/src/hooks/use-terminal-dimensions.ts +46 -0
  295. package/src/hooks/use-terminal-focus.ts +29 -0
  296. package/src/hooks/use-terminal-layout.ts +189 -0
  297. package/src/hooks/use-theme.tsx +140 -0
  298. package/src/hooks/use-timeout.ts +85 -0
  299. package/src/hooks/use-update-preference.ts +66 -0
  300. package/src/hooks/use-usage-monitor.ts +55 -0
  301. package/src/hooks/use-usage-query.ts +124 -0
  302. package/src/hooks/use-user-details-query.ts +93 -0
  303. package/src/hooks/use-why-did-you-update.ts +177 -0
  304. package/src/index.tsx +441 -0
  305. package/src/init/__tests__/init-direnv.test.ts +526 -0
  306. package/src/init/init-app.ts +34 -0
  307. package/src/init/init-direnv.ts +133 -0
  308. package/src/login/constants.ts +60 -0
  309. package/src/login/login-flow.ts +204 -0
  310. package/src/login/plain-login.ts +108 -0
  311. package/src/login/utils.ts +183 -0
  312. package/src/native/ripgrep.ts +80 -0
  313. package/src/polyfills/bun-strip-ansi.ts +12 -0
  314. package/src/pre-init/tree-sitter-wasm.ts +96 -0
  315. package/src/project-files.ts +102 -0
  316. package/src/smoke/terminal-command-broker.ts +439 -0
  317. package/src/state/__tests__/chat-store-focus.test.ts +22 -0
  318. package/src/state/__tests__/feedback-store.test.ts +257 -0
  319. package/src/state/chat-history-store.ts +30 -0
  320. package/src/state/chat-store.ts +544 -0
  321. package/src/state/feedback-store.ts +160 -0
  322. package/src/state/freebuff-model-store.ts +42 -0
  323. package/src/state/freebuff-session-store.ts +53 -0
  324. package/src/state/login-store.ts +150 -0
  325. package/src/state/message-block-store.ts +149 -0
  326. package/src/state/publish-store.ts +146 -0
  327. package/src/state/queue-panel-store.ts +26 -0
  328. package/src/state/review-store.ts +24 -0
  329. package/src/testing/env.ts +46 -0
  330. package/src/types/chat-state.ts +15 -0
  331. package/src/types/chat.ts +241 -0
  332. package/src/types/contracts/send-message.ts +12 -0
  333. package/src/types/env.ts +98 -0
  334. package/src/types/freebuff-session.ts +17 -0
  335. package/src/types/function-params.ts +28 -0
  336. package/src/types/react19-compat.d.ts +19 -0
  337. package/src/types/store.ts +111 -0
  338. package/src/types/theme-system.ts +144 -0
  339. package/src/types/utils.ts +3 -0
  340. package/src/utils/__tests__/active-run.test.ts +140 -0
  341. package/src/utils/__tests__/activity-tracker.test.ts +440 -0
  342. package/src/utils/__tests__/agent-display.test.ts +139 -0
  343. package/src/utils/__tests__/analytics-client.test.ts +262 -0
  344. package/src/utils/__tests__/anonymous-id.test.ts +74 -0
  345. package/src/utils/__tests__/anvil-api.test.ts +578 -0
  346. package/src/utils/__tests__/arrays.test.ts +264 -0
  347. package/src/utils/__tests__/bash-context-processor.test.ts +38 -0
  348. package/src/utils/__tests__/block-processor.test.ts +810 -0
  349. package/src/utils/__tests__/chat-history.test.ts +203 -0
  350. package/src/utils/__tests__/chat-input-key-intercept.test.ts +40 -0
  351. package/src/utils/__tests__/chat-meta.test.ts +130 -0
  352. package/src/utils/__tests__/clipboard-linux.test.ts +359 -0
  353. package/src/utils/__tests__/clipboard.test.ts +910 -0
  354. package/src/utils/__tests__/code-search-summary.test.ts +84 -0
  355. package/src/utils/__tests__/collapse-helpers.test.ts +1109 -0
  356. package/src/utils/__tests__/error-handling.test.ts +658 -0
  357. package/src/utils/__tests__/exit-cleanly.test.ts +110 -0
  358. package/src/utils/__tests__/feedback-helpers.test.ts +444 -0
  359. package/src/utils/__tests__/feedback-submission.test.ts +26 -0
  360. package/src/utils/__tests__/fetch-usage.test.ts +355 -0
  361. package/src/utils/__tests__/fingerprint.test.ts +144 -0
  362. package/src/utils/__tests__/fixtures/terminal-command-broker-child.ts +7 -0
  363. package/src/utils/__tests__/fixtures/terminal-command-broker-entry.ts +3 -0
  364. package/src/utils/__tests__/fixtures/terminal-command-broker-owner.ts +33 -0
  365. package/src/utils/__tests__/format-elapsed-time.test.ts +103 -0
  366. package/src/utils/__tests__/format-timeout.test.ts +87 -0
  367. package/src/utils/__tests__/freebuff-instance-owner.test.ts +69 -0
  368. package/src/utils/__tests__/freebuff-model-navigation.test.ts +102 -0
  369. package/src/utils/__tests__/freebuff-premium-reset.test.ts +79 -0
  370. package/src/utils/__tests__/freebuff-referral-cache.test.ts +116 -0
  371. package/src/utils/__tests__/freebuff-session-api.test.ts +215 -0
  372. package/src/utils/__tests__/freebuff-session-display.test.ts +23 -0
  373. package/src/utils/__tests__/freebuff-streak-line.test.ts +217 -0
  374. package/src/utils/__tests__/image-dimensions.test.ts +249 -0
  375. package/src/utils/__tests__/image-processor.test.ts +209 -0
  376. package/src/utils/__tests__/implementor-helpers.test.ts +1458 -0
  377. package/src/utils/__tests__/keyboard-actions.test.ts +660 -0
  378. package/src/utils/__tests__/layout-helpers.test.ts +36 -0
  379. package/src/utils/__tests__/lazy-response-ads.test.ts +167 -0
  380. package/src/utils/__tests__/log-shipper.test.ts +53 -0
  381. package/src/utils/__tests__/markdown-renderer.test.tsx +421 -0
  382. package/src/utils/__tests__/message-block-helpers.test.ts +1263 -0
  383. package/src/utils/__tests__/message-updater.test.ts +645 -0
  384. package/src/utils/__tests__/osc-timeout-scenarios.test.ts +172 -0
  385. package/src/utils/__tests__/pending-attachments.test.ts +277 -0
  386. package/src/utils/__tests__/polling-backoff.test.ts +51 -0
  387. package/src/utils/__tests__/queue-panel-actions.test.ts +148 -0
  388. package/src/utils/__tests__/run-state-storage.test.ts +919 -0
  389. package/src/utils/__tests__/safe-json.test.ts +77 -0
  390. package/src/utils/__tests__/sdk-event-handlers.test.ts +685 -0
  391. package/src/utils/__tests__/send-message-helpers.test.ts +1898 -0
  392. package/src/utils/__tests__/send-message-timer.test.ts +64 -0
  393. package/src/utils/__tests__/settings.test.ts +99 -0
  394. package/src/utils/__tests__/strings.test.ts +192 -0
  395. package/src/utils/__tests__/terminal-color-detection.test.ts +425 -0
  396. package/src/utils/__tests__/terminal-command-broker.test.ts +528 -0
  397. package/src/utils/__tests__/terminal-enter-detection.test.ts +77 -0
  398. package/src/utils/__tests__/terminal-io.test.ts +137 -0
  399. package/src/utils/__tests__/terminal-protocol-controller.test.ts +119 -0
  400. package/src/utils/__tests__/text-layout.test.ts +82 -0
  401. package/src/utils/__tests__/theme-platform-detection.test.ts +125 -0
  402. package/src/utils/__tests__/think-tag-parser.test.ts +159 -0
  403. package/src/utils/__tests__/trace-writer.test.ts +223 -0
  404. package/src/utils/__tests__/trim-chat-logs.test.ts +78 -0
  405. package/src/utils/__tests__/usage-banner-state.test.ts +298 -0
  406. package/src/utils/__tests__/validation-error-formatting.test.ts +126 -0
  407. package/src/utils/__tests__/windows-terminal-health.test.ts +101 -0
  408. package/src/utils/__tests__/write-file-atomic.test.ts +127 -0
  409. package/src/utils/active-run.ts +131 -0
  410. package/src/utils/activity-tracker.ts +66 -0
  411. package/src/utils/agent-display.ts +87 -0
  412. package/src/utils/agent-helpers.ts +27 -0
  413. package/src/utils/agent-id-utils.ts +14 -0
  414. package/src/utils/ai-message-id.ts +12 -0
  415. package/src/utils/analytics.ts +361 -0
  416. package/src/utils/anonymous-id.ts +87 -0
  417. package/src/utils/anvil-api.ts +612 -0
  418. package/src/utils/anvil-client.ts +217 -0
  419. package/src/utils/arrays.ts +117 -0
  420. package/src/utils/auth.ts +258 -0
  421. package/src/utils/bash-context-processor.ts +47 -0
  422. package/src/utils/bash-messages.ts +108 -0
  423. package/src/utils/block-margins.ts +35 -0
  424. package/src/utils/block-operations.ts +523 -0
  425. package/src/utils/block-processor.ts +213 -0
  426. package/src/utils/chat-history.ts +253 -0
  427. package/src/utils/chat-input-key-intercept.ts +52 -0
  428. package/src/utils/chat-meta.ts +90 -0
  429. package/src/utils/chat-scroll-accel.ts +151 -0
  430. package/src/utils/clipboard-image.ts +588 -0
  431. package/src/utils/clipboard.ts +397 -0
  432. package/src/utils/code-search-summary.ts +70 -0
  433. package/src/utils/collapse-helpers.ts +256 -0
  434. package/src/utils/config-dir.ts +24 -0
  435. package/src/utils/constants.ts +190 -0
  436. package/src/utils/create-event-handler-state.ts +73 -0
  437. package/src/utils/create-run-config.ts +121 -0
  438. package/src/utils/detect-shell.ts +116 -0
  439. package/src/utils/directory-browser.ts +72 -0
  440. package/src/utils/engagement.ts +55 -0
  441. package/src/utils/env.ts +92 -0
  442. package/src/utils/error-handling.ts +255 -0
  443. package/src/utils/error-messages.ts +62 -0
  444. package/src/utils/exit-cleanly.ts +86 -0
  445. package/src/utils/feedback-helpers.ts +103 -0
  446. package/src/utils/feedback-submission.ts +22 -0
  447. package/src/utils/fetch-usage.ts +76 -0
  448. package/src/utils/fingerprint.ts +245 -0
  449. package/src/utils/format-elapsed-time.ts +33 -0
  450. package/src/utils/format-session-units.ts +6 -0
  451. package/src/utils/format-timeout.ts +28 -0
  452. package/src/utils/format-validation-errors-for-message.ts +84 -0
  453. package/src/utils/freebuff-agent-selection.ts +33 -0
  454. package/src/utils/freebuff-instance-owner.ts +66 -0
  455. package/src/utils/freebuff-model-navigation.ts +50 -0
  456. package/src/utils/freebuff-premium-reset.ts +55 -0
  457. package/src/utils/freebuff-referral-cache.ts +51 -0
  458. package/src/utils/freebuff-session-api.ts +243 -0
  459. package/src/utils/freebuff-session-display.ts +21 -0
  460. package/src/utils/freebuff-streak-line.ts +93 -0
  461. package/src/utils/git.ts +17 -0
  462. package/src/utils/helpers.ts +47 -0
  463. package/src/utils/image-display.ts +67 -0
  464. package/src/utils/image-handler.ts +325 -0
  465. package/src/utils/image-processor.ts +133 -0
  466. package/src/utils/image-thumbnail.ts +78 -0
  467. package/src/utils/implementor-helpers.ts +787 -0
  468. package/src/utils/input-modes.ts +174 -0
  469. package/src/utils/keyboard-actions.ts +392 -0
  470. package/src/utils/keypad-keys.ts +47 -0
  471. package/src/utils/layout-helpers.ts +33 -0
  472. package/src/utils/lazy-response-ads.ts +1 -0
  473. package/src/utils/local-agent-registry.ts +474 -0
  474. package/src/utils/log-shipper.ts +126 -0
  475. package/src/utils/logger.ts +327 -0
  476. package/src/utils/markdown-renderer.tsx +1090 -0
  477. package/src/utils/math.ts +3 -0
  478. package/src/utils/message-block-helpers.ts +684 -0
  479. package/src/utils/message-history.ts +143 -0
  480. package/src/utils/message-tree-utils.ts +67 -0
  481. package/src/utils/message-updater.ts +279 -0
  482. package/src/utils/open-file.ts +146 -0
  483. package/src/utils/open-url.ts +63 -0
  484. package/src/utils/path-completion.ts +111 -0
  485. package/src/utils/path-helpers.ts +38 -0
  486. package/src/utils/pending-attachments.ts +360 -0
  487. package/src/utils/polling-backoff.ts +59 -0
  488. package/src/utils/project-picker.ts +12 -0
  489. package/src/utils/queue-panel-actions.ts +70 -0
  490. package/src/utils/recent-projects.ts +156 -0
  491. package/src/utils/renderer-cleanup.ts +233 -0
  492. package/src/utils/response-ad-positions.ts +1 -0
  493. package/src/utils/run-state-storage.ts +603 -0
  494. package/src/utils/safe-json.ts +153 -0
  495. package/src/utils/sdk-event-handlers.ts +571 -0
  496. package/src/utils/send-message-helpers.ts +210 -0
  497. package/src/utils/send-message-timer.ts +110 -0
  498. package/src/utils/settings.ts +226 -0
  499. package/src/utils/skill-registry.ts +94 -0
  500. package/src/utils/spawn-agent-matcher.ts +55 -0
  501. package/src/utils/status-indicator-state.ts +107 -0
  502. package/src/utils/stream-chunk-processor.ts +57 -0
  503. package/src/utils/strings.ts +271 -0
  504. package/src/utils/subscription.ts +31 -0
  505. package/src/utils/syntax-highlighter.tsx +19 -0
  506. package/src/utils/terminal-color-detection.ts +474 -0
  507. package/src/utils/terminal-command-broker.ts +467 -0
  508. package/src/utils/terminal-enter-detection.ts +63 -0
  509. package/src/utils/terminal-images.ts +229 -0
  510. package/src/utils/terminal-io.ts +85 -0
  511. package/src/utils/terminal-protocol-controller.ts +173 -0
  512. package/src/utils/terminal-reset-sequences.ts +33 -0
  513. package/src/utils/terminal-title.ts +84 -0
  514. package/src/utils/terminal-watchdog.ts +366 -0
  515. package/src/utils/text-layout.ts +149 -0
  516. package/src/utils/theme-config.ts +141 -0
  517. package/src/utils/theme-system.ts +1206 -0
  518. package/src/utils/think-tag-parser.ts +95 -0
  519. package/src/utils/time-format.ts +21 -0
  520. package/src/utils/trace-writer.ts +162 -0
  521. package/src/utils/ui-constants.ts +69 -0
  522. package/src/utils/update-checker.ts +110 -0
  523. package/src/utils/usage-banner-state.ts +146 -0
  524. package/src/utils/validation-error-formatting.ts +86 -0
  525. package/src/utils/validation-error-helpers.ts +19 -0
  526. package/src/utils/windows-terminal-health.ts +124 -0
  527. package/src/utils/word-wrap-utils.ts +53 -0
  528. package/src/utils/write-file-atomic.ts +54 -0
  529. package/src/utils/yield-to-event-loop.ts +9 -0
@@ -0,0 +1,1000 @@
1
+ import { describe, test, expect, beforeEach, afterEach } from 'bun:test'
2
+
3
+ import {
4
+ invalidateActivityQuery,
5
+ removeActivityQuery,
6
+ getActivityQueryData,
7
+ setActivityQueryData,
8
+ resetActivityQueryCache,
9
+ isEntryStale,
10
+ setErrorOnlyCacheEntry,
11
+ _retryTestHelpers,
12
+ } from '../use-activity-query'
13
+
14
+ describe('use-activity-query utilities', () => {
15
+ beforeEach(() => {
16
+ // Reset cache between tests
17
+ resetActivityQueryCache()
18
+ })
19
+
20
+ describe('setActivityQueryData', () => {
21
+ test('stores data in cache', () => {
22
+ setActivityQueryData(['test'], { value: 'hello' })
23
+ expect(getActivityQueryData<{ value: string }>(['test'])).toEqual({ value: 'hello' })
24
+ })
25
+
26
+ test('overwrites existing data', () => {
27
+ setActivityQueryData(['test'], { value: 'first' })
28
+ setActivityQueryData(['test'], { value: 'second' })
29
+ expect(getActivityQueryData<{ value: string }>(['test'])).toEqual({ value: 'second' })
30
+ })
31
+
32
+ test('handles complex query keys', () => {
33
+ setActivityQueryData(['users', 1], { name: 'John' })
34
+ expect(getActivityQueryData<{ name: string }>(['users', 1])).toEqual({ name: 'John' })
35
+ })
36
+
37
+ test('handles query keys with objects', () => {
38
+ setActivityQueryData(['complex', { id: 1 }], { data: 'test' })
39
+ expect(getActivityQueryData<{ data: string }>(['complex', { id: 1 }])).toEqual({
40
+ data: 'test',
41
+ })
42
+ })
43
+
44
+ test('different keys store different data', () => {
45
+ setActivityQueryData(['key1'], 'value1')
46
+ setActivityQueryData(['key2'], 'value2')
47
+ expect(getActivityQueryData<string>(['key1'])).toBe('value1')
48
+ expect(getActivityQueryData<string>(['key2'])).toBe('value2')
49
+ })
50
+ })
51
+
52
+ describe('getActivityQueryData', () => {
53
+ test('returns undefined for non-existent key', () => {
54
+ expect(getActivityQueryData(['nonexistent'])).toBeUndefined()
55
+ })
56
+
57
+ test('returns stored data for existing key', () => {
58
+ setActivityQueryData(['test'], 42)
59
+ expect(getActivityQueryData<number>(['test'])).toBe(42)
60
+ })
61
+
62
+ test('returns correct type', () => {
63
+ setActivityQueryData<string[]>(['test'], ['a', 'b', 'c'])
64
+ const data = getActivityQueryData<string[]>(['test'])
65
+ expect(data).toEqual(['a', 'b', 'c'])
66
+ })
67
+ })
68
+
69
+ describe('removeActivityQuery', () => {
70
+ test('removes existing cache entry', () => {
71
+ setActivityQueryData(['test'], 'value')
72
+ expect(getActivityQueryData<string>(['test'])).toBe('value')
73
+
74
+ removeActivityQuery(['test'])
75
+ expect(getActivityQueryData(['test'])).toBeUndefined()
76
+ })
77
+
78
+ test('does nothing for non-existent key', () => {
79
+ // Should not throw
80
+ removeActivityQuery(['nonexistent'])
81
+ expect(getActivityQueryData(['nonexistent'])).toBeUndefined()
82
+ })
83
+
84
+ test('only removes specified key', () => {
85
+ setActivityQueryData(['key1'], 'value1')
86
+ setActivityQueryData(['key2'], 'value2')
87
+
88
+ removeActivityQuery(['key1'])
89
+
90
+ expect(getActivityQueryData(['key1'])).toBeUndefined()
91
+ expect(getActivityQueryData<string>(['key2'])).toBe('value2')
92
+ })
93
+ })
94
+
95
+ describe('invalidateActivityQuery', () => {
96
+ test('marks query as stale by setting dataUpdatedAt to 0', () => {
97
+ setActivityQueryData(['test'], 'value')
98
+
99
+ // Before invalidation, data exists
100
+ expect(getActivityQueryData<string>(['test'])).toBe('value')
101
+
102
+ invalidateActivityQuery(['test'])
103
+
104
+ // Data should still exist after invalidation
105
+ expect(getActivityQueryData<string>(['test'])).toBe('value')
106
+ })
107
+
108
+ test('does nothing for non-existent key', () => {
109
+ // Should not throw
110
+ invalidateActivityQuery(['nonexistent'])
111
+ })
112
+ })
113
+
114
+ describe('query key serialization', () => {
115
+ test('same array values produce same cache key', () => {
116
+ setActivityQueryData(['test', 'key'], 'value')
117
+ expect(getActivityQueryData<string>(['test', 'key'])).toBe('value')
118
+ })
119
+
120
+ test('different array values produce different cache keys', () => {
121
+ setActivityQueryData(['test', 'key1'], 'value1')
122
+ setActivityQueryData(['test', 'key2'], 'value2')
123
+ expect(getActivityQueryData<string>(['test', 'key1'])).toBe('value1')
124
+ expect(getActivityQueryData<string>(['test', 'key2'])).toBe('value2')
125
+ })
126
+
127
+ test('object keys are serialized correctly', () => {
128
+ setActivityQueryData(['query', { page: 1, sort: 'asc' }], 'page1')
129
+ expect(getActivityQueryData<string>(['query', { page: 1, sort: 'asc' }])).toBe(
130
+ 'page1',
131
+ )
132
+ })
133
+
134
+ test('nested objects in keys work correctly', () => {
135
+ setActivityQueryData(
136
+ ['query', { filter: { status: 'active', type: 'user' } }],
137
+ 'filtered',
138
+ )
139
+ expect(
140
+ getActivityQueryData<string>([
141
+ 'query',
142
+ { filter: { status: 'active', type: 'user' } },
143
+ ]),
144
+ ).toBe('filtered')
145
+ })
146
+ })
147
+ })
148
+
149
+ describe('useActivityQuery hook behavior', () => {
150
+ // These tests verify the hook's expected behavior patterns
151
+ // We can't easily test the actual hook without a React renderer,
152
+ // but we can test the underlying cache behavior
153
+
154
+ describe('cache entry structure', () => {
155
+ test('setActivityQueryData creates proper cache entry', () => {
156
+ const testData = { users: [1, 2, 3] }
157
+ setActivityQueryData(['users'], testData)
158
+
159
+ const retrieved = getActivityQueryData<typeof testData>(['users'])
160
+ expect(retrieved).toEqual(testData)
161
+ })
162
+
163
+ test('cache preserves data types', () => {
164
+ // Numbers
165
+ setActivityQueryData(['number'], 42)
166
+ expect(getActivityQueryData<number>(['number'])).toBe(42)
167
+
168
+ // Strings
169
+ setActivityQueryData(['string'], 'hello')
170
+ expect(getActivityQueryData<string>(['string'])).toBe('hello')
171
+
172
+ // Booleans
173
+ setActivityQueryData(['boolean'], true)
174
+ expect(getActivityQueryData<boolean>(['boolean'])).toBe(true)
175
+
176
+ // Arrays
177
+ setActivityQueryData(['array'], [1, 2, 3])
178
+ expect(getActivityQueryData<number[]>(['array'])).toEqual([1, 2, 3])
179
+
180
+ // Objects
181
+ setActivityQueryData(['object'], { a: 1, b: 2 })
182
+ expect(getActivityQueryData<{ a: number; b: number }>(['object'])).toEqual({ a: 1, b: 2 })
183
+
184
+ // Null
185
+ setActivityQueryData(['null'], null)
186
+ expect(getActivityQueryData<null>(['null'])).toBeNull()
187
+ })
188
+ })
189
+
190
+ describe('invalidation behavior', () => {
191
+ test('invalidation preserves existing data', () => {
192
+ const originalData = { id: 1, name: 'Test' }
193
+ setActivityQueryData(['preserve'], originalData)
194
+
195
+ invalidateActivityQuery(['preserve'])
196
+
197
+ // Data should still be accessible
198
+ expect(getActivityQueryData<typeof originalData>(['preserve'])).toEqual(originalData)
199
+ })
200
+
201
+ test('multiple invalidations do not remove data', () => {
202
+ setActivityQueryData(['multi'], 'persistent')
203
+
204
+ invalidateActivityQuery(['multi'])
205
+ invalidateActivityQuery(['multi'])
206
+ invalidateActivityQuery(['multi'])
207
+
208
+ expect(getActivityQueryData<string>(['multi'])).toBe('persistent')
209
+ })
210
+ })
211
+
212
+ describe('remove behavior', () => {
213
+ test('remove completely clears the cache entry', () => {
214
+ setActivityQueryData(['remove-test'], 'data')
215
+ expect(getActivityQueryData<string>(['remove-test'])).toBe('data')
216
+
217
+ removeActivityQuery(['remove-test'])
218
+ expect(getActivityQueryData(['remove-test'])).toBeUndefined()
219
+
220
+ // Can set new data after removal
221
+ setActivityQueryData(['remove-test'], 'new-data')
222
+ expect(getActivityQueryData<string>(['remove-test'])).toBe('new-data')
223
+ })
224
+ })
225
+
226
+ describe('resetActivityQueryCache', () => {
227
+ test('clears all cache entries', () => {
228
+ setActivityQueryData(['key1'], 'value1')
229
+ setActivityQueryData(['key2'], 'value2')
230
+ setActivityQueryData(['key3'], 'value3')
231
+
232
+ expect(getActivityQueryData<string>(['key1'])).toBe('value1')
233
+ expect(getActivityQueryData<string>(['key2'])).toBe('value2')
234
+ expect(getActivityQueryData<string>(['key3'])).toBe('value3')
235
+
236
+ resetActivityQueryCache()
237
+
238
+ expect(getActivityQueryData(['key1'])).toBeUndefined()
239
+ expect(getActivityQueryData(['key2'])).toBeUndefined()
240
+ expect(getActivityQueryData(['key3'])).toBeUndefined()
241
+ })
242
+
243
+ test('allows setting new data after reset', () => {
244
+ setActivityQueryData(['test'], 'old')
245
+ resetActivityQueryCache()
246
+ setActivityQueryData(['test'], 'new')
247
+ expect(getActivityQueryData<string>(['test'])).toBe('new')
248
+ })
249
+ })
250
+ })
251
+
252
+ describe('staleness calculation', () => {
253
+ beforeEach(() => {
254
+ resetActivityQueryCache()
255
+ })
256
+
257
+ test('data is considered stale after staleTime has passed', () => {
258
+ const staleTime = 100 // 100ms
259
+ const testKey = ['stale-test']
260
+
261
+ // Set data with a timestamp in the past
262
+ setActivityQueryData(testKey, 'test-value')
263
+
264
+ // Immediately after setting, data should be fresh
265
+ const dataImmediately = getActivityQueryData<string>(testKey)
266
+ expect(dataImmediately).toBe('test-value')
267
+ })
268
+
269
+ test('invalidated data should be refetchable', () => {
270
+ const testKey = ['invalidate-test']
271
+
272
+ // Set initial data
273
+ setActivityQueryData(testKey, 'initial')
274
+ expect(getActivityQueryData<string>(testKey)).toBe('initial')
275
+
276
+ // Invalidate - should mark as stale (dataUpdatedAt = 0)
277
+ invalidateActivityQuery(testKey)
278
+
279
+ // Data should still exist but be stale
280
+ expect(getActivityQueryData<string>(testKey)).toBe('initial')
281
+ })
282
+ })
283
+
284
+ describe('refetch interval staleness bug fix', () => {
285
+ // This test verifies the fix for the bug where refetch intervals stopped working
286
+ // because isStale was captured in a closure and never updated.
287
+ // The fix ensures staleness is computed dynamically by reading from cache.
288
+
289
+ beforeEach(() => {
290
+ resetActivityQueryCache()
291
+ })
292
+
293
+ test('setActivityQueryData sets dataUpdatedAt to current time', () => {
294
+ const before = Date.now()
295
+ setActivityQueryData(['timing-test'], 'value')
296
+ const after = Date.now()
297
+
298
+ // The data should exist
299
+ expect(getActivityQueryData<string>(['timing-test'])).toBe('value')
300
+
301
+ // We can't directly access dataUpdatedAt, but we can verify the data was set
302
+ // and invalidation resets it to 0
303
+ invalidateActivityQuery(['timing-test'])
304
+
305
+ // Data should still exist after invalidation
306
+ expect(getActivityQueryData<string>(['timing-test'])).toBe('value')
307
+ })
308
+
309
+ test('fresh data followed by stale time passage should allow refetch', () => {
310
+ // This simulates the scenario where:
311
+ // 1. Data is fetched (fresh)
312
+ // 2. staleTime passes
313
+ // 3. Interval should refetch (was broken before fix)
314
+
315
+ const testKey = ['refetch-bug-test']
316
+
317
+ // Step 1: Set "fresh" data
318
+ setActivityQueryData(testKey, 'fresh-data')
319
+ expect(getActivityQueryData<string>(testKey)).toBe('fresh-data')
320
+
321
+ // Step 2: Invalidate to simulate staleness (sets dataUpdatedAt to 0)
322
+ invalidateActivityQuery(testKey)
323
+
324
+ // The data should still exist but be considered stale
325
+ // (dataUpdatedAt is 0, so any staleTime > 0 would make it stale)
326
+ expect(getActivityQueryData<string>(testKey)).toBe('fresh-data')
327
+
328
+ // In the old buggy code, the interval tick would check closure-captured isStale
329
+ // which was false (computed when effect ran right after fetch).
330
+ // In the fixed code, staleness is computed dynamically from cache.
331
+
332
+ // We can't easily test the hook behavior without React, but we verify
333
+ // the cache manipulation works correctly for the staleness check
334
+ })
335
+
336
+ test('multiple data updates preserve latest data', () => {
337
+ const testKey = ['multi-update-test']
338
+
339
+ setActivityQueryData(testKey, 'first')
340
+ expect(getActivityQueryData<string>(testKey)).toBe('first')
341
+
342
+ setActivityQueryData(testKey, 'second')
343
+ expect(getActivityQueryData<string>(testKey)).toBe('second')
344
+
345
+ setActivityQueryData(testKey, 'third')
346
+ expect(getActivityQueryData<string>(testKey)).toBe('third')
347
+
348
+ // Invalidate and verify data is preserved
349
+ invalidateActivityQuery(testKey)
350
+ expect(getActivityQueryData<string>(testKey)).toBe('third')
351
+ })
352
+ })
353
+
354
+ /**
355
+ * Tests for cache listener notification behavior.
356
+ * These tests verify that cache updates properly notify subscribers,
357
+ * which is critical for React components to re-render when data changes.
358
+ */
359
+ describe('cache listener notifications', () => {
360
+ beforeEach(() => {
361
+ resetActivityQueryCache()
362
+ })
363
+
364
+ test('setActivityQueryData notifies listeners', () => {
365
+ const testKey = ['listener-test']
366
+ let notificationCount = 0
367
+
368
+ // First set up some data so the cache entry exists
369
+ setActivityQueryData(testKey, 'initial')
370
+
371
+ // Now update the data - we can't directly subscribe but we can verify
372
+ // the data is updated properly
373
+ setActivityQueryData(testKey, 'updated')
374
+ expect(getActivityQueryData<string>(testKey)).toBe('updated')
375
+ })
376
+
377
+ test('invalidateActivityQuery notifies listeners', () => {
378
+ const testKey = ['invalidate-listener-test']
379
+
380
+ // Set initial data
381
+ setActivityQueryData(testKey, 'data')
382
+
383
+ // Invalidate should trigger listeners
384
+ invalidateActivityQuery(testKey)
385
+
386
+ // Data should still be there but marked stale
387
+ expect(getActivityQueryData<string>(testKey)).toBe('data')
388
+ })
389
+
390
+ test('removeActivityQuery clears data and notifies listeners', () => {
391
+ const testKey = ['remove-listener-test']
392
+
393
+ setActivityQueryData(testKey, 'data')
394
+ expect(getActivityQueryData<string>(testKey)).toBe('data')
395
+
396
+ removeActivityQuery(testKey)
397
+ expect(getActivityQueryData<string>(testKey)).toBeUndefined()
398
+ })
399
+ })
400
+
401
+ /**
402
+ * Tests simulating the polling behavior to verify refetch intervals work.
403
+ * These tests mock Date.now() to simulate time passing.
404
+ */
405
+ describe('polling and staleness simulation', () => {
406
+ let originalDateNow: typeof Date.now
407
+ let mockNow: number
408
+
409
+ beforeEach(() => {
410
+ resetActivityQueryCache()
411
+ originalDateNow = Date.now
412
+ mockNow = 1000000
413
+ Date.now = () => mockNow
414
+ })
415
+
416
+ afterEach(() => {
417
+ Date.now = originalDateNow
418
+ })
419
+
420
+ test('data becomes stale after staleTime passes', () => {
421
+ const testKey = ['stale-time-test']
422
+ const serializedKey = JSON.stringify(testKey)
423
+ const staleTime = 30000 // 30 seconds
424
+
425
+ // Set data at t=0
426
+ setActivityQueryData(testKey, 'fresh-data')
427
+
428
+ // Data was set at mockNow (1000000), so dataUpdatedAt = 1000000
429
+ expect(getActivityQueryData<string>(testKey)).toBe('fresh-data')
430
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false) // Fresh
431
+
432
+ // Advance time by 25 seconds - still fresh
433
+ mockNow += 25000
434
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
435
+
436
+ // Advance time past staleTime
437
+ mockNow += 10000 // Now 35 seconds have passed
438
+ // Data should now be considered stale (35s > 30s staleTime)
439
+ expect(isEntryStale(serializedKey, staleTime)).toBe(true)
440
+
441
+ // The data is still accessible even when stale
442
+ expect(getActivityQueryData<string>(testKey)).toBe('fresh-data')
443
+ })
444
+
445
+ test('invalidated data is immediately stale', () => {
446
+ const testKey = ['invalidate-stale-test']
447
+ const serializedKey = JSON.stringify(testKey)
448
+ const staleTime = 30000
449
+
450
+ // Set fresh data
451
+ setActivityQueryData(testKey, 'data')
452
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
453
+
454
+ // Invalidate immediately makes it stale (dataUpdatedAt = 0)
455
+ invalidateActivityQuery(testKey)
456
+ expect(isEntryStale(serializedKey, staleTime)).toBe(true)
457
+
458
+ // Data still exists but would be refetched on next access
459
+ expect(getActivityQueryData<string>(testKey)).toBe('data')
460
+ })
461
+
462
+ test('updating data resets the staleness timer', () => {
463
+ const testKey = ['reset-timer-test']
464
+ const serializedKey = JSON.stringify(testKey)
465
+ const staleTime = 30000
466
+
467
+ // Set initial data
468
+ setActivityQueryData(testKey, 'initial')
469
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
470
+
471
+ // Advance time past staleTime
472
+ mockNow += 35000
473
+ expect(isEntryStale(serializedKey, staleTime)).toBe(true)
474
+
475
+ // Update data - should reset the timer
476
+ setActivityQueryData(testKey, 'updated')
477
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false) // Fresh again
478
+
479
+ // Data is fresh again
480
+ expect(getActivityQueryData<string>(testKey)).toBe('updated')
481
+
482
+ // Advance a little bit - should still be fresh
483
+ mockNow += 10000
484
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
485
+ expect(getActivityQueryData<string>(testKey)).toBe('updated')
486
+ })
487
+
488
+ test('staleTime of 0 means always stale', () => {
489
+ const testKey = ['zero-stale-test']
490
+ const serializedKey = JSON.stringify(testKey)
491
+
492
+ // Set data
493
+ setActivityQueryData(testKey, 'data')
494
+
495
+ // With staleTime=0, data is always considered stale
496
+ // (this means refetch should happen on every interval tick)
497
+ expect(isEntryStale(serializedKey, 0)).toBe(true)
498
+ expect(getActivityQueryData<string>(testKey)).toBe('data')
499
+ })
500
+
501
+ test('non-existent key is always stale', () => {
502
+ const serializedKey = JSON.stringify(['non-existent'])
503
+ expect(isEntryStale(serializedKey, 30000)).toBe(true)
504
+ })
505
+ })
506
+
507
+ /**
508
+ * Tests for the refetch on activity feature.
509
+ * Verifies that data is refetched when user becomes active after being idle.
510
+ */
511
+ describe('refetch on activity behavior', () => {
512
+ let originalDateNow: typeof Date.now
513
+ let mockNow: number
514
+
515
+ beforeEach(() => {
516
+ resetActivityQueryCache()
517
+ originalDateNow = Date.now
518
+ mockNow = 1000000
519
+ Date.now = () => mockNow
520
+ })
521
+
522
+ afterEach(() => {
523
+ Date.now = originalDateNow
524
+ })
525
+
526
+ test('data should be refetchable when user becomes active after idle', () => {
527
+ const testKey = ['activity-refetch-test']
528
+ const staleTime = 30000
529
+ const idleThreshold = 30000
530
+
531
+ // Set initial data
532
+ setActivityQueryData(testKey, 'initial')
533
+
534
+ // Simulate time passing beyond staleTime
535
+ mockNow += 35000
536
+
537
+ // At this point, if user was idle and becomes active,
538
+ // and data is stale, a refetch should occur
539
+
540
+ // Data should still be accessible
541
+ expect(getActivityQueryData<string>(testKey)).toBe('initial')
542
+
543
+ // Update with new data (simulating what refetch would do)
544
+ setActivityQueryData(testKey, 'refetched')
545
+ expect(getActivityQueryData<string>(testKey)).toBe('refetched')
546
+ })
547
+
548
+ test('pause when idle should prevent polling updates', () => {
549
+ const testKey = ['pause-idle-test']
550
+
551
+ // Set data
552
+ setActivityQueryData(testKey, 'before-idle')
553
+
554
+ // When pauseWhenIdle is true and user is idle:
555
+ // - Polling interval fires
556
+ // - But isUserActive returns false
557
+ // - So no refetch happens
558
+
559
+ // Data remains unchanged
560
+ expect(getActivityQueryData<string>(testKey)).toBe('before-idle')
561
+ })
562
+ })
563
+
564
+ /**
565
+ * Tests for edge cases and error scenarios in the caching system.
566
+ */
567
+ describe('cache edge cases and error handling', () => {
568
+ beforeEach(() => {
569
+ resetActivityQueryCache()
570
+ })
571
+
572
+ test('setting undefined data should still create cache entry', () => {
573
+ const testKey = ['undefined-test']
574
+
575
+ setActivityQueryData(testKey, undefined)
576
+
577
+ // getActivityQueryData returns undefined for both "not in cache" and "data is undefined"
578
+ // This is expected behavior - undefined is a valid cached value
579
+ expect(getActivityQueryData(testKey)).toBeUndefined()
580
+ })
581
+
582
+ test('setting null data should store null', () => {
583
+ const testKey = ['null-test']
584
+
585
+ setActivityQueryData(testKey, null)
586
+
587
+ expect(getActivityQueryData(testKey)).toBeNull()
588
+ })
589
+
590
+ test('complex nested objects should be stored correctly', () => {
591
+ const testKey = ['complex-object-test']
592
+
593
+ const complexData = {
594
+ user: {
595
+ id: 1,
596
+ profile: {
597
+ name: 'Test',
598
+ settings: {
599
+ theme: 'dark',
600
+ notifications: [1, 2, 3],
601
+ },
602
+ },
603
+ },
604
+ timestamp: new Date('2024-01-01'),
605
+ }
606
+
607
+ setActivityQueryData(testKey, complexData)
608
+
609
+ const cached = getActivityQueryData<typeof complexData>(testKey)
610
+ expect(cached?.user.profile.settings.notifications).toEqual([1, 2, 3])
611
+ expect(cached?.timestamp).toEqual(new Date('2024-01-01'))
612
+ })
613
+
614
+ test('array data should be stored and retrieved correctly', () => {
615
+ const testKey = ['array-test']
616
+
617
+ const arrayData = [1, 2, 3, { nested: 'value' }]
618
+ setActivityQueryData(testKey, arrayData)
619
+
620
+ const cached = getActivityQueryData<typeof arrayData>(testKey)
621
+ expect(cached).toEqual(arrayData)
622
+ expect(cached?.[3]).toEqual({ nested: 'value' })
623
+ })
624
+
625
+ test('invalidating non-existent key should not throw', () => {
626
+ expect(() => {
627
+ invalidateActivityQuery(['non-existent-key'])
628
+ }).not.toThrow()
629
+ })
630
+
631
+ test('removing non-existent key should not throw', () => {
632
+ expect(() => {
633
+ removeActivityQuery(['non-existent-key'])
634
+ }).not.toThrow()
635
+ })
636
+
637
+ test('getting data after remove should return undefined', () => {
638
+ const testKey = ['remove-then-get-test']
639
+
640
+ setActivityQueryData(testKey, 'data')
641
+ removeActivityQuery(testKey)
642
+
643
+ expect(getActivityQueryData(testKey)).toBeUndefined()
644
+ })
645
+
646
+ test('setting data after remove should work', () => {
647
+ const testKey = ['remove-then-set-test']
648
+
649
+ setActivityQueryData(testKey, 'first')
650
+ removeActivityQuery(testKey)
651
+ setActivityQueryData(testKey, 'second')
652
+
653
+ expect(getActivityQueryData<string>(testKey)).toBe('second')
654
+ })
655
+ })
656
+
657
+ /**
658
+ * Tests for error-only cache entries and persistent error scenarios.
659
+ * This test suite was added to debug and fix an issue where fetchSubscriptionData
660
+ * was being called every second when the endpoint returned errors.
661
+ */
662
+ describe('error-only entries and persistent error handling', () => {
663
+ let originalDateNow: typeof Date.now
664
+ let mockNow: number
665
+
666
+ beforeEach(() => {
667
+ resetActivityQueryCache()
668
+ originalDateNow = Date.now
669
+ mockNow = 1000000
670
+ Date.now = () => mockNow
671
+ })
672
+
673
+ afterEach(() => {
674
+ Date.now = originalDateNow
675
+ })
676
+
677
+ test('setErrorOnlyCacheEntry creates entry with no data and error', () => {
678
+ const testKey = ['error-entry-test']
679
+ const error = new Error('Network error')
680
+
681
+ setErrorOnlyCacheEntry(testKey, error)
682
+
683
+ // Data should be undefined (error-only entry)
684
+ expect(getActivityQueryData(testKey)).toBeUndefined()
685
+ })
686
+
687
+ test('error-only entry with recent errorUpdatedAt should NOT be stale', () => {
688
+ // This test verifies the fix for the infinite refetch loop bug.
689
+ //
690
+ // Scenario:
691
+ // 1. Fetch fails with no prior data
692
+ // 2. Error is stored with errorUpdatedAt = now
693
+ // 3. Polling tick fires
694
+ // 4. isEntryStale should return FALSE if errorUpdatedAt is recent
695
+ // 5. This prevents immediate refetch loop
696
+
697
+ const testKey = ['error-only-fresh-test']
698
+ const serializedKey = JSON.stringify(testKey)
699
+ const staleTime = 30000 // 30 seconds
700
+ const error = new Error('API error')
701
+
702
+ // Create error-only entry at current time (mockNow = 1000000)
703
+ setErrorOnlyCacheEntry(testKey, error, mockNow)
704
+
705
+ // Entry has errorUpdatedAt = 1000000, current time = 1000000
706
+ // Time since error: 0ms, staleTime: 30000ms
707
+ // Should NOT be stale because error is recent
708
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
709
+ })
710
+
711
+ test('error-only entry becomes stale after staleTime passes', () => {
712
+ const testKey = ['error-stale-after-time-test']
713
+ const serializedKey = JSON.stringify(testKey)
714
+ const staleTime = 30000 // 30 seconds
715
+ const error = new Error('API error')
716
+
717
+ // Create error-only entry at current time
718
+ setErrorOnlyCacheEntry(testKey, error, mockNow)
719
+
720
+ // Initially not stale
721
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
722
+
723
+ // Advance time by 25 seconds - still fresh
724
+ mockNow += 25000
725
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
726
+
727
+ // Advance time past staleTime (now 35 seconds since error)
728
+ mockNow += 10000
729
+ expect(isEntryStale(serializedKey, staleTime)).toBe(true)
730
+ })
731
+
732
+ test('simulates subscription query polling with persistent errors', () => {
733
+ // This test simulates the exact bug scenario:
734
+ // - useSubscriptionQuery with staleTime=30s, refetchInterval=60s
735
+ // - Endpoint returns errors
736
+ // - Without fix: isEntryStale returns true immediately, causing rapid refetches
737
+ // - With fix: isEntryStale uses errorUpdatedAt, preventing rapid refetches
738
+
739
+ const subscriptionKey = ['subscription', 'current']
740
+ const serializedKey = JSON.stringify(subscriptionKey)
741
+ const staleTime = 30000 // 30 seconds (matches useSubscriptionQuery)
742
+ const refetchInterval = 60000 // 60 seconds
743
+ const error = new Error('Failed to fetch subscription: 500')
744
+
745
+ // Simulate first fetch failure at t=0
746
+ setErrorOnlyCacheEntry(subscriptionKey, error, mockNow)
747
+
748
+ // Immediately after error, entry should NOT be stale
749
+ // This is the critical fix - prevents immediate refetch loop
750
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
751
+
752
+ // Simulate polling interval at t=1s (as reported in bug)
753
+ mockNow += 1000
754
+ // Entry should still NOT be stale (only 1s since error, staleTime is 30s)
755
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
756
+
757
+ // Simulate many 1-second intervals - none should trigger refetch until staleTime
758
+ for (let i = 0; i < 28; i++) {
759
+ mockNow += 1000
760
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
761
+ }
762
+
763
+ // Now at t=29s - should still be fresh (29s is not > 30s)
764
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
765
+
766
+ // At t=30s - should still be fresh (30s is not > 30s, need strictly greater)
767
+ mockNow += 1000
768
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false)
769
+
770
+ // At t=31s - now stale, refetch should be allowed (31s > 30s)
771
+ mockNow += 1000
772
+ expect(isEntryStale(serializedKey, staleTime)).toBe(true)
773
+ })
774
+
775
+ test('staleTime of 0 means always stale even for error-only entries', () => {
776
+ const testKey = ['zero-stale-error-test']
777
+ const serializedKey = JSON.stringify(testKey)
778
+ const error = new Error('Some error')
779
+
780
+ setErrorOnlyCacheEntry(testKey, error, mockNow)
781
+
782
+ // With staleTime=0, entry is always considered stale
783
+ expect(isEntryStale(serializedKey, 0)).toBe(true)
784
+ })
785
+
786
+ test('error-only entry with null errorUpdatedAt is stale', () => {
787
+ // Edge case: if somehow errorUpdatedAt is null, entry should be stale
788
+ // This shouldn't happen in practice but tests defensive coding
789
+ const testKey = ['null-error-time-test']
790
+ const serializedKey = JSON.stringify(testKey)
791
+ const staleTime = 30000
792
+
793
+ // Create entry without errorUpdatedAt (using undefined which gets stored as null)
794
+ // Note: setErrorOnlyCacheEntry always sets errorUpdatedAt, so we test via regular data
795
+ // and then invalidate it
796
+
797
+ // Non-existent key is stale
798
+ expect(isEntryStale(serializedKey, staleTime)).toBe(true)
799
+ })
800
+
801
+ test('successful data takes precedence over errorUpdatedAt for staleness', () => {
802
+ const testKey = ['data-precedence-test']
803
+ const serializedKey = JSON.stringify(testKey)
804
+ const staleTime = 30000
805
+
806
+ // First, set an error-only entry
807
+ setErrorOnlyCacheEntry(testKey, new Error('Initial error'), mockNow)
808
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false) // Fresh error
809
+
810
+ // Now set successful data (this is what happens on successful retry)
811
+ setActivityQueryData(testKey, { subscription: 'active' })
812
+
813
+ // Staleness should now be based on dataUpdatedAt, not errorUpdatedAt
814
+ expect(isEntryStale(serializedKey, staleTime)).toBe(false) // Fresh data
815
+
816
+ // Advance time past staleTime
817
+ mockNow += 35000
818
+ expect(isEntryStale(serializedKey, staleTime)).toBe(true) // Stale based on dataUpdatedAt
819
+ })
820
+ })
821
+
822
+ /**
823
+ * Tests for the retry infinite loop bug.
824
+ *
825
+ * BUG: When useSubscriptionQuery fetched /api/user/subscription and got a 401,
826
+ * it would retry every ~1 second infinitely instead of respecting retry:1.
827
+ *
828
+ * ROOT CAUSE: In doFetch's catch block, when scheduling a retry:
829
+ * 1. retryCounts.set(key, next) // Sets count to 1
830
+ * 2. clearRetryState(key) // Deletes retryCounts → count back to 0!
831
+ * 3. setTimeout to retry in 1s
832
+ * When the retry fires, currentRetries reads as 0 again → thinks it still has
833
+ * retries left → schedules another retry → infinite loop.
834
+ *
835
+ * FIX: Split clearRetryState into clearRetryTimeout (only clears timeout)
836
+ * and clearRetryState (clears both). The retry scheduling block now uses
837
+ * clearRetryTimeout so the retry count is preserved.
838
+ */
839
+ describe('retry infinite loop bug fix (subscription 401 scenario)', () => {
840
+ beforeEach(() => {
841
+ resetActivityQueryCache()
842
+ })
843
+
844
+ test('retry count is preserved after scheduling a retry', () => {
845
+ const queryKey = ['subscription', 'current']
846
+ const maxRetries = 1
847
+
848
+ // Simulate a mounted component (refCount > 0)
849
+ _retryTestHelpers.setRefCount(queryKey, 1)
850
+
851
+ // Initially, no retries have been attempted
852
+ expect(_retryTestHelpers.getRetryCount(queryKey)).toBe(0)
853
+
854
+ // First fetch fails → should schedule a retry
855
+ const result1 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
856
+ expect(result1.retryScheduled).toBe(true)
857
+ expect(result1.retryCount).toBe(1)
858
+
859
+ // CRITICAL: Retry count must be preserved (not reset to 0)
860
+ expect(_retryTestHelpers.getRetryCount(queryKey)).toBe(1)
861
+ })
862
+
863
+ test('retries are exhausted after maxRetries attempts', () => {
864
+ const queryKey = ['subscription', 'current']
865
+ const maxRetries = 1
866
+
867
+ _retryTestHelpers.setRefCount(queryKey, 1)
868
+
869
+ // First fetch fails → retry scheduled (count becomes 1)
870
+ const result1 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
871
+ expect(result1.retryScheduled).toBe(true)
872
+ expect(result1.retryCount).toBe(1)
873
+
874
+ // Retry fires, also fails → retries exhausted (count = 1, not < maxRetries=1)
875
+ const result2 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
876
+ expect(result2.retryScheduled).toBe(false)
877
+ expect(result2.retryCount).toBe(0) // Reset after exhaustion
878
+ })
879
+
880
+ test('simulates full subscription 401 scenario: fetch + 1 retry + stop', () => {
881
+ // This reproduces the exact bug scenario:
882
+ // useSubscriptionQuery with retry:1 hitting a 401 on /api/user/subscription
883
+ const queryKey = ['subscription', 'current']
884
+ const maxRetries = 1
885
+
886
+ // Component is mounted
887
+ _retryTestHelpers.setRefCount(queryKey, 1)
888
+
889
+ // === Fetch #1: Initial fetch fails with 401 ===
890
+ const fetch1 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
891
+ expect(fetch1.retryScheduled).toBe(true)
892
+ expect(fetch1.retryCount).toBe(1)
893
+
894
+ // === Fetch #2: Retry fires after 1s, also fails with 401 ===
895
+ const fetch2 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
896
+ expect(fetch2.retryScheduled).toBe(false) // Retries exhausted!
897
+ expect(fetch2.retryCount).toBe(0)
898
+
899
+ // === Fetch #3: If the bug existed, this would schedule ANOTHER retry ===
900
+ // With the fix, the error is stored and no more retries are scheduled.
901
+ // A third call should also exhaust immediately since count was reset to 0
902
+ // BUT there's no retry scheduled, so this would only happen from polling.
903
+ const fetch3 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
904
+ // Even if polling triggers another fetch, retry:1 means ONE retry per fetch cycle
905
+ expect(fetch3.retryScheduled).toBe(true) // New fetch cycle starts fresh
906
+ expect(fetch3.retryCount).toBe(1)
907
+
908
+ // The retry for fetch3 fires and fails
909
+ const fetch4 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
910
+ expect(fetch4.retryScheduled).toBe(false) // Exhausted again
911
+ })
912
+
913
+ test('demonstrates the old bug: clearRetryState would reset count causing infinite loop', () => {
914
+ // This test documents the OLD buggy behavior.
915
+ // The old code called clearRetryState (which deletes retryCounts) right after
916
+ // setting the retry count, effectively resetting it to 0 every time.
917
+ const queryKey = ['subscription', 'current']
918
+
919
+ _retryTestHelpers.setRefCount(queryKey, 1)
920
+
921
+ // Step 1: Simulate first fetch failure setting retry count to 1
922
+ _retryTestHelpers.setRetryCount(queryKey, 1)
923
+ expect(_retryTestHelpers.getRetryCount(queryKey)).toBe(1)
924
+
925
+ // Step 2: OLD CODE would call clearRetryState here, which resets count to 0:
926
+ // clearRetryState(key) → retryCounts.delete(key) → count = 0
927
+ // Simulate the old bug by manually resetting:
928
+ _retryTestHelpers.setRetryCount(queryKey, 0)
929
+ expect(_retryTestHelpers.getRetryCount(queryKey)).toBe(0)
930
+
931
+ // Step 3: When the retry fires after 1s, it reads count as 0
932
+ // 0 < maxRetries(1) → true → schedules ANOTHER retry (should have been exhausted!)
933
+ const result = _retryTestHelpers.simulateFailedFetch(queryKey, 1)
934
+ expect(result.retryScheduled).toBe(true) // BUG: should have been false!
935
+ expect(result.retryCount).toBe(1) // Count set to 1 again...
936
+
937
+ // And the cycle repeats: count gets reset → retry fires → count is 0 → retry...
938
+ // With the fix (clearRetryTimeout instead of clearRetryState), count stays at 1
939
+ // so the next attempt correctly sees 1 >= maxRetries(1) → exhausted.
940
+ })
941
+
942
+ test('retry count resets to 0 when retries are exhausted', () => {
943
+ const queryKey = ['retry-reset-test']
944
+ const maxRetries = 2
945
+
946
+ _retryTestHelpers.setRefCount(queryKey, 1)
947
+
948
+ // First fail → retry scheduled, count=1
949
+ const r1 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
950
+ expect(r1).toEqual({ retryScheduled: true, retryCount: 1 })
951
+
952
+ // Second fail → retry scheduled, count=2
953
+ const r2 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
954
+ expect(r2).toEqual({ retryScheduled: true, retryCount: 2 })
955
+
956
+ // Third fail → retries exhausted, count reset to 0
957
+ const r3 = _retryTestHelpers.simulateFailedFetch(queryKey, maxRetries)
958
+ expect(r3).toEqual({ retryScheduled: false, retryCount: 0 })
959
+ })
960
+
961
+ test('no retries when retry is 0 or false', () => {
962
+ const queryKey = ['no-retry-test']
963
+ _retryTestHelpers.setRefCount(queryKey, 1)
964
+
965
+ // retry: 0 (equivalent to retry: false)
966
+ const result = _retryTestHelpers.simulateFailedFetch(queryKey, 0)
967
+ expect(result.retryScheduled).toBe(false)
968
+ expect(result.retryCount).toBe(0)
969
+ })
970
+
971
+ test('no retries when component is unmounted (refCount=0)', () => {
972
+ const queryKey = ['unmounted-test']
973
+ // Don't set refCount (defaults to 0 = no mounted components)
974
+
975
+ const result = _retryTestHelpers.simulateFailedFetch(queryKey, 1)
976
+ expect(result.retryScheduled).toBe(false)
977
+ })
978
+
979
+ test('error-only entry is created after retries exhausted', () => {
980
+ const queryKey = ['error-entry-after-retry']
981
+ _retryTestHelpers.setRefCount(queryKey, 1)
982
+
983
+ // First fail → retry
984
+ _retryTestHelpers.simulateFailedFetch(queryKey, 1)
985
+
986
+ // No cache entry yet during retry phase
987
+ expect(getActivityQueryData(queryKey)).toBeUndefined()
988
+
989
+ // Second fail → exhausted, error entry created
990
+ _retryTestHelpers.simulateFailedFetch(queryKey, 1)
991
+
992
+ // Error entry should exist (data is undefined but entry exists)
993
+ // The entry has error set, which we can verify via isEntryStale behavior
994
+ const serializedKey = JSON.stringify(queryKey)
995
+ // Entry exists (not stale due to "no entry" - stale due to other reasons)
996
+ // Since we just set errorUpdatedAt = Date.now(), it should not be stale
997
+ // for a reasonable staleTime
998
+ expect(isEntryStale(serializedKey, 30000)).toBe(false)
999
+ })
1000
+ })