mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-09-28 20:02:08 +08:00
21 lines
585 B
Go
21 lines
585 B
Go
package common
|
|
|
|
import "testing"
|
|
|
|
func TestRequestModelNamePrefersOriginalRequest(t *testing.T) {
|
|
original := []byte(`{"model":"original-model"}`)
|
|
translated := []byte(`{"model":"translated-model"}`)
|
|
|
|
if got := RequestModelName(original, translated); got != "original-model" {
|
|
t.Fatalf("model = %q, want original-model", got)
|
|
}
|
|
}
|
|
|
|
func TestRequestModelNameSupportsWrappedRequest(t *testing.T) {
|
|
request := []byte(`{"request":{"model":"wrapped-model"}}`)
|
|
|
|
if got := RequestModelName(nil, request); got != "wrapped-model" {
|
|
t.Fatalf("model = %q, want wrapped-model", got)
|
|
}
|
|
}
|