test(super-editor): cover section titlePg decoration provider

Author: luccas-harbour

packages/super-editor/src/editors/v1/core/presentation-editor/tests/HeaderFooterSessionManager.test.ts

@@ -587,9 +587,16 @@ describe('HeaderFooterSessionManager', () => {
   });
 
   describe('createDecorationProvider — resolved items', () => {
-    function buildHeaderResult(options?: { y?: number; minY?: number; blockId?: string }): HeaderFooterLayoutResult {
+    function buildHeaderResult(options?: {
+      y?: number;
+      minY?: number;
+      blockId?: string;
+      pageNumber?: number;
+      type?: HeaderFooterLayoutResult['type'];
+    }): HeaderFooterLayoutResult {
       const y = options?.y ?? 10;
       const blockId = options?.blockId ?? 'p1';
+      const pageNumber = options?.pageNumber ?? 1;
       const paraFragment: ParaFragment = {
         kind: 'para',
         blockId,
@@ -602,7 +609,7 @@ describe('HeaderFooterSessionManager', () => {
       const layout: HeaderFooterLayout = {
         height: 50,
         ...(options?.minY != null ? { minY: options.minY } : {}),
-        pages: [{ number: 1, fragments: [paraFragment] }],
+        pages: [{ number: pageNumber, fragments: [paraFragment] }],
       };
       const blocks: FlowBlock[] = [{ kind: 'paragraph', id: blockId, runs: [] }];
       const measures: Measure[] = [
@@ -612,7 +619,7 @@ describe('HeaderFooterSessionManager', () => {
           totalHeight: 18,
         },
       ];
-      return { kind: 'header', type: 'default', layout, blocks, measures };
+      return { kind: 'header', type: options?.type ?? 'default', layout, blocks, measures };
     }
 
     it('delivers items aligned 1:1 with fragments when variant layout is used', () => {
@@ -772,6 +779,88 @@ describe('HeaderFooterSessionManager', () => {
       expect(payload!.items![0]).toMatchObject({ blockId: 'p1', x: 72, y: 0 });
     });
 
+    it('uses section titlePg state when selecting decoration-provider variants', () => {
+      const deps: SessionManagerDependencies = {
+        getLayoutOptions: vi.fn(() => ({})),
+        getPageElement: vi.fn(() => null),
+        scrollPageIntoView: vi.fn(),
+        waitForPageMount: vi.fn(async () => true),
+        convertPageLocalToOverlayCoords: vi.fn(() => ({ x: 0, y: 0 })),
+        isViewLocked: vi.fn(() => false),
+        getBodyPageHeight: vi.fn(() => 800),
+        notifyInputBridgeTargetChanged: vi.fn(),
+        scheduleRerender: vi.fn(),
+        setPendingDocChange: vi.fn(),
+        getBodyPageCount: vi.fn(() => 2),
+      };
+
+      manager = new HeaderFooterSessionManager({
+        painterHost,
+        visibleHost,
+        selectionOverlay,
+        editor: {
+          ...createMainEditorStub(),
+          converter: {
+            headerIds: { titlePg: true },
+          },
+        } as unknown as Editor,
+        defaultPageSize: { w: 612, h: 792 },
+        defaultMargins: { top: 72, right: 72, bottom: 72, left: 72, header: 36, footer: 36 },
+      });
+      manager.setDependencies(deps);
+      manager.setMultiSectionIdentifier(
+        buildMultiSectionIdentifier([
+          {
+            sectionIndex: 0,
+            titlePg: true,
+            headerRefs: { first: 'rId-section0-first', default: 'rId-section0-default' },
+          },
+          {
+            sectionIndex: 1,
+            titlePg: false,
+            headerRefs: { first: 'rId-section1-first', default: 'rId-section1-default' },
+          },
+        ]),
+      );
+      manager.setLayoutResults(
+        [
+          buildHeaderResult({ type: 'first', blockId: 'first-block', pageNumber: 2 }),
+          buildHeaderResult({ type: 'default', blockId: 'default-block', pageNumber: 2 }),
+        ],
+        null,
+      );
+
+      const layout: Layout = {
+        version: 1,
+        flowMode: 'paginated',
+        pageGap: 0,
+        pageSize: { w: 612, h: 792 },
+        pages: [
+          {
+            number: 1,
+            sectionIndex: 0,
+            margins: { top: 72, right: 72, bottom: 72, left: 72, header: 36, footer: 36 },
+          },
+          {
+            number: 2,
+            sectionIndex: 1,
+            margins: { top: 72, right: 72, bottom: 72, left: 72, header: 36, footer: 36 },
+            sectionRefs: {
+              headerRefs: { first: 'rId-section1-first', default: 'rId-section1-default' },
+              footerRefs: {},
+            },
+          },
+        ] as never,
+      } as unknown as Layout;
+
+      const provider = manager.createDecorationProvider('header', layout as unknown as ResolvedLayout);
+      const payload = provider!(2, layout.pages[1]!.margins, layout.pages[1] as unknown as ResolvedPage);
+
+      expect(payload).not.toBeNull();
+      expect(payload!.sectionType).toBe('default');
+      expect(payload!.items![0]!.blockId).toBe('default-block');
+    });
+
     it('normalizes resolved items when per-rId layout minY is negative', async () => {
       mockLayoutPerRIdHeaderFooters.mockImplementation(
         async (