Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
4.2 kB · 118 lines
Go
at commit 44ea9abc
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119package executor
import ( "testing"
cliproxyauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth" "github.com/tidwall/gjson")
func TestEnsureImageGenerationTool_NoTools(t *testing.T) { body := []byte(`{"model":"gpt-5.4","input":"draw a cat"}`) result := ensureImageGenerationTool(body, "gpt-5.4", nil)
tools := gjson.GetBytes(result, "tools") if !tools.IsArray() { t.Fatalf("expected tools array, got %v", tools.Type) } arr := tools.Array() if len(arr) != 1 { t.Fatalf("expected 1 tool, got %d", len(arr)) } if arr[0].Get("type").String() != "image_generation" { t.Fatalf("expected type=image_generation, got %s", arr[0].Get("type").String()) } if arr[0].Get("output_format").String() != "png" { t.Fatalf("expected output_format=png, got %s", arr[0].Get("output_format").String()) }}
func TestEnsureImageGenerationTool_ExistingToolsWithoutImageGen(t *testing.T) { body := []byte(`{"model":"gpt-5.4","tools":[{"type":"function","name":"get_weather","parameters":{}}]}`) result := ensureImageGenerationTool(body, "gpt-5.4", nil)
tools := gjson.GetBytes(result, "tools") arr := tools.Array() if len(arr) != 2 { t.Fatalf("expected 2 tools, got %d", len(arr)) } if arr[0].Get("type").String() != "function" { t.Fatalf("expected first tool type=function, got %s", arr[0].Get("type").String()) } if arr[1].Get("type").String() != "image_generation" { t.Fatalf("expected second tool type=image_generation, got %s", arr[1].Get("type").String()) }}
func TestEnsureImageGenerationTool_AlreadyPresent(t *testing.T) { body := []byte(`{"model":"gpt-5.4","tools":[{"type":"image_generation","output_format":"webp"},{"type":"function","name":"f1"}]}`) result := ensureImageGenerationTool(body, "gpt-5.4", nil)
tools := gjson.GetBytes(result, "tools") arr := tools.Array() if len(arr) != 2 { t.Fatalf("expected 2 tools (no duplicate), got %d", len(arr)) } if arr[0].Get("output_format").String() != "webp" { t.Fatalf("expected original output_format=webp preserved, got %s", arr[0].Get("output_format").String()) }}
func TestEnsureImageGenerationTool_EmptyToolsArray(t *testing.T) { body := []byte(`{"model":"gpt-5.4","tools":[]}`) result := ensureImageGenerationTool(body, "gpt-5.4", nil)
tools := gjson.GetBytes(result, "tools") arr := tools.Array() if len(arr) != 1 { t.Fatalf("expected 1 tool, got %d", len(arr)) } if arr[0].Get("type").String() != "image_generation" { t.Fatalf("expected type=image_generation, got %s", arr[0].Get("type").String()) }}
func TestEnsureImageGenerationTool_WebSearchAndImageGen(t *testing.T) { body := []byte(`{"model":"gpt-5.4","tools":[{"type":"web_search"}]}`) result := ensureImageGenerationTool(body, "gpt-5.4", nil)
tools := gjson.GetBytes(result, "tools") arr := tools.Array() if len(arr) != 2 { t.Fatalf("expected 2 tools, got %d", len(arr)) } if arr[0].Get("type").String() != "web_search" { t.Fatalf("expected first tool type=web_search, got %s", arr[0].Get("type").String()) } if arr[1].Get("type").String() != "image_generation" { t.Fatalf("expected second tool type=image_generation, got %s", arr[1].Get("type").String()) }}
func TestEnsureImageGenerationTool_GPT53CodexSparkDoesNotInjectTool(t *testing.T) { body := []byte(`{"model":"gpt-5.3-codex-spark","input":"draw a cat"}`) result := ensureImageGenerationTool(body, "gpt-5.3-codex-spark", nil)
if string(result) != string(body) { t.Fatalf("expected body to be unchanged, got %s", string(result)) } if gjson.GetBytes(result, "tools").Exists() { t.Fatalf("expected no tools for gpt-5.3-codex-spark, got %s", gjson.GetBytes(result, "tools").Raw) }}
func TestEnsureImageGenerationTool_FreeCodexAuthDoesNotInjectTool(t *testing.T) { body := []byte(`{"model":"gpt-5.4","input":"draw a cat"}`) freeAuth := &cliproxyauth.Auth{ Provider: "codex", Attributes: map[string]string{"plan_type": "free"}, } result := ensureImageGenerationTool(body, "gpt-5.4", freeAuth)
if string(result) != string(body) { t.Fatalf("expected body to be unchanged, got %s", string(result)) } if gjson.GetBytes(result, "tools").Exists() { t.Fatalf("expected no tools for free codex auth, got %s", gjson.GetBytes(result, "tools").Raw) }}