Fixed tables in partialBlockToBlockForTesting

Author: matthewlipski

tests/src/unit/core/formatConversion/formatConversionTestUtil.ts

@@ -9,6 +9,7 @@ import {
   PartialBlock,
   PartialInlineContent,
   PartialTableCell,
+  PartialTableContent,
   StyledText,
   StyleSchema,
   TableCell,
@@ -153,10 +154,10 @@ export function partialBlockToBlockForTesting<
   );
 
   if (contentType === "inline") {
-    const content = withDefaults.content as InlineContent<I, S>[] | undefined;
+    const content = withDefaults.content as PartialInlineContent<I, S>;
     withDefaults.content = partialContentToInlineContent(content) as any;
   } else if (contentType === "table") {
-    const content = withDefaults.content as TableContent<I, S> | undefined;
+    const content = withDefaults.content as PartialTableContent<I, S>;
     withDefaults.content = {
       type: "tableContent",
       columnWidths:
@@ -167,7 +168,34 @@ export function partialBlockToBlockForTesting<
       headerCols: content?.headerCols || undefined,
       rows:
         content?.rows.map((row) => ({
-          cells: row.cells.map((cell) => partialContentToInlineContent(cell)),
+          cells: row.cells.map((cell) =>
+            typeof cell === "object" &&
+            "type" in cell &&
+            cell.type === "tableCell"
+              ? {
+                  type: "tableCell",
+                  props: {
+                    backgroundColor: "default",
+                    colspan: 1,
+                    rowspan: 1,
+                    textAlignment: "left",
+                    textColor: "default",
+                    ...cell.props,
+                  },
+                  content: partialContentToInlineContent(cell.content),
+                }
+              : {
+                  type: "tableCell",
+                  props: {
+                    backgroundColor: "default",
+                    colspan: 1,
+                    rowspan: 1,
+                    textAlignment: "left",
+                    textColor: "default",
+                  },
+                  content: partialContentToInlineContent(cell),
+                }
+          ),
         })) || [],
     } as any;
   }