117 lines
2.8 KiB
Plaintext
117 lines
2.8 KiB
Plaintext
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
func newTestFolder(portalID uint, name string) *model.Folder {
|
|
return &model.Folder{
|
|
PortalID: portalID,
|
|
Name: name,
|
|
Slug: name + "-slug",
|
|
Position: 1,
|
|
}
|
|
}
|
|
|
|
// ========== Create ==========
|
|
|
|
func TestFolderRepo_Create(t *testing.T) {
|
|
db := setupTestDB(t, &model.Folder{})
|
|
repo := NewFolderRepo(db)
|
|
|
|
folder := newTestFolder(1, "TestFolder")
|
|
err := repo.Create(context.Background(), folder)
|
|
require.NoError(t, err)
|
|
assert.NotZero(t, folder.ID)
|
|
assert.Equal(t, uint(1), folder.PortalID)
|
|
}
|
|
|
|
// ========== GetByID ==========
|
|
|
|
func TestFolderRepo_GetByID(t *testing.T) {
|
|
db := setupTestDB(t, &model.Folder{})
|
|
repo := NewFolderRepo(db)
|
|
|
|
folder := newTestFolder(1, "FindByIDFolder")
|
|
err := repo.Create(context.Background(), folder)
|
|
require.NoError(t, err)
|
|
|
|
found, err := repo.GetByID(context.Background(), folder.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, folder.ID, found.ID)
|
|
}
|
|
|
|
func TestFolderRepo_GetByID_NotFound(t *testing.T) {
|
|
db := setupTestDB(t, &model.Folder{})
|
|
repo := NewFolderRepo(db)
|
|
|
|
found, err := repo.GetByID(context.Background(), 9999)
|
|
assert.Error(t, err)
|
|
assert.Nil(t, found)
|
|
}
|
|
|
|
// ========== Update ==========
|
|
|
|
func TestFolderRepo_Update(t *testing.T) {
|
|
db := setupTestDB(t, &model.Folder{})
|
|
repo := NewFolderRepo(db)
|
|
|
|
folder := newTestFolder(1, "BeforeUpdate")
|
|
err := repo.Create(context.Background(), folder)
|
|
require.NoError(t, err)
|
|
|
|
folder.Name = "AfterUpdate"
|
|
folder.Position = 2
|
|
err = repo.Update(context.Background(), folder)
|
|
require.NoError(t, err)
|
|
|
|
found, err := repo.GetByID(context.Background(), folder.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "AfterUpdate", found.Name)
|
|
assert.Equal(t, 2, found.Position)
|
|
}
|
|
|
|
// ========== Delete ==========
|
|
|
|
func TestFolderRepo_Delete(t *testing.T) {
|
|
db := setupTestDB(t, &model.Folder{})
|
|
repo := NewFolderRepo(db)
|
|
|
|
folder := newTestFolder(1, "DeleteFolder")
|
|
err := repo.Create(context.Background(), folder)
|
|
require.NoError(t, err)
|
|
|
|
err = repo.Delete(context.Background(), folder.ID)
|
|
require.NoError(t, err)
|
|
|
|
found, err := repo.GetByID(context.Background(), folder.ID)
|
|
assert.Error(t, err)
|
|
assert.Nil(t, found)
|
|
}
|
|
|
|
// ========== FindByPortalID ==========
|
|
|
|
func TestFolderRepo_FindByPortalID(t *testing.T) {
|
|
db := setupTestDB(t, &model.Folder{})
|
|
repo := NewFolderRepo(db)
|
|
|
|
portalID := uint(10)
|
|
for i := 0; i < 3; i++ {
|
|
f := newTestFolder(portalID, "Folder"+string(rune('A'+i)))
|
|
f.Position = i + 1
|
|
err := repo.Create(context.Background(), f)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
folders, count, err := repo.FindByPortalID(context.Background(), portalID, 0, 10)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(3), count)
|
|
assert.Len(t, folders, 3)
|
|
}
|