110 lines
3.6 KiB
Plaintext
110 lines
3.6 KiB
Plaintext
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
func TestPortalMemberService_Create(t *testing.T) {
|
|
db, _, svc := setupPortalMemberService(t)
|
|
account := createTestAccount(t, db)
|
|
portal := createTestPortal(t, db, account.ID)
|
|
user := createTestUser(t, db, account.ID)
|
|
|
|
member, err := svc.Create(context.Background(), portal.ID, &CreatePortalMemberRequest{
|
|
UserID: user.ID,
|
|
Role: model.PortalMemberRoleAdministrator,
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, portal.ID, member.PortalID)
|
|
assert.Equal(t, user.ID, member.UserID)
|
|
assert.Equal(t, string(model.PortalMemberRoleAdministrator), string(member.Role))
|
|
}
|
|
|
|
func TestPortalMemberService_GetByID(t *testing.T) {
|
|
db, _, svc := setupPortalMemberService(t)
|
|
account := createTestAccount(t, db)
|
|
portal := createTestPortal(t, db, account.ID)
|
|
user := createTestUser(t, db, account.ID)
|
|
|
|
member, err := svc.Create(context.Background(), portal.ID, &CreatePortalMemberRequest{
|
|
UserID: user.ID,
|
|
Role: model.PortalMemberRoleEditor,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
found, err := svc.GetByID(context.Background(), member.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, string(model.PortalMemberRoleEditor), string(found.Role))
|
|
}
|
|
|
|
func TestPortalMemberService_GetByID_NotFound(t *testing.T) {
|
|
_, _, svc := setupPortalMemberService(t)
|
|
|
|
_, err := svc.GetByID(context.Background(), 99999)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestPortalMemberService_Update(t *testing.T) {
|
|
db, _, svc := setupPortalMemberService(t)
|
|
account := createTestAccount(t, db)
|
|
portal := createTestPortal(t, db, account.ID)
|
|
user := createTestUser(t, db, account.ID)
|
|
|
|
member, err := svc.Create(context.Background(), portal.ID, &CreatePortalMemberRequest{
|
|
UserID: user.ID,
|
|
Role: model.PortalMemberRoleReader,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
updated, err := svc.Update(context.Background(), member.ID, &UpdatePortalMemberRequest{
|
|
Role: model.PortalMemberRoleAdministrator,
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, string(model.PortalMemberRoleAdministrator), string(updated.Role))
|
|
}
|
|
|
|
func TestPortalMemberService_Delete(t *testing.T) {
|
|
db, _, svc := setupPortalMemberService(t)
|
|
account := createTestAccount(t, db)
|
|
portal := createTestPortal(t, db, account.ID)
|
|
user := createTestUser(t, db, account.ID)
|
|
|
|
member, err := svc.Create(context.Background(), portal.ID, &CreatePortalMemberRequest{
|
|
UserID: user.ID,
|
|
Role: model.PortalMemberRoleReader,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
err = svc.Delete(context.Background(), member.ID)
|
|
require.NoError(t, err)
|
|
|
|
_, err = svc.GetByID(context.Background(), member.ID)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestPortalMemberService_ListByPortalID(t *testing.T) {
|
|
db, _, svc := setupPortalMemberService(t)
|
|
account := createTestAccount(t, db)
|
|
portal := createTestPortal(t, db, account.ID)
|
|
user1 := createTestUser(t, db, account.ID)
|
|
// Create second user with different email
|
|
user2 := &model.User{AccountID: account.ID, Name: "User2", Email: "user2@test.com", Password: "pass", Role: "agent", Active: true}
|
|
require.NoError(t, db.Create(user2).Error)
|
|
|
|
_, err := svc.Create(context.Background(), portal.ID, &CreatePortalMemberRequest{UserID: user1.ID, Role: model.PortalMemberRoleReader})
|
|
require.NoError(t, err)
|
|
_, err = svc.Create(context.Background(), portal.ID, &CreatePortalMemberRequest{UserID: user2.ID, Role: model.PortalMemberRoleEditor})
|
|
require.NoError(t, err)
|
|
|
|
members, count, err := svc.ListByPortalID(context.Background(), portal.ID, 0, 10)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, int64(2), count)
|
|
assert.Len(t, members, 2)
|
|
}
|