feat: add product tags index
This commit is contained in:
parent
f6e936a760
commit
6e47895174
|
|
@ -0,0 +1,497 @@
|
||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"code.30cm.net/digimon/app-cloudep-product-service/pkg/domain/entity"
|
||||||
|
"code.30cm.net/digimon/app-cloudep-product-service/pkg/domain/product"
|
||||||
|
"code.30cm.net/digimon/app-cloudep-product-service/pkg/domain/repository"
|
||||||
|
mgo "code.30cm.net/digimon/library-go/mongo"
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"github.com/alicebob/miniredis/v2"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/cache"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/mon"
|
||||||
|
"github.com/zeromicro/go-zero/core/stores/redis"
|
||||||
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func SetupTestProductTagsRepo(db string) (repository.TagRepo, func(), error) {
|
||||||
|
h, p, tearDown, err := startMongoContainer()
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
s, _ := miniredis.Run()
|
||||||
|
|
||||||
|
conf := &mgo.Conf{
|
||||||
|
Schema: Schema,
|
||||||
|
Host: fmt.Sprintf("%s:%s", h, 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 := TagsRepositoryParam{
|
||||||
|
Conf: conf,
|
||||||
|
CacheConf: cacheConf,
|
||||||
|
CacheOpts: cacheOpts,
|
||||||
|
DBOpts: []mon.Option{
|
||||||
|
mgo.SetCustomDecimalType(),
|
||||||
|
mgo.InitMongoOptions(*conf),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
repo := NewTagsRepository(param)
|
||||||
|
_, _ = repo.IndexTags20250317001UP(context.Background())
|
||||||
|
_, _ = repo.IndexTagsBinding20250317001UP(context.Background())
|
||||||
|
|
||||||
|
return repo, tearDown, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCreateTags(t *testing.T) {
|
||||||
|
repo, tearDown, err := SetupTestProductTagsRepo("testDB")
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer tearDown()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
inputTag *entity.Tags
|
||||||
|
check func(t *testing.T, tag *entity.Tags)
|
||||||
|
expectErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Insert tag with zero ID",
|
||||||
|
inputTag: &entity.Tags{
|
||||||
|
// ID 為零值，Create 會自動生成
|
||||||
|
Types: product.ItemTypeSkill,
|
||||||
|
Name: "Tag A",
|
||||||
|
ShowType: product.ShowTypeNormal,
|
||||||
|
Cover: nil,
|
||||||
|
},
|
||||||
|
check: func(t *testing.T, tag *entity.Tags) {
|
||||||
|
require.False(t, tag.ID.IsZero(), "ID should be generated")
|
||||||
|
assert.NotZero(t, tag.CreatedAt, "CreatedAt should be set")
|
||||||
|
assert.NotZero(t, tag.UpdatedAt, "UpdatedAt should be set")
|
||||||
|
assert.Equal(t, "Tag A", tag.Name)
|
||||||
|
},
|
||||||
|
expectErr: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Insert tag with preset ID",
|
||||||
|
inputTag: &entity.Tags{
|
||||||
|
ID: primitive.NewObjectID(), // 預先設定 ID，不會被覆蓋
|
||||||
|
Types: product.ItemTypeSkill,
|
||||||
|
Name: "Tag B",
|
||||||
|
ShowType: product.ShowTypeNormal,
|
||||||
|
Cover: nil,
|
||||||
|
// 預設時間欄位
|
||||||
|
CreatedAt: 123456789,
|
||||||
|
UpdatedAt: 123456789,
|
||||||
|
},
|
||||||
|
check: func(t *testing.T, tag *entity.Tags) {
|
||||||
|
assert.False(t, tag.ID.IsZero())
|
||||||
|
assert.Equal(t, "Tag B", tag.Name)
|
||||||
|
// 保持原有的 CreatedAt 與 UpdatedAt 值
|
||||||
|
assert.Equal(t, int64(123456789), tag.CreatedAt)
|
||||||
|
assert.Equal(t, int64(123456789), tag.UpdatedAt)
|
||||||
|
},
|
||||||
|
expectErr: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tc := range tests {
|
||||||
|
tc := tc // capture range variable
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
err := repo.Create(ctx, tc.inputTag)
|
||||||
|
if tc.expectErr {
|
||||||
|
require.Error(t, err)
|
||||||
|
} else {
|
||||||
|
require.NoError(t, err)
|
||||||
|
// 驗證 Create 方法對輸入資料的影響
|
||||||
|
tc.check(t, tc.inputTag)
|
||||||
|
|
||||||
|
// 可選：利用 GetByID 再從資料庫中讀取進一步驗證
|
||||||
|
tagFromDB, err := repo.GetByID(ctx, tc.inputTag.ID.Hex())
|
||||||
|
require.NoError(t, err)
|
||||||
|
assert.Equal(t, tc.inputTag.Name, tagFromDB.Name)
|
||||||
|
assert.Equal(t, tc.inputTag.Types, tagFromDB.Types)
|
||||||
|
assert.Equal(t, tc.inputTag.ShowType, tagFromDB.ShowType)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetTagByID(t *testing.T) {
|
||||||
|
repo, tearDown, err := SetupTestProductTagsRepo("testDB")
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer tearDown()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
// 先建立一筆測試資料
|
||||||
|
tag := &entity.Tags{
|
||||||
|
Types: product.ItemTypeSkill,
|
||||||
|
Name: "Sample Tag",
|
||||||
|
ShowType: product.ShowTypeNormal,
|
||||||
|
Cover: nil,
|
||||||
|
}
|
||||||
|
err = repo.Create(ctx, tag)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
inputID string
|
||||||
|
expectedName string
|
||||||
|
expectedErr error
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Valid tag ID",
|
||||||
|
inputID: tag.ID.Hex(),
|
||||||
|
expectedName: "Sample Tag",
|
||||||
|
expectedErr: nil,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Invalid ObjectID format",
|
||||||
|
inputID: "invalid-hex",
|
||||||
|
expectedName: "",
|
||||||
|
// 這裡預期錯誤內容會包含 "hex" 字樣，故使用 nil 來做後續判斷
|
||||||
|
expectedErr: errors.New("invalid"),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Tag not found",
|
||||||
|
inputID: primitive.NewObjectID().Hex(),
|
||||||
|
expectedName: "",
|
||||||
|
expectedErr: ErrNotFound,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
tt := tt // capture range variable
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
res, err := repo.GetByID(ctx, tt.inputID)
|
||||||
|
if tt.expectedErr != nil {
|
||||||
|
require.Error(t, err)
|
||||||
|
if tt.name == "Invalid ObjectID format" {
|
||||||
|
assert.Contains(t, err.Error(), "hex")
|
||||||
|
} else {
|
||||||
|
assert.Equal(t, tt.expectedErr, err)
|
||||||
|
}
|
||||||
|
assert.Nil(t, res)
|
||||||
|
} else {
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.NotNil(t, res)
|
||||||
|
assert.Equal(t, tt.expectedName, res.Name)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUpdateTag(t *testing.T) {
|
||||||
|
repo, tearDown, err := SetupTestProductTagsRepo("testDB")
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer tearDown()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
// 先建立一筆初始的 Tag 測試資料
|
||||||
|
origTag := &entity.Tags{
|
||||||
|
Types: product.ItemTypeSkill,
|
||||||
|
ShowType: product.ShowTypeNormal,
|
||||||
|
Name: "Original Name",
|
||||||
|
}
|
||||||
|
err = repo.Create(ctx, origTag)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
id string
|
||||||
|
params repository.TagModifyParams
|
||||||
|
expectedName string
|
||||||
|
expectedTypes product.ItemType
|
||||||
|
expectedShowType product.ShowType
|
||||||
|
expectedCover *string
|
||||||
|
expectErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Update all fields",
|
||||||
|
id: origTag.ID.Hex(),
|
||||||
|
params: repository.TagModifyParams{
|
||||||
|
Name: ptr("New Name"),
|
||||||
|
Types: ptr(product.ItemTypeSkill),
|
||||||
|
ShowType: ptr(product.ShowTypeNormal),
|
||||||
|
Cover: ptr("new-cover.jpg"),
|
||||||
|
},
|
||||||
|
expectedName: "New Name",
|
||||||
|
expectedTypes: product.ItemTypeSkill,
|
||||||
|
expectedShowType: product.ShowTypeNormal,
|
||||||
|
expectedCover: ptr("new-cover.jpg"),
|
||||||
|
expectErr: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Update only name",
|
||||||
|
id: origTag.ID.Hex(),
|
||||||
|
params: repository.TagModifyParams{
|
||||||
|
Name: ptr("Another Name"),
|
||||||
|
},
|
||||||
|
// 其他欄位保持原值
|
||||||
|
expectedName: "Another Name",
|
||||||
|
expectedTypes: origTag.Types,
|
||||||
|
expectedShowType: origTag.ShowType,
|
||||||
|
expectedCover: ptr("new-cover.jpg"),
|
||||||
|
expectErr: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Invalid ObjectID",
|
||||||
|
id: "invalid-id",
|
||||||
|
params: repository.TagModifyParams{
|
||||||
|
Name: ptr("Should Not Update"),
|
||||||
|
},
|
||||||
|
expectErr: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
tt := tt // capture range variable
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
err := repo.Update(ctx, tt.id, tt.params)
|
||||||
|
if tt.expectErr {
|
||||||
|
require.Error(t, err)
|
||||||
|
} else {
|
||||||
|
require.NoError(t, err)
|
||||||
|
// 讀取更新後的資料
|
||||||
|
updated, err := repo.GetByID(ctx, tt.id)
|
||||||
|
require.NoError(t, err)
|
||||||
|
assert.Equal(t, tt.expectedName, updated.Name)
|
||||||
|
assert.Equal(t, tt.expectedTypes, updated.Types)
|
||||||
|
assert.Equal(t, tt.expectedShowType, updated.ShowType)
|
||||||
|
if tt.expectedCover == nil {
|
||||||
|
assert.Nil(t, updated.Cover)
|
||||||
|
} else {
|
||||||
|
assert.Equal(t, *tt.expectedCover, *updated.Cover)
|
||||||
|
}
|
||||||
|
// 驗證 updated_at 一定被更新（大於 0）
|
||||||
|
assert.NotZero(t, updated.UpdatedAt)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDeleteTag(t *testing.T) {
|
||||||
|
repo, tearDown, err := SetupTestProductTagsRepo("testDB")
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer tearDown()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
// 插入一筆測試資料
|
||||||
|
tag := &entity.Tags{
|
||||||
|
Types: product.ItemTypeSkill,
|
||||||
|
ShowType: product.ShowTypeNormal,
|
||||||
|
Name: "Test Tag",
|
||||||
|
// Cover 為 nil
|
||||||
|
}
|
||||||
|
err = repo.Create(ctx, tag)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
id string
|
||||||
|
expectErr error // 預期的錯誤，若為 nil 表示預期成功
|
||||||
|
checkDelete bool // 若為 true，表示刪除後需驗證資料不存在
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Delete existing tag",
|
||||||
|
id: tag.ID.Hex(),
|
||||||
|
expectErr: nil,
|
||||||
|
checkDelete: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Invalid ObjectID format",
|
||||||
|
id: "invalid-id",
|
||||||
|
expectErr: ErrInvalidObjectID,
|
||||||
|
checkDelete: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Delete non-existent tag",
|
||||||
|
id: primitive.NewObjectID().Hex(),
|
||||||
|
expectErr: nil,
|
||||||
|
checkDelete: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
tt := tt // capture range variable
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
err := repo.Delete(ctx, tt.id)
|
||||||
|
if tt.expectErr != nil {
|
||||||
|
require.Error(t, err)
|
||||||
|
assert.Equal(t, tt.expectErr, err)
|
||||||
|
} else {
|
||||||
|
require.NoError(t, err)
|
||||||
|
if tt.checkDelete {
|
||||||
|
// 刪除成功後，透過 GetByID 應查無資料
|
||||||
|
_, err := repo.GetByID(ctx, tt.id)
|
||||||
|
require.Error(t, err)
|
||||||
|
assert.Equal(t, ErrNotFound, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestListTags(t *testing.T) {
|
||||||
|
repo, tearDown, err := SetupTestProductTagsRepo("testDB")
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer tearDown()
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
now := time.Now().UnixNano()
|
||||||
|
|
||||||
|
// 建立測試資料，手動指定 ID 與時間（不讓 Create 自動覆蓋）
|
||||||
|
tag1 := &entity.Tags{
|
||||||
|
ID: primitive.NewObjectID(),
|
||||||
|
Name: "Tag A",
|
||||||
|
Types: product.ItemTypeSkill,
|
||||||
|
ShowType: product.ShowTypeNormal,
|
||||||
|
CreatedAt: now + 300,
|
||||||
|
UpdatedAt: now + 300,
|
||||||
|
}
|
||||||
|
tag2 := &entity.Tags{
|
||||||
|
ID: primitive.NewObjectID(),
|
||||||
|
Name: "Tag B",
|
||||||
|
Types: product.ItemTypeProduct,
|
||||||
|
ShowType: product.ShowTypeNormal,
|
||||||
|
CreatedAt: now + 200,
|
||||||
|
UpdatedAt: now + 200,
|
||||||
|
}
|
||||||
|
tag3 := &entity.Tags{
|
||||||
|
ID: primitive.NewObjectID(),
|
||||||
|
Types: product.ItemTypeProduct,
|
||||||
|
Name: "Tag C",
|
||||||
|
ShowType: product.ShowTypeNormal,
|
||||||
|
CreatedAt: now + 100,
|
||||||
|
UpdatedAt: now + 100,
|
||||||
|
}
|
||||||
|
|
||||||
|
// 插入測試資料
|
||||||
|
err = repo.Create(ctx, tag1)
|
||||||
|
require.NoError(t, err)
|
||||||
|
err = repo.Create(ctx, tag2)
|
||||||
|
require.NoError(t, err)
|
||||||
|
err = repo.Create(ctx, tag3)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
params repository.TagQueryParams
|
||||||
|
expectCount int64
|
||||||
|
expectIDsOrder []primitive.ObjectID
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Filter by Name",
|
||||||
|
params: repository.TagQueryParams{
|
||||||
|
Name: ptr("Tag A"),
|
||||||
|
PageSize: 10,
|
||||||
|
PageIndex: 1,
|
||||||
|
},
|
||||||
|
expectCount: 1,
|
||||||
|
expectIDsOrder: []primitive.ObjectID{tag1.ID},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Filter by Types = type1",
|
||||||
|
params: repository.TagQueryParams{
|
||||||
|
Types: ptr(product.ItemTypeProduct),
|
||||||
|
PageSize: 10,
|
||||||
|
PageIndex: 1,
|
||||||
|
},
|
||||||
|
// tag1與 tag2 符合條件
|
||||||
|
expectCount: 2,
|
||||||
|
// 排序依 updated_at 由大到小，故 tag1 (now+300) 在前，接著 tag2 (now+200)
|
||||||
|
expectIDsOrder: []primitive.ObjectID{tag1.ID, tag2.ID},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Filter by ShowType = show1",
|
||||||
|
params: repository.TagQueryParams{
|
||||||
|
ShowType: ptr(product.ShowTypeNormal),
|
||||||
|
PageSize: 10,
|
||||||
|
PageIndex: 1,
|
||||||
|
},
|
||||||
|
// tag1與 tag3 符合條件
|
||||||
|
expectCount: 3,
|
||||||
|
// 排序：tag1 (now+300) 在前，tag3 (now+100) 在後
|
||||||
|
expectIDsOrder: []primitive.ObjectID{tag1.ID, tag2.ID, tag3.ID},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Filter by Types=type1 and ShowType=show1",
|
||||||
|
params: repository.TagQueryParams{
|
||||||
|
Types: ptr(product.ItemTypeProduct),
|
||||||
|
ShowType: ptr(product.ShowTypeNormal),
|
||||||
|
PageSize: 10,
|
||||||
|
PageIndex: 1,
|
||||||
|
},
|
||||||
|
// 只有 tag1 同時符合兩個條件
|
||||||
|
expectCount: 2,
|
||||||
|
expectIDsOrder: []primitive.ObjectID{tag1.ID, tag3.ID},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Pagination works: PageIndex 1",
|
||||||
|
params: repository.TagQueryParams{
|
||||||
|
PageSize: 2,
|
||||||
|
PageIndex: 1,
|
||||||
|
},
|
||||||
|
// 全部 3 筆資料符合查詢條件，但分頁僅返回前 2 筆，排序依 updated_at 由大到小：tag1, tag2, tag3
|
||||||
|
expectCount: 3,
|
||||||
|
expectIDsOrder: []primitive.ObjectID{tag1.ID, tag2.ID},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Pagination works: PageIndex 2",
|
||||||
|
params: repository.TagQueryParams{
|
||||||
|
PageSize: 2,
|
||||||
|
PageIndex: 2,
|
||||||
|
},
|
||||||
|
// 第二頁應返回剩下的 1 筆：tag3
|
||||||
|
expectCount: 3,
|
||||||
|
expectIDsOrder: []primitive.ObjectID{tag3.ID},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
tt := tt // capture range variable
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
result, count, err := repo.List(ctx, tt.params)
|
||||||
|
require.NoError(t, err)
|
||||||
|
assert.Equal(t, tt.expectCount, count)
|
||||||
|
|
||||||
|
var gotIDs []primitive.ObjectID
|
||||||
|
for _, tag := range result {
|
||||||
|
gotIDs = append(gotIDs, tag.ID)
|
||||||
|
}
|
||||||
|
assert.Equal(t, tt.expectIDsOrder, gotIDs)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue