M0：神识 Go 网关骨架（IR 入站/SSE 出站/stub 适配器/服务令牌/健康检查）及测试与镜像

shenshi/internal/server/server_test.go

@@ -0,0 +1,110 @@
+package server
+
+import (
+	"bufio"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"hunjing/shenshi/internal/route"
+)
+
+func newTestServer(token string) *httptest.Server {
+	srv := New(Config{ServiceToken: token}, route.New())
+	return httptest.NewServer(srv)
+}
+
+func chatBody() string {
+	return `{"ir_version":1,"slot":"chat","messages":[` +
+		`{"role":"system","content":"契约占位"},{"role":"user","content":"你好"}],` +
+		`"run_id":"run_test","stream":true}`
+}
+
+func post(t *testing.T, ts *httptest.Server, token, body string) (*http.Response, string) {
+	t.Helper()
+	req, err := http.NewRequest(http.MethodPost, ts.URL+"/v1/chat/completions", strings.NewReader(body))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if token != "" {
+		req.Header.Set("Authorization", "Bearer "+token)
+	}
+	resp, err := ts.Client().Do(req)
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Cleanup(func() { _ = resp.Body.Close() })
+	var sb strings.Builder
+	scanner := bufio.NewScanner(resp.Body)
+	for scanner.Scan() {
+		sb.WriteString(scanner.Text())
+		sb.WriteString("\n")
+	}
+	return resp, sb.String()
+}
+
+func TestChatCompletionsStreamsEchoAndDone(t *testing.T) {
+	ts := newTestServer("")
+	resp, body := post(t, ts, "", chatBody())
+	if resp.StatusCode != http.StatusOK {
+		t.Fatalf("状态码 = %d", resp.StatusCode)
+	}
+	if got := resp.Header.Get("X-Accel-Buffering"); got != "no" {
+		t.Fatalf("X-Accel-Buffering = %q", got)
+	}
+	if !strings.Contains(body, `"delta":"【神识回声桩】"`) && !strings.Contains(body, "神识回声桩") {
+		t.Fatalf("缺少回声增量: %s", body)
+	}
+	if !strings.Contains(body, `"finish_reason":"stop"`) {
+		t.Fatalf("缺少终帧用量: %s", body)
+	}
+	if !strings.Contains(body, "data: [DONE]") {
+		t.Fatalf("缺少 [DONE]: %s", body)
+	}
+}
+
+func TestServiceTokenRejectsInvalidCredential(t *testing.T) {
+	ts := newTestServer("secret-token")
+	resp, body := post(t, ts, "wrong", chatBody())
+	if resp.StatusCode != http.StatusUnauthorized {
+		t.Fatalf("状态码 = %d", resp.StatusCode)
+	}
+	if !strings.Contains(body, "HJ-2003") {
+		t.Fatalf("缺少错误码: %s", body)
+	}
+}
+
+func TestInvalidIRRejected(t *testing.T) {
+	ts := newTestServer("")
+	resp, body := post(t, ts, "", `{"ir_version":99,"messages":[],"run_id":"","stream":true}`)
+	if resp.StatusCode != http.StatusBadRequest {
+		t.Fatalf("状态码 = %d", resp.StatusCode)
+	}
+	if !strings.Contains(body, "HJ-2002") {
+		t.Fatalf("缺少错误码: %s", body)
+	}
+}
+
+func TestUnknownSlotNoEndpoint(t *testing.T) {
+	ts := newTestServer("")
+	resp, body := post(t, ts, "", `{"ir_version":1,"slot":"vision","messages":[{"role":"user","content":"x"}],"run_id":"run_t","stream":true}`)
+	if resp.StatusCode != http.StatusServiceUnavailable {
+		t.Fatalf("状态码 = %d", resp.StatusCode)
+	}
+	if !strings.Contains(body, "HJ-2001") {
+		t.Fatalf("缺少错误码: %s", body)
+	}
+}
+
+func TestHealthz(t *testing.T) {
+	ts := newTestServer("")
+	resp, err := ts.Client().Get(ts.URL + "/healthz")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer func() { _ = resp.Body.Close() }()
+	if resp.StatusCode != http.StatusOK {
+		t.Fatalf("状态码 = %d", resp.StatusCode)
+	}
+}