如何Mock Go SDK内部包函数与接口以编写单元测试？

核心问题解决：测试包含net.Dial的代码段 #

Go无法直接mock包级函数，必须通过接口抽象隔离对net包的依赖，具体步骤如下：

1. 定义抽象接口
   把net.Dial的操作封装成接口，让业务代码依赖这个接口而非net包的具体实现：

// Dialer 抽象UDP连接创建逻辑
type Dialer interface {
    Dial(network, address string) (net.Conn, error)
}

2. 重构业务代码
   修改原代码，让它接收Dialer接口作为参数，不再直接调用net.Dial：

func getSourceIP(d Dialer) (net.IP, error) {
    conn, err := d.Dial("udp", "8.8.8.8:80")
    if err != nil {
        return nil, err
    }
    defer conn.Close()
    return conn.LocalAddr().(*net.UDPAddr).IP, nil
}

生产环境中，直接传入&net.Dialer{}即可（net.Dialer已实现我们定义的Dialer接口）。

3. 编写单元测试
   用mockgen生成Dialer的mock实现，然后在测试中预设返回值：

• 生成mock文件：

mockgen -source=resolver.go -package=your_test_package -destination=mock_dialer.go Dialer

• 测试代码示例：

func TestGetSourceIP(t *testing.T) {
    // 初始化mock对象
    mockDialer := &mock_dialer.MockDialer{}
    mockConn := &mocknet.MockConn{} // 若需验证Conn方法调用，可单独生成此mock

    // 预设mock行为
    testIP := net.ParseIP("192.168.1.100")
    mockDialer.On("Dial", "udp", "8.8.8.8:80").Return(mockConn, nil)
    mockConn.On("LocalAddr").Return(&net.UDPAddr{IP: testIP})
    mockConn.On("Close").Return(nil)

    // 执行测试
    ip, err := getSourceIP(mockDialer)
    if err != nil {
        t.Fatalf("unexpected error: %v", err)
    }
    if !ip.Equal(testIP) {
        t.Fatalf("expected IP %v, got %v", testIP, ip)
    }

    // 验证mock调用符合预期
    mockDialer.AssertExpectations(t)
    mockConn.AssertExpectations(t)
}

针对疑问的解答 #

1. net是包而非接口，能否Mock该调用？ #

不能直接mock包级函数，Go语言本身没有提供修改包级函数实现的机制。必须通过接口抽象的方式，把对net.Dial的依赖转化为对自定义接口的依赖，才能在测试中替换实现。

2. 是否应为SDK内部的net.Conn接口生成Mock，以及mockgen的-source参数怎么传？ #

• net.Conn本身就是Go标准库定义的接口，完全可以用mockgen生成它的mock实现。如果测试需要验证Conn的方法调用（比如Close、LocalAddr），就需要生成对应的mock。
• 使用mockgen生成net.Conn的mock时，可直接指定接口所在包和接口名：

mockgen net Conn -package=mocknet -destination=mock_conn.go

如果是针对自定义接口（比如前面的Dialer），-source参数传入你定义接口的Go文件路径即可（如resolver.go）。

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