You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1.2 KiB
38 lines
1.2 KiB
package server
|
|
|
|
import "testing"
|
|
|
|
func TestConfigFromEnvDefaultsVideoRouteToViduQ2(t *testing.T) {
|
|
t.Setenv("POPIART_DEFAULT_VIDEO_MODEL", "")
|
|
t.Setenv("POPIART_DEFAULT_IMAGE_MODEL", "")
|
|
t.Setenv("POPIART_NEWAPI_BASE_URL", "")
|
|
t.Setenv("POPIART_NEWAPI_TOKEN", "")
|
|
t.Setenv("POPIART_DATA_DIR", "")
|
|
t.Setenv("POPIART_SQLITE_PATH", "")
|
|
t.Setenv("POPIART_SESSION_SECRET", "")
|
|
t.Setenv("POPIART_SKILLHUB_DIR", "")
|
|
|
|
cfg := ConfigFromEnv()
|
|
if cfg.DefaultVideoModel != "viduq2" {
|
|
t.Fatalf("expected default video model viduq2, got %q", cfg.DefaultVideoModel)
|
|
}
|
|
if cfg.DefaultImageModel != "gemini-3.1-flash-image-preview" {
|
|
t.Fatalf("expected default image model gemini-3.1-flash-image-preview, got %q", cfg.DefaultImageModel)
|
|
}
|
|
}
|
|
|
|
func TestInferRouteKeyForModelRecognizesViduAsVideo(t *testing.T) {
|
|
cases := map[string]string{
|
|
"viduq2": "video.image2video",
|
|
"vidu2.0": "video.image2video",
|
|
"sora-2": "video.image2video",
|
|
"runway-gen4": "video.image2video",
|
|
"gpt-image-1": "image.text2image",
|
|
}
|
|
|
|
for modelID, want := range cases {
|
|
if got := inferRouteKeyForModel(modelID); got != want {
|
|
t.Fatalf("inferRouteKeyForModel(%q) = %q, want %q", modelID, got, want)
|
|
}
|
|
}
|
|
}
|
|
|