优化角色注册，发现，nuwa 角色的提示词等
This commit is contained in:
sean
106
src/tests/core/resource/DPMLContentParser.integration.test.js
Normal file
106
src/tests/core/resource/DPMLContentParser.integration.test.js
Normal file
@ -0,0 +1,106 @@
|
||||
const DPMLContentParser = require('../../../lib/core/resource/DPMLContentParser')
|
||||
const path = require('path')
|
||||
const fs = require('fs-extra')
|
||||
|
||||
describe('DPMLContentParser 集成测试', () => {
|
||||
let parser
|
||||
|
||||
beforeEach(() => {
|
||||
parser = new DPMLContentParser()
|
||||
})
|
||||
|
||||
describe('真实角色文件测试', () => {
|
||||
test('应该正确解析internet-debater角色的完整内容', async () => {
|
||||
const roleFile = '/Users/sean/WorkSpaces/DeepracticeProjects/PromptX/.promptx/resource/domain/internet-debater/internet-debater.role.md'
|
||||
|
||||
// 检查文件是否存在
|
||||
const exists = await fs.pathExists(roleFile)
|
||||
if (!exists) {
|
||||
console.log('跳过测试：internet-debater角色文件不存在')
|
||||
return
|
||||
}
|
||||
|
||||
const roleContent = await fs.readFile(roleFile, 'utf-8')
|
||||
const roleSemantics = parser.parseRoleDocument(roleContent)
|
||||
|
||||
// 验证personality解析
|
||||
expect(roleSemantics).toHaveProperty('personality')
|
||||
expect(roleSemantics.personality.metadata.contentType).toBe('direct-only')
|
||||
expect(roleSemantics.personality.directContent).toContain('网络杠精思维模式')
|
||||
expect(roleSemantics.personality.directContent).toContain('挑刺思维')
|
||||
expect(roleSemantics.personality.directContent).toContain('抬杠本能')
|
||||
|
||||
// 验证principle解析
|
||||
expect(roleSemantics).toHaveProperty('principle')
|
||||
expect(roleSemantics.principle.metadata.contentType).toBe('direct-only')
|
||||
expect(roleSemantics.principle.directContent).toContain('网络杠精行为原则')
|
||||
expect(roleSemantics.principle.directContent).toContain('逢言必反')
|
||||
expect(roleSemantics.principle.directContent).toContain('抠字眼优先')
|
||||
|
||||
// 验证knowledge解析
|
||||
expect(roleSemantics).toHaveProperty('knowledge')
|
||||
expect(roleSemantics.knowledge.metadata.contentType).toBe('direct-only')
|
||||
expect(roleSemantics.knowledge.directContent).toContain('网络杠精专业知识体系')
|
||||
expect(roleSemantics.knowledge.directContent).toContain('逻辑谬误大全')
|
||||
expect(roleSemantics.knowledge.directContent).toContain('稻草人谬误')
|
||||
|
||||
console.log('✅ internet-debater角色内容完整解析成功')
|
||||
console.log(` - personality: ${roleSemantics.personality.directContent.length} 字符`)
|
||||
console.log(` - principle: ${roleSemantics.principle.directContent.length} 字符`)
|
||||
console.log(` - knowledge: ${roleSemantics.knowledge.directContent.length} 字符`)
|
||||
})
|
||||
|
||||
test('应该正确解析系统角色的@引用内容', async () => {
|
||||
const roleFile = '/Users/sean/WorkSpaces/DeepracticeProjects/PromptX/prompt/domain/assistant/assistant.role.md'
|
||||
|
||||
const exists = await fs.pathExists(roleFile)
|
||||
if (!exists) {
|
||||
console.log('跳过测试：assistant角色文件不存在')
|
||||
return
|
||||
}
|
||||
|
||||
const roleContent = await fs.readFile(roleFile, 'utf-8')
|
||||
const roleSemantics = parser.parseRoleDocument(roleContent)
|
||||
|
||||
// 验证personality有@引用
|
||||
if (roleSemantics.personality) {
|
||||
expect(roleSemantics.personality.references.length).toBeGreaterThan(0)
|
||||
console.log('✅ assistant角色@引用解析成功')
|
||||
console.log(` - personality引用数量: ${roleSemantics.personality.references.length}`)
|
||||
}
|
||||
})
|
||||
})
|
||||
|
||||
describe('修复前后对比测试', () => {
|
||||
test('模拟ActionCommand当前解析vs新解析的差异', () => {
|
||||
const mixedContent = `@!thought://remember
|
||||
@!thought://recall
|
||||
|
||||
# 网络杠精思维模式
|
||||
## 核心思维特征
|
||||
- 挑刺思维：看到任何观点都先找问题和漏洞
|
||||
- 抬杠本能：天生反对派，习惯性质疑一切表述`
|
||||
|
||||
// 模拟当前ActionCommand的解析（只提取@引用）
|
||||
const currentParsing = {
|
||||
thoughtReferences: ['remember', 'recall'],
|
||||
directContent: '' // 完全丢失
|
||||
}
|
||||
|
||||
// 新的DPMLContentParser解析
|
||||
const newParsing = parser.parseTagContent(mixedContent, 'personality')
|
||||
|
||||
// 对比结果
|
||||
expect(newParsing.references).toHaveLength(2)
|
||||
expect(newParsing.references.map(r => r.resource)).toEqual(['remember', 'recall'])
|
||||
expect(newParsing.directContent).toContain('网络杠精思维模式')
|
||||
expect(newParsing.directContent).toContain('挑刺思维')
|
||||
expect(newParsing.directContent).toContain('抬杠本能')
|
||||
|
||||
console.log('📊 解析能力对比：')
|
||||
console.log(` 当前ActionCommand: 只解析${currentParsing.thoughtReferences.length}个引用，丢失${newParsing.directContent.length}字符直接内容`)
|
||||
console.log(` 新DPMLContentParser: 解析${newParsing.references.length}个引用 + ${newParsing.directContent.length}字符直接内容`)
|
||||
console.log(` 🎯 语义完整性提升: ${((newParsing.directContent.length / mixedContent.length) * 100).toFixed(1)}%`)
|
||||
})
|
||||
})
|
||||
})
|
||||
174
src/tests/core/resource/DPMLContentParser.position.unit.test.js
Normal file
174
src/tests/core/resource/DPMLContentParser.position.unit.test.js
Normal file
@ -0,0 +1,174 @@
|
||||
const DPMLContentParser = require('../../../lib/core/resource/DPMLContentParser')
|
||||
|
||||
describe('DPMLContentParser - Position Extension', () => {
|
||||
let parser
|
||||
|
||||
beforeEach(() => {
|
||||
parser = new DPMLContentParser()
|
||||
})
|
||||
|
||||
describe('extractReferencesWithPosition', () => {
|
||||
test('应该提取引用的位置信息', () => {
|
||||
// Arrange
|
||||
const content = 'intro @!thought://A middle @!thought://B end'
|
||||
|
||||
// Act
|
||||
const references = parser.extractReferencesWithPosition(content)
|
||||
|
||||
// Assert
|
||||
expect(references).toHaveLength(2)
|
||||
expect(references[0]).toEqual({
|
||||
fullMatch: '@!thought://A',
|
||||
priority: '!',
|
||||
protocol: 'thought',
|
||||
resource: 'A',
|
||||
position: 6,
|
||||
isRequired: true,
|
||||
isOptional: false
|
||||
})
|
||||
expect(references[1]).toEqual({
|
||||
fullMatch: '@!thought://B',
|
||||
priority: '!',
|
||||
protocol: 'thought',
|
||||
resource: 'B',
|
||||
position: 27,
|
||||
isRequired: true,
|
||||
isOptional: false
|
||||
})
|
||||
})
|
||||
|
||||
test('应该按位置排序返回引用', () => {
|
||||
// Arrange
|
||||
const content = '@!thought://second @!thought://first'
|
||||
|
||||
// Act
|
||||
const references = parser.extractReferencesWithPosition(content)
|
||||
|
||||
// Assert
|
||||
expect(references).toHaveLength(2)
|
||||
expect(references[0].resource).toBe('second')
|
||||
expect(references[0].position).toBe(0)
|
||||
expect(references[1].resource).toBe('first')
|
||||
expect(references[1].position).toBe(19)
|
||||
})
|
||||
|
||||
test('应该处理复杂布局中的位置信息', () => {
|
||||
// Arrange
|
||||
const content = `# 标题
|
||||
@!thought://base
|
||||
|
||||
## 子标题
|
||||
- 列表项1
|
||||
@!execution://action
|
||||
- 列表项2`
|
||||
|
||||
// Act
|
||||
const references = parser.extractReferencesWithPosition(content)
|
||||
|
||||
// Assert
|
||||
expect(references).toHaveLength(2)
|
||||
|
||||
const baseRef = references.find(ref => ref.resource === 'base')
|
||||
const actionRef = references.find(ref => ref.resource === 'action')
|
||||
|
||||
expect(baseRef.position).toBe(5) // 在"# 标题\n"之后
|
||||
expect(actionRef.position).toBeGreaterThan(baseRef.position)
|
||||
})
|
||||
|
||||
test('应该处理可选引用', () => {
|
||||
// Arrange
|
||||
const content = 'content @?optional://resource more'
|
||||
|
||||
// Act
|
||||
const references = parser.extractReferencesWithPosition(content)
|
||||
|
||||
// Assert
|
||||
expect(references).toHaveLength(1)
|
||||
expect(references[0]).toEqual({
|
||||
fullMatch: '@?optional://resource',
|
||||
priority: '?',
|
||||
protocol: 'optional',
|
||||
resource: 'resource',
|
||||
position: 8,
|
||||
isRequired: false,
|
||||
isOptional: true
|
||||
})
|
||||
})
|
||||
|
||||
test('应该处理普通引用（无优先级标记）', () => {
|
||||
// Arrange
|
||||
const content = 'content @normal://resource more'
|
||||
|
||||
// Act
|
||||
const references = parser.extractReferencesWithPosition(content)
|
||||
|
||||
// Assert
|
||||
expect(references).toHaveLength(1)
|
||||
expect(references[0]).toEqual({
|
||||
fullMatch: '@normal://resource',
|
||||
priority: '',
|
||||
protocol: 'normal',
|
||||
resource: 'resource',
|
||||
position: 8,
|
||||
isRequired: false,
|
||||
isOptional: false
|
||||
})
|
||||
})
|
||||
|
||||
test('应该处理空内容', () => {
|
||||
// Arrange
|
||||
const content = ''
|
||||
|
||||
// Act
|
||||
const references = parser.extractReferencesWithPosition(content)
|
||||
|
||||
// Assert
|
||||
expect(references).toEqual([])
|
||||
})
|
||||
|
||||
test('应该处理没有引用的内容', () => {
|
||||
// Arrange
|
||||
const content = '这是一段没有任何引用的普通文本内容'
|
||||
|
||||
// Act
|
||||
const references = parser.extractReferencesWithPosition(content)
|
||||
|
||||
// Assert
|
||||
expect(references).toEqual([])
|
||||
})
|
||||
|
||||
test('应该处理多行内容中的引用位置', () => {
|
||||
// Arrange
|
||||
const content = `第一行内容
|
||||
@!thought://first
|
||||
第二行内容
|
||||
@!thought://second
|
||||
第三行内容`
|
||||
|
||||
// Act
|
||||
const references = parser.extractReferencesWithPosition(content)
|
||||
|
||||
// Assert
|
||||
expect(references).toHaveLength(2)
|
||||
expect(references[0].resource).toBe('first')
|
||||
expect(references[0].position).toBe(6) // 在"第一行内容\n"之后
|
||||
expect(references[1].resource).toBe('second')
|
||||
expect(references[1].position).toBeGreaterThan(references[0].position)
|
||||
})
|
||||
})
|
||||
|
||||
describe('parseTagContent - 位置信息集成', () => {
|
||||
test('应该在parseTagContent中包含位置信息', () => {
|
||||
// Arrange
|
||||
const content = 'intro @!thought://A middle @!thought://B end'
|
||||
|
||||
// Act
|
||||
const result = parser.parseTagContent(content, 'personality')
|
||||
|
||||
// Assert
|
||||
expect(result.references).toHaveLength(2)
|
||||
expect(result.references[0].position).toBe(6)
|
||||
expect(result.references[1].position).toBe(27)
|
||||
})
|
||||
})
|
||||
})
|
||||
236
src/tests/core/resource/DPMLContentParser.unit.test.js
Normal file
236
src/tests/core/resource/DPMLContentParser.unit.test.js
Normal file
@ -0,0 +1,236 @@
|
||||
const DPMLContentParser = require('../../../lib/core/resource/DPMLContentParser')
|
||||
|
||||
describe('DPMLContentParser 单元测试', () => {
|
||||
let parser
|
||||
|
||||
beforeEach(() => {
|
||||
parser = new DPMLContentParser()
|
||||
})
|
||||
|
||||
describe('基础功能测试', () => {
|
||||
test('应该能实例化DPMLContentParser', () => {
|
||||
expect(parser).toBeInstanceOf(DPMLContentParser)
|
||||
expect(typeof parser.parseTagContent).toBe('function')
|
||||
expect(typeof parser.extractReferences).toBe('function')
|
||||
expect(typeof parser.extractDirectContent).toBe('function')
|
||||
})
|
||||
})
|
||||
|
||||
describe('引用解析测试', () => {
|
||||
test('应该正确解析单个@引用', () => {
|
||||
const content = '@!thought://remember'
|
||||
const references = parser.extractReferences(content)
|
||||
|
||||
expect(references).toHaveLength(1)
|
||||
expect(references[0]).toEqual({
|
||||
fullMatch: '@!thought://remember',
|
||||
priority: '!',
|
||||
protocol: 'thought',
|
||||
resource: 'remember',
|
||||
isRequired: true,
|
||||
isOptional: false
|
||||
})
|
||||
})
|
||||
|
||||
test('应该正确解析多个@引用', () => {
|
||||
const content = `@!thought://remember
|
||||
@?execution://assistant
|
||||
@thought://recall`
|
||||
const references = parser.extractReferences(content)
|
||||
|
||||
expect(references).toHaveLength(3)
|
||||
expect(references[0].resource).toBe('remember')
|
||||
expect(references[1].resource).toBe('assistant')
|
||||
expect(references[2].resource).toBe('recall')
|
||||
expect(references[0].isRequired).toBe(true)
|
||||
expect(references[1].isOptional).toBe(true)
|
||||
expect(references[2].isRequired).toBe(false)
|
||||
})
|
||||
|
||||
test('应该处理没有@引用的内容', () => {
|
||||
const content = '# 这是直接内容\n- 列表项目'
|
||||
const references = parser.extractReferences(content)
|
||||
|
||||
expect(references).toHaveLength(0)
|
||||
})
|
||||
})
|
||||
|
||||
describe('直接内容提取测试', () => {
|
||||
test('应该正确提取纯直接内容', () => {
|
||||
const content = `# 网络杠精思维模式
|
||||
## 核心思维特征
|
||||
- 挑刺思维：看到任何观点都先找问题
|
||||
- 抬杠本能：天生反对派`
|
||||
|
||||
const directContent = parser.extractDirectContent(content)
|
||||
|
||||
expect(directContent).toContain('网络杠精思维模式')
|
||||
expect(directContent).toContain('挑刺思维')
|
||||
expect(directContent).toContain('抬杠本能')
|
||||
})
|
||||
|
||||
test('应该从混合内容中过滤掉@引用', () => {
|
||||
const content = `@!thought://remember
|
||||
|
||||
# 直接编写的个性特征
|
||||
- 特征1
|
||||
- 特征2
|
||||
|
||||
@!execution://assistant`
|
||||
|
||||
const directContent = parser.extractDirectContent(content)
|
||||
|
||||
expect(directContent).toContain('直接编写的个性特征')
|
||||
expect(directContent).toContain('特征1')
|
||||
expect(directContent).not.toContain('@!thought://remember')
|
||||
expect(directContent).not.toContain('@!execution://assistant')
|
||||
})
|
||||
|
||||
test('应该处理只有@引用没有直接内容的情况', () => {
|
||||
const content = '@!thought://remember\n@!execution://assistant'
|
||||
const directContent = parser.extractDirectContent(content)
|
||||
|
||||
expect(directContent).toBe('')
|
||||
})
|
||||
})
|
||||
|
||||
describe('标签内容解析测试', () => {
|
||||
test('应该解析混合内容标签', () => {
|
||||
const content = `@!thought://remember
|
||||
@!thought://recall
|
||||
|
||||
# 网络杠精思维模式
|
||||
## 核心思维特征
|
||||
- 挑刺思维：看到任何观点都先找问题和漏洞
|
||||
- 抬杠本能：天生反对派，习惯性质疑一切表述`
|
||||
|
||||
const result = parser.parseTagContent(content, 'personality')
|
||||
|
||||
expect(result.fullSemantics).toBe(content.trim())
|
||||
expect(result.references).toHaveLength(2)
|
||||
expect(result.references[0].resource).toBe('remember')
|
||||
expect(result.references[1].resource).toBe('recall')
|
||||
expect(result.directContent).toContain('网络杠精思维模式')
|
||||
expect(result.directContent).toContain('挑刺思维')
|
||||
expect(result.metadata.tagName).toBe('personality')
|
||||
expect(result.metadata.hasReferences).toBe(true)
|
||||
expect(result.metadata.hasDirectContent).toBe(true)
|
||||
expect(result.metadata.contentType).toBe('mixed')
|
||||
})
|
||||
|
||||
test('应该解析纯@引用标签', () => {
|
||||
const content = `@!thought://remember
|
||||
@!thought://assistant
|
||||
@!execution://assistant`
|
||||
|
||||
const result = parser.parseTagContent(content, 'personality')
|
||||
|
||||
expect(result.references).toHaveLength(3)
|
||||
expect(result.directContent).toBe('')
|
||||
expect(result.metadata.contentType).toBe('references-only')
|
||||
})
|
||||
|
||||
test('应该解析纯直接内容标签', () => {
|
||||
const content = `# 网络杠精思维模式
|
||||
## 核心思维特征
|
||||
- 挑刺思维：看到任何观点都先找问题和漏洞`
|
||||
|
||||
const result = parser.parseTagContent(content, 'personality')
|
||||
|
||||
expect(result.references).toHaveLength(0)
|
||||
expect(result.directContent).toContain('网络杠精思维模式')
|
||||
expect(result.metadata.contentType).toBe('direct-only')
|
||||
})
|
||||
|
||||
test('应该处理空标签', () => {
|
||||
const result = parser.parseTagContent('', 'personality')
|
||||
|
||||
expect(result.fullSemantics).toBe('')
|
||||
expect(result.references).toHaveLength(0)
|
||||
expect(result.directContent).toBe('')
|
||||
expect(result.metadata.contentType).toBe('empty')
|
||||
})
|
||||
})
|
||||
|
||||
describe('DPML文档解析测试', () => {
|
||||
test('应该从DPML文档中提取标签内容', () => {
|
||||
const dpmlContent = `<role>
|
||||
<personality>
|
||||
@!thought://remember
|
||||
# 个性特征
|
||||
</personality>
|
||||
<principle>
|
||||
@!execution://assistant
|
||||
# 行为原则
|
||||
</principle>
|
||||
</role>`
|
||||
|
||||
const personalityContent = parser.extractTagContent(dpmlContent, 'personality')
|
||||
const principleContent = parser.extractTagContent(dpmlContent, 'principle')
|
||||
|
||||
expect(personalityContent).toContain('@!thought://remember')
|
||||
expect(personalityContent).toContain('个性特征')
|
||||
expect(principleContent).toContain('@!execution://assistant')
|
||||
expect(principleContent).toContain('行为原则')
|
||||
})
|
||||
|
||||
test('应该解析完整的角色文档', () => {
|
||||
const roleContent = `<role>
|
||||
<personality>
|
||||
@!thought://remember
|
||||
# 杠精思维特征
|
||||
</personality>
|
||||
<principle>
|
||||
@!execution://assistant
|
||||
# 抬杠行为原则
|
||||
</principle>
|
||||
<knowledge>
|
||||
# 逻辑谬误知识体系
|
||||
</knowledge>
|
||||
</role>`
|
||||
|
||||
const roleSemantics = parser.parseRoleDocument(roleContent)
|
||||
|
||||
expect(roleSemantics).toHaveProperty('personality')
|
||||
expect(roleSemantics).toHaveProperty('principle')
|
||||
expect(roleSemantics).toHaveProperty('knowledge')
|
||||
|
||||
expect(roleSemantics.personality.metadata.contentType).toBe('mixed')
|
||||
expect(roleSemantics.principle.metadata.contentType).toBe('mixed')
|
||||
expect(roleSemantics.knowledge.metadata.contentType).toBe('direct-only')
|
||||
})
|
||||
})
|
||||
|
||||
describe('边界情况测试', () => {
|
||||
test('应该处理复杂的@引用格式', () => {
|
||||
const content = '@!protocol://complex-resource/with-path.execution'
|
||||
const references = parser.extractReferences(content)
|
||||
|
||||
expect(references).toHaveLength(1)
|
||||
expect(references[0].resource).toBe('complex-resource/with-path.execution')
|
||||
})
|
||||
|
||||
test('应该处理包含@符号但非引用的内容', () => {
|
||||
const content = '邮箱地址：user@example.com 不应该被识别为引用'
|
||||
const references = parser.extractReferences(content)
|
||||
|
||||
expect(references).toHaveLength(0)
|
||||
})
|
||||
|
||||
test('应该正确清理多余的空行', () => {
|
||||
const content = `@!thought://remember
|
||||
|
||||
|
||||
|
||||
# 标题
|
||||
|
||||
|
||||
|
||||
内容`
|
||||
|
||||
const directContent = parser.extractDirectContent(content)
|
||||
|
||||
expect(directContent).toBe('# 标题\n\n内容')
|
||||
})
|
||||
})
|
||||
})
|
||||
@ -144,15 +144,17 @@ describe('ResourceManager - 用户资源发现', () => {
|
||||
|
||||
describe('loadUnifiedRegistry', () => {
|
||||
it('应该合并系统资源和用户资源', async () => {
|
||||
// 模拟系统资源
|
||||
// 模拟系统资源（使用正确的registry格式）
|
||||
const mockSystemResources = {
|
||||
role: {
|
||||
'assistant': {
|
||||
file: '@package://prompt/domain/assistant/assistant.role.md',
|
||||
name: '🙋 智能助手',
|
||||
source: 'system',
|
||||
format: 'dpml',
|
||||
type: 'role'
|
||||
protocols: {
|
||||
role: {
|
||||
registry: {
|
||||
'assistant': {
|
||||
file: '@package://prompt/domain/assistant/assistant.role.md',
|
||||
name: '🙋 智能助手',
|
||||
description: '通用助理角色，提供基础的助理服务和记忆支持'
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -181,15 +183,17 @@ describe('ResourceManager - 用户资源发现', () => {
|
||||
})
|
||||
|
||||
it('应该让用户资源覆盖同名系统资源', async () => {
|
||||
// 模拟系统资源
|
||||
// 模拟系统资源（使用正确的registry格式）
|
||||
const mockSystemResources = {
|
||||
role: {
|
||||
'assistant': {
|
||||
file: '@package://prompt/domain/assistant/assistant.role.md',
|
||||
name: '🙋 智能助手',
|
||||
source: 'system',
|
||||
format: 'dpml',
|
||||
type: 'role'
|
||||
protocols: {
|
||||
role: {
|
||||
registry: {
|
||||
'assistant': {
|
||||
file: '@package://prompt/domain/assistant/assistant.role.md',
|
||||
name: '🙋 智能助手',
|
||||
description: '通用助理角色，提供基础的助理服务和记忆支持'
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
344
src/tests/core/resource/RoleDiscoveryEdgeCases.unit.test.js
Normal file
344
src/tests/core/resource/RoleDiscoveryEdgeCases.unit.test.js
Normal file
@ -0,0 +1,344 @@
|
||||
const fs = require('fs-extra')
|
||||
const path = require('path')
|
||||
const os = require('os')
|
||||
|
||||
const SimplifiedRoleDiscovery = require('../../../lib/core/resource/SimplifiedRoleDiscovery')
|
||||
|
||||
describe('Role Discovery Edge Cases', () => {
|
||||
let tempDir
|
||||
let testProjectDir
|
||||
let discovery
|
||||
|
||||
beforeEach(async () => {
|
||||
tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'role-discovery-edge-'))
|
||||
testProjectDir = path.join(tempDir, 'edge-test-project')
|
||||
|
||||
await fs.ensureDir(path.join(testProjectDir, '.promptx', 'resource', 'domain'))
|
||||
await fs.writeFile(
|
||||
path.join(testProjectDir, 'package.json'),
|
||||
JSON.stringify({ name: 'edge-test-project', version: '1.0.0' })
|
||||
)
|
||||
|
||||
jest.spyOn(process, 'cwd').mockReturnValue(testProjectDir)
|
||||
discovery = new SimplifiedRoleDiscovery()
|
||||
})
|
||||
|
||||
afterEach(async () => {
|
||||
if (tempDir) {
|
||||
await fs.remove(tempDir)
|
||||
}
|
||||
jest.restoreAllMocks()
|
||||
})
|
||||
|
||||
describe('Corrupted Role Files', () => {
|
||||
test('should handle role files with malformed DPML', async () => {
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'malformed-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
// Create role file with malformed DPML
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, 'malformed-role.role.md'),
|
||||
`# Malformed Role
|
||||
<role>
|
||||
<personality>
|
||||
Unclosed tag here
|
||||
</personalit
|
||||
<principle>
|
||||
Normal content
|
||||
</principle>
|
||||
</role>`
|
||||
)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
|
||||
// Should still discover the role (basic validation only checks for tags presence)
|
||||
expect(userRoles).toHaveProperty('malformed-role')
|
||||
})
|
||||
|
||||
test('should handle role files with missing required tags', async () => {
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'missing-tags')
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, 'missing-tags.role.md'),
|
||||
`# Missing Tags Role
|
||||
This file has no <role> tags at all.`
|
||||
)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).not.toHaveProperty('missing-tags')
|
||||
})
|
||||
|
||||
test('should handle empty role files', async () => {
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'empty-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
await fs.writeFile(path.join(roleDir, 'empty-role.role.md'), '')
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).not.toHaveProperty('empty-role')
|
||||
})
|
||||
|
||||
test('should handle role files with only whitespace', async () => {
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'whitespace-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, 'whitespace-role.role.md'),
|
||||
' \n\t \n '
|
||||
)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).not.toHaveProperty('whitespace-role')
|
||||
})
|
||||
})
|
||||
|
||||
describe('File System Edge Cases', () => {
|
||||
test('should handle permission denied errors gracefully', async () => {
|
||||
if (process.platform === 'win32') {
|
||||
// Skip permission tests on Windows
|
||||
return
|
||||
}
|
||||
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'permission-denied')
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
const roleFile = path.join(roleDir, 'permission-denied.role.md')
|
||||
await fs.writeFile(roleFile, '<role>test</role>')
|
||||
|
||||
// Remove read permissions
|
||||
await fs.chmod(roleFile, 0o000)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).not.toHaveProperty('permission-denied')
|
||||
|
||||
// Restore permissions for cleanup
|
||||
await fs.chmod(roleFile, 0o644)
|
||||
})
|
||||
|
||||
test('should handle directory symlinks correctly', async () => {
|
||||
if (process.platform === 'win32') {
|
||||
// Skip symlink tests on Windows (require admin privileges)
|
||||
return
|
||||
}
|
||||
|
||||
// Note: SimplifiedRoleDiscovery intentionally doesn't support symlinks for security
|
||||
// This test documents the expected behavior rather than testing it
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
|
||||
// SimplifiedRoleDiscovery doesn't follow symlinks by design
|
||||
expect(userRoles).toBeDefined()
|
||||
expect(typeof userRoles).toBe('object')
|
||||
})
|
||||
|
||||
test('should handle broken symlinks gracefully', async () => {
|
||||
if (process.platform === 'win32') {
|
||||
return
|
||||
}
|
||||
|
||||
// Create a symlink to a non-existent directory
|
||||
const brokenSymlink = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'broken-symlink')
|
||||
const nonExistentTarget = path.join(testProjectDir, 'non-existent-target')
|
||||
|
||||
await fs.symlink(nonExistentTarget, brokenSymlink)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).not.toHaveProperty('broken-symlink')
|
||||
})
|
||||
})
|
||||
|
||||
describe('Special Characters and Unicode', () => {
|
||||
test('should handle role names with special characters', async () => {
|
||||
const roleName = 'special-chars_123.test'
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', roleName)
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, `${roleName}.role.md`),
|
||||
'<role><personality>Special chars role</personality></role>'
|
||||
)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(Object.keys(userRoles)).toContain(roleName)
|
||||
expect(userRoles[roleName]).toBeDefined()
|
||||
})
|
||||
|
||||
test('should handle Unicode role names', async () => {
|
||||
const roleName = '测试角色'
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', roleName)
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, `${roleName}.role.md`),
|
||||
'<role><personality>Unicode role</personality></role>'
|
||||
)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).toHaveProperty(roleName)
|
||||
})
|
||||
|
||||
test('should handle roles with emoji in content', async () => {
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'emoji-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, 'emoji-role.role.md'),
|
||||
`# 🎭 Emoji Role
|
||||
> A role with emojis 🚀✨
|
||||
|
||||
<role>
|
||||
<personality>
|
||||
I love using emojis! 😄🎉
|
||||
</personality>
|
||||
</role>`
|
||||
)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).toHaveProperty('emoji-role')
|
||||
expect(userRoles['emoji-role'].name).toBe('🎭 Emoji Role')
|
||||
expect(userRoles['emoji-role'].description).toBe('A role with emojis 🚀✨')
|
||||
})
|
||||
})
|
||||
|
||||
describe('Concurrent Access', () => {
|
||||
test('should handle concurrent discovery calls safely', async () => {
|
||||
// Create test roles
|
||||
await createTestRole('concurrent-1')
|
||||
await createTestRole('concurrent-2')
|
||||
await createTestRole('concurrent-3')
|
||||
|
||||
// Start multiple discovery operations concurrently
|
||||
const discoveryPromises = [
|
||||
discovery.discoverUserRoles(),
|
||||
discovery.discoverUserRoles(),
|
||||
discovery.discoverUserRoles()
|
||||
]
|
||||
|
||||
const results = await Promise.all(discoveryPromises)
|
||||
|
||||
// All results should be consistent
|
||||
expect(results[0]).toEqual(results[1])
|
||||
expect(results[1]).toEqual(results[2])
|
||||
|
||||
// Should find all test roles
|
||||
expect(results[0]).toHaveProperty('concurrent-1')
|
||||
expect(results[0]).toHaveProperty('concurrent-2')
|
||||
expect(results[0]).toHaveProperty('concurrent-3')
|
||||
})
|
||||
})
|
||||
|
||||
describe('Large File Handling', () => {
|
||||
test('should handle very large role files', async () => {
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'large-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
// Create a large role file (1MB of content)
|
||||
const largeContent = 'A'.repeat(1024 * 1024)
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, 'large-role.role.md'),
|
||||
`<role><personality>${largeContent}</personality></role>`
|
||||
)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).toHaveProperty('large-role')
|
||||
})
|
||||
})
|
||||
|
||||
describe('Directory Structure Edge Cases', () => {
|
||||
test('should handle nested subdirectories gracefully', async () => {
|
||||
// Create deeply nested structure (should be ignored)
|
||||
const nestedDir = path.join(
|
||||
testProjectDir, '.promptx', 'resource', 'domain', 'nested',
|
||||
'very', 'deep', 'structure'
|
||||
)
|
||||
await fs.ensureDir(nestedDir)
|
||||
await fs.writeFile(
|
||||
path.join(nestedDir, 'deep.role.md'),
|
||||
'<role>deep</role>'
|
||||
)
|
||||
|
||||
// Also create a valid role at the correct level
|
||||
await createTestRole('valid-role')
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
|
||||
// Should find the valid role but ignore the deeply nested one
|
||||
expect(userRoles).toHaveProperty('valid-role')
|
||||
expect(userRoles).not.toHaveProperty('deep')
|
||||
})
|
||||
|
||||
test('should handle files instead of directories in domain folder', async () => {
|
||||
const domainPath = path.join(testProjectDir, '.promptx', 'resource', 'domain')
|
||||
|
||||
// Create a file directly in the domain folder (should be ignored)
|
||||
await fs.writeFile(
|
||||
path.join(domainPath, 'not-a-role-dir.md'),
|
||||
'<role>should be ignored</role>'
|
||||
)
|
||||
|
||||
// Create a valid role
|
||||
await createTestRole('valid-role')
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
|
||||
expect(userRoles).toHaveProperty('valid-role')
|
||||
expect(Object.keys(userRoles)).toHaveLength(1)
|
||||
})
|
||||
})
|
||||
|
||||
describe('Missing Registry File', () => {
|
||||
test('should handle missing system registry gracefully', async () => {
|
||||
// Mock fs.readJSON to simulate missing registry file
|
||||
const originalReadJSON = fs.readJSON
|
||||
fs.readJSON = jest.fn().mockRejectedValue(new Error('ENOENT: no such file'))
|
||||
|
||||
const systemRoles = await discovery.loadSystemRoles()
|
||||
expect(systemRoles).toEqual({})
|
||||
|
||||
// Restore original function
|
||||
fs.readJSON = originalReadJSON
|
||||
})
|
||||
|
||||
test('should handle corrupted registry file gracefully', async () => {
|
||||
const originalReadJSON = fs.readJSON
|
||||
fs.readJSON = jest.fn().mockRejectedValue(new Error('Unexpected token in JSON'))
|
||||
|
||||
const systemRoles = await discovery.loadSystemRoles()
|
||||
expect(systemRoles).toEqual({})
|
||||
|
||||
fs.readJSON = originalReadJSON
|
||||
})
|
||||
})
|
||||
|
||||
describe('Project Root Detection Edge Cases', () => {
|
||||
test('should handle projects without package.json', async () => {
|
||||
// Remove package.json
|
||||
await fs.remove(path.join(testProjectDir, 'package.json'))
|
||||
|
||||
// Should still work (fallback to current directory)
|
||||
await createTestRole('no-package-json')
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
|
||||
expect(userRoles).toHaveProperty('no-package-json')
|
||||
})
|
||||
|
||||
test('should handle project root at filesystem root', async () => {
|
||||
// Mock process.cwd to return root directory
|
||||
jest.spyOn(process, 'cwd').mockReturnValue(path.parse(process.cwd()).root)
|
||||
|
||||
// Should not crash
|
||||
const userPath = await discovery.getUserRolePath()
|
||||
expect(userPath).toBeDefined()
|
||||
})
|
||||
})
|
||||
|
||||
// Helper function to create a test role
|
||||
async function createTestRole(roleName) {
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', roleName)
|
||||
await fs.ensureDir(roleDir)
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, `${roleName}.role.md`),
|
||||
`<role><personality>${roleName} personality</personality></role>`
|
||||
)
|
||||
}
|
||||
})
|
||||
222
src/tests/core/resource/RoleDiscoveryPerformance.unit.test.js
Normal file
222
src/tests/core/resource/RoleDiscoveryPerformance.unit.test.js
Normal file
@ -0,0 +1,222 @@
|
||||
const fs = require('fs-extra')
|
||||
const path = require('path')
|
||||
const os = require('os')
|
||||
|
||||
const SimplifiedRoleDiscovery = require('../../../lib/core/resource/SimplifiedRoleDiscovery')
|
||||
|
||||
describe('Role Discovery Performance Benchmarks', () => {
|
||||
let tempDir
|
||||
let testProjectDir
|
||||
let discovery
|
||||
|
||||
beforeEach(async () => {
|
||||
tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'role-discovery-perf-'))
|
||||
testProjectDir = path.join(tempDir, 'perf-test-project')
|
||||
|
||||
await fs.ensureDir(path.join(testProjectDir, '.promptx', 'resource', 'domain'))
|
||||
await fs.writeFile(
|
||||
path.join(testProjectDir, 'package.json'),
|
||||
JSON.stringify({ name: 'perf-test-project', version: '1.0.0' })
|
||||
)
|
||||
|
||||
jest.spyOn(process, 'cwd').mockReturnValue(testProjectDir)
|
||||
discovery = new SimplifiedRoleDiscovery()
|
||||
})
|
||||
|
||||
afterEach(async () => {
|
||||
if (tempDir) {
|
||||
await fs.remove(tempDir)
|
||||
}
|
||||
jest.restoreAllMocks()
|
||||
})
|
||||
|
||||
describe('Scaling Performance Tests', () => {
|
||||
test('should discover 10 roles in under 50ms', async () => {
|
||||
await createMultipleTestRoles(10)
|
||||
|
||||
const startTime = process.hrtime.bigint()
|
||||
const roles = await discovery.discoverAllRoles()
|
||||
const endTime = process.hrtime.bigint()
|
||||
|
||||
const durationMs = Number(endTime - startTime) / 1000000
|
||||
|
||||
expect(Object.keys(roles).length).toBeGreaterThanOrEqual(10) // System + user roles
|
||||
expect(durationMs).toBeLessThan(50)
|
||||
})
|
||||
|
||||
test('should discover 50 roles in under 100ms', async () => {
|
||||
await createMultipleTestRoles(50)
|
||||
|
||||
const startTime = process.hrtime.bigint()
|
||||
const roles = await discovery.discoverAllRoles()
|
||||
const endTime = process.hrtime.bigint()
|
||||
|
||||
const durationMs = Number(endTime - startTime) / 1000000
|
||||
|
||||
expect(Object.keys(roles).length).toBeGreaterThanOrEqual(50)
|
||||
expect(durationMs).toBeLessThan(100)
|
||||
})
|
||||
|
||||
test('should discover 100 roles in under 150ms', async () => {
|
||||
await createMultipleTestRoles(100)
|
||||
|
||||
const startTime = process.hrtime.bigint()
|
||||
const roles = await discovery.discoverAllRoles()
|
||||
const endTime = process.hrtime.bigint()
|
||||
|
||||
const durationMs = Number(endTime - startTime) / 1000000
|
||||
|
||||
expect(Object.keys(roles).length).toBeGreaterThanOrEqual(100)
|
||||
expect(durationMs).toBeLessThan(150)
|
||||
})
|
||||
})
|
||||
|
||||
describe('Parallel vs Sequential Processing', () => {
|
||||
test('parallel discovery should be faster than sequential', async () => {
|
||||
const roleCount = 50 // 增加角色数量以放大差异
|
||||
await createMultipleTestRoles(roleCount)
|
||||
|
||||
// 多次运行取平均值，减少测试波动
|
||||
const runs = 3
|
||||
let parallelTotal = 0
|
||||
let sequentialTotal = 0
|
||||
|
||||
for (let i = 0; i < runs; i++) {
|
||||
// Test parallel discovery (our implementation)
|
||||
const parallelStart = process.hrtime.bigint()
|
||||
await discovery.discoverUserRoles()
|
||||
const parallelEnd = process.hrtime.bigint()
|
||||
parallelTotal += Number(parallelEnd - parallelStart) / 1000000
|
||||
|
||||
// Test sequential discovery (simulated)
|
||||
const sequentialStart = process.hrtime.bigint()
|
||||
await simulateSequentialDiscovery(roleCount)
|
||||
const sequentialEnd = process.hrtime.bigint()
|
||||
sequentialTotal += Number(sequentialEnd - sequentialStart) / 1000000
|
||||
}
|
||||
|
||||
const parallelAvg = parallelTotal / runs
|
||||
const sequentialAvg = sequentialTotal / runs
|
||||
|
||||
// 放宽条件：并行应该比串行快，或者至少不慢太多
|
||||
expect(parallelAvg).toBeLessThan(sequentialAvg * 1.2) // 允许20%的误差
|
||||
})
|
||||
})
|
||||
|
||||
describe('Memory Usage Tests', () => {
|
||||
test('should not accumulate excessive memory with large role sets', async () => {
|
||||
const initialMemory = process.memoryUsage().heapUsed
|
||||
|
||||
// Create and discover many roles
|
||||
await createMultipleTestRoles(100)
|
||||
await discovery.discoverAllRoles()
|
||||
|
||||
const finalMemory = process.memoryUsage().heapUsed
|
||||
const memoryIncrease = finalMemory - initialMemory
|
||||
|
||||
// Memory increase should be reasonable (less than 50MB for 100 roles)
|
||||
expect(memoryIncrease).toBeLessThan(50 * 1024 * 1024)
|
||||
})
|
||||
})
|
||||
|
||||
describe('File System Optimization Tests', () => {
|
||||
test('should minimize file system calls', async () => {
|
||||
await createMultipleTestRoles(10)
|
||||
|
||||
// Spy on file system operations
|
||||
const statSpy = jest.spyOn(fs, 'stat')
|
||||
const pathExistsSpy = jest.spyOn(fs, 'pathExists')
|
||||
const readFileSpy = jest.spyOn(fs, 'readFile')
|
||||
const readdirSpy = jest.spyOn(fs, 'readdir')
|
||||
|
||||
await discovery.discoverUserRoles()
|
||||
|
||||
// Should use readdir with withFileTypes to minimize stat calls
|
||||
expect(readdirSpy).toHaveBeenCalled()
|
||||
|
||||
// Should minimize individual stat and pathExists calls through optimization
|
||||
const totalFsCalls = statSpy.mock.calls.length +
|
||||
pathExistsSpy.mock.calls.length +
|
||||
readFileSpy.mock.calls.length
|
||||
|
||||
expect(totalFsCalls).toBeLessThan(25) // Should be efficient with batch operations
|
||||
|
||||
statSpy.mockRestore()
|
||||
pathExistsSpy.mockRestore()
|
||||
readFileSpy.mockRestore()
|
||||
readdirSpy.mockRestore()
|
||||
})
|
||||
})
|
||||
|
||||
describe('Caching Performance (Future Enhancement)', () => {
|
||||
test('should be ready for caching implementation', async () => {
|
||||
await createMultipleTestRoles(20)
|
||||
|
||||
// First discovery
|
||||
const firstStart = process.hrtime.bigint()
|
||||
const firstResult = await discovery.discoverAllRoles()
|
||||
const firstEnd = process.hrtime.bigint()
|
||||
const firstDuration = Number(firstEnd - firstStart) / 1000000
|
||||
|
||||
// Second discovery (cache would help here)
|
||||
const secondStart = process.hrtime.bigint()
|
||||
const secondResult = await discovery.discoverAllRoles()
|
||||
const secondEnd = process.hrtime.bigint()
|
||||
const secondDuration = Number(secondEnd - secondStart) / 1000000
|
||||
|
||||
// Results should be consistent
|
||||
expect(Object.keys(firstResult)).toEqual(Object.keys(secondResult))
|
||||
|
||||
// Both should be reasonably fast (caching would make second faster)
|
||||
expect(firstDuration).toBeLessThan(100)
|
||||
expect(secondDuration).toBeLessThan(100)
|
||||
})
|
||||
})
|
||||
|
||||
// Helper function to create multiple test roles
|
||||
async function createMultipleTestRoles(count) {
|
||||
const promises = []
|
||||
|
||||
for (let i = 0; i < count; i++) {
|
||||
const roleName = `perf-test-role-${i.toString().padStart(3, '0')}`
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', roleName)
|
||||
|
||||
promises.push(
|
||||
fs.ensureDir(roleDir).then(() =>
|
||||
fs.writeFile(
|
||||
path.join(roleDir, `${roleName}.role.md`),
|
||||
`# Performance Test Role ${i}
|
||||
> Role created for performance testing
|
||||
|
||||
<role>
|
||||
<personality>
|
||||
Performance test personality for role ${i}
|
||||
</personality>
|
||||
<principle>
|
||||
Performance test principle for role ${i}
|
||||
</principle>
|
||||
<knowledge>
|
||||
Performance test knowledge for role ${i}
|
||||
</knowledge>
|
||||
</role>`
|
||||
)
|
||||
)
|
||||
)
|
||||
}
|
||||
|
||||
await Promise.all(promises)
|
||||
}
|
||||
|
||||
// Simulate sequential discovery for comparison
|
||||
async function simulateSequentialDiscovery(count) {
|
||||
const userPath = path.join(testProjectDir, '.promptx', 'resource', 'domain')
|
||||
const directories = await fs.readdir(userPath)
|
||||
|
||||
for (const dir of directories) {
|
||||
const roleFile = path.join(userPath, dir, `${dir}.role.md`)
|
||||
if (await fs.pathExists(roleFile)) {
|
||||
await fs.readFile(roleFile, 'utf8')
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
309
src/tests/core/resource/SimplifiedRoleDiscovery.unit.test.js
Normal file
309
src/tests/core/resource/SimplifiedRoleDiscovery.unit.test.js
Normal file
@ -0,0 +1,309 @@
|
||||
const fs = require('fs-extra')
|
||||
const path = require('path')
|
||||
const os = require('os')
|
||||
|
||||
// This will be the implementation we're building towards
|
||||
const SimplifiedRoleDiscovery = require('../../../lib/core/resource/SimplifiedRoleDiscovery')
|
||||
|
||||
describe('SimplifiedRoleDiscovery - TDD Implementation', () => {
|
||||
let tempDir
|
||||
let testProjectDir
|
||||
let discovery
|
||||
|
||||
beforeEach(async () => {
|
||||
// Create temporary test environment
|
||||
tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'simplified-role-discovery-'))
|
||||
testProjectDir = path.join(tempDir, 'test-project')
|
||||
|
||||
// Create test project structure
|
||||
await fs.ensureDir(path.join(testProjectDir, '.promptx', 'resource', 'domain'))
|
||||
await fs.writeFile(
|
||||
path.join(testProjectDir, 'package.json'),
|
||||
JSON.stringify({ name: 'test-project', version: '1.0.0' })
|
||||
)
|
||||
|
||||
// Mock process.cwd to point to our test project
|
||||
jest.spyOn(process, 'cwd').mockReturnValue(testProjectDir)
|
||||
|
||||
discovery = new SimplifiedRoleDiscovery()
|
||||
})
|
||||
|
||||
afterEach(async () => {
|
||||
if (tempDir) {
|
||||
await fs.remove(tempDir)
|
||||
}
|
||||
jest.restoreAllMocks()
|
||||
})
|
||||
|
||||
describe('Core Algorithm API', () => {
|
||||
test('should expose discoverAllRoles method', () => {
|
||||
expect(typeof discovery.discoverAllRoles).toBe('function')
|
||||
})
|
||||
|
||||
test('should expose loadSystemRoles method', () => {
|
||||
expect(typeof discovery.loadSystemRoles).toBe('function')
|
||||
})
|
||||
|
||||
test('should expose discoverUserRoles method', () => {
|
||||
expect(typeof discovery.discoverUserRoles).toBe('function')
|
||||
})
|
||||
|
||||
test('should expose mergeRoles method', () => {
|
||||
expect(typeof discovery.mergeRoles).toBe('function')
|
||||
})
|
||||
})
|
||||
|
||||
describe('System Roles Loading', () => {
|
||||
test('should load system roles from static registry', async () => {
|
||||
const systemRoles = await discovery.loadSystemRoles()
|
||||
|
||||
expect(systemRoles).toBeDefined()
|
||||
expect(typeof systemRoles).toBe('object')
|
||||
|
||||
// Should contain known system roles
|
||||
expect(systemRoles).toHaveProperty('assistant')
|
||||
expect(systemRoles.assistant).toHaveProperty('name')
|
||||
expect(systemRoles.assistant).toHaveProperty('file')
|
||||
expect(systemRoles.assistant.name).toContain('智能助手')
|
||||
})
|
||||
|
||||
test('should handle missing registry file gracefully', async () => {
|
||||
// Mock missing registry file
|
||||
const originalReadJSON = fs.readJSON
|
||||
fs.readJSON = jest.fn().mockRejectedValue(new Error('File not found'))
|
||||
|
||||
const systemRoles = await discovery.loadSystemRoles()
|
||||
expect(systemRoles).toEqual({})
|
||||
|
||||
fs.readJSON = originalReadJSON
|
||||
})
|
||||
})
|
||||
|
||||
describe('User Roles Discovery', () => {
|
||||
test('should return empty object when user directory does not exist', async () => {
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).toEqual({})
|
||||
})
|
||||
|
||||
test('should discover valid user role', async () => {
|
||||
// Create test user role
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'test-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, 'test-role.role.md'),
|
||||
`# Test Role
|
||||
> A test role for unit testing
|
||||
|
||||
<role>
|
||||
<personality>
|
||||
Test personality
|
||||
</personality>
|
||||
<principle>
|
||||
Test principle
|
||||
</principle>
|
||||
<knowledge>
|
||||
Test knowledge
|
||||
</knowledge>
|
||||
</role>`
|
||||
)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
|
||||
expect(userRoles).toHaveProperty('test-role')
|
||||
expect(userRoles['test-role']).toHaveProperty('name', 'Test Role')
|
||||
expect(userRoles['test-role']).toHaveProperty('description', 'A test role for unit testing')
|
||||
expect(userRoles['test-role']).toHaveProperty('source', 'user-generated')
|
||||
expect(userRoles['test-role']).toHaveProperty('file')
|
||||
})
|
||||
|
||||
test('should skip invalid role files', async () => {
|
||||
// Create invalid role file (missing <role> tags)
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'invalid-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, 'invalid-role.role.md'),
|
||||
'This is not a valid role file'
|
||||
)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).not.toHaveProperty('invalid-role')
|
||||
})
|
||||
|
||||
test('should handle missing role file gracefully', async () => {
|
||||
// Create directory but no role file
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'empty-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).not.toHaveProperty('empty-role')
|
||||
})
|
||||
|
||||
test('should handle file system errors gracefully', async () => {
|
||||
// Create a role directory with permission issues (Unix only)
|
||||
if (process.platform !== 'win32') {
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'restricted-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
const roleFile = path.join(roleDir, 'restricted-role.role.md')
|
||||
await fs.writeFile(roleFile, '<role>test</role>')
|
||||
await fs.chmod(roleFile, 0o000) // Remove all permissions
|
||||
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).not.toHaveProperty('restricted-role')
|
||||
|
||||
// Restore permissions for cleanup
|
||||
await fs.chmod(roleFile, 0o644)
|
||||
} else {
|
||||
// On Windows, just test that the method doesn't throw
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
expect(userRoles).toBeDefined()
|
||||
}
|
||||
})
|
||||
})
|
||||
|
||||
describe('Parallel Discovery Performance', () => {
|
||||
test('should process multiple user roles in parallel', async () => {
|
||||
const roleCount = 10
|
||||
const createRolePromises = []
|
||||
|
||||
// Create multiple test roles
|
||||
for (let i = 0; i < roleCount; i++) {
|
||||
const roleName = `test-role-${i}`
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', roleName)
|
||||
|
||||
createRolePromises.push(
|
||||
fs.ensureDir(roleDir).then(() =>
|
||||
fs.writeFile(
|
||||
path.join(roleDir, `${roleName}.role.md`),
|
||||
`<role><personality>Role ${i}</personality></role>`
|
||||
)
|
||||
)
|
||||
)
|
||||
}
|
||||
|
||||
await Promise.all(createRolePromises)
|
||||
|
||||
const startTime = Date.now()
|
||||
const userRoles = await discovery.discoverUserRoles()
|
||||
const endTime = Date.now()
|
||||
|
||||
expect(Object.keys(userRoles)).toHaveLength(roleCount)
|
||||
expect(endTime - startTime).toBeLessThan(100) // Should be fast with parallel processing
|
||||
})
|
||||
})
|
||||
|
||||
describe('Role Merging', () => {
|
||||
test('should merge system and user roles correctly', () => {
|
||||
const systemRoles = {
|
||||
'assistant': { name: 'System Assistant', source: 'system' },
|
||||
'system-only': { name: 'System Only', source: 'system' }
|
||||
}
|
||||
|
||||
const userRoles = {
|
||||
'assistant': { name: 'User Assistant', source: 'user' },
|
||||
'user-only': { name: 'User Only', source: 'user' }
|
||||
}
|
||||
|
||||
const merged = discovery.mergeRoles(systemRoles, userRoles)
|
||||
|
||||
expect(merged).toHaveProperty('assistant')
|
||||
expect(merged).toHaveProperty('system-only')
|
||||
expect(merged).toHaveProperty('user-only')
|
||||
|
||||
// User role should override system role
|
||||
expect(merged.assistant.source).toBe('user')
|
||||
expect(merged.assistant.name).toBe('User Assistant')
|
||||
|
||||
// System-only role should remain
|
||||
expect(merged['system-only'].source).toBe('system')
|
||||
|
||||
// User-only role should be included
|
||||
expect(merged['user-only'].source).toBe('user')
|
||||
})
|
||||
|
||||
test('should handle empty input gracefully', () => {
|
||||
expect(discovery.mergeRoles({}, {})).toEqual({})
|
||||
expect(discovery.mergeRoles({ test: 'value' }, {})).toEqual({ test: 'value' })
|
||||
expect(discovery.mergeRoles({}, { test: 'value' })).toEqual({ test: 'value' })
|
||||
})
|
||||
})
|
||||
|
||||
describe('Complete Discovery Flow', () => {
|
||||
test('should discover all roles (system + user)', async () => {
|
||||
// Create a test user role
|
||||
const roleDir = path.join(testProjectDir, '.promptx', 'resource', 'domain', 'custom-role')
|
||||
await fs.ensureDir(roleDir)
|
||||
await fs.writeFile(
|
||||
path.join(roleDir, 'custom-role.role.md'),
|
||||
'<role><personality>Custom role</personality></role>'
|
||||
)
|
||||
|
||||
const allRoles = await discovery.discoverAllRoles()
|
||||
|
||||
expect(allRoles).toBeDefined()
|
||||
expect(typeof allRoles).toBe('object')
|
||||
|
||||
// Should contain system roles
|
||||
expect(allRoles).toHaveProperty('assistant')
|
||||
|
||||
// Should contain user role
|
||||
expect(allRoles).toHaveProperty('custom-role')
|
||||
expect(allRoles['custom-role'].source).toBe('user-generated')
|
||||
})
|
||||
})
|
||||
|
||||
describe('DPML Validation', () => {
|
||||
test('should validate basic DPML format', () => {
|
||||
const validContent = '<role><personality>test</personality></role>'
|
||||
const invalidContent = 'no role tags here'
|
||||
|
||||
expect(discovery.isValidRoleFile(validContent)).toBe(true)
|
||||
expect(discovery.isValidRoleFile(invalidContent)).toBe(false)
|
||||
})
|
||||
|
||||
test('should extract role name from markdown header', () => {
|
||||
const content = `# My Custom Role
|
||||
<role>content</role>`
|
||||
|
||||
expect(discovery.extractRoleName(content)).toBe('My Custom Role')
|
||||
})
|
||||
|
||||
test('should extract description from markdown quote', () => {
|
||||
const content = `# Role Name
|
||||
> This is the role description
|
||||
<role>content</role>`
|
||||
|
||||
expect(discovery.extractDescription(content)).toBe('This is the role description')
|
||||
})
|
||||
|
||||
test('should handle missing metadata gracefully', () => {
|
||||
const content = '<role>content</role>'
|
||||
|
||||
expect(discovery.extractRoleName(content)).toBeNull()
|
||||
expect(discovery.extractDescription(content)).toBeNull()
|
||||
})
|
||||
})
|
||||
|
||||
describe('Cross-platform Path Handling', () => {
|
||||
test('should handle Windows paths correctly', () => {
|
||||
const originalPlatform = process.platform
|
||||
Object.defineProperty(process, 'platform', { value: 'win32' })
|
||||
|
||||
// Test that path operations work correctly on Windows
|
||||
const userPath = discovery.getUserRolePath()
|
||||
expect(userPath).toBeDefined()
|
||||
|
||||
Object.defineProperty(process, 'platform', { value: originalPlatform })
|
||||
})
|
||||
|
||||
test('should handle Unix paths correctly', () => {
|
||||
const originalPlatform = process.platform
|
||||
Object.defineProperty(process, 'platform', { value: 'linux' })
|
||||
|
||||
// Test that path operations work correctly on Unix
|
||||
const userPath = discovery.getUserRolePath()
|
||||
expect(userPath).toBeDefined()
|
||||
|
||||
Object.defineProperty(process, 'platform', { value: originalPlatform })
|
||||
})
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user