tbphp
7cbd2e672d
* feat(responses): 实现响应状态续接路由 * fix(responses): 限制响应 ID 长度并兼容缺失字段删除 * fix(responses): 按协议存储能力匹配续接路由
66 lines
2.7 KiB
Go
66 lines
2.7 KiB
Go
package dialect
|
||
|
||
import (
|
||
"bytes"
|
||
"testing"
|
||
)
|
||
|
||
func TestExtractJSONRequestFieldsStrictDecisionContract(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
body string
|
||
wantModel string
|
||
wantStream bool
|
||
wantErr bool
|
||
}{
|
||
{name: "minimal", body: `{"model":"gpt-4o"}`, wantModel: "gpt-4o"},
|
||
{name: "stream true", body: `{"model":"gpt-4o","stream":true}`, wantModel: "gpt-4o", wantStream: true},
|
||
{name: "unknown nested fields", body: `{"metadata":{"model":"ignored","stream":true},"model":"gpt-4o"}`, wantModel: "gpt-4o"},
|
||
{name: "escaped field name", body: `{"\u006dodel":"gpt-4o"}`, wantModel: "gpt-4o"},
|
||
{name: "missing model", body: `{}`, wantErr: true},
|
||
{name: "model null", body: `{"model":null}`, wantErr: true},
|
||
{name: "model number", body: `{"model":1}`, wantErr: true},
|
||
{name: "model boundary whitespace", body: `{"model":" gpt-4o"}`, wantErr: true},
|
||
{name: "duplicate model", body: `{"model":"a","model":"b"}`, wantErr: true},
|
||
{name: "case alias", body: `{"Model":"gpt-4o"}`, wantErr: true},
|
||
{name: "case collision first exact", body: `{"model":"forbidden","Model":"allowed"}`, wantErr: true},
|
||
{name: "case collision first alias", body: `{"Model":"allowed","model":"forbidden"}`, wantErr: true},
|
||
{name: "stream null", body: `{"model":"gpt-4o","stream":null}`, wantErr: true},
|
||
{name: "stream string", body: `{"model":"gpt-4o","stream":"true"}`, wantErr: true},
|
||
{name: "duplicate stream", body: `{"model":"gpt-4o","stream":false,"stream":true}`, wantErr: true},
|
||
{name: "stream case alias", body: `{"model":"gpt-4o","Stream":true}`, wantErr: true},
|
||
{name: "stream unicode simple-fold alias", body: `{"model":"gpt-4o","ſtream":true}`, wantErr: true},
|
||
{name: "stream unicode simple-fold collision", body: `{"model":"gpt-4o","stream":false,"ſtream":true}`, wantErr: true},
|
||
{name: "array root", body: `[{"model":"gpt-4o"}]`, wantErr: true},
|
||
{name: "trailing value", body: `{"model":"gpt-4o"}{}`, wantErr: true},
|
||
{name: "invalid utf8", body: string([]byte{'{', '"', 'm', 'o', 'd', 'e', 'l', '"', ':', '"', 0xff, '"', '}'}), wantErr: true},
|
||
}
|
||
|
||
for _, test := range tests {
|
||
t.Run(test.name, func(t *testing.T) {
|
||
body := []byte(test.body)
|
||
before := bytes.Clone(body)
|
||
metadata, err := inspectJSONRequestFields(body, true, false)
|
||
model := ""
|
||
if metadata.Model != nil {
|
||
model = *metadata.Model
|
||
}
|
||
if test.wantErr {
|
||
if err == nil {
|
||
t.Fatalf("inspectJSONRequestFields() = %#v, nil", metadata)
|
||
}
|
||
} else if err != nil || model != test.wantModel ||
|
||
metadata.Stream != test.wantStream {
|
||
t.Fatalf(
|
||
"inspectJSONRequestFields() = %#v, %v",
|
||
metadata,
|
||
err,
|
||
)
|
||
}
|
||
if !bytes.Equal(body, before) {
|
||
t.Fatal("parser mutated wire body")
|
||
}
|
||
})
|
||
}
|
||
}
|