mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-10-21 14:52:05 +02:00
fix: validate login and role inputs and protect task access
This commit is contained in:
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)
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user