Kiểm Tra Golang với Thư Viện Stretchr/Testify và Mockery

/project
│
├── main.go
├── service.go
├── service_test.go
├── user_client.go
├── mocks/
│   └── UserClient.go (được tạo ra bởi mockery)
└── go.mod

package project

type User struct {
    ID   int
    Name string
}

type UserClient interface {
    GetUserByID(id int) (*User, error)
}

package project

import "fmt"

type UserService struct {
    client UserClient
}

func NewUserService(client UserClient) *UserService {
    return &UserService{client: client}
}

func (s *UserService) GetUserDetails(id int) (string, error) {
    user, err := s.client.GetUserByID(id)
    if err != nil {
        return "", fmt.Errorf("failed to get user: %w", err)
    }

    return fmt.Sprintf("User: %s (ID: %d)", user.Name, user.ID), nil
}

mockery --name=UserClient --output=./mocks

package project_test

import (
    "errors"
    "testing"

    "github.com/stretchr/testify/assert"
    "github.com/stretchr/testify/require"
    "github.com/stretchr/testify/mock"
    "project"
    "project/mocks"
)

func TestUserService_GetUserDetails_Success(t *testing.T) {
    // Tạo một khách hàng giả lập mới
    mockClient := new(mocks.UserClient)

    // Định nghĩa những gì giả lập nên trả về khi `GetUserByID` được gọi
    mockClient.On("GetUserByID", 1).Return(&project.User{
        ID:   1,
        Name: "John Doe",
    }, nil)

    // Tạo UserService với khách hàng giả lập
    service := project.NewUserService(mockClient)

    // Thử nghiệm phương thức GetUserDetails
    result, err := service.GetUserDetails(1)

    // Sử dụng `require` cho các kiểm tra lỗi
    require.NoError(t, err)
    require.NotEmpty(t, result)

    // Sử dụng `assert` cho các kiểm tra giá trị
    assert.Equal(t, "User: John Doe (ID: 1)", result)

    // Đảm bảo rằng phương thức `GetUserByID` được gọi đúng một lần
    mockClient.AssertExpectations(t)
}

func TestUserService_GetUserDetails_Error(t *testing.T) {
    // Tạo một khách hàng giả lập mới
    mockClient := new(mocks.UserClient)

    // Định nghĩa những gì giả lập nên trả về khi `GetUserByID` được gọi với lỗi
    mockClient.On("GetUserByID", 2).Return(nil, errors.New("user not found"))

    // Tạo UserService với khách hàng giả lập
    service := project.NewUserService(mockClient)

    // Thử nghiệm phương thức GetUserDetails
    result, err := service.GetUserDetails(2)

    // Sử dụng `require` cho các kiểm tra lỗi
    require.Error(t, err)
    assert.Contains(t, err.Error(), "user not found")

    // Đảm bảo rằng kết quả là rỗng
    assert.Empty(t, result)

    // Đảm bảo rằng phương thức `GetUserByID` được gọi đúng một lần
    mockClient.AssertExpectations(t)
}