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)
		}
	}
}