如何使用 Go 标准库进行单元测试

go 标准库通过 testing 包提供了单元测试功能，只需创建 _test.go 文件并编写测试函数即可。测试函数使用断言函数，如 assertequal 和 asserttrue，比较预期结果和实际结果。测试通过或失败的信息将通过 go test 命令显示。

如何使用 Go 标准库进行单元测试

Go 标准库提供了 testing 包，它提供了丰富的功能，用于编写和运行单元测试。本文将指导你如何使用 testing 包来测试你的 Go 代码。

设置单元测试

要为 Go 程序设置单元测试，请按照以下步骤操作：

1. 在项目的根目录中创建一个名为 your_package_name_test.go 的文件。
2. 导入 testing 包：

import "testing"

编写测试函数

每个单元测试都作为一个独立的函数编写，名为 func TestX(t *testing.T)。其中 X 是测试案例的名称，而 t 是一个 *testing.T 类型的指针，它提供了用于执行测试、报告结果和记录错误的方法。

func TestPassingTest(t *testing.T) {
    // 测试用例的代码...
}

断言

testing 包提供了多种断言函数来比较测试结果和预期值。常用的断言函数包括：

• AssertEqual(t, expected, actual)：检查 expected 和 actual 是否相等
• AssertNotEqual(t, value1, value2)：检查 value1 和 value2 是否不相等
• AssertTrue(t, condition)：断言条件 condition 为真
• AssertFalse(t, condition)：断言条件 condition 为假

实战案例

让我们为一个计算两个数字之和的函数编写一个单元测试。

知识画家

AI交互知识生成引擎，一句话生成知识视频、动画和应用

下载

package my_package

import (
    "testing"
)

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

单元测试

import (
    "testing"

    "my_package"
)

func TestSum(t *testing.T) {
    tests := []struct {
        a        int
        b        int
        expected int
    }{
        {1, 2, 3},
        {5, 7, 12},
    }

    for _, tc := range tests {
        result := my_package.Sum(tc.a, tc.b)
        if result != tc.expected {
            t.Errorf("Sum(%d, %d) returned %d, expected %d", tc.a, tc.b, result, tc.expected)
        }
    }
}

在单元测试中，我们创建了一个 []struct 类型的测试用例列表，其中包含输入值和预期的输出。我们使用循环遍历测试用例，并针对每个用例使用 AssertEqual 断言来验证计算的结果与预期结果是否一致。

运行单元测试

要运行单元测试，可以在命令行中运行以下命令：

go test

它将在项目目录及其子目录中查找所有 _test.go 文件并运行其中的测试。成功运行的测试将显示为绿色通过信息，而失败的测试将显示为红色错误信息。