Files
multica/server/internal/handler/skill_list_test.go
Tom Qiao b9602adabe fix(handler): validate skill id UUID at request boundary (#3025)
loadSkillForUser was passing chi.URLParam(r, "id") directly into
parseUUID, the panic-on-invalid helper reserved for trusted UUID
round-trips. A malformed `/api/skills/{notuuid}` request panicked
in util.MustParseUUID; chi's middleware.Recoverer turned it into a
500 instead of a 400.

This violates the documented convention (CLAUDE.md → "Backend Handler
UUID Parsing Convention"): pure-UUID request inputs must use
parseUUIDOrBadRequest, which writes a 400 and short-circuits.

Switch loadSkillForUser to parseUUIDOrBadRequest. Behaviour for valid
UUIDs is unchanged; malformed input now returns 400 with a clear
"invalid skill id" message.

Test:
- TestGetSkill_MalformedUUIDReturns400 asserts GET /api/skills/not-a-uuid
  returns 400.

Co-authored-by: Tom Qiao <tomqiaozc@users.noreply.github.com>
2026-05-22 12:22:07 +08:00

149 lines
5.4 KiB
Go

package handler
import (
"context"
"encoding/json"
"net/http/httptest"
"strings"
"testing"
)
// 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)
}
}
}
// 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
}