本文介绍通过接口抽象与自定义实现的方式，为依赖 net.conn 的代码编写可并行、可控制超时行为的单元测试，避免使用实际 socket 或不支持 deadline 的 io.pipe。
本文介绍通过接口抽象与自定义实现的方式，为依赖 net.conn 的代码编写可并行、可控制超时行为的单元测试，避免使用实际 socket 或不支持 deadline 的 io.pipe。
在 Go 中测试依赖 net.Conn 的业务逻辑时，核心原则是：不测试标准库，只测试自己的逻辑。net.Dial、net.Listen 等底层网络操作属于 net 包职责，其正确性由 Go 团队保障；我们真正需要验证的是：当传入一个符合 net.Conn 接口的对象时，你的代码是否能正确调用 SetReadDeadline/SetWriteDeadline、是否能妥善处理读写错误、是否在超时路径下按预期行为退出等。
因此，关键在于解耦依赖——将 net.Conn 作为参数注入，而非在函数内部直接调用 net.Dial()。例如，将原本紧耦合的代码：
func handleConnection() error {
conn, err := net.Dial("tcp", "example.com:80")
if err != nil {
return err
}
defer conn.Close()
conn.SetReadDeadline(time.Now().Add(5 * time.Second))
// ... 业务逻辑
}
重构为面向接口的设计：
func handleConnection(conn net.Conn) error {
defer conn.Close()
if err := conn.SetReadDeadline(time.Now().Add(5 * time.Second)); err != nil {
return fmt.Errorf("failed to set read deadline: %w", err)
}
// ... 业务逻辑（如读取请求、写入响应）
return nil
}
这样，测试时即可传入一个完全可控的 net.Conn 实现，精准模拟 deadline 行为。
自定义 Conn 实现：支持 Deadline 的内存连接
下面是一个轻量、线程安全、完整实现 net.Conn 接口（含所有 deadline 方法）的测试用结构体：
type MockConn struct {
mu sync.RWMutex
readDeadline, writeDeadline, deadline time.Time
closed bool
}
func (m *MockConn) Read(b []byte) (int, error) {
m.mu.RLock()
defer m.mu.RUnlock()
if m.closed {
return 0, io.EOF
}
// 可在此注入模拟延迟或错误，例如：
// if time.Now().After(m.readDeadline) { return 0, os.ErrDeadline }
return 0, nil // 模拟空读
}
func (m *MockConn) Write(b []byte) (int, error) {
m.mu.RLock()
defer m.mu.RUnlock()
if m.closed {
return 0, io.ErrClosedPipe
}
return len(b), nil
}
func (m *MockConn) Close() error {
m.mu.Lock()
defer m.mu.Unlock()
m.closed = true
return nil
}
func (m *MockConn) LocalAddr() net.Addr { return &mockAddr{"mock", "127.0.0.1:12345"} }
func (m *MockConn) RemoteAddr() net.Addr { return &mockAddr{"mock", "10.0.0.1:80"} }
func (m *MockConn) SetDeadline(t time.Time) error {
m.mu.Lock()
defer m.mu.Unlock()
m.deadline = t
m.readDeadline = t
m.writeDeadline = t
return nil
}
func (m *MockConn) SetReadDeadline(t time.Time) error {
m.mu.Lock()
defer m.mu.Unlock()
m.readDeadline = t
return nil
}
func (m *MockConn) SetWriteDeadline(t time.Time) error {
m.mu.Lock()
defer m.mu.Unlock()
m.writeDeadline = t
return nil
}
type mockAddr struct{ network, addr string }
func (a *mockAddr) Network() string { return a.network }
func (a *mockAddr) String() string { return a.addr }
该实现不仅满足 net.Conn 接口全部方法，还通过 sync.RWMutex 保证并发安全，并允许你在 Read/Write 中主动检查 readDeadline/writeDeadline 时间戳，从而精确模拟超时场景（例如返回 os.ErrDeadline）。
编写可验证 deadline 行为的测试用例
func TestHandleConnection_SetsReadDeadline(t *testing.T) {
mock := &MockConn{}
expected := time.Now().Add(3 * time.Second)
// 调用被测函数
err := handleConnection(mock)
if err != nil {
t.Fatalf("unexpected error: %v", err)
}
// 验证 SetReadDeadline 是否被正确调用
mock.mu.RLock()
actual := mock.readDeadline
mock.mu.RUnlock()
if actual.IsZero() {
t.Error("expected read deadline to be set, but it was zero")
} else if time.Until(actual) 4*time.Second {
t.Errorf("expected deadline ~3s from now, got %v", actual)
}
}
func TestHandleConnection_ReadTimeout(t *testing.T) {
mock := &MockConn{}
mock.readDeadline = time.Now().Add(-1 * time.Second) // 已过期
// 在 Read 中主动检查 deadline 并返回超时错误
// （需在 MockConn.Read 中补充逻辑，见上文注释）
err := handleConnection(mock)
if !errors.Is(err, os.ErrDeadline) {
t.Errorf("expected os.ErrDeadline, got %v", err)
}
}
✅ 关键优势：
- ✅ 完全无网络 I/O，100% 纯内存执行，可并行运行；
- ✅ 支持任意 deadline 控制（过去/未来/零值），覆盖超时、非超时、未设置等边界；
- ✅ 可扩展模拟阻塞、中断、部分读写等复杂网络异常；
- ✅ 符合 Go “interface-first” 设计哲学，提升代码可测试性与可维护性。
总结
测试 net.Conn 依赖代码的本质，不是寻找“替代 socket 的神奇类型”，而是践行依赖倒置（DIP）与接口隔离（ISP）：将具体实现（*net.TCPConn）替换为抽象接口（net.Conn），再为测试提供符合契约的轻量实现。io.Pipe 不适用，因其明确拒绝 deadline；而自定义 MockConn 不仅满足接口，更赋予你对超时行为的完全掌控权——这才是健壮网络服务单元测试的正确起点。
