在Go语言中，即使没有定义接口，我们仍然可以实施测试桩（test stubs）。测试桩是在单元测试中用来模拟外部依赖或未实现部分的代码。以下是几种在Go中实现测试桩的方法：
1. 手工编写测试桩： 最直接的方式是手工编写测试桩代码。你可以创建一个单独的文件或在测试文件中定义一个结构体或函数，模拟外部依赖的行为。这种方法简单直接，但可能需要较多的手动编码工作。
2. 使用gomock工具： gomock是Go官方提供的测试框架，它可以帮助我们对接口类型进行mock。通过使用gomock，我们可以创建一个模拟对象来替代真实的依赖。首先需要使用 mockgen 工具生成mock代码，然后在测试中使用这个mock对象。这种方法适合于有接口定义的情况，可以自动生成mock代码，减少手工编码的工作量。
3. 使用monkey打桩工具： monkey是一个强大的打桩工具，它允许你在运行时通过汇编语言重写可执行文件，将目标函数或方法的实现跳转到桩实现。monkey支持为任意函数及方法进行打桩，不仅限于接口。使用monkey时，需要注意它不支持内联函数，并且在并发测试中不可用。安装monkey的命令是 go get bou.ke/monkey 。使用示例如下：
func TestMyFunc(t *testing.T) {
monkey.Patch(varys.GetInfoByUID, func(int64)(*varys.UserInfo, error) {
return &varys.UserInfo{Name: "liwenzhou"}, nil
})
ret := MyFunc(123)
if !strings.Contains(ret, "liwenzhou") {
t.Fatal()
}
}
执行单元测试时，需要添加 -gcflags=-l 参数以防止内联优化。
4. 使用GoStub打桩工具： GoStub是另一个打桩工具，它支持为全局变量、函数等打桩。GoStub通过反射来实现打桩，因此需要对函数进行侵入式修改。对于没有接口定义的情况，可以使用GoStub来模拟函数的行为。安装GoStub的命令是 go get github.com/prashantv/gostub 。使用示例如下：
var GreetFunc = Greet
fmt.Println("Before stub:", GreetFunc("axiaoxin"))
stubs := gostub.Stub(&GreetFunc, func(name string) string {
return "fuck u," + name
})
defer stubs.Reset()
fmt.Println("After stub:", GreetFunc("axiaoxin"))
5. 使用gomonkey框架： gomonkey是另一个类似于monkey的打桩工具，它提供了更多灵活的打桩方式，包括为函数、方法、全局变量等打桩。gomonkey的使用方式与monkey类似，但提供了更多的API来控制打桩行为。
通过上述方法，即使在没有定义接口的情况下，我们也可以有效地实现测试桩，以进行单元测试。这些工具和方法可以帮助我们隔离外部依赖，确保单元测试的独立性和可维护性。
Gomonkey 是一个在 Go 语言单元测试中用于打桩（monkey patching）的库，它允许你在运行时动态地替换函数、方法、全局变量等的行为。以下是一些使用 gomonkey 的示例：
1. Mock 普通函数：
import (
"github.com/agiledragon/gomonkey"
"testing"
)
// 假设有如下函数
func someFunctionToMock() string {
return "original"
}
// 测试函数
func TestSomeFunction(t *testing.T) {
patches := gomonkey.ApplyFunc(someFunctionToMock, func() string {
return "mocked"
})
defer patches.Reset()
// 现在 someFunctionToMock() 将返回 "mocked"
result := someFunctionToMock()
if result != "mocked" {
t.Errorf("Expected 'mocked', got '%s'", result)
}
}
2. Mock 成员方法：
import (
"reflect"
"github.com/agiledragon/gomonkey"
"testing"
)
type MyStruct struct{}
func (ms *MyStruct) MethodToMock() string {
return "original"
}
func TestMyStructMethod(t *testing.T) {
var ms *MyStruct
patches := gomonkey.ApplyMethod(reflect.TypeOf(ms), "MethodToMock", func(_ *MyStruct) string {
return "mocked"
})
defer patches.Reset()
// 现在 ms.MethodToMock() 将返回 "mocked"
result := ms.MethodToMock()
if result != "mocked" {
t.Errorf("Expected 'mocked', got '%s'", result)
}
}
3. Mock 全局变量：
import (
"github.com/agiledragon/gomonkey"
"testing"
)
var globalVar string = "original"
func TestGlobalVar(t *testing.T) {
patches := gomonkey.ApplyGlobalVar(&globalVar, "mocked")
defer patches.Reset()
// 现在 globalVar 将返回 "mocked"
if globalVar != "mocked" {
t.Errorf("Expected 'mocked', got '%s'", globalVar)
}
}
4. Mock 函数序列：
import (
"github.com/agiledragon/gomonkey"
"testing"
)
func functionToMockSequence(a, b int) int {
return a + b
}
func TestFunctionSequence(t *testing.T) {
outputs := []gomonkey.OutputCell{
{Values: gomonkey.Params{2}},
{Values: gomonkey.Params{4}},
{Values: gomonkey.Params{6}},
}
patches := gomonkey.ApplyFuncSeq(functionToMockSequence, outputs)
defer patches.Reset()
// 函数将按顺序返回 2, 4, 6
result1 := functionToMockSequence(1, 1)
if result1 != 2 {
t.Errorf("Expected 2, got %d", result1)
}
result2 := functionToMockSequence(2, 2)
if result2 != 4 {
t.Errorf("Expected 4, got %d", result2)
}
result3 := functionToMockSequence(3, 3)
if result3 != 6 {
t.Errorf("Expected 6, got %d", result3)
}
}
在使用 gomonkey 时，需要注意的是，由于 Go 语言编译器会进行内联优化，可能会影响打桩的效果。为了确保打桩有效，可以在执行测试时加上 -gcflags=all=-l 参数来禁用内联优化。
这些示例展示了如何使用 gomonkey 来打桩不同的代码实体，从而在单元测试中隔离依赖和模拟行为。
gomonkey 是一个 Go 语言的库，它提供了在单元测试中进行 Monkey Patching 的能力，即在运行时动态替换函数、方法、全局变量等的行为。这种能力对于单元测试尤其有用，因为它允许测试者隔离依赖项并模拟它们的行为。
gomonkey 模拟替换函数的原理
在 Go 语言中，函数调用是通过函数指针来实现的。每个函数在编译后都会在内存中有一个确定的地址，函数调用实际上是跳转到这个内存地址执行指令。gomonkey 利用了这一特性，通过以下步骤实现函数的模拟替换：
1. 获取原函数地址：使用 reflect 包获取原函数的内存地址。
2. 生成跳转指令：构造汇编指令，使得当调用原函数时，实际上是跳转到模拟函数（桩函数）的地址执行。
3. 修改原函数指令：将原函数开头的机器指令替换为跳转到模拟函数的指令。
4. 保存原函数指令：在替换原函数指令时，保存原函数的指令，以便在测试结束后可以恢复原函数的行为。
具体来说，gomonkey 通过以下技术手段实现打桩：
汇编指令：在不同操作系统下，gomonkey 使用不同的汇编指令来实现函数的跳转。例如，在 amd64 架构下，使用 jmp 指令实现跳转 。
内存修改：使用系统调用（如 syscall.Mprotect ）修改内存保护，允许修改原函数的机器代码 。
反射：使用 Go 语言的反射机制来获取函数的地址和修改函数的行为 。
注意事项
禁用内联优化：由于 Go 编译器可能会对简短的函数进行内联优化，这会导致无法通过地址替换来打桩。因此，在运行测试时需要禁用内联优化，可以通过添加编译参数 -gcflags=all=-l 来实现 。
线程安全：gomonkey 不是线程安全的，如果在并发环境下使用，可能会导致问题 。
通过上述原理和方法，gomonkey 为 Go 语言的单元测试提供了强大的模拟和打桩能力，使得测试更加灵活和可控。