40 lines
2.1 KiB
Go
40 lines
2.1 KiB
Go
package protocol
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestMiniMaxVideoFailureMessages(t *testing.T) {
|
|
adapter := officialPackageAdapter(t, "minimax-hailuo-video-v2.yingce-plugin", "minimax-video")
|
|
const dimensionError = "content[1].image_url: media dimensions must be between 256 and 5760 pixels"
|
|
for _, tc := range []struct {
|
|
name, payload, message string
|
|
status Status
|
|
}{
|
|
{"nested task error", `{"task":{"id":"video-1","status":"failed","error":{"code":"2013","message":"` + dimensionError + `"}}}`, dimensionError, StatusFailed},
|
|
{"nested error without status", `{"task":{"id":"video-1","error":{"code":"2013","message":"` + dimensionError + `"}}}`, dimensionError, StatusFailed},
|
|
{"specific error precedes envelope", `{"base_resp":{"status_code":0,"status_msg":"success"},"task":{"id":"video-1","status":"failed","message":"task failed","error":{"code":"2013","message":"` + dimensionError + `"}}}`, dimensionError, StatusFailed},
|
|
{"base response error", `{"task_id":"video-1","base_resp":{"status_code":2013,"status_msg":"invalid input"}}`, "invalid input", StatusFailed},
|
|
{"task message", `{"task":{"id":"video-1","status":"failed","message":"task rejected"}}`, "task rejected", StatusFailed},
|
|
{"successful task", `{"task":{"id":"video-1","status":"succeeded","error":{"code":0},"content":{"url":"https://cdn.example/video.mp4"}}}`, "", StatusSucceeded},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
created, err := adapter.ParseCreate(context.Background(), []byte(tc.payload))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if created.TaskID != "video-1" || created.Status != tc.status || created.Message != tc.message {
|
|
t.Fatalf("create = %#v, want status %s and message %q", created, tc.status, tc.message)
|
|
}
|
|
polled, err := adapter.ParsePoll(context.Background(), PollContext{TaskID: "video-1"}, []byte(tc.payload))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if polled.TaskID != "video-1" || polled.Status != tc.status || polled.Message != tc.message {
|
|
t.Fatalf("poll = %#v, want status %s and message %q", polled, tc.status, tc.message)
|
|
}
|
|
})
|
|
}
|
|
}
|