106 lines
5.1 KiB
Go
106 lines
5.1 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestKemeiSeedreamRequestDefaultsAndReferences(t *testing.T) {
|
|
adapter := officialPackageAdapter(t, "km-kemei-seedream.yingce-plugin", "km-kemei-seedream")
|
|
for _, name := range []string{"doubao-seedream-5-0-pro-260628", "doubao-seedream-5-0-flash-260915"} {
|
|
request := GenerationRequest{Model: name, Prompt: "暖色书房中的橘猫", ImageCount: 1, Quality: "high"}
|
|
spec, err := adapter.BuildCreate(context.Background(), RequestContext{Request: request})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
body := manifestTestBody(t, spec)
|
|
if spec.Path != "/v1/images/generations" || body["model"] != name || body["n"] != float64(1) || body["size"] != "2K" || body["response_format"] != "url" || body["output_format"] != "jpeg" || body["watermark"] != false {
|
|
t.Fatalf("defaults = %#v", body)
|
|
}
|
|
for _, key := range []string{"image", "quality", "resolution", "seed", "stream", "web_search", "sequential_image_generation"} {
|
|
if _, exists := body[key]; exists {
|
|
t.Fatalf("unexpected %s: %#v", key, body)
|
|
}
|
|
}
|
|
request.Images = []MediaReference{{URL: "https://example.com/b.png", Order: 2}, {DataURL: "data:image/png;base64,aA==", Order: 1}}
|
|
request.AspectRatio = "16:9"
|
|
spec, err = adapter.BuildCreate(context.Background(), RequestContext{Request: request})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
body = manifestTestBody(t, spec)
|
|
images := body["image"].([]any)
|
|
if body["size"] != "2816x1584" || len(images) != 2 || images[0] != request.Images[1].DataURL {
|
|
t.Fatalf("references = %#v", body)
|
|
}
|
|
for _, size := range []string{"1K", "1.5K", "2K", "2352x1008", "2816x1584", "1536x1536"} {
|
|
request.AspectRatio = size
|
|
if _, err := adapter.BuildCreate(context.Background(), RequestContext{Request: request}); err != nil {
|
|
t.Fatalf("size %s: %v", size, err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestKemeiSeedreamRejectsUnsupportedAndOutOfRangeRequests(t *testing.T) {
|
|
adapter := officialPackageAdapter(t, "km-kemei-seedream.yingce-plugin", "km-kemei-seedream")
|
|
base := GenerationRequest{Model: "doubao-seedream-5-0-pro-260628", Prompt: "橘猫", ImageCount: 1, AspectRatio: "2048x2048"}
|
|
for _, tc := range []struct {
|
|
name string
|
|
change func(*GenerationRequest)
|
|
}{
|
|
{"wrong model", func(r *GenerationRequest) { r.Model = "midjourney-1k" }},
|
|
{"multiple outputs", func(r *GenerationRequest) { r.ImageCount = 2 }},
|
|
{"empty prompt", func(r *GenerationRequest) { r.Prompt = " " }},
|
|
{"too small", func(r *GenerationRequest) { r.AspectRatio = "512x512" }},
|
|
{"4k", func(r *GenerationRequest) { r.AspectRatio = "4K" }},
|
|
{"too large", func(r *GenerationRequest) { r.AspectRatio = "2880x2880" }},
|
|
{"bad dimensions", func(r *GenerationRequest) { r.AspectRatio = "2048x2048x1" }},
|
|
{"mask", func(r *GenerationRequest) {
|
|
r.Images = []MediaReference{{URL: "https://example.com/a.png", Role: "mask"}}
|
|
}},
|
|
{"too many references", func(r *GenerationRequest) {
|
|
for i := 0; i < 11; i++ {
|
|
r.Images = append(r.Images, MediaReference{URL: "https://example.com/a.png"})
|
|
}
|
|
}},
|
|
{"oversize reference", func(r *GenerationRequest) {
|
|
r.Images = []MediaReference{{URL: "https://example.com/a.png", Metadata: map[string]any{"bytes": 30*1024*1024 + 1}}}
|
|
}},
|
|
{"unsupported format", func(r *GenerationRequest) {
|
|
r.ProviderOptions = map[string]map[string]any{"km-kemei-seedream": {"output_format": "webp"}}
|
|
}},
|
|
{"group generation", func(r *GenerationRequest) {
|
|
r.ProviderOptions = map[string]map[string]any{"km-kemei-seedream": {"sequential_image_generation": "auto"}}
|
|
}},
|
|
{"transparent text", func(r *GenerationRequest) {
|
|
r.ProviderOptions = map[string]map[string]any{"km-kemei-seedream": {"background": "transparent", "output_format": "png"}}
|
|
}},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
request := base
|
|
tc.change(&request)
|
|
if _, err := adapter.BuildCreate(context.Background(), RequestContext{Request: request}); err == nil {
|
|
t.Fatal("invalid request accepted")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestKemeiSeedreamLayerResultsAndErrors(t *testing.T) {
|
|
adapter := officialPackageAdapter(t, "km-kemei-seedream.yingce-plugin", "km-kemei-seedream")
|
|
request := GenerationRequest{Model: "doubao-seedream-5-0-flash-260915", AspectRatio: "auto", Images: []MediaReference{{URL: "https://example.com/source.png"}}, ProviderOptions: map[string]map[string]any{"km-kemei-seedream": {"layer_decomposition": true, "output_format": "png"}}}
|
|
spec, err := adapter.BuildCreate(context.Background(), RequestContext{Request: request})
|
|
if err != nil || manifestTestBody(t, spec)["size"] != "auto" {
|
|
t.Fatalf("layer request: %#v %v", spec, err)
|
|
}
|
|
result, err := adapter.ParseCreate(context.Background(), []byte(`{"data":[{"url":"https://example.com/base.png","z_index":0},{"b64_json":"aA==","z_index":1}],"usage":{"generated_images":2}}`))
|
|
if err != nil || result.Status != StatusSucceeded || result.Result == nil || len(result.Result.Images) != 2 {
|
|
t.Fatalf("results: %#v %v", result, err)
|
|
}
|
|
result, err = adapter.ParseCreate(context.Background(), []byte(`{"error":{"code":"invalid_size","message":"尺寸不合法"}}`))
|
|
if err != nil || result.Status != StatusFailed || result.Message != "尺寸不合法" {
|
|
t.Fatalf("error: %#v %v", result, err)
|
|
}
|
|
}
|