Golang如何测试多接口组合调用_Golang 多接口调用单元测试实践

type UserRepo interface {
    Create(user *User) error
}

type EmailService interface {
    SendWelcomeEmail(email string) error
}

type UserService struct {
    userRepo     UserRepo
    emailService EmailService
}

type MockUserRepo struct {
    CreateUserFunc func(*User) error
}

func (m *MockUserRepo) Create(user *User) error {
    return m.CreateUserFunc(user)
}

type MockEmailService struct {
    SendWelcomeEmailFunc func(string) error
}

func (m *MockEmailService) SendWelcomeEmail(email string) error {
    return m.SendWelcomeEmailFunc(email)
}

func TestUserService_Register_Success(t *testing.T) {
    mockRepo := &MockUserRepo{
        CreateUserFunc: func(user *User) error {
            if user.Email != "test@example.com" {
                t.Fatal("expected test@example.com")
            }
            return nil
        },
    }

    mockEmail := &MockEmailService{
        SendWelcomeEmailFunc: func(email string) error {
            if email != "test@example.com" {
                t.Fatal("expected test@example.com")
            }
            return nil
        },
    }

    svc := &UserService{userRepo: mockRepo, emailService: mockEmail}
    err := svc.Register("test@example.com")

    if err != nil {
        t.Fatalf("expected no error, got %v", err)
    }
}

mockgen -source=user.go -destination=mocks/mock_user.go