mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-10-01 05:30:19 +02:00
Ed1s0nZandCursor
38b96ec67a
Let audit_agent approve or reject with one System One call instead of chat JSON, while keeping the OpenAI-compatible backend as an option. Co-authored-by: Cursor <cursoragent@cursor.com>
70 lines
2.2 KiB
Go
70 lines
2.2 KiB
Go
package handler
|
||
|
||
import (
|
||
"testing"
|
||
|
||
"cyberstrike-ai/internal/config"
|
||
)
|
||
|
||
func TestHitlAuditEngineInfoTypeSafe(t *testing.T) {
|
||
h := &AgentHandler{config: &config.Config{
|
||
OpenAI: config.OpenAIConfig{Model: "gpt-4o"},
|
||
Hitl: config.HitlConfig{AuditBackend: "typesafe"},
|
||
}}
|
||
backend, model := h.hitlAuditEngineInfo()
|
||
if backend != config.HitlAuditBackendTypeSafe {
|
||
t.Fatalf("backend=%q", backend)
|
||
}
|
||
if model != config.TypeSafeDefaultModel {
|
||
t.Fatalf("model=%q, want %s", model, config.TypeSafeDefaultModel)
|
||
}
|
||
}
|
||
|
||
func TestHitlAuditEngineInfoOpenAIInheritsMainModel(t *testing.T) {
|
||
h := &AgentHandler{config: &config.Config{
|
||
OpenAI: config.OpenAIConfig{Model: "gpt-4o-mini"},
|
||
Hitl: config.HitlConfig{AuditBackend: "openai"},
|
||
}}
|
||
backend, model := h.hitlAuditEngineInfo()
|
||
if backend != config.HitlAuditBackendOpenAI {
|
||
t.Fatalf("backend=%q", backend)
|
||
}
|
||
if model != "gpt-4o-mini" {
|
||
t.Fatalf("model=%q", model)
|
||
}
|
||
}
|
||
|
||
func TestHitlAuditBackendFromRecordPrefersPayload(t *testing.T) {
|
||
backend, model := hitlAuditBackendFromRecord("audit_agent", "audit agent: 实际操作：探测", `{
|
||
"hitlApproval": {"auditBackend": "typesafe", "auditModel": "jev-latest"}
|
||
}`)
|
||
if backend != config.HitlAuditBackendTypeSafe || model != "jev-latest" {
|
||
t.Fatalf("backend=%q model=%q", backend, model)
|
||
}
|
||
}
|
||
|
||
func TestHitlAuditBackendFromRecordInfersJevComment(t *testing.T) {
|
||
backend, _ := hitlAuditBackendFromRecord("audit_agent",
|
||
"audit agent: 未命中破坏性规则，默认放行；最高破坏分=破坏业务可用性 0.12；choice=approve(0.90)",
|
||
`{}`)
|
||
if backend != config.HitlAuditBackendTypeSafe {
|
||
t.Fatalf("backend=%q", backend)
|
||
}
|
||
}
|
||
|
||
func TestHitlAuditBackendFromRecordInfersOpenAIComment(t *testing.T) {
|
||
backend, _ := hitlAuditBackendFromRecord("audit_agent",
|
||
"audit agent: 实际操作：读取 /etc/passwd；命中规则：A3",
|
||
`{}`)
|
||
if backend != config.HitlAuditBackendOpenAI {
|
||
t.Fatalf("backend=%q", backend)
|
||
}
|
||
}
|
||
|
||
func TestHitlAuditBackendFromRecordIgnoresHuman(t *testing.T) {
|
||
backend, model := hitlAuditBackendFromRecord("human", "人工通过", `{"hitlApproval":{"auditBackend":"typesafe"}}`)
|
||
if backend != "" || model != "" {
|
||
t.Fatalf("backend=%q model=%q", backend, model)
|
||
}
|
||
}
|