66 lines
2.2 KiB
TypeScript
66 lines
2.2 KiB
TypeScript
import { describe, it, expect } from 'vitest'
|
|
import { getXmlTag, getXmlChildren } from '../src/prompts/xml.js'
|
|
|
|
describe('getXmlTag', () => {
|
|
it('extracts simple tag content', () => {
|
|
expect(getXmlTag('<title>Hello World</title>', 'title')).toBe('Hello World')
|
|
})
|
|
|
|
it('extracts multiline content', () => {
|
|
expect(getXmlTag('<narrative>\nLine 1\nLine 2\n</narrative>', 'narrative')).toBe('Line 1\nLine 2')
|
|
})
|
|
|
|
it('returns empty string for missing tag', () => {
|
|
expect(getXmlTag('<title>Hello</title>', 'missing')).toBe('')
|
|
})
|
|
|
|
it('returns first match for duplicate tags', () => {
|
|
expect(getXmlTag('<title>First</title><title>Second</title>', 'title')).toBe('First')
|
|
})
|
|
|
|
it('returns empty string for empty tag', () => {
|
|
expect(getXmlTag('<title></title>', 'title')).toBe('')
|
|
})
|
|
|
|
it('trims whitespace', () => {
|
|
expect(getXmlTag('<title> trimmed </title>', 'title')).toBe('trimmed')
|
|
})
|
|
|
|
it('returns empty for invalid tag names', () => {
|
|
expect(getXmlTag('<foo>bar</foo>', '.*')).toBe('')
|
|
})
|
|
|
|
it('returns empty for tag with special regex chars', () => {
|
|
expect(getXmlTag('<foo>bar</foo>', 'a(b')).toBe('')
|
|
})
|
|
})
|
|
|
|
describe('getXmlChildren', () => {
|
|
it('extracts child elements', () => {
|
|
const xml = '<facts><fact>One</fact><fact>Two</fact></facts>'
|
|
expect(getXmlChildren(xml, 'facts', 'fact')).toEqual(['One', 'Two'])
|
|
})
|
|
|
|
it('returns empty array for missing parent', () => {
|
|
expect(getXmlChildren('<foo>bar</foo>', 'facts', 'fact')).toEqual([])
|
|
})
|
|
|
|
it('returns empty array for missing children', () => {
|
|
expect(getXmlChildren('<facts></facts>', 'facts', 'fact')).toEqual([])
|
|
})
|
|
|
|
it('trims child content', () => {
|
|
const xml = '<facts><fact> trimmed </fact></facts>'
|
|
expect(getXmlChildren(xml, 'facts', 'fact')).toEqual(['trimmed'])
|
|
})
|
|
|
|
it('handles multiline children', () => {
|
|
const xml = '<decisions><decision>Use JWT\nfor auth</decision></decisions>'
|
|
expect(getXmlChildren(xml, 'decisions', 'decision')).toEqual(['Use JWT\nfor auth'])
|
|
})
|
|
|
|
it('returns empty for invalid parent tag name', () => {
|
|
expect(getXmlChildren('<facts><fact>A</fact></facts>', '.*', 'fact')).toEqual([])
|
|
})
|
|
})
|