Go单元测试中Mock函数返回值的简易实现方法咨询

针对你的需求，不用结构体或接口也能轻松Mock foo() 函数，核心思路是把函数改成可赋值的变量，测试时临时替换它的实现。

第一步：调整原代码 #

把原本的foo函数改成函数类型变量，保留原逻辑作为默认实现：

package main

import (
    "fmt"
    "math/rand"
)

// 将foo定义为函数类型变量，默认指向原判断奇偶的逻辑
var foo = func(i int) bool {
    return i%2 == 0
}

func main() {
    i := rand.Intn(10)
    fooResult := foo(i)

    if fooResult {
        fmt.Printf("%d is even!", i)
        // more code ... 
    } else {
        fmt.Printf("%d is odd!", i)
        // more code ... 
    }
}

第二步：编写单元测试 #

在同目录下创建main_test.go，通过替换foo变量的实现来Mock返回值，同时捕获标准输出验证逻辑：

package main

import (
    "bytes"
    "fmt"
    "testing"
)

func TestMain_EvenOutput(t *testing.T) {
    // 保存原foo实现，测试结束后恢复，避免影响其他测试
    originalFoo := foo
    defer func() { foo = originalFoo }()

    // Mock foo：不管输入什么都返回true
    foo = func(int) bool { return true }

    // 捕获fmt的输出内容
    var outputBuf bytes.Buffer
    originalStdout := fmt.Stdout
    defer func() { fmt.Stdout = originalStdout }()
    fmt.Stdout = &outputBuf

    // 执行main函数
    main()

    // 验证输出是否符合预期
    if !bytes.Contains(outputBuf.Bytes(), []byte("is even!")) {
        t.Fatalf("期望输出包含'is even!'，实际输出：%s", outputBuf.String())
    }
}

func TestMain_OddOutput(t *testing.T) {
    originalFoo := foo
    defer func() { foo = originalFoo }()

    // Mock foo：不管输入什么都返回false
    foo = func(int) bool { return false }

    var outputBuf bytes.Buffer
    originalStdout := fmt.Stdout
    defer func() { fmt.Stdout = originalStdout }()
    fmt.Stdout = &outputBuf

    main()

    if !bytes.Contains(outputBuf.Bytes(), []byte("is odd!")) {
        t.Fatalf("期望输出包含'is odd!'，实际输出：%s", outputBuf.String())
    }
}

方法说明 #

• 这种方式不需要引入任何第三方库，完全用Go原生语法实现
• 通过defer确保测试后恢复原函数和标准输出，保证测试的独立性
• 既可以Mock固定返回值，也可以根据输入定制Mock逻辑（比如特定输入返回特定值）

内容的提问来源于stack exchange，提问作者Qaldak