包中的模拟方法

Question

我希望模拟一些用于单元测试的方法。不幸的是，代码的结构不是很好。

var config = struct { abc *abc }

func Init(arg1) {
    // config.abc = newAbc(arg2, arg3)
}   

func UnitTestThis() { 
    //some code here 

    config.abc.Search(arg4,arg5) 

    //code here 
}

如何对 UnitTestThis 函数进行单元测试，模拟 Search 方法的结果？我一直在尝试创建一个接口并模拟方法，但无法这样做。

Answer 1

如果config.abc 字段是具体类型（或指向具体类型的指针），则不能真正模拟它。

你需要一些重构。

带接口

最好将config.abc 的类型更改为接口类型，因此在测试中您可以创建自己的实现并分配给它，您可以在其中做任何您想做的事情。

在config.abc.Search 上使用方法值

另一种选择是创建一个函数类型的变量来保存config.abc.Search() 的method value，并在测试中为其分配一个新值，例如function literal。

这就是它的样子：

var config = struct{ abc *abc }{}

var searchFunc func(arg4Type, arg5Type)

func Init(arg1) {
    config.abc = newAbc(arg2,arg3)
    searchFunc = config.abc.Search
}

func UnitTestThis() {
    //some code here
    searchFunc(arg4, arg5)
    //code here
}

在此处了解更多信息：Is it possible to mock a function imported from a package in golang?