dela
0335d572de
ci / go (push) Waiting to run
ci / go-db (agent) (push) Waiting to run
ci / go-db (config) (push) Waiting to run
ci / go-db (db) (push) Waiting to run
ci / go-db (evidence) (push) Waiting to run
ci / go-db (llmrec) (push) Waiting to run
ci / go-db (server) (push) Waiting to run
detections / detections (push) Waiting to run
web / web (push) Waiting to run
docs / links (push) Canceled after 0s
94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
package server
|
||
|
||
import (
|
||
"context"
|
||
"os"
|
||
"path/filepath"
|
||
"testing"
|
||
|
||
actool "github.com/Autumn-27/norma/tool"
|
||
"github.com/Autumn-27/artex/agent"
|
||
"github.com/Autumn-27/artex/db"
|
||
)
|
||
|
||
// hasSkillTool reports whether the packed tool set contains the Skill meta-tool.
|
||
func hasSkillTool(tools []actool.CoreTool) bool {
|
||
for _, t := range tools {
|
||
if t.Name() == "Skill" {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// TestAssembleVisibleSkill verifies a filesystem skill made visible to an agent is
|
||
// packed into that agent's tool set (as the single Skill meta-tool) via the wired
|
||
// ToolAugment hook. Skills live on disk under skillDir; visibility is a per-agent
|
||
// (agent × skill_name) row keyed by the skill's directory name.
|
||
func TestAssembleVisibleSkill(t *testing.T) {
|
||
dsn, _, err := db.DSN()
|
||
if err != nil {
|
||
t.Skipf("no database config (%v) — skipping", err)
|
||
}
|
||
pg, err := db.Open(dsn)
|
||
if err != nil {
|
||
t.Skipf("postgres unavailable (%v) — skipping", err)
|
||
}
|
||
defer pg.Close()
|
||
|
||
// Filesystem skill fixture: <skillDir>/t-assemble/SKILL.md. The directory name
|
||
// (t-assemble) is the visibility key matched against AgentSkillNames.
|
||
skillDir := t.TempDir()
|
||
const skillName = "t-assemble"
|
||
if err := os.MkdirAll(filepath.Join(skillDir, skillName), 0o755); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
md := "---\nname: t-assemble\ndescription: do the thing\n---\ndo the thing"
|
||
if err := os.WriteFile(filepath.Join(skillDir, skillName, "SKILL.md"), []byte(md), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
wireAgentAugment(pg, skillDir, nil)
|
||
t.Cleanup(func() { agent.ToolAugment = nil })
|
||
|
||
ag, _ := pg.GetAgentByKey("planner")
|
||
if ag == nil {
|
||
t.Fatal("planner agent missing")
|
||
}
|
||
|
||
// Isolate this test from whatever skill visibility the planner already has in the
|
||
// shared DB: clear it now, restore it on cleanup.
|
||
orig, _ := pg.AgentSkillNames(ag.ID)
|
||
if err := pg.SetAgentSkillVisibility(ag.ID, nil); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
t.Cleanup(func() { _ = pg.SetAgentSkillVisibility(ag.ID, orig) })
|
||
|
||
// before: skill not visible → no Skill meta-tool
|
||
extra, _, cleanup := agent.ToolAugment(context.Background(), "planner")
|
||
cleanup()
|
||
if hasSkillTool(extra) {
|
||
t.Fatalf("expected no Skill meta-tool before the skill is made visible")
|
||
}
|
||
|
||
// make the skill visible to planner
|
||
if err := pg.ToggleSkillVisibility(ag.ID, skillName, true); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
t.Cleanup(func() { _ = pg.ToggleSkillVisibility(ag.ID, skillName, false) })
|
||
|
||
// after: the Skill meta-tool is packed in
|
||
extra, _, cleanup = agent.ToolAugment(context.Background(), "planner")
|
||
defer cleanup()
|
||
if !hasSkillTool(extra) {
|
||
t.Fatalf("expected the Skill meta-tool after making the skill visible")
|
||
}
|
||
|
||
// AugmentTools appends extra to base without filtering base
|
||
combined, _, c2 := agent.AugmentTools(context.Background(), "planner", nil)
|
||
defer c2()
|
||
if !hasSkillTool(combined) {
|
||
t.Fatalf("AugmentTools should surface the Skill meta-tool")
|
||
}
|
||
}
|