dpml-prompt/src/tests/core/resource/resourceRegistry.unit.test.js

DPML-powered AI prompt framework - Revolutionary AI-First CLI system based on Deepractice Prompt Markup Language. Build sophisticated AI agents with structured prompts, memory systems, and execution frameworks.

const ResourceRegistry = require('../../../lib/core/resource/resourceRegistry')
const { ProtocolInfo } = require('../../../lib/core/resource/types')

describe('ResourceRegistry - Unit Tests', () => {
  let registry

  beforeEach(() => {
    registry = new ResourceRegistry()
  })

  describe('内置协议', () => {
    test('应该包含内置协议', () => {
      const protocols = registry.listProtocols()

      expect(protocols).toContain('prompt')
      expect(protocols).toContain('file')
      expect(protocols).toContain('memory')
    })

    test('应该正确获取prompt协议信息', () => {
      const protocolInfo = registry.getProtocolInfo('prompt')

      expect(protocolInfo).toBeDefined()
      expect(protocolInfo.name).toBe('prompt')
      expect(protocolInfo.description).toContain('PromptX内置提示词资源协议')
      expect(protocolInfo.location).toContain('prompt://')
    })

    test('应该为协议提供资源注册表', () => {
      const protocolInfo = registry.getProtocolInfo('memory')

      expect(protocolInfo.registry).toBeDefined()
      expect(protocolInfo.registry.size).toBeGreaterThan(0)
      expect(protocolInfo.registry.has('declarative')).toBe(true)
      expect(protocolInfo.registry.has('procedural')).toBe(true)
    })
  })

  describe('资源解析', () => {
    test('应该解析prompt协议的资源ID', () => {
      const resolved = registry.resolve('prompt', 'protocols')

      expect(resolved).toBe('@package://prompt/protocol/**/*.md')
    })

    test('应该解析memory协议的资源ID', () => {
      const resolved = registry.resolve('memory', 'declarative')

      expect(resolved).toBe('@project://.promptx/memory/declarative.md')
    })

    test('应该解析未注册协议的资源路径', () => {
      const resolved = registry.resolve('file', 'any/path.md')

      expect(resolved).toBe('any/path.md')
    })

    test('应该在资源ID不存在时抛出错误', () => {
      expect(() => registry.resolve('prompt', 'nonexistent')).toThrow('Resource ID \'nonexistent\' not found in prompt protocol registry')
    })
  })

  describe('自定义协议注册', () => {
    test('应该注册新的自定义协议', () => {
      const customProtocol = {
        description: '测试协议',
        location: 'test://{resource_id}',
        registry: {
          test1: '@file://test1.md',
          test2: '@file://test2.md'
        }
      }

      registry.register('test', customProtocol)

      expect(registry.hasProtocol('test')).toBe(true)
      expect(registry.resolve('test', 'test1')).toBe('@file://test1.md')
    })

    test('应该列出自定义协议的资源', () => {
      const customProtocol = {
        registry: {
          resource1: '@file://r1.md',
          resource2: '@file://r2.md'
        }
      }

      registry.register('custom', customProtocol)
      const resources = registry.listProtocolResources('custom')

      expect(resources).toContain('resource1')
      expect(resources).toContain('resource2')
    })
  })

  describe('验证功能', () => {
    test('应该验证有效的协议和资源ID', () => {
      expect(registry.validateReference('prompt', 'protocols')).toBe(true)
      expect(registry.validateReference('file', 'any-path.md')).toBe(true)
      expect(registry.validateReference('memory', 'declarative')).toBe(true)
    })

    test('应该拒绝无效的协议和资源ID', () => {
      expect(registry.validateReference('unknown', 'test')).toBe(false)
      expect(registry.validateReference('prompt', 'nonexistent')).toBe(false)
    })
  })

  describe('注册表信息', () => {
    test('应该返回完整的注册表信息', () => {
      const info = registry.getRegistryInfo()

      expect(info.builtin).toHaveProperty('prompt')
      expect(info.builtin).toHaveProperty('file')
      expect(info.builtin).toHaveProperty('memory')
      expect(info.custom).toEqual({})
    })

    test('应该返回协议的资源列表', () => {
      const resources = registry.listProtocolResources('prompt')

      expect(resources).toContain('protocols')
      expect(resources).toContain('core')
      expect(resources).toContain('domain')
      expect(resources).toContain('bootstrap')
    })

    test('应该为无注册表的协议返回空列表', () => {
      const resources = registry.listProtocolResources('file')

      expect(resources).toEqual([])
    })
  })
})