mirror of
https://github.com/rohitg00/agentmemory.git
synced 2026-09-14 20:16:33 +08:00
Rohit Ghumare
c85ee6d1e5
- XML tag validation: reject invalid tag names before regex construction - API auth: optional bearer token (AGENTMEMORY_SECRET) on POST endpoints - Conservative token estimation: length/3 for code-heavy content - XML-escape project name in context output - Prefix-only search matching (startsWith vs includes) - Dynamic viewer URLs from window.location - CSP header on viewer endpoint - Importance class sanitization in viewer - 45 tests: privacy, XML parsing, search index, schema
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([])
|
|
})
|
|
})
|