如何使用 stretr/testify 模拟 class 字段？

Question

我正在使用https://github.com/stretchr/testify go 库来模拟 class。我想为 aws 模拟一个类的字段 sdk package：github： github.com/aws/aws-sdk-go-v2/service/apigatewaymanagementapi 。

我的代码是：

func publishData(ctx context.Context, api *apigatewaymanagementapi.Client) {

   ...
   api.PostToConnection(...)
}

我需要模拟PostToConnection方法。 我像这样设置模拟：

type MockApi struct {
    apigatewaymanagementapi.Client
    mock.Mock
}

func (m *MockApi) PostToConnection() (*apigatewaymanagementapi.PostToConnectionOutput, error) {
    m.Called()
    return nil, nil
}

在我的测试用例中，

api := MockApi{}
api.On("PostToConnection").Return(nil, nil)

模拟不起作用，代码仍然使用真实的Client 。 我认为问题是模拟只模拟根级方法而不是Client.PostToConnection方法。

我如何模拟Client下的方法？

Answer 1

将 *apigatewaymanagementapi.Client 替换为您自己的界面。 对于这个接口，在测试中，你可以自己写一个实现（存根）或者使用一个库来生成它（比如https://github.com/golang/mock ）。

例子：

type (
    PublisherClient interface {
        PostToConnection(ctx context.Context, someArgs ... interface{})
    }
)

func publishData(ctx context.Context, api PublisherClient) {
    api.PostToConnection(ctx)
}

type (
    PublisherClientMock struct {
        Args []interface{}
    }
)

func (c PublisherClientMock) PostToConnection(ctx context.Context, someArgs ... interface{}) {
    c.Args = someArgs
}