mirror of
https://github.com/multica-ai/multica.git
synced 2026-08-02 18:13:27 +02:00
Co-authored-by: Eve <eve@multica-ai.local> Co-authored-by: multica-agent <github@multica.ai>
199 lines
7.3 KiB
Go
199 lines
7.3 KiB
Go
package handler
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
)
|
|
|
|
// TestListSkills_OmitsContent guards the fix for GH multica-ai/multica#2174:
|
|
// the workspace skill list endpoint must not ship the SKILL.md `content`
|
|
// blob, which used to bloat the payload past CLI timeouts on workspaces with
|
|
// many large skills. The detail endpoint still returns content (covered by
|
|
// TestGetSkill_IncludesContent below).
|
|
func TestListSkills_OmitsContent(t *testing.T) {
|
|
skillID := insertHandlerTestSkill(t, "list-omits-content", strings.Repeat("a", 4096))
|
|
|
|
w := httptest.NewRecorder()
|
|
req := newRequest("GET", "/api/skills?workspace_id="+testWorkspaceID, nil)
|
|
testHandler.ListSkills(w, req)
|
|
if w.Code != 200 {
|
|
t.Fatalf("ListSkills: expected 200, got %d: %s", w.Code, w.Body.String())
|
|
}
|
|
|
|
// Decode into a generic shape so we can prove the wire format has no
|
|
// `content` field at all — not "content present but empty", which would
|
|
// still leave the bytes on the wire.
|
|
var rows []map[string]any
|
|
if err := json.Unmarshal(w.Body.Bytes(), &rows); err != nil {
|
|
t.Fatalf("ListSkills: failed to decode body: %v", err)
|
|
}
|
|
|
|
var found bool
|
|
for _, row := range rows {
|
|
if row["id"] != skillID {
|
|
continue
|
|
}
|
|
found = true
|
|
if _, ok := row["content"]; ok {
|
|
t.Fatalf("ListSkills: response must not include `content` field, got: %v", row)
|
|
}
|
|
// Other expected list fields should still be present.
|
|
for _, key := range []string{"id", "name", "description", "config", "created_at", "updated_at", "workspace_id"} {
|
|
if _, ok := row[key]; !ok {
|
|
t.Fatalf("ListSkills: missing expected field %q in response: %v", key, row)
|
|
}
|
|
}
|
|
}
|
|
if !found {
|
|
t.Fatalf("ListSkills: inserted skill %s not in response", skillID)
|
|
}
|
|
}
|
|
|
|
// TestGetSkill_IncludesContent confirms the detail endpoint still ships the
|
|
// full SKILL.md body — the list-summary change must not regress single-skill
|
|
// reads.
|
|
func TestGetSkill_IncludesContent(t *testing.T) {
|
|
body := "# detail body\nstill served on /api/skills/{id}"
|
|
skillID := insertHandlerTestSkill(t, "detail-includes-content", body)
|
|
|
|
w := httptest.NewRecorder()
|
|
req := newRequest("GET", "/api/skills/"+skillID, nil)
|
|
req = withURLParam(req, "id", skillID)
|
|
testHandler.GetSkill(w, req)
|
|
if w.Code != 200 {
|
|
t.Fatalf("GetSkill: expected 200, got %d: %s", w.Code, w.Body.String())
|
|
}
|
|
|
|
var resp map[string]any
|
|
if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil {
|
|
t.Fatalf("GetSkill: failed to decode body: %v", err)
|
|
}
|
|
if got, _ := resp["content"].(string); got != body {
|
|
t.Fatalf("GetSkill: expected content %q, got %q", body, got)
|
|
}
|
|
}
|
|
|
|
// TestListAgentSkills_OmitsContent: same constraint for the agent-scoped
|
|
// listing — gpt-boy review of the original fix flagged this as a sister case
|
|
// because `multica agent skills list` follows the same shape rules.
|
|
func TestListAgentSkills_OmitsContent(t *testing.T) {
|
|
agentID := createHandlerTestAgent(t, "Handler Skill Summary Test", nil)
|
|
skillID := insertHandlerTestSkill(t, "agent-skill-omits-content", strings.Repeat("b", 1024))
|
|
if _, err := testPool.Exec(context.Background(),
|
|
`INSERT INTO agent_skill (agent_id, skill_id) VALUES ($1, $2)`,
|
|
agentID, skillID,
|
|
); err != nil {
|
|
t.Fatalf("attach skill to agent: %v", err)
|
|
}
|
|
|
|
w := httptest.NewRecorder()
|
|
req := newRequest("GET", "/api/agents/"+agentID+"/skills", nil)
|
|
req = withURLParam(req, "id", agentID)
|
|
testHandler.ListAgentSkills(w, req)
|
|
if w.Code != 200 {
|
|
t.Fatalf("ListAgentSkills: expected 200, got %d: %s", w.Code, w.Body.String())
|
|
}
|
|
|
|
var rows []map[string]any
|
|
if err := json.Unmarshal(w.Body.Bytes(), &rows); err != nil {
|
|
t.Fatalf("ListAgentSkills: failed to decode body: %v", err)
|
|
}
|
|
if len(rows) == 0 {
|
|
t.Fatalf("ListAgentSkills: expected at least 1 skill")
|
|
}
|
|
for _, row := range rows {
|
|
if _, ok := row["content"]; ok {
|
|
t.Fatalf("ListAgentSkills: response must not include `content` field, got: %v", row)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSetAgentSkillEnabledControlsExecutionWithoutRemovingAssignment(t *testing.T) {
|
|
agentID := createHandlerTestAgent(t, "Handler Skill Toggle Test", nil)
|
|
skillID := insertHandlerTestSkill(t, "agent-skill-toggle", "# Toggle me")
|
|
if _, err := testPool.Exec(context.Background(),
|
|
`INSERT INTO agent_skill (agent_id, skill_id) VALUES ($1, $2)`,
|
|
agentID, skillID,
|
|
); err != nil {
|
|
t.Fatalf("attach skill to agent: %v", err)
|
|
}
|
|
|
|
setEnabled := func(enabled bool) *httptest.ResponseRecorder {
|
|
t.Helper()
|
|
w := httptest.NewRecorder()
|
|
req := newRequest("PUT", "/api/agents/"+agentID+"/skills/"+skillID+"/enabled", map[string]any{
|
|
"enabled": enabled,
|
|
})
|
|
rctx := chi.NewRouteContext()
|
|
rctx.URLParams.Add("id", agentID)
|
|
rctx.URLParams.Add("skillId", skillID)
|
|
req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))
|
|
testHandler.SetAgentSkillEnabled(w, req)
|
|
return w
|
|
}
|
|
|
|
if w := setEnabled(false); w.Code != 200 {
|
|
t.Fatalf("disable skill: expected 200, got %d: %s", w.Code, w.Body.String())
|
|
}
|
|
|
|
// Disabled assignments remain visible to management surfaces.
|
|
rows, err := testHandler.Queries.ListAgentSkillSummaries(context.Background(), parseUUID(agentID))
|
|
if err != nil || len(rows) != 1 || rows[0].Enabled {
|
|
t.Fatalf("disabled assignment not preserved: rows=%+v err=%v", rows, err)
|
|
}
|
|
// Execution only receives enabled skills.
|
|
active, err := testHandler.Queries.ListAgentSkills(context.Background(), parseUUID(agentID))
|
|
if err != nil || len(active) != 0 {
|
|
t.Fatalf("disabled skill leaked into execution: skills=%+v err=%v", active, err)
|
|
}
|
|
|
|
if w := setEnabled(true); w.Code != 200 {
|
|
t.Fatalf("enable skill: expected 200, got %d: %s", w.Code, w.Body.String())
|
|
}
|
|
active, err = testHandler.Queries.ListAgentSkills(context.Background(), parseUUID(agentID))
|
|
if err != nil || len(active) != 1 || active[0].Name == "" {
|
|
t.Fatalf("re-enabled skill missing from execution: skills=%+v err=%v", active, err)
|
|
}
|
|
}
|
|
|
|
// TestGetSkill_MalformedUUIDReturns400 guards the handler UUID parsing
|
|
// convention (CLAUDE.md → "Backend Handler UUID Parsing Convention"): raw
|
|
// `id` URL params on the request boundary must be validated with
|
|
// parseUUIDOrBadRequest, not the panic-prone parseUUID. Before the fix
|
|
// the malformed input panicked in MustParseUUID and was rescued by the
|
|
// chi Recoverer middleware as a 500.
|
|
func TestGetSkill_MalformedUUIDReturns400(t *testing.T) {
|
|
w := httptest.NewRecorder()
|
|
req := newRequest("GET", "/api/skills/not-a-uuid", nil)
|
|
req = withURLParam(req, "id", "not-a-uuid")
|
|
testHandler.GetSkill(w, req)
|
|
if w.Code != 400 {
|
|
t.Fatalf("GetSkill malformed uuid: expected 400, got %d: %s", w.Code, w.Body.String())
|
|
}
|
|
}
|
|
|
|
// insertHandlerTestSkill writes a skill row directly via SQL and registers a
|
|
// cleanup hook. We bypass the create handler to keep the test focused on the
|
|
// list/detail wire shape and to make it easy to inject a large body.
|
|
func insertHandlerTestSkill(t *testing.T, namePrefix, content string) string {
|
|
t.Helper()
|
|
name := namePrefix + "-" + t.Name()
|
|
var id string
|
|
if err := testPool.QueryRow(context.Background(), `
|
|
INSERT INTO skill (workspace_id, name, description, content, config, created_by)
|
|
VALUES ($1, $2, $3, $4, '{}'::jsonb, $5)
|
|
RETURNING id
|
|
`, testWorkspaceID, name, "fixture", content, testUserID).Scan(&id); err != nil {
|
|
t.Fatalf("insert skill: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
testPool.Exec(context.Background(), `DELETE FROM skill WHERE id = $1`, id)
|
|
})
|
|
return id
|
|
}
|