Go 单元测试入门：轻松掌握 Golang 单元测试基础

2023-02-28 15:03:37

Go 单元测试：代码质量和可靠性的保证

介绍

软件开发是一个复杂的流程，其中包含编写、测试和维护代码等多个步骤。单元测试是软件开发生命周期中不可或缺的一部分，它有助于识别和隔离代码中的错误，确保其质量和可靠性。Go 语言提供了全面的单元测试框架，名为 testing，它提供了丰富的函数和类型，使开发者能够轻松编写和执行测试。

Go 单元测试基础

创建测试文件

要创建单元测试，首先需要在项目中创建一个新的文件，并以 _test.go 作为后缀名。例如，如果你要测试 my_module.go，那么你的测试文件应该命名为 my_module_test.go。

导入 testing 包

在测试文件中，你需要导入 testing 包，它包含了所有必要的函数和类型来编写和运行测试。

测试函数

测试函数是用于测试特定代码逻辑的函数。它必须以 Test 开头，后面跟一个性名称。例如，以下函数测试 AddNumbers 函数：

func TestAddNumbers(t *testing.T) {
	// ...
}

断言

断言用于验证测试结果是否符合预期。如果断言失败，测试就会失败。testing 包提供了多种断言函数，例如：

t.Equal(a, b)：检查 a 和 b 是否相等
t.Assert(a != nil)：检查 a 是否不为 nil
t.Run("subtest", func(t *testing.T) { ... })：子测试

运行测试

可以使用 go test 命令来运行测试。在项目根目录下运行以下命令：

go test

Go 单元测试高级用法

基准测试

基准测试用于衡量代码的性能。要编写基准测试，请使用 Benchmark 前缀代替 Test 前缀。例如：

func BenchmarkAddNumbers(b *testing.B) {
	// ...
}

代码覆盖率

代码覆盖率衡量了测试所执行的代码量。要生成代码覆盖率报告，请使用 -cover 标志。例如：

go test -cover

表驱动测试

表驱动测试使你能够使用一组输入和预期的输出来编写测试。要编写表驱动测试，请使用 testing.T.Run 函数。例如：

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

	for _, test := range tests {
		t.Run("case_"+strconv.Itoa(test.a), func(t *testing.T) {
			actual := AddNumbers(test.a, test.b)
			if actual != test.expected {
				t.Errorf("Expected %d, got %d", test.expected, actual)
			}
		})
	}
}

集成测试

集成测试用于测试多个组件如何协同工作。要编写集成测试，可以使用 testing.T.Run 函数创建子测试，并调用其他模块或服务。例如：

func TestDatabaseIntegration(t *testing.T) {
	t.Run("create_record", func(t *testing.T) {
		db := ConnectDatabase()
		record := NewRecord("name", 10)
		err := db.CreateRecord(record)
		if err != nil {
			t.Errorf("Error creating record: %s", err)
		}
	})
}