在 go 函数链中进行单元测试至关重要,以确保链中各个函数的预期行为。为此,需要设置函数链和测试函数,并使用模拟来测试函数链中函数的行为。具体步骤包括:创建函数链及其测试函数。使用模拟函数来测试链中其他函数的行为。针对不同输入和预期输出编写测试用例。循环执行测试,并检查实际输出与预期输出是否一致。
Go 函数链中的单元测试
在 Go 中,函数链可以通过将函数的输出作为另一个函数的输入来创建。在单元测试中,测试这种函数链至关重要,以确保整个链的预期行为。
设置
立即学习“go语言免费学习笔记(深入)”;
为了进行单元测试,我们需要一个函数链以及一个测试函数。例如,我们创建一个计算三角形面积的程序:
import "math"
func area(base, height float64) float64 {
return 0.5 * base * height
}
func triangleArea(base, height float64) float64 {
return area(base, height)
}
测试函数
接下来,我们编写一个 TestTriangleArea 函数来测试 triangleArea 函数:
import "testing"
func TestTriangleArea(t *testing.T) {
// 测试输入和预期输出
tests := []struct {
base, height, expectedArea float64
}{
{3, 4, 6},
{5, 7, 17.5},
{-1, 2, 0}, // 输入无效时返回 0
}
// 循环执行测试
for _, test := range tests {
actualArea := triangleArea(test.base, test.height)
if actualArea != test.expectedArea {
t.Errorf("triangleArea(%g, %g) = %g, want %g",
test.base, test.height, actualArea, test.expectedArea)
}
}
}
模拟
在 TestTriangleArea 中,我们使用模拟来测试 area 函数的行为。具体来说,我们可以创建一个模拟的 area 函数,返回我们自己的值。这允许我们在不修改实际 area 函数的情况下测试函数链:
func TestTriangleAreaWithMock(t *testing.T) {
// 创建一个模拟 area 函数
mockArea := func(base, height float64) float64 {
return base * height
}
// 使用模拟函数调用 triangleArea
actualArea := triangleArea(3, 4)
// 断言预期输出
if actualArea != 12 {
t.Errorf("triangleArea(%g, %g) = %g, want 12", 3, 4, actualArea)
}
}
实战案例
这是一个更复杂的函数链的实战案例:
import "strings"
func toUpper(s string) string {
return strings.ToUpper(s)
}
func removeWhitespace(s string) string {
return strings.ReplaceAll(s, " ", "")
}
func prepareString(s string) string {
return toUpper(removeWhitespace(s))
}
测试函数链
可以使用类似于前面的示例的技术测试这种函数链:
import "testing"
func TestPrepareString(t *testing.T) {
tests := []struct {
input, expected string
}{
{"hello world", "HELLOWORLD"},
{" test string ", "TESTSTRING"},
{"", ""},
}
for _, test := range tests {
actual := prepareString(test.input)
if actual != test.expected {
t.Errorf("prepareString(%s) = %s, want %s",
test.input, actual, test.expected)
}
}
}