package runner import "testing" func TestDistributedAdmissionModelTypeBoundary(t *testing.T) { for _, modelType := range []string{ "image_generate", "image_edit", "image_analysis", "image_vectorize", "video_generate", "video_enhance", "image_to_video", "text_to_video", "video_edit", "video_reference", "video_first_last_frame", "video_understanding", "omni_video", "omni", "audio_generate", "audio_understanding", "text_to_speech", "voice_clone", } { if !distributedAdmissionModelType(modelType) { t.Errorf("%s should use distributed admission", modelType) } } for _, modelType := range []string{"text_generate", "embedding", "rerank", "", "unknown"} { if distributedAdmissionModelType(modelType) { t.Errorf("%s should preserve immediate execution/rate-limit behavior", modelType) } } }