35 lines
1.2 KiB
Go
35 lines
1.2 KiB
Go
package app
|
|
|
|
import "testing"
|
|
|
|
func TestResolveWorkbenchProviderConfigMapsWorkbenchParameters(t *testing.T) {
|
|
service := &Service{}
|
|
|
|
image, err := service.resolveWorkbenchProviderConfig(WorkbenchKindImage, "image-model", "logical-image", "", map[string]any{
|
|
"size": "1024x1024", "quality": "high", "imageResolution": "2k", "count": 2,
|
|
}, map[string]any{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for key, want := range map[string]any{"model": "image-model", "size": "1024x1024", "quality": "high", "imageResolution": "2k", "count": "2"} {
|
|
if image[key] != want {
|
|
t.Errorf("image config[%q] = %#v, want %#v", key, image[key], want)
|
|
}
|
|
}
|
|
|
|
video, err := service.resolveWorkbenchProviderConfig(WorkbenchKindVideo, "video-model", "logical-video", "", map[string]any{
|
|
"seconds": 8, "ratio": "9:16", "resolution": "1080P", "generateAudio": true, "watermark": false,
|
|
}, map[string]any{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for key, want := range map[string]any{
|
|
"model": "video-model", "videoSeconds": "8", "size": "9:16", "vquality": "1080P",
|
|
"videoGenerateAudio": "true", "videoWatermark": "false",
|
|
} {
|
|
if video[key] != want {
|
|
t.Errorf("video config[%q] = %#v, want %#v", key, video[key], want)
|
|
}
|
|
}
|
|
}
|