packages/muya/src/__tests__/crossBlockParagraph.spec.ts - develop - githubmirror/marktext

packages/muya/src/__tests__/crossBlockParagraph.spec.ts
143 строки6 KB
Ran Luo
chore: audit Claude-authored comments against the commenting guidelines (#4740)
26 июн 2026, 13:54
26 июн 2026, 13:54e416a99
// @vitest-environment happy-dom
import type Parent from '../block/base/parent';
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
import { Muya } from '../muya';

interface IStateBlock { name: string; text: string; children: IStateBlock[] }

const hosts: HTMLElement[] = [];
beforeEach(() => {
    window.MUYA_VERSION = 'test';
});
afterEach(() => {
    while (hosts.length)
        hosts.pop()!.remove();
});
function boot(md: string): Muya {
    const host = document.createElement('div');
    document.body.appendChild(host);
    const muya = new Muya(host, { markdown: md } as ConstructorParameters<typeof Muya>[1]);
    muya.init();
    hosts.push(muya.domNode);
    return muya;
}
// Select a range spanning the first two top-level paragraphs.
function selectFirstTwoBlocks(muya: Muya) {
    const sp = muya.editor.scrollPage!;
    const first = sp.firstContentInDescendant()!;
    const second = (sp.firstChild!.next as Parent).firstContentInDescendant()!;
    muya.editor.activeContentBlock = second;
    muya.editor.selection.setSelection(
        { offset: 0, block: first, path: first.path },
        { offset: second.text.length, block: second, path: second.path },
    );
}

describe('cross-block paragraph wrapping', () => {
    it('wraps selected paragraphs into a bullet list (one item per block)', async () => {
        const muya = boot('alpha\n\nbravo\n');
        selectFirstTwoBlocks(muya);
        muya.updateParagraph('ul-bullet');
        await vi.waitFor(() => {
            const s = muya.getState() as unknown as IStateBlock[];
            expect(s[0].name).toBe('bullet-list');
            expect(s[0].children.length).toBe(2);
            expect(s[0].children[0].children[0].text).toBe('alpha');
            expect(s[0].children[1].children[0].text).toBe('bravo');
        });
    });

    it('wraps selected paragraphs into an ordered list', async () => {
        const muya = boot('alpha\n\nbravo\n');
        selectFirstTwoBlocks(muya);
        muya.updateParagraph('ol-order');
        await vi.waitFor(() => {
            const s = muya.getState() as unknown as IStateBlock[];
            expect(s[0].name).toBe('order-list');
            expect(s[0].children.length).toBe(2);
        });
    });

    it('wraps selected paragraphs into a task list', async () => {
        const muya = boot('alpha\n\nbravo\n');
        selectFirstTwoBlocks(muya);
        muya.updateParagraph('ul-task');
        await vi.waitFor(() => {
            const s = muya.getState() as unknown as IStateBlock[];
            expect(s[0].name).toBe('task-list');
            expect(s[0].children.length).toBe(2);
        });
    });

    it('wraps selected paragraphs into a single block-quote', async () => {
        const muya = boot('alpha\n\nbravo\n');
        selectFirstTwoBlocks(muya);
        muya.updateParagraph('blockquote');
        await vi.waitFor(() => {
            const s = muya.getState() as unknown as IStateBlock[];
            expect(s[0].name).toBe('block-quote');
            expect(s[0].children.length).toBe(2);
            expect(s[0].children[0].text).toBe('alpha');
            expect(s[0].children[1].text).toBe('bravo');
        });
    });

    it('joins selected paragraphs into a single code block', async () => {
        const muya = boot('alpha\n\nbravo\n');
        selectFirstTwoBlocks(muya);
        muya.updateParagraph('pre');
        await vi.waitFor(() => {
            const s = muya.getState() as unknown as IStateBlock[];
            expect(s.length).toBe(1);
            expect(s[0].name).toBe('code-block');
            const text = (s[0].text ?? '') as string;
            expect(text).toContain('alpha');
            expect(text).toContain('bravo');
        });
    });

    it('keeps the selection spanning the wrapped content after list wrap', async () => {
        const muya = boot('alpha\n\nbravo\n');
        selectFirstTwoBlocks(muya);
        muya.updateParagraph('ul-bullet');
        await vi.waitFor(() => {
            expect(muya.getState()[0].name).toBe('bullet-list');
        });
        // happy-dom collapses the live DOM selection, so assert via the cached
        // endpoints that setSelection records (the same ones the menu round-trip
        // relies on).
        const sel = muya.editor.selection;
        expect(sel.anchorBlock!.text).toBe('alpha');
        expect(sel.focusBlock!.text).toBe('bravo');
    });

    it('preserves markdown syntax when wrapping a cross-block selection into a code block', async () => {
        const muya = boot('# Title\n\nbody\n');
        selectFirstTwoBlocks(muya);
        muya.updateParagraph('pre');
        await vi.waitFor(() => {
            const s = muya.getState() as unknown as IStateBlock[];
            expect(s.length).toBe(1);
            expect(s[0].name).toBe('code-block');
            expect(s[0].text).toContain('# Title');
            expect(s[0].text).toContain('body');
        });
    });

    it('preserves a multi-block selection across wrap then unwrap (paragraph + list)', async () => {
        const muya = boot('alpha\n\n- bravo\n');
        selectFirstTwoBlocks(muya);
        muya.updateParagraph('ol-order'); // wrap both into an ordered list
        await vi.waitFor(() => expect(muya.getState()[0].name).toBe('order-list'));

        muya.updateParagraph('ol-order'); // click ordered again -> unwrap
        await vi.waitFor(() => {
            const s = muya.getState() as unknown as IStateBlock[];
            expect(s.some(b => b.name === 'order-list')).toBe(false);
        });
        // the original span (alpha .. bravo) is restored
        const sel = muya.editor.selection;
        expect(sel.anchorBlock!.text).toBe('alpha');
        expect(sel.focusBlock!.text).toBe('bravo');
    });
});