首页 > 文章列表 > 如何使用单元测试框架对 Golang 函数进行测试?

如何使用单元测试框架对 Golang 函数进行测试?

单元测试 golang
186 2024-04-23

Go 中使用单元测试框架进行单元测试:导入 testing 包。编写以 Test 为前缀的单元测试函数。使用断言函数(如 assertEqual())验证测试结果。运行单元测试(go test),验证函数的正确性。

如何使用单元测试框架对 Golang 函数进行测试?

如何使用单元测试框架对 Go 函数进行测试

单元测试是验证单个函数或方法正确性的至关重要的软件开发实践。在 Go 中,使用单元测试框架(如 testing 包)进行单元测试非常简单。

安装单元测试框架

testing 包是 Go 标准库的一部分。要使用它,请在你的项目中导入 testing 包:

import "testing"

编写单元测试

每个单元测试都是一个以 Test 为前缀的函数。它接受一个指针 *testing.T 作为参数,该参数用于报告测试结果和记录失败信息。

基本的单元测试函数如下所示:

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

断言

testing 包提供了一系列断言函数,用于验证测试结果。常用的断言包括:

  • assertEqual()
  • assertNotEqual()
  • assertNil()
  • assertError()

实战案例

考虑一个计算两个数字之和的 Add() 函数:

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

我们可以编写一个单元测试来验证此函数:

import "testing"

func TestAdd(t *testing.T) {
    tests := []struct {
        a, b, want int
    }{
        {1, 2, 3},
        {0, 0, 0},
        {-1, -1, -2},
    }

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

运行测试

要运行单元测试,请使用 go test 命令。它将查找并运行项目中所有 Test* 函数。

go test

如果测试通过,它将输出以下结果:

PASS
ok  my_project/my_package  1.730s

否则,它将报告失败的测试,包括错误消息。