Golang模拟数据与mock使用详解

type User struct {
    ID   int
    Name string
}

func GetUserInfo(users []User, id int) *User {
    for _, u := range users {
        if u.ID == id {
            return &u
        }
    }
    return nil
}

func TestGetUserInfo(t *testing.T) {
    mockUsers := []User{
        {ID: 1, Name: "Alice"},
        {ID: 2, Name: "Bob"},
    }

    user := GetUserInfo(mockUsers, 1)
    if user == nil {
        t.Fatal("expected user with ID 1, got nil")
    }
    if user.Name != "Alice" {
        t.Errorf("expected Alice, got %s", user.Name)
    }
}

go get github.com/stretchr/testify/mock

type PaymentGateway interface {
    Charge(amount float64) (string, error)
}

type OrderService struct {
    Gateway PaymentGateway
}

func (s *OrderService) CreateOrder(amount float64) (string, error) {
    if amount 

type MockPaymentGateway struct {
    mock.Mock
}

func (m *MockPaymentGateway) Charge(amount float64) (string, error) {
    args := m.Called(amount)
    return args.String(0), args.Error(1)
}

func TestOrderService_CreateOrder(t *testing.T) {
    mockGateway := new(MockPaymentGateway)
    service := &OrderService{Gateway: mockGateway}

    // 设定期望行为
    mockGateway.On("Charge", 100.0).Return("txn_123", nil)

    txnID, err := service.CreateOrder(100.0)

    assert.NoError(t, err)
    assert.Equal(t, "txn_123", txnID)
    mockGateway.AssertExpectations(t)
}