148 lines
4.2 KiB
Go
148 lines
4.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
func newTestCategory(portalID uint, name string) *model.Category {
|
|
return &model.Category{
|
|
PortalID: portalID,
|
|
Name: name,
|
|
Slug: name + "-slug",
|
|
Position: 1,
|
|
Locale: "en",
|
|
}
|
|
}
|
|
|
|
// ========== Create ==========
|
|
|
|
func TestCategoryRepo_Create(t *testing.T) {
|
|
db := setupTestDB(t, &model.Category{}, &model.Folder{}, &model.RelatedCategory{}, &model.Portal{}, &model.Account{})
|
|
repo := NewCategoryRepo(db)
|
|
|
|
cat := newTestCategory(1, "TestCategory")
|
|
err := repo.Create(context.Background(), cat)
|
|
require.NoError(t, err)
|
|
assert.NotZero(t, cat.ID)
|
|
assert.Equal(t, uint(1), cat.PortalID)
|
|
}
|
|
|
|
// ========== GetByID ==========
|
|
|
|
func TestCategoryRepo_GetByID(t *testing.T) {
|
|
db := setupTestDB(t, &model.Category{}, &model.Folder{}, &model.RelatedCategory{}, &model.Portal{}, &model.Account{})
|
|
repo := NewCategoryRepo(db)
|
|
|
|
cat := newTestCategory(1, "FindByIDCat")
|
|
err := repo.Create(context.Background(), cat)
|
|
require.NoError(t, err)
|
|
|
|
found, err := repo.GetByID(context.Background(), cat.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, cat.ID, found.ID)
|
|
}
|
|
|
|
func TestCategoryRepo_GetByID_NotFound(t *testing.T) {
|
|
db := setupTestDB(t, &model.Category{}, &model.Folder{}, &model.RelatedCategory{}, &model.Portal{}, &model.Account{})
|
|
repo := NewCategoryRepo(db)
|
|
|
|
found, err := repo.GetByID(context.Background(), 9999)
|
|
assert.Error(t, err)
|
|
assert.Nil(t, found)
|
|
}
|
|
|
|
// ========== Update ==========
|
|
|
|
func TestCategoryRepo_Update(t *testing.T) {
|
|
db := setupTestDB(t, &model.Category{}, &model.Folder{}, &model.RelatedCategory{}, &model.Portal{}, &model.Account{})
|
|
repo := NewCategoryRepo(db)
|
|
|
|
cat := newTestCategory(1, "BeforeUpdate")
|
|
err := repo.Create(context.Background(), cat)
|
|
require.NoError(t, err)
|
|
|
|
cat.Name = "AfterUpdate"
|
|
cat.Position = 2
|
|
err = repo.Update(context.Background(), cat)
|
|
require.NoError(t, err)
|
|
|
|
found, err := repo.GetByID(context.Background(), cat.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "AfterUpdate", found.Name)
|
|
assert.Equal(t, 2, found.Position)
|
|
}
|
|
|
|
// ========== Delete ==========
|
|
|
|
func TestCategoryRepo_Delete(t *testing.T) {
|
|
db := setupTestDB(t, &model.Category{}, &model.Folder{}, &model.RelatedCategory{}, &model.Portal{}, &model.Account{})
|
|
repo := NewCategoryRepo(db)
|
|
|
|
cat := newTestCategory(1, "DeleteCat")
|
|
err := repo.Create(context.Background(), cat)
|
|
require.NoError(t, err)
|
|
|
|
err = repo.Delete(context.Background(), cat.ID)
|
|
require.NoError(t, err)
|
|
|
|
found, err := repo.GetByID(context.Background(), cat.ID)
|
|
assert.Error(t, err)
|
|
assert.Nil(t, found)
|
|
}
|
|
|
|
// ========== FindByPortalID ==========
|
|
|
|
func TestCategoryRepo_FindByPortalID(t *testing.T) {
|
|
db := setupTestDB(t, &model.Category{}, &model.Folder{}, &model.RelatedCategory{}, &model.Portal{}, &model.Account{})
|
|
repo := NewCategoryRepo(db)
|
|
|
|
portalID := uint(10)
|
|
for i := 0; i < 3; i++ {
|
|
cat := newTestCategory(portalID, "Cat"+string(rune('A'+i)))
|
|
cat.Position = i + 1
|
|
err := repo.Create(context.Background(), cat)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
categories, count, err := repo.FindByPortalID(context.Background(), portalID, 0, 10)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(3), count)
|
|
assert.Len(t, categories, 3)
|
|
}
|
|
|
|
// ========== FindByPortalIDAndParentID ==========
|
|
|
|
func TestCategoryRepo_FindByPortalIDAndParentID(t *testing.T) {
|
|
db := setupTestDB(t, &model.Category{}, &model.Folder{}, &model.RelatedCategory{}, &model.Portal{}, &model.Account{})
|
|
repo := NewCategoryRepo(db)
|
|
|
|
portalID := uint(10)
|
|
parentID := uint(5)
|
|
|
|
// Create child categories
|
|
for i := 0; i < 2; i++ {
|
|
cat := newTestCategory(portalID, "ChildCat"+string(rune('A'+i)))
|
|
cat.ParentID = &parentID
|
|
err := repo.Create(context.Background(), cat)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
// Create root category (no parent)
|
|
rootCat := newTestCategory(portalID, "RootCat")
|
|
err := repo.Create(context.Background(), rootCat)
|
|
require.NoError(t, err)
|
|
|
|
categories, count, err := repo.FindByPortalIDAndParentID(context.Background(), portalID, parentID, 0, 10)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(2), count)
|
|
assert.Len(t, categories, 2)
|
|
for _, c := range categories {
|
|
assert.Equal(t, parentID, *c.ParentID)
|
|
}
|
|
} |