@semiont/react-ui 0.5.27 → 0.5.29

package/src/components/modals/__tests__/ReferenceWizardModal.test.tsx

@@ -19,7 +19,7 @@
  * settings.
  */
 import { describe, it, expect, vi, beforeEach, beforeAll } from 'vitest';
-import { screen } from '@testing-library/react';
+import { screen, fireEvent, waitFor } from '@testing-library/react';
 import userEvent from '@testing-library/user-event';
 import '@testing-library/jest-dom';
 import type { GatheredContext } from '@semiont/core';
@@ -37,7 +37,7 @@ const CONTEXT = {
   focus: {
     kind: 'annotation',
     annotation: { id: 'ann-1' },
-    sourceResource: { id: 'res-1', name: 'Scythian steppe' },
+    sourceResource: { '@id': 'res-1', name: 'Scythian steppe' },
     selected: { text: 'Caspian Sea' },
   },
   graph: { nodes: [], edges: [] },
@@ -45,24 +45,45 @@ const CONTEXT = {
 } as unknown as GatheredContext;
 
 const T = {
-  gatherTitle: 'Gathered Context', configureGenerationTitle: 'Configure Generation',
-  configureSearchTitle: 'Configure Search', searchResultsTitle: 'Search Results',
+  resolveTitle: 'Resolve Reference',
   sourceContextLabel: 'Source', connectionsLabel: 'Connections', citedByLabel: 'Cited by',
-  userHintLabel: 'Hint', userHintPlaceholder: 'Describe what this refers to…',
+  userHintLabel: 'Hint', userHintEffect: 'steers Search and Generate',
+  userHintPlaceholder: 'Describe what this refers to…',
+  graphPaneTitle: 'In the graph', graphEmpty: 'No links yet.', resourceLinkLabel: 'Resource link',
+  corpusPaneTitle: 'In the corpus', corpusEmpty: 'Nothing similar.',
+  excludedReceipt: '{{types}} excluded', machineRead: 'OCR',
   loadingContext: 'Loading…', failedContext: 'Failed',
-  cancel: 'Cancel', search: 'Search', searching: 'Searching…', generate: 'Generate',
+  search: 'Search', searching: 'Searching…', generate: 'Generate',
   compose: 'Compose', resolutionStrategyLabel: 'Resolution Strategy', back: 'Back',
   link: 'Link', score: 'Score', noResults: 'No results',
-  resourceTitle: 'Resource Title', resourceTitlePlaceholder: 'Title…',
+  resourceTitle: 'Resource Title', resourceTitlePlaceholder: 'Title…', saveLocation: 'Save location',
   additionalInstructions: 'Additional Instructions', additionalInstructionsPlaceholder: '…',
   language: 'Language', languageHelp: '', creativity: 'Creativity',
   creativityFocused: 'Focused', creativityCreative: 'Creative',
   maxLength: 'Max length', maxLengthHelp: '', maxLengthCeiling: 'Limited to {{maxOutputTokens}} by {{model}}.',
+  outputFormat: 'Format',
+  formatExtensionMismatch: 'Save location must end in {{extension}} to match the selected format.',
   maxResults: 'Max Results', semanticScoring: 'Semantic Scoring', semanticScoringHelp: '',
+  searchFailed: 'Search failed',
+  contentLabel: 'Content', entityTypes: 'Entity types',
+  createAndLink: 'Create & Link', creatingAndLinking: 'Creating…',
+  discardDraftPrompt: 'Discard this draft?', discardDraft: 'Discard', keepEditing: 'Keep editing',
 };
 
+// ComposeStep embeds the editor; same mock the page's own tests use.
+vi.mock('../../CodeMirrorRenderer', () => ({
+  CodeMirrorRenderer: ({ content, onChange, editable }: any) => (
+    <textarea
+      data-testid="code-editor"
+      value={content}
+      disabled={!editable}
+      onChange={(e) => onChange?.(e.target.value)}
+    />
+  ),
+}));
+
 function renderWizard(over: Partial<React.ComponentProps<typeof ReferenceWizardModal>> = {}) {
-  const onComposeNavigate = vi.fn();
+  const onComposeSubmit = vi.fn<(referenceId: string, params: unknown) => Promise<void>>().mockResolvedValue(undefined);
   const onGenerateSubmit = vi.fn();
   const onLinkResource = vi.fn();
   const onClose = vi.fn();
@@ -75,13 +96,14 @@ function renderWizard(over: Partial<React.ComponentProps<typeof ReferenceWizardM
       resourceId="res-1"
       defaultTitle="Caspian Sea"
       entityTypes={['Location']}
+      entityTypeOptions={['Person', 'Topic', 'Location']}
       locale="en"
       context={CONTEXT}
       contextLoading={false}
       contextError={null}
       onGenerateSubmit={onGenerateSubmit}
       onLinkResource={onLinkResource}
-      onComposeNavigate={onComposeNavigate}
+      onComposeSubmit={onComposeSubmit}
       translations={T}
       {...over}
     />,
@@ -92,7 +114,7 @@ function renderWizard(over: Partial<React.ComponentProps<typeof ReferenceWizardM
   const client = utils.session!.client;
   const searchSpy = vi.spyOn(client.match, 'requestSearch');
 
-  return { ...utils, client, searchSpy, onComposeNavigate, onGenerateSubmit, onLinkResource, onClose };
+  return { ...utils, client, searchSpy, onComposeSubmit, onGenerateSubmit, onLinkResource, onClose };
 }
 
 /** Type the hint into the gather step's textarea. */
@@ -103,17 +125,32 @@ async function typeHint(text: string) {
 describe('ReferenceWizardModal — the Hint reaches every strategy, at focus.userHint', () => {
   beforeEach(() => { vi.clearAllMocks(); });
 
-  it('compose carries the hint inside focus, not at the top level', async () => {
-    const { onComposeNavigate } = renderWizard();
-    await typeHint('the lake, not the myth');
-    await userEvent.click(screen.getByText(new RegExp(T.compose)));
+  it('the gather panel carries the widened class (GEP P2, D2)', () => {
+    const { baseElement } = renderWizard();
+    const panel = baseElement.querySelector('.semiont-search-modal__panel--gather');
+    expect(panel).not.toBeNull();
+    expect(panel!.className).toContain('semiont-search-modal__panel--wide');
+  });
 
-    expect(onComposeNavigate).toHaveBeenCalledTimes(1);
-    const context = onComposeNavigate.mock.calls[0]![0] as Record<string, unknown>;
-    expect((context.focus as Record<string, unknown>).userHint).toBe('the lake, not the myth');
-    // The shape the old code produced. `GatheredContext` has no top-level
-    // `userHint`, so anything reading the schema would never look here.
-    expect(context).not.toHaveProperty('userHint');
+  it('the typed hint stays visible after stepping into configure-search (GEP P1c, D8)', async () => {
+    // The thing being steered must not vanish while you steer it.
+    const { baseElement } = renderWizard();
+    await typeHint('the ancient city');
+    await userEvent.click(screen.getByRole('button', { name: /Search…/ }));
+    const echo = baseElement.querySelector('.semiont-wizard__hint-echo');
+    expect(echo).not.toBeNull();
+    expect(echo!.textContent).toContain('the ancient city');
+  });
+
+  it('compose is a step now — choosing it navigates nowhere (COMPOSE-IN-MODAL D1)', async () => {
+    // The old flow stashed the context in sessionStorage and navigated to the
+    // compose page; both are gone. Choosing ✍️ Compose unfolds the compose
+    // form below the evidence like the other strategies.
+    renderWizard();
+    await userEvent.click(screen.getByText(new RegExp(T.compose)));
+    expect(screen.getByRole('button', { name: T.createAndLink })).toBeInTheDocument(); // the compose form is open
+    expect(screen.getByText(T.resolveTitle)).toBeInTheDocument(); // ONE title, every step
+    expect(screen.getByLabelText(T.resourceTitle)).toHaveValue('Caspian Sea'); // seeded draft
   });
 
   it('search sends the same enriched context over the wire', async () => {
@@ -131,20 +168,311 @@ describe('ReferenceWizardModal — the Hint reaches every strategy, at focus.use
   it('generation gets it too — the path that used to drop it silently', async () => {
     renderWizard();
     await typeHint('focus on hydrology');
-    await userEvent.click(screen.getByText(new RegExp(T.generate)));
+    await userEvent.click(screen.getByRole('button', { name: `✨ ${T.generate}…` }));
 
     // The step renders the context it was handed; the hint is visible in the
     // gathered-context panel it shows, which is only true if the ENRICHED
     // object was passed down rather than the raw prop.
-    expect(screen.getByText(T.configureGenerationTitle)).toBeInTheDocument();
+    expect(screen.getByLabelText(T.resourceTitle)).toBeInTheDocument();
+  });
+
+});
+
+describe('ReferenceWizardModal — the context stays in view on the strategy steps', () => {
+  beforeEach(() => { vi.clearAllMocks(); });
+
+  // Choosing Generate or Search unfolds that strategy's form BELOW the
+  // evidence — it must not navigate away from it. The gather step stays the
+  // decision surface (chooser footer, hint textarea); the strategy steps keep
+  // the RESOLUTION STRATEGY band for continuity, collapsed from three choices
+  // to the chosen one — a passive echo, not a second chooser: the OTHER
+  // strategies' buttons are gone, and there is no second hint textarea.
+  it('configure-search renders the evidence above the form, the strategy band collapsed to Search', async () => {
+    const { baseElement } = renderWizard();
+    await userEvent.click(screen.getByText(new RegExp(`^🔍? ?${T.search}`)));
+
+    expect(baseElement.querySelector('.semiont-gather__source-box')).not.toBeNull(); // quotation strip
+    expect(screen.getByText(T.graphPaneTitle)).toBeInTheDocument();                  // graph pane
+    expect(screen.getByText(T.maxResults)).toBeInTheDocument();                      // the form, below
+    // The band survives the step: header + the chosen strategy, nothing clickable.
+    const footer = baseElement.querySelector('.semiont-gather__footer');
+    expect(footer).not.toBeNull();
+    expect(footer!.textContent).toContain(T.resolutionStrategyLabel);
+    expect(footer!.textContent).toContain(`🔍 ${T.search}`);
+    expect(footer!.querySelectorAll('button')).toHaveLength(0);
+    expect(screen.queryByRole('button', { name: `✨ ${T.generate}…` })).not.toBeInTheDocument();
+    expect(screen.queryByRole('button', { name: `✍️ ${T.compose}` })).not.toBeInTheDocument();
+    expect(screen.queryByPlaceholderText(T.userHintPlaceholder)).not.toBeInTheDocument();
+    const panel = baseElement.querySelector('.semiont-search-modal__panel')!;
+    expect(panel.className).toContain('semiont-search-modal__panel--wide');          // evidence needs the width
+  });
+
+  it('search-results keeps the evidence and the collapsed band above the results', async () => {
+    // Same grammar as the configure steps — the results replace the FORM
+    // region, not the evidence. The step itself is pure results now (D10
+    // amended); the host stacks the full display above it.
+    const { client, baseElement } = renderWizard();
+    await userEvent.click(screen.getByText(new RegExp(`^🔍? ?${T.search}`)));
+    await userEvent.click(screen.getByRole('button', { name: T.search }));
+    client.bus.get('match:search-results').next({
+      referenceId: 'ann-1',
+      response: [{ '@id': 'res-9', name: 'Caspian Sea', score: 54.2 }],
+    } as never);
+    await screen.findByRole('button', { name: new RegExp(T.link) }); // results are up
+
+    expect(baseElement.querySelector('.semiont-gather__source-box')).not.toBeNull(); // quotation strip
+    expect(screen.getByText(T.graphPaneTitle)).toBeInTheDocument();                  // graph pane
+    expect(screen.getByText(T.corpusPaneTitle)).toBeInTheDocument();                 // full display, not the redux
+    const footer = baseElement.querySelector('.semiont-gather__footer');
+    expect(footer).not.toBeNull();
+    expect(footer!.textContent).toContain(`🔍 ${T.search}`);
+    expect(footer!.querySelectorAll('button')).toHaveLength(0);
+    expect(screen.getByRole('button', { name: `🔗 ${T.link}` })).toBeInTheDocument(); // the results
+    const panel = baseElement.querySelector('.semiont-search-modal__panel')!;
+    expect(panel.className).toContain('semiont-search-modal__panel--gather');
+  });
+
+  it('configure-generation renders the evidence above the form, the strategy band collapsed to Generate', async () => {
+    const { baseElement } = renderWizard();
+    await userEvent.click(screen.getByRole('button', { name: `✨ ${T.generate}…` }));
+
+    expect(baseElement.querySelector('.semiont-gather__source-box')).not.toBeNull();
+    expect(screen.getByText(T.graphPaneTitle)).toBeInTheDocument();
+    expect(screen.getByText(T.resourceTitle)).toBeInTheDocument();
+    const footer = baseElement.querySelector('.semiont-gather__footer');
+    expect(footer).not.toBeNull();
+    expect(footer!.textContent).toContain(T.resolutionStrategyLabel);
+    expect(footer!.textContent).toContain(`✨ ${T.generate}`);
+    expect(footer!.querySelectorAll('button')).toHaveLength(0);
+    // The echo is a receipt LINE, not a control row: label and value share
+    // the --echo band, and the chooser's button-row wrapper is gone — the
+    // value must never sit where controls sit (it wore a control's costume:
+    // inherited base size + link blue).
+    expect(footer!.className).toContain('semiont-gather__footer--echo');
+    expect(footer!.querySelector('.semiont-gather__actions')).toBeNull();
+    expect(footer!.querySelector('.semiont-gather__chosen-strategy')).not.toBeNull();
+    expect(screen.queryByText(new RegExp(`^🔍 ${T.search}…`))).not.toBeInTheDocument();
+    const panel = baseElement.querySelector('.semiont-search-modal__panel')!;
+    expect(panel.className).toContain('semiont-search-modal__panel--wide');
+
+    // ONE scroll pane for the whole step: evidence and parameters scroll
+    // together, the form in full at the bottom, the evidence tucking up under
+    // the modal top. An independently-scrollable form inside the stack is what
+    // squeezed the parameters out of view.
+    const scroll = baseElement.querySelector('.semiont-wizard__step-scroll');
+    expect(scroll).not.toBeNull();
+    expect(scroll!.querySelector('.semiont-gather__outer')).not.toBeNull();
+    const form = scroll!.querySelector('form.semiont-form');
+    expect(form).not.toBeNull();
+    expect(form!.className).not.toContain('semiont-form--scrollable');
   });
+});
+
+describe('ReferenceWizardModal — compose is the fourth in-modal strategy (COMPOSE-IN-MODAL P2)', () => {
+  beforeEach(() => { vi.clearAllMocks(); });
 
-  it('leaves the context untouched when no hint was typed', async () => {
-    const { onComposeNavigate } = renderWizard();
+  async function enterCompose() {
     await userEvent.click(screen.getByText(new RegExp(T.compose)));
+  }
+
+  it('compose renders the evidence above the form, the strategy band collapsed to Compose', async () => {
+    const { baseElement } = renderWizard();
+    await enterCompose();
+
+    expect(baseElement.querySelector('.semiont-gather__source-box')).not.toBeNull();
+    expect(screen.getByText(T.graphPaneTitle)).toBeInTheDocument();
+    const footer = baseElement.querySelector('.semiont-gather__footer');
+    expect(footer).not.toBeNull();
+    expect(footer!.textContent).toContain(T.resolutionStrategyLabel);
+    expect(footer!.textContent).toContain(`✍️ ${T.compose}`);
+    expect(footer!.querySelectorAll('button')).toHaveLength(0);
+    // ONE scroll pane holding display + form (same grammar as the others).
+    const scroll = baseElement.querySelector('.semiont-wizard__step-scroll');
+    expect(scroll).not.toBeNull();
+    expect(scroll!.querySelector('.semiont-gather__outer')).not.toBeNull();
+    expect(scroll!.querySelector('form.semiont-form')).not.toBeNull();
+  });
+
+  it('the draft survives Back and re-entry (WIZARD-NAVIGATION D3)', async () => {
+    renderWizard();
+    await enterCompose();
+    fireEvent.change(screen.getByTestId('code-editor'), { target: { value: 'A Sauk leader.' } });
+    await userEvent.click(screen.getByRole('button', { name: `◀ ${T.back}` }));
+    expect(screen.getByText(new RegExp(T.compose))).toBeInTheDocument(); // back on gather
+    await enterCompose();
+    expect(screen.getByTestId('code-editor')).toHaveValue('A Sauk leader.');
+  });
+
+  it('submit emits against the annotation with the reference-fixed entity types, then closes', async () => {
+    const { onComposeSubmit, onClose } = renderWizard();
+    await enterCompose();
+    fireEvent.change(screen.getByLabelText(T.saveLocation), { target: { value: 'people/caspian-sea.md' } });
+    fireEvent.change(screen.getByTestId('code-editor'), { target: { value: 'The lake.' } });
+    await userEvent.click(screen.getByRole('button', { name: T.createAndLink }));
+
+    await waitFor(() => expect(onComposeSubmit).toHaveBeenCalledTimes(1));
+    expect(onComposeSubmit).toHaveBeenCalledWith('ann-1', {
+      name: 'Caspian Sea',
+      storagePath: 'file://people/caspian-sea.md',
+      content: 'The lake.',
+      entityTypes: ['Location'],
+      language: 'en',
+    });
+    await waitFor(() => expect(onClose).toHaveBeenCalledTimes(1));
+  });
+
+  it('a rejected submit keeps the modal open with the footer re-enabled', async () => {
+    const onComposeSubmit = vi.fn<(referenceId: string, params: unknown) => Promise<void>>()
+      .mockRejectedValue(new Error('boom'));
+    const { onClose } = renderWizard({ onComposeSubmit });
+    await enterCompose();
+    fireEvent.change(screen.getByLabelText(T.saveLocation), { target: { value: 'people/x.md' } });
+    await userEvent.click(screen.getByRole('button', { name: T.createAndLink }));
+
+    await waitFor(() => expect(onComposeSubmit).toHaveBeenCalled());
+    expect(onClose).not.toHaveBeenCalled();
+    expect(await screen.findByRole('button', { name: T.createAndLink })).toBeEnabled();
+  });
 
-    const context = onComposeNavigate.mock.calls[0]![0];
-    expect(context).toBe(CONTEXT); // same reference: nothing was rebuilt
+  // D4: a modal dies on ✕/Escape/backdrop; a non-empty draft must not die
+  // with it. The guard is an INLINE prompt — never window.confirm — and the
+  // footer stays dismissal-free (A4).
+  it('dismissing a dirty draft asks first; Keep editing stays, Discard closes', async () => {
+    const { onClose } = renderWizard();
+    await enterCompose();
+    fireEvent.change(screen.getByTestId('code-editor'), { target: { value: 'typed work' } });
+
+    await userEvent.click(screen.getByLabelText('Close'));
+    expect(onClose).not.toHaveBeenCalled();
+    expect(screen.getByText(T.discardDraftPrompt)).toBeInTheDocument();
+
+    await userEvent.click(screen.getByRole('button', { name: T.keepEditing }));
+    expect(screen.queryByText(T.discardDraftPrompt)).not.toBeInTheDocument();
+    expect(onClose).not.toHaveBeenCalled();
+
+    await userEvent.click(screen.getByLabelText('Close'));
+    await userEvent.click(screen.getByRole('button', { name: T.discardDraft }));
+    expect(onClose).toHaveBeenCalledTimes(1);
+  });
+
+  it('a clean draft dismisses without any prompt', async () => {
+    const { onClose } = renderWizard();
+    await enterCompose();
+    await userEvent.click(screen.getByLabelText('Close'));
+    expect(screen.queryByText(T.discardDraftPrompt)).not.toBeInTheDocument();
+    expect(onClose).toHaveBeenCalledTimes(1);
+  });
+});
+
+// GATHER-AT-THE-TOP P1: the dirty guard widens. D4 — dirtiness is
+// step-independent (typed work guards dismissal wherever the user currently
+// is); D5 — typed text only (title beyond seed, save location, instructions,
+// content). The generation draft gets the same protection compose has.
+describe('ReferenceWizardModal — the dirty guard widens (GATHER-AT-THE-TOP D4/D5)', () => {
+  beforeEach(() => { vi.clearAllMocks(); });
+
+  it('a typed generation draft guards dismissal', async () => {
+    const { onClose } = renderWizard();
+    await userEvent.click(screen.getByRole('button', { name: /Generate…/ }));
+    fireEvent.change(screen.getByLabelText(T.saveLocation), { target: { value: 'people/x.md' } });
+
+    await userEvent.click(screen.getByLabelText('Close'));
+    expect(onClose).not.toHaveBeenCalled();
+    expect(screen.getByText(T.discardDraftPrompt)).toBeInTheDocument();
+  });
+
+  it('a defaultTitle drifting while open does not fake a dirty draft', async () => {
+    // The wizard's defaultTitle derives from observable state and can re-emit
+    // while the modal is open. The baseline moves, the user has typed nothing —
+    // dirtiness compares against what was SEEDED, not the live prop.
+    const { rerender, onClose, onComposeSubmit, onGenerateSubmit, onLinkResource } = renderWizard();
+    rerender(
+      <ReferenceWizardModal
+        isOpen
+        onClose={onClose}
+        annotationId="ann-1"
+        resourceId="res-1"
+        defaultTitle="Caspian Sea (moved)"
+        entityTypes={['Location']}
+        entityTypeOptions={['Person', 'Topic', 'Location']}
+        locale="en"
+        context={CONTEXT}
+        contextLoading={false}
+        contextError={null}
+        onGenerateSubmit={onGenerateSubmit}
+        onLinkResource={onLinkResource}
+        onComposeSubmit={onComposeSubmit}
+        translations={T}
+      />,
+    );
+
+    await userEvent.click(screen.getByLabelText('Close'));
+    expect(screen.queryByText(T.discardDraftPrompt)).not.toBeInTheDocument();
+    expect(onClose).toHaveBeenCalledTimes(1);
+  });
+
+  it('a compose draft still guards after stepping Back to the evidence — D4 kills the step gate', async () => {
+    const { onClose } = renderWizard();
+    await userEvent.click(screen.getByText(new RegExp(T.compose)));
+    fireEvent.change(screen.getByTestId('code-editor'), { target: { value: 'typed work' } });
+    await userEvent.click(screen.getByRole('button', { name: new RegExp(T.back) }));
+
+    await userEvent.click(screen.getByLabelText('Close'));
+    expect(onClose).not.toHaveBeenCalled();
+    expect(screen.getByText(T.discardDraftPrompt)).toBeInTheDocument();
+  });
+});
+
+// The gather-loading state is not a blank modal: the annotation's exact text
+// and entity types are wizard props — known before any wire round-trip — and
+// the pane headers are static. Render them immediately with skeleton bars so
+// the loading screen is the loaded screen minus the data.
+describe('ReferenceWizardModal — the gather skeleton', () => {
+  beforeEach(() => { vi.clearAllMocks(); });
+
+  it('while gathering, the known annotation facts and per-cell animations render immediately', () => {
+    const { baseElement } = renderWizard({ context: null, contextLoading: true });
+
+    expect(screen.getByText(/"Caspian Sea"/)).toBeInTheDocument();      // quoted exact text
+    expect(screen.getByText('Location')).toBeInTheDocument();           // entity chip
+    expect(screen.getByText(T.graphPaneTitle)).toBeInTheDocument();     // pane headers in place
+    expect(screen.getByText(T.corpusPaneTitle)).toBeInTheDocument();
+    // One dots animation PER zone — source excerpt, Neighborhood, Similar
+    // passages — each where its content will land; no central block, no bars.
+    expect(baseElement.querySelectorAll('.semiont-gather__skeleton-dots')).toHaveLength(3);
+    expect(baseElement.querySelectorAll('.semiont-gather__skeleton-bar')).toHaveLength(0);
+    // The sentence survives for screen readers only.
+    expect(screen.getByText(T.loadingContext)).toHaveClass('semiont-sr-only');
+  });
+
+  it('the source line, Hint, and strategy chooser render during loading — buttons grayed until context lands', () => {
+    // The resource name is a page fact (the wizard opens FROM the source
+    // resource); the Hint is wizard state, safely typed while the gather
+    // runs; the chooser's buttons already gate on contextReady.
+    renderWizard({ context: null, contextLoading: true, resourceName: 'Cedar County, Iowa' });
+
+    expect(screen.getByText(/Source "Cedar County, Iowa"/)).toBeInTheDocument();
+    expect(screen.getByPlaceholderText(T.userHintPlaceholder)).toBeInTheDocument();
+    expect(screen.getByRole('button', { name: /Search…/ })).toBeDisabled();
+    expect(screen.getByRole('button', { name: /Generate…/ })).toBeDisabled();
+    expect(screen.getByRole('button', { name: new RegExp(T.compose) })).toBeDisabled();
+  });
+
+  it('a hint typed during the gather survives context arrival', () => {
+    const { rerender, onClose, onComposeSubmit, onGenerateSubmit, onLinkResource } = renderWizard({ context: null, contextLoading: true });
+    fireEvent.change(screen.getByPlaceholderText(T.userHintPlaceholder), { target: { value: 'early steer' } });
+
+    rerender(
+      <ReferenceWizardModal
+        isOpen onClose={onClose} annotationId="ann-1" resourceId="res-1"
+        defaultTitle="Caspian Sea" entityTypes={['Location']}
+        entityTypeOptions={['Person', 'Topic', 'Location']} locale="en"
+        context={CONTEXT} contextLoading={false} contextError={null}
+        onGenerateSubmit={onGenerateSubmit} onLinkResource={onLinkResource}
+        onComposeSubmit={onComposeSubmit} translations={T}
+      />,
+    );
+    expect(screen.getByPlaceholderText(T.userHintPlaceholder)).toHaveValue('early steer');
   });
 });
 
@@ -158,7 +486,8 @@ describe('ReferenceWizardModal — a new run starts clean (D3 flip side)', () =>
       isOpen: false, onClose: vi.fn(), annotationId: 'ann-1', resourceId: 'res-1',
       defaultTitle: 'Caspian Sea', entityTypes: ['Location'], locale: 'en',
       context: CONTEXT, contextLoading: false, contextError: null,
-      onGenerateSubmit: vi.fn(), onLinkResource: vi.fn(), onComposeNavigate: vi.fn(),
+      onGenerateSubmit: vi.fn(), onLinkResource: vi.fn(),
+      onComposeSubmit: vi.fn<(r: string, p: unknown) => Promise<void>>().mockResolvedValue(undefined),
       translations: T,
     };
     rerender(<ReferenceWizardModal {...props} />);
@@ -166,7 +495,7 @@ describe('ReferenceWizardModal — a new run starts clean (D3 flip side)', () =>
 
     // Back is lossless WITHIN a run (D3); a new run inherits nothing.
     expect(screen.getByPlaceholderText(T.userHintPlaceholder)).toHaveValue('');
-    expect(screen.getByText(T.gatherTitle)).toBeInTheDocument();
+    expect(screen.getByRole('button', { name: `🔍 ${T.search}…` })).toBeInTheDocument(); // back on the gather step
   });
 });
 
@@ -189,13 +518,65 @@ describe('ReferenceWizardModal — the three strategies complete', () => {
       response: [{ '@id': 'res-9', name: 'Caspian Sea', score: 54.2 }],
     } as never);
 
-    expect(await screen.findByText(T.searchResultsTitle)).toBeInTheDocument();
-    await userEvent.click(screen.getByRole('button', { name: new RegExp(T.link) }));
+    await userEvent.click(await screen.findByRole('button', { name: new RegExp(T.link) }));
 
     expect(onLinkResource).toHaveBeenCalledWith('ann-1', 'res-9');
     expect(onClose).toHaveBeenCalledTimes(1);
   });
 
+  it('a search failure over the bus settles the spinner and surfaces the error — no eternal "Searching…"', async () => {
+    // The measured hang (.plans/bugs/match-search-hangs-on-neo4j-datetime-annotations.md):
+    // /bus/emit 400s the request, the match machinery publishes
+    // match:search-failed — and the wizard listened only for results, so the
+    // failure fired into an empty room while the button spun forever.
+    const { client } = renderWizard();
+    await userEvent.click(screen.getByText(new RegExp(`^🔍? ?${T.search}`)));
+    await userEvent.click(screen.getByRole('button', { name: T.search }));
+    expect(screen.getByRole('button', { name: T.searching })).toBeDisabled();
+
+    client.bus.get('match:search-failed').next({
+      correlationId: 'c-1',
+      referenceId: 'ann-1',
+      error: '/bus/emit 400: Bus emit validation failed',
+    } as never);
+
+    // Still on configure-search, failure visible, retry available.
+    expect(await screen.findByText(new RegExp(T.searchFailed))).toBeInTheDocument();
+    expect(screen.getByText(/Bus emit validation failed/)).toBeInTheDocument();
+    expect(screen.getByRole('button', { name: T.search })).toBeEnabled();
+  });
+
+  it('a failure addressed to a different annotation changes nothing', async () => {
+    // Same scoping rule as results: a late failure from another annotation's
+    // run must not settle THIS spinner or paint an error that isn't ours.
+    const { client } = renderWizard();
+    await userEvent.click(screen.getByText(new RegExp(`^🔍? ?${T.search}`)));
+    await userEvent.click(screen.getByRole('button', { name: T.search }));
+
+    client.bus.get('match:search-failed').next({
+      correlationId: 'c-2',
+      referenceId: 'someone-elses-annotation',
+      error: 'not ours',
+    } as never);
+
+    expect(screen.getByRole('button', { name: T.searching })).toBeDisabled();
+    expect(screen.queryByText(new RegExp(T.searchFailed))).not.toBeInTheDocument();
+  });
+
+  it('retrying after a failure clears the error while the new search runs', async () => {
+    const { client } = renderWizard();
+    await userEvent.click(screen.getByText(new RegExp(`^🔍? ?${T.search}`)));
+    await userEvent.click(screen.getByRole('button', { name: T.search }));
+    client.bus.get('match:search-failed').next({
+      correlationId: 'c-1', referenceId: 'ann-1', error: 'boom',
+    } as never);
+    expect(await screen.findByText(new RegExp(T.searchFailed))).toBeInTheDocument();
+
+    await userEvent.click(screen.getByRole('button', { name: T.search }));
+    expect(screen.queryByText(new RegExp(T.searchFailed))).not.toBeInTheDocument();
+    expect(screen.getByRole('button', { name: T.searching })).toBeDisabled();
+  });
+
   it('ignores results addressed to a different annotation', () => {
     // Two wizards can never be open at once, but one reply can outlive the
     // annotation it was asked for — a late answer must not hijack the step.
@@ -204,27 +585,49 @@ describe('ReferenceWizardModal — the three strategies complete', () => {
       referenceId: 'someone-elses-annotation',
       response: [{ '@id': 'res-9', name: 'Caspian Sea' }],
     } as never);
-    expect(screen.getByText(T.gatherTitle)).toBeInTheDocument();
+    expect(screen.getByRole('button', { name: `🔍 ${T.search}…` })).toBeInTheDocument(); // still the gather step
   });
 
   it('generation submits the step\'s config against this annotation, then closes', async () => {
     const { onGenerateSubmit, onClose } = renderWizard();
-    await userEvent.click(screen.getByText(new RegExp(T.generate)));
-    await userEvent.type(screen.getByLabelText(/Save location/i), 'generated/out.md');
+    await userEvent.click(screen.getByRole('button', { name: `✨ ${T.generate}…` }));
+    // The field arrives PRE-FILLED with the proposal now (D11). Overriding it
+    // is select-all-then-type, which the DOM delivers as ONE change carrying
+    // the replacement — `clear()` then `type()` would instead empty the field
+    // (restoring the proposal, by design) and append to it.
+    fireEvent.change(screen.getByLabelText(/Save location/i), {
+      target: { value: 'generated/out.md' },
+    });
     await userEvent.click(screen.getByRole('button', { name: T.generate }));
 
     expect(onGenerateSubmit).toHaveBeenCalledTimes(1);
     expect(onGenerateSubmit.mock.calls[0]![0]).toBe('ann-1');
     expect(onGenerateSubmit.mock.calls[0]![1].title).toBe('Caspian Sea');
+    // D8/D2: the payload names the URI what it is, and carries the default
+    // format explicitly rather than leaving the worker to assume it.
+    expect(onGenerateSubmit.mock.calls[0]![1]).toMatchObject({
+      storageUri: 'file://generated/out.md',
+      outputMediaType: 'text/markdown',
+    });
     expect(onClose).toHaveBeenCalledTimes(1);
   });
 
+  it('the annotation flow can choose a format too — one control, both hosts (D4)', async () => {
+    const { onGenerateSubmit } = renderWizard();
+    await userEvent.click(screen.getByRole('button', { name: `✨ ${T.generate}…` }));
+    await userEvent.type(screen.getByLabelText(/Save location/i), 'generated/out.pdf');
+    fireEvent.change(screen.getByLabelText(T.outputFormat), { target: { value: 'application/pdf' } });
+    await userEvent.click(screen.getByRole('button', { name: T.generate }));
+
+    expect(onGenerateSubmit.mock.calls[0]![1]).toMatchObject({ outputMediaType: 'application/pdf' });
+  });
+
   it('retreats from a configure step back to the gather step', async () => {
     renderWizard();
-    await userEvent.click(screen.getByText(new RegExp(T.generate)));
-    expect(screen.getByText(T.configureGenerationTitle)).toBeInTheDocument();
+    await userEvent.click(screen.getByRole('button', { name: `✨ ${T.generate}…` }));
+    expect(screen.getByLabelText(T.resourceTitle)).toBeInTheDocument();
     await userEvent.click(screen.getByRole('button', { name: /Back/ }));
-    expect(screen.getByText(T.gatherTitle)).toBeInTheDocument();
+    expect(screen.getByRole('button', { name: `🔍 ${T.search}…` })).toBeInTheDocument();
   });
 });
 
@@ -236,11 +639,11 @@ describe('ReferenceWizardModal — nothing fires without an annotation to resolv
   beforeEach(() => { vi.clearAllMocks(); });
 
   it('compose, search and generate all no-op', async () => {
-    const { onComposeNavigate, onGenerateSubmit, searchSpy, client } =
+    const { onGenerateSubmit, searchSpy, client } =
       renderWizard({ annotationId: null, resourceId: null });
 
     await userEvent.click(screen.getByText(new RegExp(T.compose)));
-    expect(onComposeNavigate).not.toHaveBeenCalled();
+    expect(screen.queryByRole('button', { name: T.createAndLink })).not.toBeInTheDocument();
 
     await userEvent.click(screen.getByText(new RegExp(`^🔍? ?${T.search}`)));
     await userEvent.click(screen.getByRole('button', { name: T.search }));
@@ -250,7 +653,7 @@ describe('ReferenceWizardModal — nothing fires without an annotation to resolv
     client.bus.get('match:search-results').next({
       referenceId: 'ann-1', response: [{ '@id': 'res-9', name: 'X' }],
     } as never);
-    expect(screen.queryByText(T.searchResultsTitle)).not.toBeInTheDocument();
+    expect(screen.queryByRole('button', { name: new RegExp(T.link) })).not.toBeInTheDocument();
     expect(onGenerateSubmit).not.toHaveBeenCalled();
   });
 });