perf(tests): trim sweep, copilot, and timer-bound suites; run three CI shards (#7375)

* perf(ci): run apps/sim as three test shards

* chore(deps): drop unused @types/mdx devDependency from apps/docs

Nothing in apps/docs imports mdx/types or a bare *.mdx module; the
generated .source imports carry a ?collection= suffix that the *.mdx
wildcard does not match, and @types/mdx still resolves transitively via
@mdx-js/mdx for fumadocs. The docs type-check passes without the direct
edge.

* perf(tests): move DOM-free jsdom suites to the node environment

Audited the 229 test files declaring @vitest-environment jsdom. 217 need a
real DOM (React rendering, TipTap/ProseMirror, window/document/localStorage,
DOMParser, ProgressEvent) and stay on jsdom; 12 exercise pure logic (yjs
collab-doc conversion, http tool URL building, panel editor store, log
utils, attachment upload, headless TipTap table commands, terminal-session
tooltip helpers) and now run on node, skipping the jsdom boot.

Dropped one redundant test in apply-streamed-markdown.test.ts: the
hand-rolled one-shot peer case is a strict subset of the NON-OVERLAPPING
two-peer test in apply-streamed-markdown.concurrent.test.ts, which also
asserts CRDT convergence and exactly-once survival of the peer edit.

* perf(tests): drive retry backoff and scheduled flushes with fake timers

Real sleeps and repeated setup were the run-phase cost in seven files:

- tools/index.test.ts: `retryDelayMs: 0` falls through to the 500ms default, so
  every retry test slept for real. Fake the clock for the retry describe.
- embeddings/client.test.ts: two retry tests waited the 1s initial backoff.
- hosted-key-rate-limiter.test.ts: the queue-head test slept two 200ms polls;
  the abort test slept 20ms wall clock and asserted on Date.now.
- event-buffer.test.ts: sleep(60)/sleep(5) waited on the 15ms flush timer.
- events.attribution.test.ts: walked and read the whole app twice, serially.
  Walk once in a beforeAll with batched concurrent reads.
- sso-trust.test.ts: imported the Better Auth graph in each of three tests.
  Import once in a beforeAll that carries the 30s budget.
- markdown-parse.test.ts: the one-shot helper built and destroyed a tiptap
  Editor per document; reuse one across the 400-seed property corpus.

No production timing constants, corpus sizes, or assertions changed.

* perf(tests): drop the executable tool registry and full block map from non-sweep tests

The vitest suite is bounded by module loading, and the biggest single cost is a
test file opting out of the global `@/tools/registry` or `@/blocks/registry`
mock. Trim that where the test never needed the whole thing:

- `lib/internal/tool-operations/registry.server.test.ts` checked registration
  against `getToolIds()` but still imported the executable registry for one
  sweep. That sweep — every operation-backed tool has no `request` and a
  registered handler — is the in-process half of the transport partition, so
  it moves next to the external-URL sweep in `tools/request-transport.test.ts`,
  which already pays for the registry. Handler loads now run in parallel.
  Collect 12.5s -> 0.1s, run 6s -> 4.3s; request-transport unchanged (~8s).
- New `partialBlockRegistry` in `@sim/testing/mocks/block-registry.mock`
  builds the `@/blocks/registry-maps` shape from the block modules a test
  actually reads, so the real registry code runs over those blocks alone.
  Applied to the six non-sweep tests that unmocked the block registry for one
  or two blocks (webhook, slack, whatsapp, condition/function,
  knowledge/start_trigger): collect 3-8s -> 0.05-2s each. Registry sweeps
  keep the real map.
- `tools/index.test.ts` retry cases set `retryDelayMs: 0`, which the retry
  config replaces with the 500ms default; `1` is the smallest delay honored.
  Run 4.0s -> 0.45s.
- `tools/metadata.test.ts` collects sweep offenders into one assertion instead
  of one `expect` per registry entry, and indexes ids once. Run 1.1s -> 0.03s.
- The `vi.unmock('@/tools/registry')` in the search-replace tests was dead:
  nothing in their graph imports the registry any more, and all 106 tests pass
  without it, contrary to the comment that claimed eight would fail.

* perf(tests): stop copilot handler tests from loading the executor, block, and trigger registries

The router test stubs every handler implementation module so the handler map
binds without loading the executor and block registry; the 90s timeout goes.
Handler, VFS, payload, and request tests mock the execution-side leaves and
catalog registries they never exercise. Source edges that dragged in barrels
for a single symbol now import the defining module: the Mothership block icon
from @sim/emcn/icons, contract schemas from their contract file, and autolayout
constants from new @sim/workflow-renderer/dimensions and /note-content subpaths.

lib/copilot: 141 files, per-file collect+run 74.2s -> 37.4s, wall 19.1s -> 15.0s
router.test.ts: 15.1s -> 0.45s

* fix(tests): import ContractBodyInput from the contract types module

* fix(tests): bind the real cancellation handler in the router test; mock the narrowed contract path

* fix(emcn): wait for focus restoration instead of a fixed 20ms sleep in the modal test

Author: waleedlatif1

.github/workflows/test-build.yml

@@ -182,17 +182,17 @@ jobs:
       # without `--coverage`. See the Codecov note below.
       #
       # apps/sim runs only its first shard here; `test-shard` below runs the
-      # other. That suite is bound by the single Vite server thread that feeds
+      # others. That suite is bound by the single Vite server thread that feeds
       # every worker — wall time is flat from 4 to 13 workers — so a bigger
-      # runner buys nothing and a second runner halves it.
+      # runner buys nothing and each extra runner takes a proportional slice.
       - name: Run tests
         env:
           NODE_OPTIONS: '--no-warnings --max-old-space-size=8192'
           NEXT_PUBLIC_APP_URL: 'https://www.sim.ai'
           DATABASE_URL: 'postgresql://postgres:postgres@localhost:5432/simstudio'
           ENCRYPTION_KEY: '0000000000000000000000000000000000000000000000000000000000000000' # dummy key for CI only
           TURBO_CACHE_DIR: .turbo
-          SIM_TEST_SHARD: 1/2
+          SIM_TEST_SHARD: 1/3
         run: bun run test
 
       - name: Check schema and migrations are in sync
@@ -208,15 +208,20 @@ jobs:
           fi
           echo "✅ Schema and migrations are in sync"
 
-  # The second half of apps/sim's Vitest suite. Everything else — lint, the
-  # audits, type-check, the other workspaces' suites — lives in `test-build`
-  # with shard 1; this job exists only because that suite cannot go faster on
-  # one machine (see the "Run tests" note there). The Turbo cache disk gets
-  # its own key so the two shards' entries do not evict each other.
+  # The remaining shards of apps/sim's Vitest suite. Everything else — lint,
+  # the audits, type-check, the other workspaces' suites — lives in
+  # `test-build` with shard 1; these jobs exist only because that suite cannot
+  # go faster on one machine (see the "Run tests" note there). Three shards
+  # put each runner at roughly the fixed cost of checkout + install. The Turbo
+  # cache disk gets its own key so the shards' entries do not evict each other.
   test-shard:
-    name: Test (shard 2)
+    name: Test (shard ${{ matrix.shard }})
     runs-on: ${{ (vars.CI_PROVIDER == '' || vars.CI_PROVIDER == 'blacksmith') && 'blacksmith-8vcpu-ubuntu-2404' || 'ubuntu-latest' }}
     timeout-minutes: 15
+    strategy:
+      fail-fast: false
+      matrix:
+        shard: [2, 3]
 
     steps:
       - name: Checkout code
@@ -250,7 +255,7 @@ jobs:
         uses: ./.github/actions/cache-mount
         with:
           provider: ${{ vars.CI_PROVIDER }}
-          key: ${{ github.repository }}-turbo-cache-shard-${{ github.event_name }}${{ github.event.pull_request.head.repo.fork && '-fork' || '' }}
+          key: ${{ github.repository }}-turbo-cache-shard-${{ matrix.shard }}-${{ github.event_name }}${{ github.event.pull_request.head.repo.fork && '-fork' || '' }}
           path: ./.turbo
 
       - name: Install dependencies
@@ -259,14 +264,14 @@ jobs:
       - name: Install ripgrep
         run: command -v rg || (sudo apt-get update && sudo apt-get install -y ripgrep)
 
-      - name: Run tests (apps/sim shard 2/2)
+      - name: Run tests (apps/sim shard ${{ matrix.shard }}/3)
         env:
           NODE_OPTIONS: '--no-warnings --max-old-space-size=8192'
           NEXT_PUBLIC_APP_URL: 'https://www.sim.ai'
           DATABASE_URL: 'postgresql://postgres:postgres@localhost:5432/simstudio'
           ENCRYPTION_KEY: '0000000000000000000000000000000000000000000000000000000000000000' # dummy key for CI only
           TURBO_CACHE_DIR: .turbo
-          SIM_TEST_SHARD: 2/2
+          SIM_TEST_SHARD: ${{ matrix.shard }}/3
         run: bunx turbo run test --filter=@sim/app
 
   # Next.js production build, in parallel with lint + tests. Sticky disks are

apps/docs/package.json

@@ -42,7 +42,6 @@
   "devDependencies": {
     "@sim/tsconfig": "workspace:*",
     "@tailwindcss/postcss": "^4.0.12",
-    "@types/mdx": "^2.0.13",
     "@types/node": "24.2.1",
     "@types/react": "^19.1.2",
     "@types/react-dom": "^19.0.4",

apps/sim/app/workspace/[workspaceId]/files/components/file-viewer/rich-markdown-editor/code-languages.test.ts

@@ -1,5 +1,5 @@
 /**
- * @vitest-environment jsdom
+ * @vitest-environment node
  *
  * Guards against drift between the code-block language picker and the Prism grammars actually
  * registered by CodeBlockHighlight: every selectable language must have a registered grammar, or it

apps/sim/app/workspace/[workspaceId]/files/components/file-viewer/rich-markdown-editor/collaboration/apply-streamed-markdown.test.ts

@@ -157,41 +157,6 @@ describe('agent-stream applier', () => {
     expect(freshText).toContain('Gamma paragraph')
   })
 
-  it('preserves a concurrent peer edit to a region the agent snapshot does not include', () => {
-    // This is the core "AI as a CRDT peer" guarantee: the agent relays only its OWN delta (computed
-    // against a private shadow), never a whole-document reconcile that would revert a peer's edit.
-    const { editor, doc } = track(makeCollabEditor())
-
-    const session = beginAgentStream(editor)!
-    applyAgentStreamFrame(editor, session, 'Alpha paragraph.\n\nBeta paragraph.')
-
-    // A peer edits the FIRST paragraph directly on the shared doc — the agent's later snapshot still
-    // carries the ORIGINAL first paragraph (it was built from the base, before this edit).
-    const peer = new Y.Doc()
-    Y.applyUpdate(peer, Y.encodeStateAsUpdate(doc))
-    const peerFrag = peer.getXmlFragment('default')
-    peer.transact(() => {
-      const firstPara = peerFrag.get(0) as Y.XmlElement
-      const textNode = firstPara.get(0) as Y.XmlText
-      textNode.insert(textNode.toString().length, ' EDITED')
-    })
-    Y.applyUpdate(doc, Y.encodeStateAsUpdate(peer, Y.encodeStateVector(doc)))
-    peer.destroy()
-
-    // The agent appends a third paragraph. Its snapshot's first paragraph is the stale original, but the
-    // shadow-relayed delta only inserts the new paragraph — so the peer's " EDITED" must survive.
-    applyAgentStreamFrame(
-      editor,
-      session,
-      'Alpha paragraph.\n\nBeta paragraph.\n\nGamma paragraph.'
-    )
-    endAgentStream(session)
-
-    const live = doc.getXmlFragment('default').toString()
-    expect(live).toContain('EDITED')
-    expect(live).toContain('Gamma paragraph')
-  })
-
   it('reuses cached binding metadata across frames, still emitting minimal per-frame deltas', () => {
     // The binding `meta` is built ONCE (first frame) and reused — `updateYFragment` maintains it in place,
     // so we skip an O(doc) `initProseMirrorDoc` rebuild per frame. This guards that caching preserves the

apps/sim/app/workspace/[workspaceId]/files/components/file-viewer/rich-markdown-editor/image-drag-move.test.ts

@@ -1,5 +1,5 @@
 /**
- * @vitest-environment jsdom
+ * @vitest-environment node
  *
  * Dragging an image to reposition it inside a document must MOVE it, not import it again.
  *

apps/sim/app/workspace/[workspaceId]/files/components/file-viewer/rich-markdown-editor/image.test.ts

@@ -1,5 +1,5 @@
 /**
- * @vitest-environment jsdom
+ * @vitest-environment node
  */
 import { describe, expect, it } from 'vitest'
 import {

apps/sim/app/workspace/[workspaceId]/files/components/file-viewer/rich-markdown-editor/markdown-parse.test.ts

@@ -2,7 +2,7 @@
  * @vitest-environment jsdom
  */
 import { Editor } from '@tiptap/core'
-import { afterEach, describe, expect, it } from 'vitest'
+import { afterAll, describe, expect, it } from 'vitest'
 import { createMarkdownContentExtensions } from './extensions'
 import { parseMarkdownToDoc, serializeMarkdownBody, splitMarkdownBlocks } from './markdown-parse'
 import { isRoundTripSafe } from './round-trip-safety'
@@ -12,19 +12,20 @@ const isEmptyPara = (n: { type?: string; content?: unknown[] }): boolean =>
   n.type === 'paragraph' && !n.content?.length
 
 let editor: Editor | null = null
-afterEach(() => {
+afterAll(() => {
   editor?.destroy()
   editor = null
 })
 
-/** The current whole-document path: parse markdown in one shot, serialize back. */
+/**
+ * The current whole-document path: parse markdown in one shot, serialize back. One editor serves
+ * every call — `setContent` replaces the document wholesale, so a fresh instance per call only adds
+ * the cost of building the view, which the property tests below paid hundreds of times over.
+ */
 function oneShot(body: string): string {
-  editor = new Editor({ extensions: createMarkdownContentExtensions() })
+  editor ??= new Editor({ extensions: createMarkdownContentExtensions() })
   editor.commands.setContent(body, { contentType: 'markdown' })
-  const out = editor.getMarkdown()
-  editor.destroy()
-  editor = null
-  return out
+  return editor.getMarkdown()
 }
 
 /**

apps/sim/app/workspace/[workspaceId]/home/components/mothership-view/components/resource-content/components/terminal-session/terminal-session.test.ts

@@ -1,5 +1,5 @@
 /**
- * @vitest-environment jsdom
+ * @vitest-environment node
  */
 
 import { resolveDesktopZoom } from '@sim/desktop-bridge'

apps/sim/app/workspace/[workspaceId]/logs/utils.test.ts

@@ -1,5 +1,5 @@
 /**
- * @vitest-environment jsdom
+ * @vitest-environment node
  */
 import { describe, expect, it } from 'vitest'
 import { resolveLogWorkflowId, workflowEditorPath } from './utils'

apps/sim/app/workspace/[workspaceId]/w/[workflowId]/utils/workflow-attachment-upload.test.ts

@@ -1,5 +1,5 @@
 /**
- * @vitest-environment jsdom
+ * @vitest-environment node
  */
 import { beforeEach, describe, expect, it, vi } from 'vitest'