单元测试是测试单个功能或组件的隔离性测试，使用 testing 包编写；集成测试是测试多个组件或系统交互的端到端测试，可以使用 httptest 包进行。1. 单元测试：编写单元测试来验证单个功能或组件的正确性。2. 集成测试：编写集成测试来验证多个组件或系统交互的正确性。3. 实战案例：展示单元测试和集成测试在验证 getuser 函数中的应用。

Go 框架的单元测试与集成测试

单元测试

单元测试是测试单个功能或组件的隔离性测试。在 Go 中，我们可以使用 testing 包编写单元测试。

立即学习“go语言免费学习笔记（深入）”；

package mypackage

import (
    "testing"
)

func TestAdd(t *testing.T) {
    sum := Add(1, 2)
    if sum != 3 {
        t.Errorf("Add(1, 2) = %d; want 3", sum)
    }
}

集成测试

Golang Spf13 Viper

Go 配置库，使用 spf13/viper — 分层优先级（flag > env >file > KV > default），提供 BindPFlag/BindPFlags、SetEnvPrefix + SetEnvKeyReplace 等功能。

下载

集成测试是测试多个组件或系统交互的端到端测试。它们比单元测试复杂，但可以确保整个系统的正确性。在 Go 中，我们可以使用 httptest 包进行集成测试。

package mypackage

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

func TestHelloWorldHandler(t *testing.T) {
    req, err := http.NewRequest(http.MethodGet, "/", nil)
    if err != nil {
        t.Fatalf("http.NewRequest: %v", err)
    }
    rr := httptest.NewRecorder()
    HelloWorldHandler(rr, req)
    if rr.Code != http.StatusOK {
        t.Errorf("HelloWorldHandler returned status %d; want %d", rr.Code, http.StatusOK)
    }
    if rr.Body.String() != "Hello, World!" {
        t.Errorf("HelloWorldHandler returned body %q; want %q", rr.Body.String(), "Hello, World!")
    }
}

实战案例

假设我们有一个 GetUser 函数，可以从数据库中获取用户。我们可以编写单元测试和集成测试来验证其正确性。

单元测试

package mypackage

import (
    "database/sql"
    "testing"
)

func TestGetUser(t *testing.T) {
    db, err := sql.Open("postgres", "host=localhost user=root password=root")
    if err != nil {
        t.Fatalf("sql.Open: %v", err)
    }
    user, err := GetUser(db, 1)
    if err != nil {
        t.Fatalf("GetUser: %v", err)
    }
    if user.ID != 1 || user.Name != "John Doe" {
        t.Errorf("GetUser returned unexpected user: %v", user)
    }
}

集成测试

package mypackage

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

func TestGetUserHandler(t *testing.T) {
    db, err := sql.Open("postgres", "host=localhost user=root password=root")
    if err != nil {
        t.Fatalf("sql.Open: %v", err)
    }
    req, err := http.NewRequest(http.MethodGet, "/users/1", nil)
    if err != nil {
        t.Fatalf("http.NewRequest: %v", err)
    }
    rr := httptest.NewRecorder()
    GetUserHandler(db, rr, req)
    if rr.Code != http.StatusOK {
        t.Errorf("GetUserHandler returned status %d; want %d", rr.Code, http.StatusOK)
    }
    if rr.Body.String() != "{\"id\":1,\"name\":\"John Doe\"}" {
        t.Errorf("GetUserHandler returned body %q; want %q", rr.Body.String(), "{\"id\":1,\"name\":\"John Doe\"}")
    }
}