单元测试框架设计注重简洁、可拓展和并发性。框架围绕 testing 软件包、test 函数和辅助子构建。辅助子提供断言和记录功能，而 test 函数设置测试，执行代码并断言。框架支持并发测试，使用 waitgroup 和 t.parallel() 确保顺序执行。

Go 单元测试框架的设计：深入剖析

单元测试是软件开发的一个关键方面，它有助于确保代码的正确性和健壮性。Go 语言提供了强大的单元测试框架，该框架的设计考虑了简单性、可扩展性和并发性。

框架结构

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

Go 单元测试框架围绕以下组件构建：

• testing 软件包：定义了单元测试的 API 和实用程序。
• 测试函数：以 Test 开头并接受一个 *testing.T 指针。
• 辅助子：由 *testing.T 提供的用于断言、记录和基准测试的函数。

测试函数

所有单元测试都被组织成 _test.go 文件中以 Test 开头的函数。这些函数负责设置测试所需的条件，执行代码并使用辅助子进行断言。

import "testing"

func TestAdd(t *testing.T) {
    result := Add(2, 3)
    if result != 5 {
        t.Errorf("Add(2, 3) should be 5, got %d instead", result)
    }
}

辅助子

*testing.T 指针提供了许多可用于断言和记录测试信息的辅助子。常用辅助子包括：

• Error(): 记录测试错误。
• Errorf(): 记录测试错误并格式化输出。
• Fail(): 断言失败。
• FailNow(): 立即断言失败并停止测试。

并发测试

Go 单元测试框架支持并发测试。为了确保并发测试的正确顺序执行，框架使用了 sync.WaitGroup 和 t.Parallel()。

import (
    "sync/atomic"
    "sync/WaitGroup"
    "testing"
)

func TestConcurrentAdd(t *testing.T) {
    const (
        numThreads = 10
        numIterations = 1000
    )
    var wg WaitGroup

    atomic.StoreInt64(&counter, 0)

    for i := 0; i < numThreads; i++ {
        wg.Add(1)

        go func() {
            for j := 0; j < numIterations; j++ {
                Add(1)
            }

            wg.Done()
        }()
    }

    wg.Wait()
    if counter != int64(numThreads*numIterations) {
        t.Errorf("ConcurrentAdd failed, got %d instead", counter)
    }
}

实战案例

以下是一个如何编写和运行单元测试的简单实战案例：

// my_test.go
package mypackage

import "testing"

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

func TestAdd(t *testing.T) {
    result := Add(2, 3)
    if result != 5 {
        t.Errorf("Add(2, 3) should be 5, got %d instead", result)
    }
}

// main.go
package main

import (
    "mypackage"
    "testing"
)

func main() {
    testing.Main(, mypackage.matchAll())
}

要运行测试，请执行以下命令：

go test mypackage