Go语言Mock单元测试的实现示例

// ProductService 商品服务接口（抽象依赖）
type ProductService interface {
    GetProduct(id string) (*Product, error)   // 获取商品
    UpdateStock(id string, num int) error     // 更新库存
}

// PaymentService 支付服务接口（抽象依赖）
type PaymentService interface {
    ProcessPayment(amount float64, orderID string) (*PaymentResult, error) // 处理支付
}

// OrderService 被测试的订单服务（依赖接口，不依赖具体实现）
type OrderService struct {
    productService ProductService  // 依赖商品服务接口
    paymentService PaymentService  // 依赖支付服务接口
}

// MockProductService 商品服务的Mock实现
type MockProductService struct {
    mock.Mock  // 嵌入testify的Mock结构体，获得Mock能力
}

// GetProduct 实现ProductService接口的GetProduct方法
func (m *MockProductService) GetProduct(id string) (*Product, error) {
    args := m.Called(id)  // 记录方法调用的参数
    if args.Get(0) == nil {
        return nil, args.Error(1)
    }
    return args.Get(0).(*Product), args.Error(1)
}

// 预设"查询prod1商品"的行为
mockProduct.On("GetProduct", "prod1").Return(testProduct, nil)
// 预设"支付服务未开启"的行为
mockPayment.On("ProcessPayment", 99.99, mock.Anything).Return(nil, errors.New("支付服务未开启"))

// 注入Mock对象
orderService := NewOrderService(mockProduct, mockPayment)

// 调用被测试方法
order, err := orderService.CreateOrder("prod1", "user1")

// 验证业务结果
assert.Error(t, err)
assert.Nil(t, order)

// 验证Mock调用
mockProduct.AssertExpectations(t)
mockPayment.AssertExpectations(t)

package main

import "github.com/stretchr/testify/mock"

// MockProductService 商品服务的Mock实现
type MockProductService struct {
    mock.Mock  // 嵌入mock.Mock结构体，继承核心功能
}

// GetProduct 实现ProductService接口的GetProduct方法
func (m *MockProductService) GetProduct(id string) (*Product, error) {
    args := m.Called(id)  // 记录方法调用的参数
    if args.Get(0) == nil {
        return nil, args.Error(1)
    }
    return args.Get(0).(*Product), args.Error(1)
}

// UpdateStock 实现ProductService接口的UpdateStock方法
func (m *MockProductService) UpdateStock(id string, num int) error {
    args := m.Called(id, num)  // 记录方法调用的参数
    return args.Error(0)  // 返回预设的错误
}

// MockPaymentService 支付服务的Mock实现
type MockPaymentService struct {
    mock.Mock  // 嵌入Mock结构体，获得Mock能力
}

// ProcessPayment 实现PaymentService接口的ProcessPayment方法
func (m *MockPaymentService) ProcessPayment(amount float64, orderID string) (*PaymentResult, error) {
    args := m.Called(amount, orderID)  // 记录方法调用的参数
    if args.Get(0) == nil {
        return nil, args.Error(1)
    }
    return args.Get(0).(*PaymentResult), args.Error(1)
}

package main

import (
    "errors"
    "testing"
    "github.com/stretchr/testify/assert"
)

// TestCreateOrder_PaymentServiceUnavailable 测试支付服务未开启场景
func TestCreateOrder_PaymentServiceUnavailable(t *testing.T) {
    // 创建Mock对象
    mockProduct := new(MockProductService)
    mockPayment := new(MockPaymentService)

    // 定义测试数据
    testProduct := &Product{
        ID:    "prod1",
        Name:  "测试商品",
        Price: 99.99,
        Stock: 100,
    }
    
    // 预设Mock行为
    mockProduct.On("GetProduct", "prod1").Return(testProduct, nil)
    mockProduct.On("UpdateStock", "prod1", -1).Return(nil)
    mockPayment.On("ProcessPayment", 99.99, mock.Anything).Return(nil, errors.New("支付服务未开启"))

    // 初始化被测试服务
    orderService := NewOrderService(mockProduct, mockPayment)
    
    // 调用被测试方法
    order, err := orderService.CreateOrder("prod1", "user1")

    // 验证结果
    assert.Error(t, err)
    assert.Nil(t, order)
    mockProduct.AssertExpectations(t)
    mockPayment.AssertExpectations(t)
}