Golang测试中如何将Mock客户端作为参数传入被测函数？

我正在测试以下代码：

func listObjects(cli *client.Client, options clientOptions) ([]BlobObjects, error) {
    objects, err := cli.ListBlobObjects(...)
}

在测试设置中我编写了如下Mock客户端：

type MockClient struct {
    MockListBlobObjects func() ([]BlobObjects, error)
}

func (m *MockClient) ListBlobObjects(....) ([]BlobObjects, error) {
    // return some mock response
}

测试用例代码如下：

func TestBlobObjects(t *testing.T) {
    tests := map[string]struct {
        client         *MockClient
        ...
    }{
        "Test case 1": {
            client: &MockClient{
                MockListBlobObjects: ....,
            },
            ....
        },

      ....
   for testName, test := range tests {
      blobs, err := listObjects(test.client, clientOptions{})
      // make assertions here
   }
}

编译器提示错误：

cannot use test.client (variable of type *MockClient) as *client.Client value in argument to listObjects

我原本希望传入Mock客户端后，调用被测函数时会执行Mock的ListBlobObjects方法，这是我在Python中的常用做法，请问在Golang中该如何解决这个问题？

Go是静态类型语言，不支持Python那样的鸭子类型，要实现Mock注入，核心是通过定义接口实现多态，让真实客户端和Mock客户端都适配同一个接口。

步骤1：提取接口 #

先把listObjects用到的client.Client方法抽成接口，只保留需要的方法：

type BlobClient interface {
    ListBlobObjects(/* 填入原方法的完整参数列表 */) ([]BlobObjects, error)
}

步骤2：修改被测函数参数 #

把listObjects的第一个参数类型从*client.Client改成上面定义的BlobClient：

func listObjects(cli BlobClient, options clientOptions) ([]BlobObjects, error) {
    objects, err := cli.ListBlobObjects(...)
}

步骤3：确保MockClient适配接口 #

你的MockClient已经实现了ListBlobObjects方法，只要方法签名和接口完全一致，Go会自动认定它实现了BlobClient接口，不需要额外声明。

步骤4：调整测试用例 #

把测试用例里的client字段类型改成BlobClient，这样就能同时兼容*MockClient和真实的*client.Client：

func TestBlobObjects(t *testing.T) {
    tests := map[string]struct {
        client         BlobClient
        // 其他测试字段...
    }{
        "Test case 1": {
            client: &MockClient{
                MockListBlobObjects: func() ([]BlobObjects, error) {
                    // 返回自定义的Mock数据
                    return []BlobObjects{{Name: "test-file"}}, nil
                },
            },
            // 其他测试配置...
        },
        // 更多测试用例...
    }

    for testName, test := range tests {
        t.Run(testName, func(t *testing.T) {
            blobs, err := listObjects(test.client, clientOptions{})
            // 执行断言逻辑
            if err != nil {
                t.Fatalf("意外错误：%v", err)
            }
            if len(blobs) != 1 {
                t.Errorf("期望1个Blob，实际得到%d个", len(blobs))
            }
        })
    }
}

补充说明 #

• Go的接口是隐式实现的，只要类型的方法集完全覆盖接口方法，就自动实现了该接口，不需要类似Java的implements关键字。
• 这种方式遵循依赖倒置原则，让被测函数依赖抽象而非具体实现，既方便测试，也提升了代码的扩展性。

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