Golang单元测试和压力测试

package main

import (
    "net/http/httptest"
    "testing"
)

// 模拟的HTTP处理函数
func mockHandler(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Hello, World!"))
}

// 单元测试示例
func TestMockHandler(t *testing.T) {
    // 创建一个模拟的HTTP请求
    req := httptest.NewRequest("GET", "http://example.com/test", nil)
    // 创建一个模拟的HTTP响应记录器
    w := httptest.NewRecorder()

    // 调用我们的处理函数
    mockHandler(w, req)

    // 检查响应是否符合预期
    if w.Code != http.StatusOK {
        t.Errorf("Expected status code %d, got %d", http.StatusOK, w.Code)
    }
    if w.Body.String() != "Hello, World!" {
        t.Errorf("Expected body 'Hello, World!', got '%s'", w.Body.String())
    }
}

// 压力测试示例
func BenchmarkMockHandler(b *testing.B) {
    req := httptest.NewRequest("GET", "http://example.com/test", nil)
    w := httptest.NewRecorder()

    for i := 0; i < b.N; i++ {
        mockHandler(w, req)
        w.Body.Reset()
    }
}