fix(image): 图片生成 - 打通分辨率选择并补充回归测试

This commit is contained in:
ddcat666 committed 2026-10-14 09:34:41 +08:00
1 parent 6a118ae93b
commit db8a301474
9 files changed
+125 -20

No files matched your search

+16
View File
@@ -59,6 +59,22 @@ func TestTaskOutputIncludesOnlySafeWorkbenchRestoreData(t *testing.T) {
}
}
func TestTaskOutputRestoresWorkbenchImageResolution(t *testing.T) {
task := model.Task{InputJSON: `{"mode":"image","metadata":{"source":"workbench","workbenchKind":"image","workbench":{"kind":"image","parameters":{"size":"1824x1024","imageResolution":"2k","quality":"auto","count":1}}},"config":{"size":"1824x1024","imageResolution":"2k","quality":"auto","count":"1"}}`}
var output map[string]any
if err := json.Unmarshal([]byte(publicTaskInputJSON(task.InputJSON)), &output); err != nil {
t.Fatal(err)
}
workbench, ok := output["workbench"].(map[string]any)
if !ok {
t.Fatalf("workbench restore data missing: %#v", output)
}
parameters, ok := workbench["parameters"].(map[string]any)
if !ok || parameters["imageResolution"] != "2k" {
t.Fatalf("image resolution not preserved for restore: %#v", workbench)
}
}
func TestTaskMediaPreviewUsesSafeMediaURLs(t *testing.T) {
previewURL, previewKind := taskMediaPreview(`{"images":["data:image/png;base64,AAAA","/api/resources/resource-1/file"],"video":"https://cdn.example.com/output.mp4"}`, "video")
if previewURL != "/api/resources/resource-1/file" || previewKind != "image" {