43 lines
1.5 KiB
Go
43 lines
1.5 KiB
Go
package app
|
|
|
|
import "testing"
|
|
|
|
func TestProviderProgressFromResponse(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
body string
|
|
want int
|
|
ok bool
|
|
}{
|
|
{name: "direct number", body: `{"progress":35}`, want: 35, ok: true},
|
|
{name: "nested percent string", body: `{"data":{"task":{"progress_percent":"68%"}}}`, want: 68, ok: true},
|
|
{name: "fraction", body: `{"result":{"progress":0.42}}`, want: 42, ok: true},
|
|
{name: "fractional explicit percent", body: `{"progress":"0.5%"}`, want: 1, ok: true},
|
|
{name: "camel case", body: `{"operation":{"progressPercentage":87.6}}`, want: 88, ok: true},
|
|
{name: "unrelated completion tokens", body: `{"usage":{"completion_tokens":120}}`, ok: false},
|
|
{name: "invalid range", body: `{"progress":140}`, ok: false},
|
|
{name: "not json", body: `not-json`, ok: false},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, ok := providerProgressFromResponse([]byte(tt.body))
|
|
if got != tt.want || ok != tt.ok {
|
|
t.Fatalf("providerProgressFromResponse() = (%d, %v), want (%d, %v)", got, ok, tt.want, tt.ok)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestTaskUsesUpstreamReportedProgress(t *testing.T) {
|
|
for _, taskType := range []string{"canvas_image", "canvas_video", "video_shot"} {
|
|
if !taskUsesUpstreamReportedProgress(taskType) {
|
|
t.Fatalf("task type %q should use upstream progress", taskType)
|
|
}
|
|
}
|
|
for _, taskType := range []string{"canvas_text", "canvas_audio", "text_generation"} {
|
|
if taskUsesUpstreamReportedProgress(taskType) {
|
|
t.Fatalf("task type %q should keep internal progress", taskType)
|
|
}
|
|
}
|
|
}
|