CyberStrikeAI/internal/agents/markdown_orchestrator_test.go

package agents

import (
	"os"
	"path/filepath"
	"testing"
)

func TestLoadMarkdownAgentsDir_OrchestratorExcludedFromSubs(t *testing.T) {
	dir := t.TempDir()
	orch := filepath.Join(dir, OrchestratorMarkdownFilename)
	if err := os.WriteFile(orch, []byte(`---
id: cyberstrike-deep
name: Main
description: Test desc
---

Hello orchestrator
`), 0644); err != nil {
		t.Fatal(err)
	}
	subPath := filepath.Join(dir, "worker.md")
	if err := os.WriteFile(subPath, []byte(`---
id: worker
name: Worker
description: W
---

Do work
`), 0644); err != nil {
		t.Fatal(err)
	}
	load, err := LoadMarkdownAgentsDir(dir)
	if err != nil {
		t.Fatal(err)
	}
	if load.Orchestrator == nil || load.Orchestrator.EinoName != "cyberstrike-deep" {
		t.Fatalf("orchestrator: %+v", load.Orchestrator)
	}
	if len(load.SubAgents) != 1 || load.SubAgents[0].ID != "worker" {
		t.Fatalf("subs: %+v", load.SubAgents)
	}
	if len(load.FileEntries) != 2 {
		t.Fatalf("file entries: %d", len(load.FileEntries))
	}
	var orchFile *FileAgent
	for i := range load.FileEntries {
		if load.FileEntries[i].IsOrchestrator {
			orchFile = &load.FileEntries[i]
			break
		}
	}
	if orchFile == nil || orchFile.Filename != OrchestratorMarkdownFilename {
		t.Fatal("missing orchestrator file entry")
	}
}

func TestLoadMarkdownAgentsDir_DuplicateOrchestrator(t *testing.T) {
	dir := t.TempDir()
	_ = os.WriteFile(filepath.Join(dir, OrchestratorMarkdownFilename), []byte("---\nname: A\n---\n\nx\n"), 0644)
	_ = os.WriteFile(filepath.Join(dir, "b.md"), []byte("---\nname: B\nkind: orchestrator\n---\n\ny\n"), 0644)
	_, err := LoadMarkdownAgentsDir(dir)
	if err == nil {
		t.Fatal("expected duplicate orchestrator error")
	}
}