如何在Go中优雅地测试包装多个客户端的方法？

package main

import (
    "fmt"
    "net/http"
    "testing"
)

// 定义一个接口，包含需要测试的方法
type Client interface {
    Get(url string) (*http.Response, error)
}

// 实现一个具体的客户端
type HTTPClient struct{}

func (c *HTTPClient) Get(url string) (*http.Response, error) {
    // 发送HTTP GET请求并返回响应
    return http.Get(url)
}

// 包装多个客户端的方法
func FetchData(client Client, url string) (string, error) {
    resp, err := client.Get(url)
    if err != nil {
        return "", err
    }
    defer resp.Body.Close()

    // 处理响应并返回数据
    // 这里只是简单地读取响应的内容并转换为字符串
    // 实际应用中可能会有更复杂的处理逻辑
    data := make([]byte, resp.ContentLength)
    resp.Body.Read(data)
    return string(data), nil
}

// 测试FetchData方法
func TestFetchData(t *testing.T) {
    // 创建一个模拟的客户端
    mockClient := &ClientMock{}

    // 设置模拟客户端的期望行为
    mockClient.On("Get", "https://example.com").Return(
        &http.Response{
            StatusCode: http.StatusOK,
            Body:       ioutil.NopCloser(bytes.NewBufferString("Hello, World!")),
        },
        nil,
    )

    // 调用FetchData方法
    result, err := FetchData(mockClient, "https://example.com")
    if err != nil {
        t.Errorf("FetchData returned an error: %v", err)
    }

    // 验证结果是否符合预期
    expectedResult := "Hello, World!"
    if result != expectedResult {
        t.Errorf("FetchData returned %s, expected %s", result, expectedResult)
    }

    // 验证模拟客户端的方法是否被调用
    mockClient.AssertCalled(t, "Get", "https://example.com")
}

// 定义一个模拟客户端，用于测试
type ClientMock struct {
    mock.Mock
}

func (m *ClientMock) Get(url string) (*http.Response, error) {
    args := m.Called(url)
    return args.Get(0).(*http.Response), args.Error(1)
}

func main() {
    // 创建一个实际的HTTP客户端
    httpClient := &HTTPClient{}

    // 使用实际的客户端调用FetchData方法
    result, err := FetchData(httpClient, "https://example.com")
    if err != nil {
        fmt.Printf("FetchData returned an error: %v", err)
        return
    }

    fmt.Println(result)
}