fix: validate login and role inputs and protect task access

This commit is contained in:
Ed1s0nZ committed 2026-10-06 01:36:31 +08:00
1 parent 17498368f6
commit 079ba0d2b3
18 files changed
+328 -8

No files matched your search

@@ -0,0 +1,29 @@
package handler
import (
"context"
"path/filepath"
"testing"
"cyberstrike-ai/internal/database"
"cyberstrike-ai/internal/multiagent"
"go.uber.org/zap"
)
func TestRobotDeliversPlainTextWithoutToolEvidence(t *testing.T) {
db, err := database.NewDB(filepath.Join(t.TempDir(), "robot.db"), zap.NewNop())
if err != nil {
t.Fatal(err)
}
t.Cleanup(func() { _ = db.Close() })
conv, err := db.CreateConversation("text", database.ConversationCreateMeta{})
if err != nil {
t.Fatal(err)
}
h := &AgentHandler{db: db, logger: zap.NewNop()}
result := &multiagent.RunResult{Response: "OK"}
text, _, err := h.finalizeRobotAgentSuccess(context.Background(), "", conv.ID, result)
if err != nil || text != "OK" || !result.Finalized {
t.Fatalf("text=%q finalized=%v error=%v", text, result.Finalized, err)
}
}