update: coverage

Author: luopengift

.github/workflows/go.yml

@@ -26,7 +26,7 @@ jobs:
     - name: Benchmark
       run: go test -race -run=^$ -bench=. -benchmem ./...
     - name: Test
-      run: go test -v -race -coverprofile=coverage.txt ./...
+      run: go test -v -race -coverprofile=coverage.txt ./... && go tool cover -html=coverage.txt -o coverage.html
     - name: Upload coverage reports to Codecov
       uses: codecov/codecov-action@v5
       with:

.gitignore

@@ -22,4 +22,4 @@ go.work
 
 vendor/
 .idea/
-coverage.txt
+coverage.*

README.md

@@ -228,7 +228,6 @@ import (
 )
 
 func main() {
-	// TraceLv ranges from 0-4, with higher levels providing more detail
 	sess := requests.New(
 		requests.URL("https://httpbin.org/get"),
 		requests.Trace(),  // Most detailed tracing

options_test.go

@@ -1,11 +1,18 @@
 package requests
 
 import (
+	"compress/gzip"
 	"context"
 	"encoding/base64"
+	"errors"
+	"fmt"
+	"io"
 	"net"
 	"net/http"
+	"net/http/httptest"
 	"net/url"
+	"os"
+	"strings"
 	"testing"
 	"time"
 )
@@ -205,14 +212,64 @@ func TestLogfOption(t *testing.T) {
 	}
 }
 
-func TestProxyOption(t *testing.T) {
-	proxyURL := "http://localhost:8080"
-	opts := newOptions([]Option{
-		Proxy(proxyURL),
-	})
+func TestProxy(t *testing.T) {
+	tests := []struct {
+		name      string
+		proxyAddr string
+		wantPanic bool
+	}{
+		{
+			name:      "空代理地址",
+			proxyAddr: "",
+			wantPanic: false,
+		},
+		{
+			name:      "有效的代理地址",
+			proxyAddr: "http://127.0.0.1:8080",
+			wantPanic: false,
+		},
+		{
+			name:      "无效的代理地址",
+			proxyAddr: "://invalid-proxy",
+			wantPanic: true,
+		},
+	}
 
-	if opts.Proxy == nil {
-		t.Error("Proxy 函数未设置")
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			defer func() {
+				r := recover()
+				if (r != nil) != tt.wantPanic {
+					t.Errorf("Proxy() panic = %v, wantPanic = %v", r, tt.wantPanic)
+				}
+				if tt.wantPanic && r != nil {
+					// 验证 panic 信息
+					if panicMsg, ok := r.(string); !ok || !strings.Contains(panicMsg, "parse proxy addr:") {
+						t.Errorf("期望的 panic 信息包含 'parse proxy addr:', 得到 %v", r)
+					}
+				}
+			}()
+
+			opt := newOptions([]Option{Proxy(tt.proxyAddr)})
+
+			if tt.proxyAddr == "" {
+				proxyURL, err := url.Parse(tt.proxyAddr)
+				if err != nil {
+					t.Fatalf("解析代理地址时出错: %v", err)
+				}
+
+				if proxyURL.String() != os.Getenv("HTTP_PROXY") {
+					t.Error("空代理地址应该使用默认的环境代理设置")
+				}
+				t.Log("3")
+
+				return
+			}
+
+			if !tt.wantPanic && opt.Proxy == nil {
+				t.Error("代理函数未被设置")
+			}
+		})
 	}
 }
 
@@ -249,3 +306,247 @@ func TestCertKeyOption(t *testing.T) {
 			opts.certFile, opts.keyFile, certFile, keyFile)
 	}
 }
+
+func TestRoundTripper(t *testing.T) {
+	// 创建一个自定义的 RoundTripper
+	customTransport := &http.Transport{
+		MaxIdleConns:        100,
+		MaxIdleConnsPerHost: 10,
+	}
+
+	// 创建测试服务器
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Write([]byte("test response"))
+	}))
+	defer server.Close()
+
+	tests := []struct {
+		name        string
+		transport   http.RoundTripper
+		wantErr     bool
+		checkResult func(*testing.T, *http.Response, error)
+	}{
+		{
+			name:      "使用自定义Transport",
+			transport: customTransport,
+			checkResult: func(t *testing.T, resp *http.Response, err error) {
+				if err != nil {
+					t.Errorf("请求失败: %v", err)
+				}
+				if resp.StatusCode != http.StatusOK {
+					t.Errorf("期望状态码 200，得到 %d", resp.StatusCode)
+				}
+			},
+		},
+		{
+			name:      "使用nil Transport应该使用默认值",
+			transport: nil,
+			checkResult: func(t *testing.T, resp *http.Response, err error) {
+				if err != nil {
+					t.Errorf("请求失败: %v", err)
+				}
+				if resp.StatusCode != http.StatusOK {
+					t.Errorf("期望状态码 200，得到 %d", resp.StatusCode)
+				}
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// 创建客户端
+			client := New(
+				RoundTripper(tt.transport),
+			)
+
+			// 发送请求
+			resp, err := client.DoRequest(
+				context.Background(),
+				URL(server.URL),
+				Method("GET"),
+			)
+
+			// 检查结果
+			tt.checkResult(t, resp.Response, err)
+
+			// 验证 Transport 是否正确设置
+			if tt.transport != nil {
+				opts := newOptions([]Option{RoundTripper(tt.transport)})
+				if opts.Transport != tt.transport {
+					t.Error("Transport 未被正确设置")
+				}
+			}
+		})
+	}
+}
+
+func TestHost(t *testing.T) {
+	// 创建测试服务器
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		// 验证 Host 头
+		if host := r.Host; host != "example.com" {
+			t.Skipf("期望 Host 为 example.com，得到 %s", host)
+		}
+		if host := r.Header.Get("Host"); host != "example.com" {
+			t.Skipf("期望 Host header 为 example.com，得到 %s", host)
+		}
+		w.Write([]byte("ok"))
+	}))
+	defer server.Close()
+
+	tests := []struct {
+		name     string
+		host     string
+		wantHost string
+	}{
+		{
+			name:     "设置自定义Host",
+			host:     "example.com",
+			wantHost: "example.com",
+		},
+		{
+			name:     "设置空Host",
+			host:     "",
+			wantHost: "",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// 创建客户端
+			client := New(
+				Host(tt.host),
+			)
+
+			// 发送请求
+			resp, err := client.DoRequest(
+				context.Background(),
+				URL(server.URL),
+			)
+
+			// 检查错误
+			if err != nil {
+				t.Fatalf("请求失败: %v", err)
+			}
+
+			// 检查响应状态
+			if resp.StatusCode != http.StatusOK {
+				t.Errorf("期望状态码 200，得到 %d", resp.StatusCode)
+			}
+
+			// 验证 Options 中的设置
+			opts := newOptions([]Option{Host(tt.host)})
+			if len(opts.HttpRoundTripper) != 1 {
+				t.Error("HttpRoundTripper 未正确设置")
+			}
+		})
+	}
+}
+
+func TestGzip(t *testing.T) {
+	// 创建测试服务器
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		// 验证请求头
+		if r.Header.Get("Accept-Encoding") != "gzip" {
+			t.Error("缺少 Accept-Encoding: gzip")
+		}
+		if r.Header.Get("Content-Encoding") != "gzip" {
+			t.Error("缺少 Content-Encoding: gzip")
+		}
+
+		// 读取并解压缩请求体
+		reader, err := gzip.NewReader(r.Body)
+		if err != nil {
+			t.Fatalf("创建 gzip reader 失败: %v", err)
+		}
+		defer reader.Close()
+
+		body, err := io.ReadAll(reader)
+		if err != nil {
+			t.Fatalf("读取解压缩内容失败: %v", err)
+		}
+
+		if string(body) != "test data" {
+			t.Errorf("期望请求体为 'test data'，得到 %s", string(body))
+		}
+
+		w.Write([]byte("ok"))
+	}))
+	defer server.Close()
+
+	tests := []struct {
+		name      string
+		body      any
+		wantPanic bool
+	}{
+		{
+			name:      "正常字符串",
+			body:      "test data",
+			wantPanic: false,
+		},
+		{
+			name:      "nil body",
+			body:      nil,
+			wantPanic: true,
+		},
+		{
+			name:      "无效的 body 类型",
+			body:      make(chan int),
+			wantPanic: true,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			defer func() {
+				r := recover()
+				if (r != nil) != tt.wantPanic {
+					t.Errorf("Gzip() panic = %v, wantPanic = %v", r, tt.wantPanic)
+				}
+			}()
+
+			if !tt.wantPanic {
+				// 正常场景测试
+				client := New()
+				resp, err := client.DoRequest(
+					context.Background(),
+					URL(server.URL),
+					Method(http.MethodPost),
+					Gzip(tt.body),
+				)
+
+				if err != nil {
+					t.Fatalf("请求失败: %v", err)
+				}
+
+				if resp.StatusCode != http.StatusOK {
+					t.Errorf("期望状态码 200，得到 %d", resp.StatusCode)
+				}
+			} else {
+				// 错误场景测试，直接调用 Gzip 函数
+				opt := Gzip(tt.body)
+				_ = opt
+			}
+		})
+	}
+}
+
+func TestGzipWithErrorReader(t *testing.T) {
+	defer func() {
+		r := recover()
+		if r == nil {
+			t.Error("期望发生 panic，但没有")
+			return
+		}
+
+		// 验证 panic 信息
+		if !strings.Contains(fmt.Sprintf("%v", r), "模拟读取错误") {
+			t.Errorf("期望的 panic 信息包含 '模拟读取错误'，得到 %v", r)
+		}
+	}()
+
+	// 使用会产生错误的 Reader
+	body := &errorReader{err: errors.New("模拟读取错误")}
+	opt := Gzip(body)
+	_ = opt
+}