如何在Golang的testify/mock中从Mock返回另一个Mock？

错误原因 #

你遇到的panic是因为mockStorage.Bucket方法声明返回*storage.BucketHandle，但实际返回的是*mockBucketHandle——这是两个完全不同的指针类型，Go无法直接完成类型转换，导致转换失败。

解决方案 #

要在Mock调用中返回Mock结构体，核心是让Mock类型满足目标接口的要求，而非直接返回与原结构体类型不兼容的Mock对象。具体步骤如下：

1. 定义依赖接口 #

在业务代码中，不要直接依赖storage.Client和storage.BucketHandle这类具体结构体，而是定义对应的抽象接口。这既符合依赖倒置原则，也能让Mock更易实现：

// 存储客户端抽象接口
type StorageClientInterface interface {
    Bucket(bucket string) BucketHandleInterface
}

// Bucket句柄抽象接口
type BucketHandleInterface interface {
    Object(name string) *storage.ObjectHandle
}

原storage.Client和storage.BucketHandle会自动实现这些接口（只要它们拥有接口中声明的方法）。

2. 修改Mock结构体实现接口 #

调整Mock类型，让它们实现上面定义的接口：

type mockStorage struct {
    mock.Mock
}

// 实现StorageClientInterface的Bucket方法
func (m *mockStorage) Bucket(bucket string) BucketHandleInterface {
    args := m.Called(bucket)
    return args.Get(0).(BucketHandleInterface)
}

type mockBucketHandle struct {
    mock.Mock
}

// 实现BucketHandleInterface的Object方法
func (m *mockBucketHandle) Object(name string) *storage.ObjectHandle {
    args := m.Called(name)
    return args.Get(0).(*storage.ObjectHandle)
}

3. 调整测试代码 #

现在可以直接返回mockBucketHandle实例，因为它已经满足BucketHandleInterface的类型要求：

func TestStorageClient_LoadChangelog(t *testing.T) {
    // Given
    logger := zerolog.New(os.Stdout).With().Timestamp().Logger()
    zerolog.DefaultContextLogger = &logger
    
    bucketHandleMock := &mockBucketHandle{}
    bucketHandleMock.On("Object", mock.Anything).Return(&storage.ObjectHandle{})
    
    storageMock := &mockStorage{}
    storageMock.On("Bucket", mock.Anything).Return(bucketHandleMock)
    
    // 注意：StorageClient的Client字段需改为StorageClientInterface类型
    client := NewStorageClient(storageMock)

    // When
    handle := client.getObjectHandle(uuid.NewString())
    
    // Then 
    assert.NotNil(t, handle)
}

4. 调整待测试代码的依赖 #

确保StorageClient依赖的是接口而非具体结构体：

type StorageClient struct {
    Client StorageClientInterface // 使用接口类型替代*storage.Client
    // 其他字段...
}

func (s *StorageClient) getObjectHandle(url string) *storage.ObjectHandle {
    bucket, name := urlToBucketAndName(url)
    return s.Client.Bucket(bucket).Object(name)
}

关键说明 #

Go的类型系统中，结构体之间没有继承关系，只有接口实现。因此不能直接把Mock结构体当成原结构体返回，必须通过抽象接口实现类型兼容。这种方式不仅解决了Mock返回的问题，还能提升代码的扩展性和可测试性。

内容的提问来源于stack exchange，提问作者Moritz Schmitz v. Hülst