如何在golang中构建模拟类

Question

我试图在golang中构建单元测试的模拟类;有人知道怎么做吗?例如，在下面的代码片中，我想打印FakeMyClass。对象的返回值。

package main
import (
    "fmt"
)
type MyClass struct {
}
func (*MyClass) Object() (int) {
    return 0
}
func (mc *MyClass) PrintInfo() {
    fmt.Printf("%vn", mc.Object())
}
type FakeMyClass struct {
    MyClass
}
func (*FakeMyClass) Object() (int) {
    return 1
}
func main() {
    mc := &FakeMyClass{}
    mc.PrintInfo()
}

Answer 1

我更喜欢用github.com/stretchr/testify/tree/master/mock

package main
import (
    "fmt"
    "github.com/stretchr/testify/mock"
)
type MyClass struct {
}
func (*MyClass) Object() (int) {
    return 0
}
func (mc *MyClass) PrintInfo() {
    fmt.Printf("%vn", mc.Object())
}
type FakeMyClass struct {
    mock.Mock
}
func (fmc *FakeMyClass) Object() (int) {
    args := fmc.Called()
    return args.Int(0)
}
func (fmc *FakeMyClass) PrintInfo() {
    fmt.Printf("%vn", fmc.Object())
}
func main() {
    mc := &FakeMyClass{}
    mc.On("Object").Return(1)
    mc.PrintInfo()
    //mc.AssertExpectations(t)
}

Answer 2

我使用了mockgen:

mockgen -source myModule。

你可以安装它:得到github.com/golang/mock

Answer 3

您可以创建一个返回模拟函数的结构体，您可以在每个测试用例中传递您想要返回的值。例如:

package main
import (
    "fmt"
)
type MyClass struct{}
func (*MyClass) Object() int {
    return 0
}
func (mc *MyClass) PrintInfo() {
    fmt.Printf("%vn", mc.Object())
}
type FakeMyClass struct {
     ObjectMock: func() int
}
func (fm *FakeMyClass) Object() int {
    return fm.ObjectMock()
}
func (fm *FakeMyClass) PrintInfo() {
    fmt.Printf("%vn", fm.Object())
}
func main() {
    mc := &FakeMyClass{
        // Create the mocked function here, to be called when the Object() function is called.
        ObjectMock: func() int {
            return 0
        }
    }
    mc.PrintInfo()
}

Answer 4

您正在寻找Object()方法上的虚拟调度。在Go中，我所知道的唯一一种虚拟分派形式是使用接口。

PrintInfo将是一个使用该接口类型的函数。

例如:

package main
import (
    "fmt"
)
type WithObject interface {
    Object() int
}
func PrintInfo(wo WithObject) {
    fmt.Printf("%vn", wo.Object())
}
type MyClass struct {
}
func (*MyClass) Object() (int) {
    return 0
}
type FakeMyClass struct {
    MyClass
}
func (*FakeMyClass) Object() (int) {
    return 1
}
func main() {
    fmc := &FakeMyClass{}
    PrintInfo(fmc)
}