chore: test fixes (#2503)

* chore: fix tests

* chore: type fix

* chore: fix wrapper regressino

Author: harbournick

apps/cli/src/__tests__/command-examples.test.ts

@@ -36,6 +36,7 @@ const WRAPPER_EXTRA_SPECS: Partial<Record<string, OptionSpec[]>> = {
     { name: 'pattern', type: 'string' },
     { name: 'mode', type: 'string' },
     { name: 'case-sensitive', type: 'boolean' },
+    { name: 'select-json', type: 'string' },
     { name: 'query-json', type: 'string' },
     { name: 'query-file', type: 'string' },
     { name: 'within-json', type: 'string' },

apps/cli/src/lib/operation-wrapper-input.ts

@@ -85,6 +85,7 @@ export async function parseWrapperOperationInput(
         { name: 'pattern', type: 'string' },
         { name: 'mode', type: 'string' },
         { name: 'case-sensitive', type: 'boolean' },
+        { name: 'select-json', type: 'string' },
         { name: 'query-json', type: 'string' },
         { name: 'query-file', type: 'string' },
         { name: 'within-json', type: 'string' },

packages/document-api/package.json

@@ -17,7 +17,7 @@
     }
   },
   "scripts": {
-    "test": "vitest run"
+    "test": "bun test"
   },
   "files": [
     "src"

packages/layout-engine/layout-bridge/tsconfig.types.json

@@ -2,7 +2,9 @@
   "extends": "./tsconfig.json",
   "compilerOptions": {
     "composite": true,
-    "outDir": "dist-types"
+    "emitDeclarationOnly": true,
+    "outDir": "dist",
+    "tsBuildInfoFile": "./dist/tsconfig.types.tsbuildinfo"
   },
   "references": [
     { "path": "../contracts/tsconfig.json" },

packages/super-editor/src/core/commands/insertContent.test.js

@@ -196,7 +196,7 @@ describe('insertContent (integration) list export', () => {
     helpers = await import('../../tests/helpers/helpers.js');
     cachedDocxData = await helpers.loadTestDataForEditorTests('blank-doc.docx');
     exportHelpers = await import('../../tests/export/export-helpers/index.js');
-  });
+  }, 30000);
 
   const setupEditor = () => {
     const { docx, media, mediaFiles, fonts } = cachedDocxData;
@@ -368,7 +368,7 @@ describe.skipIf(!process.env.CI)('insertContent (integration) horizontal rule',
     vi.doUnmock('../helpers/contentProcessor.js');
     helpers = await import('../../tests/helpers/helpers.js');
     cachedDocxData = await helpers.loadTestDataForEditorTests('blank-doc.docx');
-  });
+  }, 30000);
 
   const setupEditor = () => {
     const { docx, media, mediaFiles, fonts } = cachedDocxData;

packages/super-editor/src/document-api-adapters/info-adapter.ts

@@ -62,8 +62,9 @@ function extractTextFormatting(node: import('prosemirror-model').Node): { fontFa
   // to get consistent "first run" formatting rather than mixed properties.
   node.descendants((child) => {
     if (fontFamily !== undefined || fontSize !== undefined) return false;
-    if (!child.isText || child.marks.length === 0) return;
-    for (const mark of child.marks) {
+    const marks = child.marks ?? [];
+    if (!child.isText || marks.length === 0) return;
+    for (const mark of marks) {
       const attrs = mark.attrs as Record<string, unknown>;
       if (typeof attrs.fontFamily === 'string' && attrs.fontFamily) {
         fontFamily = attrs.fontFamily;
@@ -85,6 +86,14 @@ function extractTextFormatting(node: import('prosemirror-model').Node): { fontFa
  */
 function collectDocumentStyles(editor: Editor): { styles: DocumentStyles; defaults: DocumentDefaults } {
   const headingPattern = HEADING_STYLE_PATTERN;
+  const doc = editor.state?.doc;
+
+  if (!doc?.descendants) {
+    return {
+      styles: { paragraphStyles: [] },
+      defaults: { styleId: 'Normal' },
+    };
+  }
 
   // Per-style data
   const styleData = new Map<string, { count: number; fontFamily?: string; fontSize?: number }>();
@@ -93,7 +102,7 @@ function collectDocumentStyles(editor: Editor): { styles: DocumentStyles; defaul
   const fontCounts = new Map<string, number>();
   const sizeCounts = new Map<number, number>();
 
-  editor.state.doc.descendants((node) => {
+  doc.descendants((node) => {
     if (node.type.name !== 'paragraph') return;
 
     const props = node.attrs.paragraphProperties as { styleId?: string } | undefined;

packages/super-editor/src/document-api-adapters/plan-engine/blocks-wrappers.ts

@@ -86,8 +86,9 @@ function extractBlockFormatting(node: ProseMirrorNode): {
 
   node.descendants((child) => {
     if (fontFamily !== undefined) return false;
-    if (!child.isText || child.marks.length === 0) return;
-    for (const mark of child.marks) {
+    const marks = child.marks ?? [];
+    if (!child.isText || marks.length === 0) return;
+    for (const mark of marks) {
       const attrs = mark.attrs as Record<string, unknown>;
       if (typeof attrs.fontFamily === 'string' && attrs.fontFamily) fontFamily = attrs.fontFamily;
       if (attrs.fontSize != null) {

packages/super-editor/src/document-api-adapters/plan-engine/query-match-adapter.ts

@@ -513,9 +513,13 @@ export function queryMatchAdapter(editor: Editor, input: QueryMatchInput): Query
       // Include a short document text preview so the model can see the actual
       // text formatting (quote style, whitespace, etc.) and retry with the
       // correct pattern instead of guessing blindly.
-      const docSize = editor.state.doc.content.size;
+      const doc = storyEditor.state?.doc;
+      const docSize = typeof doc?.content?.size === 'number' ? doc.content.size : 0;
       const previewLength = Math.min(docSize, 300);
-      const textPreview = previewLength > 0 ? editor.state.doc.textBetween(0, previewLength, '\n', '\n') : '';
+      const textPreview =
+        previewLength > 0 && typeof doc?.textBetween === 'function'
+          ? doc.textBetween(0, previewLength, '\n', '\n')
+          : '';
       throw planError(
         'MATCH_NOT_FOUND',
         `selector matched zero ranges. Document starts with: "${textPreview.slice(0, 200)}..."`,