|
1 | | -import { describe, expect, it } from 'vitest'; |
| 1 | +import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'; |
2 | 2 | import { unified } from 'unified'; |
3 | 3 | import remarkParse from 'remark-parse'; |
4 | 4 | import remarkGfm from 'remark-gfm'; |
@@ -39,3 +39,89 @@ describe('immutable Markdown parse cache', () => { |
39 | 39 | expect(JSON.stringify(first)).not.toContain('onError'); |
40 | 40 | }); |
41 | 41 | }); |
| 42 | + |
| 43 | +describe('Markdown worker prewarming', () => { |
| 44 | + const text = 'Paragraph with © and **emphasis**.\n\n'.repeat(80); |
| 45 | + let workers: TestWorker[]; |
| 46 | + class TestWorker { |
| 47 | + onerror?: () => void; |
| 48 | + onmessage?: (event: { data: { tree?: unknown } }) => void; |
| 49 | + postMessage = vi.fn(); |
| 50 | + terminate = vi.fn(); |
| 51 | + constructor() { workers.push(this); } |
| 52 | + } |
| 53 | + |
| 54 | + beforeEach(() => { |
| 55 | + vi.resetModules(); |
| 56 | + vi.useFakeTimers(); |
| 57 | + workers = []; |
| 58 | + vi.stubGlobal('Worker', TestWorker); |
| 59 | + }); |
| 60 | + afterEach(() => { |
| 61 | + vi.clearAllTimers(); |
| 62 | + vi.useRealTimers(); |
| 63 | + vi.unstubAllGlobals(); |
| 64 | + }); |
| 65 | + |
| 66 | + it('stops restarting a failed worker and preserves main-thread parsing', async () => { |
| 67 | + const { prewarmMarkdownParse, remarkCachedParse: cachedParse } = await import('./markdownParseCache'); |
| 68 | + prewarmMarkdownParse(text); |
| 69 | + prewarmMarkdownParse(`${text} queued`); |
| 70 | + workers[0].onerror!(); |
| 71 | + prewarmMarkdownParse(`${text} streamed update`); |
| 72 | + vi.advanceTimersByTime(60_000); |
| 73 | + prewarmMarkdownParse(`${text} later update`); |
| 74 | + expect(workers).toHaveLength(1); |
| 75 | + expect(workers[0].terminate).toHaveBeenCalledOnce(); |
| 76 | + expect(workers[0].postMessage).toHaveBeenCalledTimes(1); |
| 77 | + const processor = unified().use(remarkParse).use(remarkGfm).use(cachedParse); |
| 78 | + expect(processor.parse(text)).toEqual(unified().use(remarkParse).use(remarkGfm).parse(text)); |
| 79 | + }); |
| 80 | + |
| 81 | + it('does not retry a worker whose constructor throws', async () => { |
| 82 | + const constructor = vi.fn(function () { throw new Error('Worker unavailable'); }); |
| 83 | + vi.stubGlobal('Worker', constructor); |
| 84 | + const { prewarmMarkdownParse } = await import('./markdownParseCache'); |
| 85 | + expect(() => prewarmMarkdownParse(text)).not.toThrow(); |
| 86 | + prewarmMarkdownParse(`${text} update`); |
| 87 | + expect(constructor).toHaveBeenCalledOnce(); |
| 88 | + }); |
| 89 | + |
| 90 | + it('disables and terminates the worker if posting fails', async () => { |
| 91 | + const { prewarmMarkdownParse } = await import('./markdownParseCache'); |
| 92 | + prewarmMarkdownParse(text); |
| 93 | + workers[0].onmessage!({ data: {} }); |
| 94 | + workers[0].postMessage.mockImplementation(() => { throw new Error('Cannot post'); }); |
| 95 | + expect(() => prewarmMarkdownParse(`${text} next`)).not.toThrow(); |
| 96 | + prewarmMarkdownParse(`${text} update`); |
| 97 | + expect(workers).toHaveLength(1); |
| 98 | + expect(workers[0].terminate).toHaveBeenCalledOnce(); |
| 99 | + expect(workers[0].postMessage).toHaveBeenCalledTimes(2); |
| 100 | + }); |
| 101 | + |
| 102 | + it('caches successful replies and can restart after normal idle termination', async () => { |
| 103 | + const { prewarmMarkdownParse, remarkCachedParse: cachedParse } = await import('./markdownParseCache'); |
| 104 | + const tree = unified().use(remarkParse).use(remarkGfm).parse(text); |
| 105 | + prewarmMarkdownParse(text); |
| 106 | + workers[0].onmessage!({ data: { tree } }); |
| 107 | + const processor = unified().use(remarkParse).use(remarkGfm).use(cachedParse); |
| 108 | + expect(processor.parse(text)).toEqual(tree); |
| 109 | + prewarmMarkdownParse(text); |
| 110 | + expect(workers[0].postMessage).toHaveBeenCalledOnce(); |
| 111 | + vi.advanceTimersByTime(30_000); |
| 112 | + expect(workers[0].terminate).toHaveBeenCalledOnce(); |
| 113 | + prewarmMarkdownParse(`${text} next`); |
| 114 | + expect(workers).toHaveLength(2); |
| 115 | + }); |
| 116 | + |
| 117 | + it('handles posting failures while advancing the queue after a reply', async () => { |
| 118 | + const { prewarmMarkdownParse } = await import('./markdownParseCache'); |
| 119 | + prewarmMarkdownParse(text); |
| 120 | + prewarmMarkdownParse(`${text} queued`); |
| 121 | + workers[0].postMessage.mockImplementation(() => { throw new Error('Cannot post'); }); |
| 122 | + expect(() => workers[0].onmessage!({ data: {} })).not.toThrow(); |
| 123 | + prewarmMarkdownParse(`${text} update`); |
| 124 | + expect(workers).toHaveLength(1); |
| 125 | + expect(workers[0].terminate).toHaveBeenCalledOnce(); |
| 126 | + }); |
| 127 | +}); |
0 commit comments