如何在 golang 中构建模拟类？

Question

我正在尝试为 golang 中的单元测试构建模拟类；有谁知道这是怎么做到的吗？例如，在下面的代码片段中，我想打印 FakeMyClass.Object 的返回值。

package main

import (

    "fmt"

)

type MyClass struct {

}

func (*MyClass) Object() (int) {

    return 0

}

func (mc *MyClass) PrintInfo() {

    fmt.Printf("%v\n", 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/mockpackage mainimport (    "fmt"    "github.com/stretchr/testify/mock")type MyClass struct {}func (*MyClass) Object() (int) {    return 0}func (mc *MyClass) PrintInfo() {    fmt.Printf("%v\n", 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("%v\n", fmc.Object())}func main() {    mc := &FakeMyClass{}    mc.On("Object").Return(1)    mc.PrintInfo()    //mc.AssertExpectations(t)}

Answer 2

我为此使用了 mockgen：mockgen -source myModule.go -package myPackage -destination myModuleMock.go您可以通过以下方式安装它：get get github.com/golang/mock

Answer 3

您正在寻找该Object()方法的虚拟调度。我所知道的 Go 中唯一的虚拟调度形式是使用接口。PrintInfo然后将是一个使用该接口类型的函数。例如：package mainimport (    "fmt")type WithObject interface {    Object() int}func PrintInfo(wo WithObject) {    fmt.Printf("%v\n", 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)}