Files
easyai-ai-gateway/apps/api/internal/store/private_task_state_test.go
T
easyai 3056cf8fca
ci / verify (pull_request) Successful in 15m34s
feat(gateway): 补齐桌面端高级媒体直连接口
新增图片矢量化、视频超分、每日用量、计价与任务隔离能力,并通过环境变量解析平台凭据。

已通过 Go 全量门禁、迁移检查、镜像构建以及 Vectorizer 五格式和 Topaz 3 秒视频真实 DEV 验收。
2026-07-22 14:02:53 +08:00

26 lines
834 B
Go

package store
import (
"encoding/json"
"strings"
"testing"
)
func TestGatewayTaskJSONOmitsPrivateRemoteTaskPayload(t *testing.T) {
raw, err := json.Marshal(GatewayTask{
ID: "task-private-state",
RemoteTaskID: "safe-upstream-id",
RemoteTaskPayload: map[string]any{"imageToken": "private-image-token", "receipt": "private-receipt"},
})
if err != nil {
t.Fatal(err)
}
serialized := strings.ToLower(string(raw))
if strings.Contains(serialized, "private-image-token") || strings.Contains(serialized, "private-receipt") || strings.Contains(serialized, "remotetaskpayload") {
t.Fatalf("private provider recovery state leaked into task JSON: %s", serialized)
}
if !strings.Contains(serialized, "safe-upstream-id") {
t.Fatalf("public upstream task id should remain available: %s", serialized)
}
}