feat(osi): 打通 Go 侧真实请求验收(T-005)

This commit is contained in:
ila
2026-07-06 23:03:14 +08:00
parent 2a77a4dbc1
commit def95efb1c
12 changed files with 411 additions and 54 deletions
+67
View File
@@ -3,8 +3,10 @@ package osi
import (
"context"
"encoding/json"
"net"
"net/http"
"net/http/httptest"
"strings"
"testing"
"time"
)
@@ -77,3 +79,68 @@ func TestNewTransportRejectsInvalidSOCKS5Proxy(t *testing.T) {
t.Fatal("NewTransport accepted invalid SOCKS5 proxy")
}
}
func TestTransportPreservesHeaderNameCasing(t *testing.T) {
listener, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatalf("listen: %v", err)
}
defer listener.Close()
received := make(chan string, 1)
go func() {
conn, err := listener.Accept()
if err != nil {
received <- ""
return
}
defer conn.Close()
buf := make([]byte, 4096)
n, _ := conn.Read(buf)
received <- string(buf[:n])
_, _ = conn.Write([]byte("HTTP/1.1 200 OK\r\nContent-Length: 2\r\n\r\n{}"))
}()
transport, err := NewTransport(TransportConfig{Timeout: time.Second})
if err != nil {
t.Fatalf("NewTransport: %v", err)
}
_, _, err = transport.PostJSON(
context.Background(),
"http://"+listener.Addr().String(),
map[string]string{"serviceId": "JKDA00002"},
map[string]string{"orgCode": "org-001", "deviceSN": "", "userName": "dyytgw"},
)
if err != nil {
t.Fatalf("PostJSON: %v", err)
}
raw := <-received
for _, header := range []string{"orgCode:", "deviceSN:", "userName:"} {
if !strings.Contains(raw, "\r\n"+header) {
t.Fatalf("raw request missing exact header %q:\n%s", header, raw)
}
}
}
func TestTransportAddsRequestsCompatibleBaseHeaders(t *testing.T) {
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if r.Header.Get("User-Agent") == "" {
t.Fatal("missing User-Agent")
}
if r.Header.Get("Accept") != "*/*" {
t.Fatalf("Accept = %q", r.Header.Get("Accept"))
}
if r.Header.Get("Connection") != "keep-alive" {
t.Fatalf("Connection = %q", r.Header.Get("Connection"))
}
_, _ = w.Write([]byte(`{}`))
}))
defer server.Close()
transport, err := NewTransport(TransportConfig{Timeout: time.Second})
if err != nil {
t.Fatalf("NewTransport: %v", err)
}
if _, _, err := transport.PostJSON(context.Background(), server.URL, map[string]string{"serviceId": "JKDA00002"}, nil); err != nil {
t.Fatalf("PostJSON: %v", err)
}
}