48 lines
1.4 KiB
Go
48 lines
1.4 KiB
Go
package stream
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"testing"
|
|
)
|
|
|
|
// TestErrorEventJSONIncludesErrorInfo 校验 ErrorInfo 经 StreamEvent 序列化后
|
|
// 前端可从 error_info 字段读取到分类元数据（含 request_id）。
|
|
func TestErrorEventJSONIncludesErrorInfo(t *testing.T) {
|
|
info := ClassifyError(errors.New("context deadline exceeded"), "dialogue_generating", "req_x", "sess_y", "")
|
|
ev := StreamEvent{Type: EventError, Error: info.UserMessage, ErrorInfo: &info}
|
|
|
|
b, err := json.Marshal(ev)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var raw map[string]any
|
|
if err := json.Unmarshal(b, &raw); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
ei, ok := raw["error_info"].(map[string]any)
|
|
if !ok {
|
|
t.Fatalf("missing error_info in %s", string(b))
|
|
}
|
|
if ei["request_id"] != "req_x" {
|
|
t.Fatalf("request_id=%v", ei["request_id"])
|
|
}
|
|
if ei["phase_id"] != "dialogue_generating" {
|
|
t.Fatalf("phase_id=%v", ei["phase_id"])
|
|
}
|
|
if ei["class"] != string(ErrorClassTransient) {
|
|
t.Fatalf("class=%v", ei["class"])
|
|
}
|
|
}
|
|
|
|
// TestWithRequestID_RoundTrip 校验请求 ID 可经 Context 注入并读回。
|
|
func TestWithRequestID_RoundTrip(t *testing.T) {
|
|
ctx := WithRequestID(t.Context(), "req_abc")
|
|
if got := RequestIDFromContext(ctx); got != "req_abc" {
|
|
t.Fatalf("got=%s want=req_abc", got)
|
|
}
|
|
if got := RequestIDFromContext(nil); got != "" {
|
|
t.Fatalf("nil ctx should return empty, got=%s", got)
|
|
}
|
|
}
|