@deimoscloud/coreai 0.1.0

package/src/agents/compiler.test.ts

@@ -0,0 +1,399 @@
+import { mkdtempSync, mkdirSync, writeFileSync, rmSync, readFileSync, existsSync } from 'fs';
+import { join, dirname } from 'path';
+import { tmpdir } from 'os';
+import { fileURLToPath } from 'url';
+import {
+  generateAgentMarkdown,
+  compileAgent,
+  loadAllAgents,
+  filterAgentsByTeam,
+  compileAgents,
+} from './compiler.js';
+import type { AgentDefinition, AgentMetadata } from './types.js';
+import type { CoreAIConfig } from '../config/types.js';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+describe('Agent Compiler', () => {
+  let tempDir: string;
+
+  beforeEach(() => {
+    tempDir = mkdtempSync(join(tmpdir(), 'coreai-compiler-test-'));
+  });
+
+  afterEach(() => {
+    rmSync(tempDir, { recursive: true, force: true });
+  });
+
+  const minimalAgent: AgentDefinition = {
+    role: 'test-agent',
+    type: 'ic-engineer',
+    display_name: 'Test Agent',
+    description: 'A test agent for unit testing.',
+  };
+
+  const fullAgent: AgentDefinition = {
+    role: 'backend-engineer',
+    type: 'ic-engineer',
+    display_name: 'Backend Engineer',
+    description: 'Senior backend engineer specializing in API design.',
+    responsibilities: ['Design APIs', 'Write tests', 'Review code'],
+    expertise: {
+      primary: ['API design', 'Database design'],
+      tech_stack: 'TypeScript, Node.js',
+    },
+    skills: ['Code review', 'Debugging'],
+    principles: {
+      code_quality: ['Write clean code', 'Follow SOLID'],
+      testing: ['Test first', 'High coverage'],
+    },
+    behaviors: {
+      workflow: 'ticket-implementation',
+      quality_gates: 'lint, test, build',
+    },
+    context_sources: {
+      shared: ['docs/architecture'],
+      personal: ['KnowledgeLibrary/backend-engineer/context'],
+    },
+    communication: {
+      inbox: 'KnowledgeLibrary/backend-engineer/inbox',
+      outbox: 'KnowledgeLibrary/backend-engineer/outbox',
+    },
+  };
+
+  describe('generateAgentMarkdown', () => {
+    it('should generate markdown for minimal agent', () => {
+      const markdown = generateAgentMarkdown(minimalAgent);
+
+      expect(markdown).toContain('# Test Agent');
+      expect(markdown).toContain('**Role:** test-agent');
+      expect(markdown).toContain('**Type:** ic-engineer');
+      expect(markdown).toContain('## Description');
+      expect(markdown).toContain('A test agent for unit testing.');
+      expect(markdown).toContain('*Generated by CoreAI*');
+    });
+
+    it('should generate markdown for full agent', () => {
+      const markdown = generateAgentMarkdown(fullAgent);
+
+      // Header
+      expect(markdown).toContain('# Backend Engineer');
+      expect(markdown).toContain('**Role:** backend-engineer');
+
+      // Responsibilities
+      expect(markdown).toContain('## Responsibilities');
+      expect(markdown).toContain('- Design APIs');
+      expect(markdown).toContain('- Write tests');
+
+      // Expertise
+      expect(markdown).toContain('## Expertise');
+      expect(markdown).toContain('### Primary Areas');
+      expect(markdown).toContain('- API design');
+      expect(markdown).toContain('### Tech Stack');
+      expect(markdown).toContain('TypeScript, Node.js');
+
+      // Skills
+      expect(markdown).toContain('## Skills');
+      expect(markdown).toContain('- Code review');
+
+      // Principles
+      expect(markdown).toContain('## Principles');
+      expect(markdown).toContain('### Code Quality');
+      expect(markdown).toContain('- Write clean code');
+      expect(markdown).toContain('### Testing');
+      expect(markdown).toContain('- Test first');
+
+      // Behaviors
+      expect(markdown).toContain('## Behaviors');
+      expect(markdown).toContain('**Workflow:** ticket-implementation');
+
+      // Context Sources
+      expect(markdown).toContain('## Context Sources');
+      expect(markdown).toContain('### Shared');
+      expect(markdown).toContain('- docs/architecture');
+      expect(markdown).toContain('### Personal');
+
+      // Communication
+      expect(markdown).toContain('## Communication');
+      expect(markdown).toContain('**Inbox:**');
+      expect(markdown).toContain('**Outbox:**');
+    });
+
+    it('should format snake_case titles correctly', () => {
+      const markdown = generateAgentMarkdown(fullAgent);
+      expect(markdown).toContain('### Code Quality');
+    });
+  });
+
+  describe('compileAgent', () => {
+    it('should compile agent with variable resolution', () => {
+      const agent: AgentDefinition = {
+        role: 'test-agent',
+        type: 'ic-engineer',
+        display_name: 'Test Agent',
+        description: 'Agent for ${config.project.name}',
+      };
+
+      const config: CoreAIConfig = {
+        version: '1.0',
+        project: { name: 'my-project' },
+      };
+
+      const markdown = compileAgent(agent, config);
+
+      expect(markdown).toContain('Agent for my-project');
+    });
+
+    it('should compile agent without config', () => {
+      const markdown = compileAgent(minimalAgent);
+      expect(markdown).toContain('# Test Agent');
+    });
+  });
+
+  describe('loadAllAgents', () => {
+    it('should load agents from core directory', () => {
+      // Create core agents
+      const coreDir = join(tempDir, 'core');
+      mkdirSync(coreDir);
+      writeFileSync(
+        join(coreDir, 'agent-one.yaml'),
+        `
+role: agent-one
+type: ic-engineer
+display_name: Agent One
+description: First agent
+`
+      );
+
+      const agents = loadAllAgents({ coreAgentsDir: coreDir });
+
+      expect(agents.size).toBe(1);
+      expect(agents.get('agent-one')?.source).toBe('core');
+    });
+
+    it('should load agents from custom directory', () => {
+      const customDir = join(tempDir, 'custom');
+      mkdirSync(customDir);
+      writeFileSync(
+        join(customDir, 'custom-agent.yaml'),
+        `
+role: custom-agent
+type: specialist
+display_name: Custom Agent
+description: A custom agent
+`
+      );
+
+      const agents = loadAllAgents({ customAgentsDir: customDir });
+
+      expect(agents.size).toBe(1);
+      expect(agents.get('custom-agent')?.source).toBe('custom');
+    });
+
+    it('should mark overridden agents correctly', () => {
+      const coreDir = join(tempDir, 'core');
+      const customDir = join(tempDir, 'custom');
+      mkdirSync(coreDir);
+      mkdirSync(customDir);
+
+      // Core agent
+      writeFileSync(
+        join(coreDir, 'shared-agent.yaml'),
+        `
+role: shared-agent
+type: ic-engineer
+display_name: Core Version
+description: Core version of agent
+`
+      );
+
+      // Custom override
+      writeFileSync(
+        join(customDir, 'shared-agent.yaml'),
+        `
+role: shared-agent
+type: ic-engineer
+display_name: Custom Version
+description: Custom version of agent
+`
+      );
+
+      const agents = loadAllAgents({
+        coreAgentsDir: coreDir,
+        customAgentsDir: customDir,
+      });
+
+      expect(agents.size).toBe(1);
+      expect(agents.get('shared-agent')?.source).toBe('override');
+      expect(agents.get('shared-agent')?.definition.display_name).toBe('Custom Version');
+    });
+  });
+
+  describe('filterAgentsByTeam', () => {
+    it('should return all agents when no team config', () => {
+      const agents = new Map<string, AgentMetadata>([
+        ['agent-one', { definition: minimalAgent, source: 'core', filePath: '' }],
+        ['agent-two', { definition: fullAgent, source: 'core', filePath: '' }],
+      ]);
+
+      const filtered = filterAgentsByTeam(agents);
+      expect(filtered.size).toBe(2);
+    });
+
+    it('should filter agents by team config', () => {
+      const agents = new Map<string, AgentMetadata>([
+        ['agent-one', { definition: minimalAgent, source: 'core', filePath: '' }],
+        ['agent-two', { definition: fullAgent, source: 'core', filePath: '' }],
+      ]);
+
+      const config: CoreAIConfig = {
+        version: '1.0',
+        project: { name: 'test' },
+        team: { agents: ['agent-one'] },
+      };
+
+      const filtered = filterAgentsByTeam(agents, config);
+      expect(filtered.size).toBe(1);
+      expect(filtered.has('agent-one')).toBe(true);
+      expect(filtered.has('agent-two')).toBe(false);
+    });
+
+    it('should handle empty team agents array', () => {
+      const agents = new Map<string, AgentMetadata>([
+        ['agent-one', { definition: minimalAgent, source: 'core', filePath: '' }],
+      ]);
+
+      const config: CoreAIConfig = {
+        version: '1.0',
+        project: { name: 'test' },
+        team: { agents: [] },
+      };
+
+      const filtered = filterAgentsByTeam(agents, config);
+      expect(filtered.size).toBe(1);
+    });
+  });
+
+  describe('compileAgents', () => {
+    it('should compile agents to output directory', () => {
+      const coreDir = join(tempDir, 'core');
+      const outputDir = join(tempDir, 'output');
+      mkdirSync(coreDir);
+
+      writeFileSync(
+        join(coreDir, 'test-agent.yaml'),
+        `
+role: test-agent
+type: ic-engineer
+display_name: Test Agent
+description: A test agent
+`
+      );
+
+      const result = compileAgents(undefined, {
+        coreAgentsDir: coreDir,
+        outputDir,
+        projectRoot: tempDir,
+      });
+
+      expect(result.compiled.length).toBe(1);
+      expect(result.errors.length).toBe(0);
+      expect(result.compiled[0].role).toBe('test-agent');
+
+      // Verify file was created
+      const outputPath = join(outputDir, 'test-agent.md');
+      expect(existsSync(outputPath)).toBe(true);
+
+      const content = readFileSync(outputPath, 'utf-8');
+      expect(content).toContain('# Test Agent');
+    });
+
+    it('should resolve variables during compilation', () => {
+      const coreDir = join(tempDir, 'core');
+      const outputDir = join(tempDir, 'output');
+      mkdirSync(coreDir);
+
+      writeFileSync(
+        join(coreDir, 'test-agent.yaml'),
+        `
+role: test-agent
+type: ic-engineer
+display_name: Test Agent
+description: Agent for \${config.project.name}
+`
+      );
+
+      const config: CoreAIConfig = {
+        version: '1.0',
+        project: { name: 'my-awesome-project' },
+      };
+
+      compileAgents(config, {
+        coreAgentsDir: coreDir,
+        outputDir,
+        projectRoot: tempDir,
+      });
+
+      const content = readFileSync(join(outputDir, 'test-agent.md'), 'utf-8');
+      expect(content).toContain('Agent for my-awesome-project');
+    });
+
+    it('should create output directory if it does not exist', () => {
+      const coreDir = join(tempDir, 'core');
+      const outputDir = join(tempDir, 'nested', 'output', 'dir');
+      mkdirSync(coreDir);
+
+      writeFileSync(
+        join(coreDir, 'test-agent.yaml'),
+        `
+role: test-agent
+type: ic-engineer
+display_name: Test Agent
+description: A test agent
+`
+      );
+
+      compileAgents(undefined, {
+        coreAgentsDir: coreDir,
+        outputDir,
+        projectRoot: tempDir,
+      });
+
+      expect(existsSync(outputDir)).toBe(true);
+    });
+  });
+
+  describe('Core Agents Compilation', () => {
+    it('should compile all core agents successfully', () => {
+      const coreAgentsDir = join(__dirname, '../../agents');
+      const outputDir = join(tempDir, 'compiled');
+
+      const result = compileAgents(undefined, {
+        coreAgentsDir,
+        outputDir,
+        projectRoot: tempDir,
+      });
+
+      expect(result.errors.length).toBe(0);
+      expect(result.compiled.length).toBe(4);
+
+      // Verify each agent was compiled
+      const roles = result.compiled.map((c) => c.role).sort();
+      expect(roles).toEqual([
+        'backend-engineer',
+        'devops-engineer',
+        'engineering-manager',
+        'frontend-engineer',
+      ]);
+
+      // Verify files exist and have content
+      for (const compiled of result.compiled) {
+        const content = readFileSync(compiled.outputPath, 'utf-8');
+        expect(content.length).toBeGreaterThan(0);
+        expect(content).toContain('# ');
+        expect(content).toContain('*Generated by CoreAI*');
+      }
+    });
+  });
+});