fix(osi): 加固 Phase 0 验收实现(T-006)

This commit is contained in:
ila
2026-07-06 23:20:42 +08:00
parent def95efb1c
commit 9db2e84476
11 changed files with 110 additions and 22 deletions
+40 -2
View File
@@ -69,8 +69,8 @@ func TestNewTransportKeepsTimeout(t *testing.T) {
if err != nil {
t.Fatalf("NewTransport: %v", err)
}
if transport.client.Timeout != 3*time.Second {
t.Fatalf("timeout = %s, want 3s", transport.client.Timeout)
if transport.timeout != 3*time.Second {
t.Fatalf("timeout = %s, want 3s", transport.timeout)
}
}
@@ -79,6 +79,7 @@ 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 {
@@ -121,6 +122,7 @@ func TestTransportPreservesHeaderNameCasing(t *testing.T) {
}
}
}
func TestTransportAddsRequestsCompatibleBaseHeaders(t *testing.T) {
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if r.Header.Get("User-Agent") == "" {
@@ -144,3 +146,39 @@ func TestTransportAddsRequestsCompatibleBaseHeaders(t *testing.T) {
t.Fatalf("PostJSON: %v", err)
}
}
func TestTransportRequestsIdentityEncoding(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"}, nil)
if err != nil {
t.Fatalf("PostJSON: %v", err)
}
raw := <-received
if !strings.Contains(raw, "\r\nAccept-Encoding: identity\r\n") {
t.Fatalf("raw request missing identity Accept-Encoding:\n%s", raw)
}
}