如何在 Golang 单元测试中集成第三方库？

在 Golang 单元测试中集成第三方库可通过依赖注入或使用存根实现：依赖注入：使用模拟或存根代替实际库实现。示例：使用 MockDependency 模拟第三方库，并将其注入到待测函数中。存根：提供对真实库实现的访问。示例：使用 stubFunc 存根第三方库中的实际函数，并覆盖其行为。

如何在 Golang 单元测试中集成第三方库

在 Golang 单元测试中集成第三方库需要一些技巧。本文将指导你如何完成这项任务，并提供一个实战案例来说明如何实现。

1. 使用依赖注入

依赖注入是一种在测试中隔离第三方库的有效方式。它允许你在测试中提供模拟或存根，而不是使用实际的库实现。

import (
    "testing"

    "github.com/stretchr/testify/assert"
)

// MockDependency 模拟第三方库
type MockDependency struct {
    result int
}

// Method 模拟第三方库的方法
func (m *MockDependency) Method() int {
    return m.result
}

func TestFunctionUnderTest(t *testing.T) {
    // 使用模拟的依赖项
    dependency := &MockDependency{result: 42}

    // 调用待测函数
    result := FunctionUnderTest(dependency)

    // 断言结果
    assert.Equal(t, 42, result)
}

2. 使用存根

存根是另一种隔离第三方库的方法。与模拟不同，存根提供对真实库实现的访问，但允许你通过拦截或修改其行为来控制其调用。

import (
    "testing"

    "github.com/stretchr/testify/assert"
    "github.com/google/go-cmp/cmp"
)

func TestFunctionUnderTest(t *testing.T) {
    // 使用存根函数
    stubFunc := func() (string, error) {
        return "stubbed result", nil
    }
    originalFunc := library.Func

    // 覆盖实际函数
    library.Func = stubFunc

    // 调用待测函数
    result, err := FunctionUnderTest()

    // 还原实际函数
    library.Func = originalFunc

    // 断言结果
    assert.NoError(t, err)
    diff := cmp.Diff("stubbed result", result)
    assert.Empty(t, diff)
}

实战案例

假设我们有一个函数 NewService，它从第三方库 github.com/example/service 中获取一个服务实例。我们可以使用依赖注入来测试这个函数：

import (
    "testing"

    "github.com/stretchr/testify/assert"
    "github.com/example/service"
)

// MockService 模拟 service 库
type MockService struct {
    result *service.Service
}

// NewMockService 返回一个模拟的服务实例
func NewMockService(result *service.Service) *MockService {
    return &MockService{result: result}
}

// Service 返回模拟的服务实例
func (m *MockService) Service() *service.Service {
    return m.result
}

func TestNewService(t *testing.T) {
    // 使用模拟的服务
    mockService := NewMockService(&service.Service{})

    // 调用待测函数
    newService, err := NewService(mockService)

    // 断言结果
    assert.NoError(t, err)
    assert.Equal(t, &service.Service{}, newService)
}