在 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)
}

测试函数

Golang Spf13 Viper

Go 配置库，使用 spf13/viper — 分层优先级（flag > env >file > KV > default），提供 BindPFlag/BindPFlags、SetEnvPrefix + SetEnvKeyReplace 等功能。

下载

接下来，我们编写一个 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)
        }
    }
}