t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
package response
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/models"
|
|
)
|
|
|
|
func TestBuildAIConfigResponseOmitsAPIKey(t *testing.T) {
|
|
payload, err := json.Marshal(BuildAIConfigResponse(&models.AIConfig{
|
|
ID: 1,
|
|
Name: "test",
|
|
APIKey: "sk-secret",
|
|
}))
|
|
if err != nil {
|
|
t.Fatalf("marshal response error = %v", err)
|
|
}
|
|
|
|
var decoded map[string]any
|
|
if err := json.Unmarshal(payload, &decoded); err != nil {
|
|
t.Fatalf("unmarshal response error = %v", err)
|
|
}
|
|
if _, ok := decoded["api_key"]; ok {
|
|
t.Fatalf("api_key should not be exposed: %s", payload)
|
|
}
|
|
if got, ok := decoded["has_api_key"].(bool); !ok || !got {
|
|
t.Fatalf("has_api_key = %v, want true: %s", decoded["has_api_key"], payload)
|
|
}
|
|
}
|
|
|
|
func TestBuildAIConfigDetailResponseIncludesAPIKey(t *testing.T) {
|
|
payload, err := json.Marshal(BuildAIConfigDetailResponse(&models.AIConfig{
|
|
ID: 1,
|
|
Name: "test",
|
|
APIKey: "sk-secret",
|
|
}))
|
|
if err != nil {
|
|
t.Fatalf("marshal response error = %v", err)
|
|
}
|
|
|
|
var decoded map[string]any
|
|
if err := json.Unmarshal(payload, &decoded); err != nil {
|
|
t.Fatalf("unmarshal response error = %v", err)
|
|
}
|
|
if got := decoded["api_key"]; got != "sk-secret" {
|
|
t.Fatalf("api_key = %v, want sk-secret: %s", got, payload)
|
|
}
|
|
}
|