|
| 1 | +/** |
| 2 | + * @vitest-environment node |
| 3 | + */ |
| 4 | +import { describe, expect, it, vi } from 'vitest' |
| 5 | + |
| 6 | +vi.mock('@/lib/knowledge/embeddings', () => ({ |
| 7 | + generateEmbeddings: vi.fn(async () => ({ embeddings: [] })), |
| 8 | + getConfiguredEmbeddingModel: vi.fn(() => 'test-model'), |
| 9 | +})) |
| 10 | + |
| 11 | +import { DocsChunker } from '@/lib/chunkers/docs-chunker' |
| 12 | + |
| 13 | +function cleanContent(content: string): string { |
| 14 | + const chunker = new DocsChunker() |
| 15 | + return (chunker as unknown as { cleanContent(content: string): string }).cleanContent.call( |
| 16 | + chunker, |
| 17 | + content |
| 18 | + ) |
| 19 | +} |
| 20 | + |
| 21 | +describe('cleanContent FAQ extraction', () => { |
| 22 | + it('keeps FAQ question/answer prose that the tag and brace strips would otherwise delete', () => { |
| 23 | + const cleaned = cleanContent( |
| 24 | + [ |
| 25 | + 'Some intro prose.', |
| 26 | + '', |
| 27 | + 'import { FAQ } from "@/components/ui/faq"', |
| 28 | + '', |
| 29 | + '<FAQ items={[', |
| 30 | + ' { question: "What is the maximum file size for uploads?", answer: "The maximum file size for files processed during a workflow run is 20 MB." },', |
| 31 | + ' { question: "How are files passed between blocks internally?", answer: "Files are represented as standardized UserFile objects." },', |
| 32 | + ']} />', |
| 33 | + ].join('\n') |
| 34 | + ) |
| 35 | + |
| 36 | + expect(cleaned).toContain('What is the maximum file size for uploads?') |
| 37 | + expect(cleaned).toContain('20 MB') |
| 38 | + expect(cleaned).toContain('standardized UserFile objects') |
| 39 | + expect(cleaned).toContain('Some intro prose.') |
| 40 | + expect(cleaned).not.toContain('items=') |
| 41 | + expect(cleaned).not.toContain('question:') |
| 42 | + }) |
| 43 | + |
| 44 | + it('survives braces and angle-bracket tokens inside answer strings', () => { |
| 45 | + const cleaned = cleanContent( |
| 46 | + [ |
| 47 | + '<FAQ items={[', |
| 48 | + ` { question: "What input formats work?", answer: "Use a data URI with the format 'data:{mime};base64,{data}' or a URL." },`, |
| 49 | + ' { question: "Do I extract base64 manually?", answer: "No. Pass the entire file reference (e.g., <gmail.attachments[0]>) and the block extracts what it needs." },', |
| 50 | + ']} />', |
| 51 | + ].join('\n') |
| 52 | + ) |
| 53 | + |
| 54 | + // Brace placeholders keep their token text; the wrapper chars are dropped |
| 55 | + // so the later brace strip cannot punch holes in the sentence. |
| 56 | + expect(cleaned).toContain("'data:mime;base64,data'") |
| 57 | + // Angle brackets are dropped so the tag strip cannot re-eat the sentence. |
| 58 | + expect(cleaned).toContain('(e.g., gmail.attachments[0]) and the block extracts') |
| 59 | + }) |
| 60 | + |
| 61 | + it('extracts items formatted across multiple lines', () => { |
| 62 | + const cleaned = cleanContent( |
| 63 | + [ |
| 64 | + '<FAQ items={[', |
| 65 | + ' {', |
| 66 | + ' question: "Is SSO supported?",', |
| 67 | + ' answer: "Yes, on enterprise plans."', |
| 68 | + ' },', |
| 69 | + ']} />', |
| 70 | + ].join('\n') |
| 71 | + ) |
| 72 | + |
| 73 | + expect(cleaned).toContain('Is SSO supported?') |
| 74 | + expect(cleaned).toContain('Yes, on enterprise plans.') |
| 75 | + }) |
| 76 | + |
| 77 | + it('unescapes escaped quotes in extracted strings', () => { |
| 78 | + const cleaned = cleanContent( |
| 79 | + '<FAQ items={[ { question: "What does \\"draft\\" mean?", answer: "An unsaved workflow." } ]} />' |
| 80 | + ) |
| 81 | + |
| 82 | + expect(cleaned).toContain('What does "draft" mean?') |
| 83 | + }) |
| 84 | +}) |
| 85 | + |
| 86 | +describe('cleanContent scaffolding strips', () => { |
| 87 | + it('still strips imports, exports, comments, and code-ish brace expressions', () => { |
| 88 | + const cleaned = cleanContent( |
| 89 | + [ |
| 90 | + 'import { Callout } from "fumadocs-ui/components/callout"', |
| 91 | + 'export const dynamic = "force-static"', |
| 92 | + '{/* editorial note */}', |
| 93 | + 'Visible prose {props.title} continues here.', |
| 94 | + '<Callout>Inside text stays</Callout>', |
| 95 | + ].join('\n') |
| 96 | + ) |
| 97 | + |
| 98 | + expect(cleaned).not.toContain('import') |
| 99 | + expect(cleaned).not.toContain('force-static') |
| 100 | + expect(cleaned).not.toContain('editorial note') |
| 101 | + expect(cleaned).not.toContain('props.title') |
| 102 | + expect(cleaned).toContain('Visible prose') |
| 103 | + expect(cleaned).toContain('Inside text stays') |
| 104 | + }) |
| 105 | +}) |
0 commit comments