如何在Go中Mock Google Cloud Speech-v2 gRPC语音客户端？

1. 抽象客户端接口 #

先把Google Speech客户端的核心业务方法抽象成接口，让Mock实现可以完全替代真实客户端：

import (
    "context"
    speechpb "cloud.google.com/go/speech/apiv2/speechpb"
)

// SpeechClient 抽象业务用到的STT客户端方法
type SpeechClient interface {
    Recognize(ctx context.Context, req *speechpb.RecognizeRequest, opts ...interface{}) (*speechpb.RecognizeResponse, error)
    // 按需添加其他用到的方法，比如LongRunningRecognize等
}

2. 重构客户端创建逻辑 #

修改原有的客户端创建函数，让它返回抽象接口，同时修正原代码的返回值问题：

import (
    "context"
    "fmt"

    speech "cloud.google.com/go/speech/apiv2"
    "google.golang.org/api/option"
)

// GSTTClient 创建真实的STT客户端，返回抽象接口
func GSTTClient(ctx context.Context) (SpeechClient, error) {
    opts := []option.ClientOption{}
    client, err := speech.NewClient(ctx, opts...)
    if err != nil {
        return nil, fmt.Errorf("failed to create new client: %w", err)
    }
    return client, nil
}

3. 实现Mock客户端 #

用testify/mock库创建Mock实现，匹配上面定义的SpeechClient接口：

import (
    "context"
    speechpb "cloud.google.com/go/speech/apiv2/speechpb"
    "github.com/stretchr/testify/mock"
)

// MockSpeechClient 是SpeechClient的Mock实现
type MockSpeechClient struct {
    mock.Mock
}

// Recognize 实现接口的Recognize方法
func (m *MockSpeechClient) Recognize(ctx context.Context, req *speechpb.RecognizeRequest, opts ...interface{}) (*speechpb.RecognizeResponse, error) {
    args := m.Called(ctx, req, opts)
    return args.Get(0).(*speechpb.RecognizeResponse), args.Error(1)
}

4. 单元测试中使用Mock客户端 #

在测试代码里初始化Mock客户端，预设方法返回值，注入到业务逻辑中：

import (
    "context"
    "testing"
    speechpb "cloud.google.com/go/speech/apiv2/speechpb"
    "github.com/stretchr/testify/assert"
    "github.com/stretchr/testify/mock"
)

func TestYourBusinessLogic(t *testing.T) {
    // 初始化Mock客户端
    mockClient := new(MockSpeechClient)

    // 预设Recognize方法的返回值
    mockResp := &speechpb.RecognizeResponse{
        // 构造测试用的响应数据
    }
    mockClient.On("Recognize", mock.Anything, mock.Anything, mock.Anything).Return(mockResp, nil)

    // 将mockClient传入你的业务函数（业务函数需接收SpeechClient接口作为参数）
    result, err := YourBusinessFunction(context.Background(), mockClient)

    // 断言结果符合预期
    assert.NoError(t, err)
    assert.Equal(t, mockResp, result)

    // 验证Mock方法被正确调用
    mockClient.AssertExpectations(t)
}

核心要点 #

• 接口隔离：通过抽象接口切断业务逻辑与真实客户端的强依赖，让Mock可以无缝替换
• 依赖注入：业务逻辑必须依赖SpeechClient接口而非具体的speech.Client类型，测试时才能传入Mock
• 按需Mock：只实现业务中实际用到的客户端方法，无需全部覆盖

内容的提问来源于stack exchange，提问作者Shashanka