Golang Go语言中写了一个 mock 任何变量的函数

发布于 1周前 作者 nodeper 来自 Go语言

Golang Go语言中写了一个 mock 任何变量的函数

在写测试的时候, 有一种 mock 方法是这样的

var foo = func(){}

这个时候可以替换 foo, 然后跑测试, 最后在还原(隔离)

如果这么写要很长很长....

有了 patch 方法, 只需要两行

func TestH(t *testing.T) {
 say := func() {
  fmt.Println("say")
 }
 rec := patchFunction(&say, func() { fmt.Println("replace") })
 say()
 defer rec() # 还原
}

func patchFunction(item interface{}, new interface{}) func() {
old := reflect.ValueOf(item).Elem().Interface()
rec := func() {
reflect.ValueOf(item).Elem().Set(reflect.ValueOf(old))
}
reflect.ValueOf(item).Elem().Set(reflect.ValueOf(new))
return rec
}

另外求教一下, reflect.valueOf 一个 interface 拿到的直接就是里面的包的元素?? interface 内存里不是两部分吗, 我总觉得应该是 .Elem()拿到包裹的元素, 然后在.Elem().Set()改指针值.


更多关于Golang Go语言中写了一个 mock 任何变量的函数的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html

Harmonyos Next教程
1 回复

更多关于Golang Go语言中写了一个 mock 任何变量的函数的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html


在 Go 语言中，创建一个可以 mock 任何变量的函数确实是一个高级话题，通常用于测试目的。不过，由于 Go 语言的静态类型和编译时检查机制，直接创建一个通用的 mock 函数是有限制的。但你可以使用一些技术来近似实现这一目标，比如使用接口和反射。

首先，你可以定义一个通用的接口，然后在测试中使用这个接口来 mock 具体的实现。例如：

type Mockable interface {
    MockMethod()
}

func someFunction(m Mockable) {
    m.MockMethod()
}

// 测试时
type MockStruct struct{}

func (m MockStruct) MockMethod() {
    // Mock 实现
}

func TestSomeFunction(t *testing.T) {
    mock := MockStruct{}
    someFunction(mock)
}

如果你需要更动态的行为，可以考虑使用反射（reflect 包），但这通常会使代码更加复杂且难以维护。反射允许你在运行时检查和操作变量的类型和值，但滥用反射可能会导致代码可读性下降和性能问题。

另一种方法是使用 Go 的测试框架（如 testify 或 gomock），这些框架提供了更高级别的抽象和工具来创建和管理 mock 对象。

总之，虽然 Go 语言没有内置的通用 mock 机制，但通过接口、反射和测试框架，你可以实现灵活的 mock 逻辑来支持你的测试需求。在实际项目中，应根据具体情况选择最适合的方法。

回到顶部