Go 语言测试：单元测试、集成测试、端到端测试完全指南

引言：为何在 Go 中进行测试很重要

Go（Golang）是一种从设计之初就专注于简洁性、性能和可靠性的语言。内置的 testing 包以及 go test 等强大的工具，使得编写测试成为开发过程中不可或缺的一部分。没有高质量的测试，即使是最优雅的代码，最终也会变成一个“脆弱”的系统，一处改动就可能引发全面崩溃。

在本文中，我们将深入探讨测试的三个关键级别：单元测试、集成测试和 端到端测试（E2E）。您将了解它们之间的区别、何时应用它们，以及如何在 Go 中编写有效的测试。本材料对于希望系统化自身知识的初学者和经验丰富的开发者都将有所裨益。

1. 单元测试

单元测试是测试金字塔的基石。它们以隔离的方式（不依赖数据库、网络、文件系统等外部依赖）检查单个函数、方法或模块的运行情况。

Go 中单元测试的基本原则

• 隔离性： 被测试的代码不应依赖于外部服务。
• 速度快： 单元测试应在毫秒级执行完毕。
• 确定性： 测试结果不应因运行环境或运行次数不同而改变。

简单单元测试示例

假设我们有一个两数相加的函数：

// math.gopackage math

func Add(a, b int) int {    return a + b}

该函数的测试代码如下：

// math_test.gopackage math

import "testing"

func TestAdd(t *testing.T) {    result := Add(2, 3)    expected := 5

    if result != expected {        t.Errorf("Add(2, 3) = %d; want %d", result, expected)    }}

通过以下命令运行测试：go test ./...

表格驱动测试

Go 社区更倾向于使用表格驱动测试来检查多种测试用例：

func TestAddTable(t *testing.T) {    tests := []struct {        a, b, expected int    }{        {1, 1, 2},        {0, 0, 0},        {-1, 1, 0},        {100, 200, 300},    }

    for _, tt := range tests {        result := Add(tt.a, tt.b)        if result != tt.expected {            t.Errorf("Add(%d, %d) = %d; want %d", tt.a, tt.b, result, tt.expected)        }    }}

小贴士： 使用 t.Run() 创建名称清晰的嵌套测试，这有助于提高测试报告的可读性。

2. 集成测试

集成测试用于检查系统各组件（如数据库、API、外部服务）之间的交互。与单元测试不同，集成测试会启动真实（或接近真实）的依赖项来运行。

Go 中集成测试的特点

• 通常需要搭建环境（例如，一个测试用数据库）。
• 可以使用 构建标签（build tags） 来区分单元测试和集成测试。
• 运行速度较慢，因此通常单独执行。

示例：测试数据库操作（PostgreSQL）

我们创建一个带有 integration 构建标签的文件：

// db_test.go//go:build integration

package db

import (    "database/sql"    "testing")

func TestSaveUser(t *testing.T) {    // 连接到测试数据库    dsn := "host=localhost user=test password=test dbname=testdb sslmode=disable"    db, err := sql.Open("postgres", dsn)    if err != nil {        t.Fatalf("failed to connect to db: %v", err)    }    defer db.Close()

    // 在测试前清理数据    _, _ = db.Exec("TRUNCATE TABLE users")

    // 执行被测试的操作    user := User{Name: "Alice", Email: "alice@example.com"}