Cyrene_v2/internal/channel/adapter/httpclient/leak_test.go

package httpclient

// 本文件钉住本客户端的两条出镜边界（M41 #10/#11 那一族）：
//
//   - **非 2xx 的响应体原文不进错误**：旧实现把体截到 512B 拼进 error，而调用方
//     （KOOK 适配器）会把这条 error 直接落日志 —— 上游在体里回显什么，日志里就有
//     什么。现在只留状态码、来源主机，以及信封的 code/message（已清洗、已截断）。
//   - **请求地址的凭据不进错误**：target 可能带 `?access_token=…` 或 `user:pass@`，
//     而 net/http 的 *url.Error 会把整条地址拼进错误串。
//
// 每条用例都是「把它改回去就变红」的形状：断言的是**错误文本**，也就是调用方实际
// 打进日志的那段字符串。

import (
	"context"
	"errors"
	"net/http"
	"net/http/httptest"
	"net/url"
	"strings"
	"testing"
)

// failingTransport 让 http.Client.Do 以传输层错误收场。选定它而不是拨一个真端口，
// 是因为只有传输层失败才会让 net/http 包出 **带完整地址** 的 *url.Error ——
// 那正是要钉住的泄漏形态。
type failingTransport struct{}

func (failingTransport) RoundTrip(*http.Request) (*http.Response, error) {
	return nil, errors.New("dial tcp: connection refused")
}

// TestDoJSON_StatusErrorDoesNotEchoBodySentinel 4xx 体里放哨兵 ⇒ 错误里搜不到。
//
// 用非信封体（代理返回的 HTML 错误页就是这样）：信封类的体按设计会保留
// code/message，非信封体则一个字节都不留。
func TestDoJSON_StatusErrorDoesNotEchoBodySentinel(t *testing.T) {
	t.Parallel()
	const sentinel = "SENTINEL-4XX-BODY"
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusBadGateway)
		_, _ = w.Write([]byte("<html><body>bad gateway " + sentinel + "</body></html>"))
	}))
	defer srv.Close()

	c := New(WithBaseURL(srv.URL), WithHTTPClient(srv.Client()), WithLogger(discardLogger()))
	_, err := c.DoJSON(context.Background(), http.MethodPost, "/message/create", map[string]any{"a": 1})
	if err == nil {
		t.Fatal("502 却报成功")
	}
	if !errors.Is(err, ErrAPI) {
		t.Fatalf("err = %v, want ErrAPI", err)
	}
	if strings.Contains(err.Error(), sentinel) {
		t.Fatalf("响应体原文进了错误（会随调用方落日志）：%v", err)
	}
	// 可诊断性不能一起丢：来源主机与状态码必须还在，且带得出「体有多大」。
	host, _, ok := strings.Cut(strings.TrimPrefix(srv.URL, "http://"), ":")
	if !ok || !strings.Contains(err.Error(), host) {
		t.Fatalf("错误里没有来源主机 %q：%v", host, err)
	}
	if !strings.Contains(err.Error(), "status=502") {
		t.Fatalf("错误里没有状态码：%v", err)
	}
	if !strings.Contains(err.Error(), "body_bytes=") {
		t.Fatalf("错误里没有响应体体积（相关性判断要用）：%v", err)
	}
}

// TestDoJSON_StatusErrorKeepsEnvelopeCode 信封体仍要给出 code/message：这是本客户端
// 唯一认识的诊断面，替换掉体原文之后不能连它一起丢。
func TestDoJSON_StatusErrorKeepsEnvelopeCode(t *testing.T) {
	t.Parallel()
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusUnauthorized)
		_, _ = w.Write([]byte(`{"code":40101,"message":"invalid token","data":null}`))
	}))
	defer srv.Close()

	c := New(WithBaseURL(srv.URL), WithHTTPClient(srv.Client()), WithLogger(discardLogger()))
	_, err := c.DoJSON(context.Background(), http.MethodGet, "/gateway/index", nil)
	if err == nil {
		t.Fatal("401 却报成功")
	}
	if !strings.Contains(err.Error(), "code 40101") || !strings.Contains(err.Error(), "invalid token") {
		t.Fatalf("信封的 code/message 丢了：%v", err)
	}
}

// TestDoJSON_NonZeroCodeRedactsMessageSecrets 信封 message 是上游自由文本面：里面的
// 凭据形状必须被清洗掉，长度必须被截断。改回原样透传就变红。
func TestDoJSON_NonZeroCodeRedactsMessageSecrets(t *testing.T) {
	t.Parallel()
	const secret = "big-secret-value-123456"
	body := `{"code":40101,"message":"invalid access_token=` + secret + `","data":null}`
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		_, _ = w.Write([]byte(body))
	}))
	defer srv.Close()

	c := New(WithBaseURL(srv.URL), WithHTTPClient(srv.Client()), WithLogger(discardLogger()), WithToken(secret))
	_, err := c.DoJSON(context.Background(), http.MethodGet, "/gateway/index", nil)
	if err == nil {
		t.Fatal("非零 code 却报成功")
	}
	if strings.Contains(err.Error(), secret) {
		t.Fatalf("信封 message 里的凭据原样进了错误：%v", err)
	}
}

// TestGetJSON_TransportErrorOmitsTargetCredentials 传输层失败时，错误里不得出现目标
// 地址（含 query 里的凭据）—— net/http 会把整条 URL 放在 *url.Error 里。
func TestGetJSON_TransportErrorOmitsTargetCredentials(t *testing.T) {
	t.Parallel()
	const secret = "QUERY-SECRET-VALUE"
	c := New(
		WithBaseURL("http://api.example.com"),
		WithHTTPClient(&http.Client{Transport: failingTransport{}}),
		WithLogger(discardLogger()),
	)
	_, err := c.GetJSON(context.Background(), "/gateway/index", url.Values{"access_token": {secret}})
	if err == nil {
		t.Fatal("传输层失败却报成功")
	}
	if strings.Contains(err.Error(), secret) {
		t.Fatalf("目标地址的凭据进了错误：%v", err)
	}
	if !strings.Contains(err.Error(), "api.example.com") || !strings.Contains(err.Error(), "connection refused") {
		t.Fatalf("剥掉地址时把来源或原因一起丢了：%v", err)
	}
}

// TestGetJSON_UserinfoNeverEntersError 绝对地址带 user:pass@ 时同样不得出镜。
func TestGetJSON_UserinfoNeverEntersError(t *testing.T) {
	t.Parallel()
	c := New(
		WithHTTPClient(&http.Client{Transport: failingTransport{}}),
		WithLogger(discardLogger()),
	)
	_, err := c.GetJSON(context.Background(), "https://user:pass@api.example.com/x", nil)
	if err == nil {
		t.Fatal("传输层失败却报成功")
	}
	if strings.Contains(err.Error(), "user:pass") {
		t.Fatalf("user:pass@ 进了错误：%v", err)
	}
}