renjue
c4b01ebe2b
CI / docker (push) Successful in 2m52s
Co-authored-by: Cursor <cursoragent@cursor.com>
28 lines
613 B
Go
28 lines
613 B
Go
package invoke
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
)
|
|
|
|
func TestBuildPredictionBody(t *testing.T) {
|
|
body, err := buildPredictionBody("comfyui/z-image", map[string]any{"prompt": "cat"})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var payload map[string]any
|
|
if err := json.Unmarshal(body, &payload); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if payload["version"] != "comfyui/z-image" {
|
|
t.Fatalf("version: %v", payload["version"])
|
|
}
|
|
if _, ok := payload["model"]; ok {
|
|
t.Fatal("should not send model field")
|
|
}
|
|
input, _ := payload["input"].(map[string]any)
|
|
if input["prompt"] != "cat" {
|
|
t.Fatalf("input: %v", input)
|
|
}
|
|
}
|