387 lines
10 KiB
Go
387 lines
10 KiB
Go
|
package repository
|
||
|
|
|
||
|
|
import (
|
||
|
|
"app-cloudep-member-server/pkg/domain/entity"
|
||
|
|
"app-cloudep-member-server/pkg/domain/member"
|
||
|
|
"app-cloudep-member-server/pkg/domain/repository"
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
mgo "code.30cm.net/digimon/library-go/mongo"
|
||
|
|
"github.com/alicebob/miniredis/v2"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/zeromicro/go-zero/core/stores/cache"
|
||
|
|
"github.com/zeromicro/go-zero/core/stores/redis"
|
||
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
||
|
|
"google.golang.org/protobuf/proto"
|
||
|
|
)
|
||
|
|
|
||
|
|
func SetupTestUserRepository(db string) (repository.UserRepository, func(), error) {
|
||
|
|
h, p, tearDown, err := startMongoContainer()
|
||
|
|
if err != nil {
|
||
|
|
return nil, nil, err
|
||
|
|
}
|
||
|
|
s, _ := miniredis.Run()
|
||
|
|
|
||
|
|
conf := &mgo.Conf{
|
||
|
|
Schema: Schema,
|
||
|
|
Host: h,
|
||
|
|
Port: p,
|
||
|
|
Database: db,
|
||
|
|
MaxStaleness: 300,
|
||
|
|
MaxPoolSize: 100,
|
||
|
|
MinPoolSize: 100,
|
||
|
|
MaxConnIdleTime: 300,
|
||
|
|
Compressors: []string{},
|
||
|
|
EnableStandardReadWriteSplitMode: false,
|
||
|
|
ConnectTimeoutMs: 3000,
|
||
|
|
}
|
||
|
|
|
||
|
|
cacheConf := cache.CacheConf{
|
||
|
|
cache.NodeConf{
|
||
|
|
RedisConf: redis.RedisConf{
|
||
|
|
Host: s.Addr(),
|
||
|
|
Type: redis.NodeType,
|
||
|
|
},
|
||
|
|
Weight: 100,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
cacheOpts := []cache.Option{
|
||
|
|
cache.WithExpiry(1000 * time.Microsecond),
|
||
|
|
cache.WithNotFoundExpiry(1000 * time.Microsecond),
|
||
|
|
}
|
||
|
|
|
||
|
|
param := UserRepositoryParam{
|
||
|
|
Conf: conf,
|
||
|
|
CacheConf: cacheConf,
|
||
|
|
CacheOpts: cacheOpts,
|
||
|
|
}
|
||
|
|
|
||
|
|
repo := NewUserRepository(param)
|
||
|
|
_, _ = repo.Index20241226001UP(context.Background())
|
||
|
|
|
||
|
|
return repo, tearDown, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCustomUserModel_UpdateEmailVerifyStatus(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
initialUser := &entity.User{
|
||
|
|
UID: "test_uid_email",
|
||
|
|
Email: proto.String("old_email@example.com"),
|
||
|
|
}
|
||
|
|
err = repo.Insert(context.Background(), initialUser)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
err = repo.UpdateEmailVerifyStatus(context.Background(), "test_uid_email", "new_email@example.com")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
updatedUser, err := repo.FindOneByUID(context.Background(), "test_uid_email")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "new_email@example.com", *updatedUser.Email)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCustomUserModel_UpdatePhoneVerifyStatus(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
initialUser := &entity.User{
|
||
|
|
UID: "test_uid_phone",
|
||
|
|
PhoneNumber: proto.String("123456789"),
|
||
|
|
}
|
||
|
|
err = repo.Insert(context.Background(), initialUser)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
err = repo.UpdatePhoneVerifyStatus(context.Background(), "test_uid_phone", "987654321")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
updatedUser, err := repo.FindOneByUID(context.Background(), "test_uid_phone")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "987654321", *updatedUser.PhoneNumber)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCustomUserModel_UpdateUserDetailsByUID(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
initialData *entity.User
|
||
|
|
updateData *repository.UpdateUserInfoRequest
|
||
|
|
expectError bool
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "Update user details by UID",
|
||
|
|
initialData: &entity.User{
|
||
|
|
UID: "test_uid_1",
|
||
|
|
Nickname: proto.String("old_nickname"),
|
||
|
|
FullName: proto.String("Old Full Name"),
|
||
|
|
},
|
||
|
|
updateData: &repository.UpdateUserInfoRequest{
|
||
|
|
UID: "test_uid_1",
|
||
|
|
Nickname: proto.String("new_nickname"),
|
||
|
|
FullName: proto.String("New Full Name"),
|
||
|
|
},
|
||
|
|
expectError: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "User not found",
|
||
|
|
initialData: nil,
|
||
|
|
updateData: &repository.UpdateUserInfoRequest{
|
||
|
|
UID: "non_existent_uid",
|
||
|
|
Nickname: proto.String("nickname"),
|
||
|
|
},
|
||
|
|
expectError: true,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
if tt.initialData != nil {
|
||
|
|
err := repo.Insert(context.Background(), tt.initialData)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
err := repo.UpdateUserDetailsByUID(context.Background(), tt.updateData)
|
||
|
|
if tt.expectError {
|
||
|
|
assert.Error(t, err)
|
||
|
|
} else {
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
updatedUser, err := repo.FindOneByUID(context.Background(), tt.updateData.UID)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, *tt.updateData.Nickname, *updatedUser.Nickname)
|
||
|
|
assert.Equal(t, *tt.updateData.FullName, *updatedUser.FullName)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCustomUserModel_FindOneByUID(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
initialUser := &entity.User{
|
||
|
|
UID: "test_uid",
|
||
|
|
}
|
||
|
|
|
||
|
|
err = repo.Insert(context.Background(), initialUser)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
user, err := repo.FindOneByUID(context.Background(), "test_uid")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "test_uid", user.UID)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCustomUserModel_ListMembers(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
users := []*entity.User{
|
||
|
|
{UID: "uid1", UserStatus: member.AccountStatusActive},
|
||
|
|
{UID: "uid2", UserStatus: member.AccountStatusActive},
|
||
|
|
{UID: "uid3", UserStatus: member.AccountStatusUnverified},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, user := range users {
|
||
|
|
err := repo.Insert(context.Background(), user)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
activeStatus := member.AccountStatusActive
|
||
|
|
params := &repository.UserQueryParams{
|
||
|
|
UserStatus: &activeStatus,
|
||
|
|
PageSize: 2,
|
||
|
|
PageIndex: 1,
|
||
|
|
}
|
||
|
|
|
||
|
|
result, count, err := repo.ListMembers(context.Background(), params)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, int64(2), count)
|
||
|
|
assert.Len(t, result, 2)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCustomUserModel_FindOne(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
// 準備測試資料
|
||
|
|
initialUser := &entity.User{
|
||
|
|
UID: "test_uid",
|
||
|
|
}
|
||
|
|
|
||
|
|
err = repo.Insert(context.Background(), initialUser)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
id string
|
||
|
|
expectError bool
|
||
|
|
expectedUID string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "Valid FindOne",
|
||
|
|
id: initialUser.ID.Hex(),
|
||
|
|
expectError: false,
|
||
|
|
expectedUID: initialUser.UID,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "Invalid ObjectID",
|
||
|
|
id: "invalid_id",
|
||
|
|
expectError: true,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "Non-existent ObjectID",
|
||
|
|
id: primitive.NewObjectID().Hex(),
|
||
|
|
expectError: true,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
result, err := repo.FindOne(context.Background(), tt.id)
|
||
|
|
if tt.expectError {
|
||
|
|
assert.Error(t, err)
|
||
|
|
} else {
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, tt.expectedUID, result.UID, "找到的 UID 應符合預期")
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCustomUserModel_Update(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
// 準備測試資料
|
||
|
|
initialUser := &entity.User{
|
||
|
|
UID: "test_uid",
|
||
|
|
}
|
||
|
|
|
||
|
|
err = repo.Insert(context.Background(), initialUser)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
updatedUID := "test_uid"
|
||
|
|
initialUser.UID = updatedUID
|
||
|
|
|
||
|
|
// 執行更新操作
|
||
|
|
_, err = repo.Update(context.Background(), initialUser)
|
||
|
|
assert.NoError(t, err, "應成功更新資料")
|
||
|
|
|
||
|
|
// 確認更新結果
|
||
|
|
updatedAccountUid, err := repo.FindOne(context.Background(), initialUser.ID.Hex())
|
||
|
|
assert.NoError(t, err, "應能找到更新後的資料")
|
||
|
|
assert.Equal(t, updatedUID, updatedAccountUid.UID, "更新後的 UID 應符合預期")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCustomUserModel_Delete(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
// 準備測試資料
|
||
|
|
initialUser := &entity.User{
|
||
|
|
UID: "test_uid",
|
||
|
|
}
|
||
|
|
|
||
|
|
err = repo.Insert(context.Background(), initialUser)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
// 執行刪除操作
|
||
|
|
count, err := repo.Delete(context.Background(), initialUser.ID.Hex())
|
||
|
|
assert.NoError(t, err, "應成功刪除資料")
|
||
|
|
assert.Equal(t, int64(1), count, "刪除數量應為 1")
|
||
|
|
|
||
|
|
// 確認資料已被刪除
|
||
|
|
_, err = repo.FindOne(context.Background(), initialUser.ID.Hex())
|
||
|
|
assert.Error(t, err, "應無法找到已刪除的資料")
|
||
|
|
assert.Equal(t, ErrNotFound, err, "應返回 ErrNotFound 錯誤")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUserRepository_UpdateStatus(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
// 插入測試數據
|
||
|
|
e := &entity.User{
|
||
|
|
UID: "test_uid",
|
||
|
|
UserStatus: member.AccountStatusActive,
|
||
|
|
}
|
||
|
|
err = repo.Insert(context.TODO(), e)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
// 測試用例
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
uid string
|
||
|
|
status member.Status
|
||
|
|
expectError bool
|
||
|
|
expectedErr error
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "Valid UID - Successful update",
|
||
|
|
uid: e.UID,
|
||
|
|
status: member.AccountStatusActive,
|
||
|
|
expectError: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "Invalid UID - No mapping to ID",
|
||
|
|
uid: "invalid_uid",
|
||
|
|
status: 3,
|
||
|
|
expectError: true,
|
||
|
|
expectedErr: errors.New("invalid uid"),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "Non-existent UID",
|
||
|
|
uid: "non_existent_uid",
|
||
|
|
status: 4,
|
||
|
|
expectError: true,
|
||
|
|
expectedErr: errors.New("invalid uid"),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
err := repo.UpdateStatus(context.TODO(), tt.uid, tt.status.ToInt32())
|
||
|
|
if tt.expectError {
|
||
|
|
assert.Error(t, err)
|
||
|
|
assert.Equal(t, tt.expectedErr.Error(), err.Error())
|
||
|
|
} else {
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
// 驗證數據是否正確更新
|
||
|
|
user, err := repo.FindOneByUID(context.TODO(), tt.uid)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, tt.status, user.UserStatus, "UserStatus 應更新為預期值")
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCustomUserModel_FindOneByNickName(t *testing.T) {
|
||
|
|
repo, tearDown, err := SetupTestUserRepository("testDB")
|
||
|
|
defer tearDown()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
initialUser := &entity.User{
|
||
|
|
UID: "test_uid",
|
||
|
|
Nickname: proto.String("my_name"),
|
||
|
|
}
|
||
|
|
|
||
|
|
err = repo.Insert(context.Background(), initialUser)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
user, err := repo.FindOneByNickName(context.Background(), "my_name")
|
||
|
|
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "test_uid", user.UID)
|
||
|
|
}
|