库安装
首先，安装 Mock 类生成工具 Mockery：
go install github.com/vektra/mockery/v2@v2.45.1
实际上，你也可以手动创建 Mock 类。
生成 Mock 类
假设你在 internal/metrics 包下有如下定义的接口：
package metricstype Getter[T any] interface {Get() (T, error)
}
在项目根目录，可以使用以下命令生成 Mock 类：
mockery --name=Getter --dir=internal/metrics
生成的 Mock 类会在 mocks 目录下的 getter.go 文件中。
编写用例
package metricsimport ("testing"mocks "xxx/mock/internal_/metrics""github.com/stretchr/testify/suite"
)type GetterTestSuite struct {suite.Suite
}func TestGetter(t *testing.T) {suite.Run(t, new(GetterTestSuite))
}func (t *GetterTestSuite) TestGetterInt() {t.T().Logf("TestGetterInt run")getter := new(mocks.Getter[int])getter.On("Get").Return(1, nil)val, err := getter.Get()t.Nil(err)t.Equal(1, val)
}说明：
- GetterTestSuite 是测试集的名称，每个method都会作为测试用例调用。TestGetter 函数运行时，会调用 TestGetterInt。
- TestGetterInt 中引用的
t是TestSuite，包含许多有用的断言函数，如Equal和Nil等。 - 创建 Mock 实例后，可以使用
On方法来标记方法对应的返回值。假设Get方法可以传递参数，则可以根据不同的参数选择不同的返回值。
Mock 常见用法：
假设 mockObj 是 Mock 类的实例：
mockObj.On("GetApiKey", mock.Anything).Return("dummy_api_key")：GetApiKey有一个参数，且无论传什么，都会返回dummy_api_key。mockObj.On("GetAllClusterInfo").Maybe().Return(GenerateTestClustersInfo())：如果使用Maybe，则GetAllClusterInfo不一定必须被调用；如未使用Maybe且函数未被调用，则断言将失败。mockObj.On("RunCleanup", true, true).Once().Return(nil, nil)：RunCleanup有两个参数，所以需要传递两个 Mock 的值进入。Once表示这个函数只应该被调用一次。mockObj.AssertNumberOfCalls(t.T(), "RunCleanup", 4)：可以检查方法的调用次数。
通过这些用法，用户可以完全控制 Mock 类的每个方法的行为，并进行一些检查以完善整个测试。
