mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-09-11 14:47:58 +08:00
64291120e7
- Added tests to ensure no-copy payload reuse for various translators (Codex, Gemini, OpenAI, Claude, etc.). - Covered scenarios for field normalization, tool parameter cleaning, and thought signature sanitization. - Included a benchmark for SanitizeGeminiRequestThoughtSignatures with large normalized payloads.
42 lines
1.3 KiB
Go
42 lines
1.3 KiB
Go
package gemini
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
func TestCleanGeminiCodexToolParametersPreservesCanonicalSchema(t *testing.T) {
|
|
input := []byte(`{"type":"object","properties":{"value":{"type":"string"}},"additionalProperties":false}`)
|
|
|
|
output := cleanGeminiCodexToolParameters(gjson.ParseBytes(input))
|
|
|
|
if string(output) != string(input) {
|
|
t.Fatalf("canonical schema changed:\n got: %s\nwant: %s", output, input)
|
|
}
|
|
}
|
|
|
|
func TestSetCodexToolChoiceFromGeminiToolConfigReusesAutoChoice(t *testing.T) {
|
|
input := []byte(`{"tool_choice":"auto","input":[]}`)
|
|
config := gjson.Parse(`{"mode":"AUTO"}`)
|
|
|
|
output := setCodexToolChoiceFromGeminiToolConfig(input, config)
|
|
|
|
if &output[0] != &input[0] {
|
|
t.Fatal("AUTO tool choice caused a payload copy")
|
|
}
|
|
}
|
|
|
|
func TestCleanGeminiCodexToolParametersNormalizesSchema(t *testing.T) {
|
|
input := []byte(`{"type":"object","$schema":"draft","additionalProperties":true}`)
|
|
|
|
output := cleanGeminiCodexToolParameters(gjson.ParseBytes(input))
|
|
|
|
if gjson.GetBytes(output, "$schema").Exists() {
|
|
t.Fatal("$schema should be removed")
|
|
}
|
|
if additionalProperties := gjson.GetBytes(output, "additionalProperties"); additionalProperties.Type != gjson.False {
|
|
t.Fatalf("additionalProperties = %s, want false", additionalProperties.Raw)
|
|
}
|
|
}
|