96 lines
3.3 KiB
Go
96 lines
3.3 KiB
Go
package service
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"affiliate_dash/internal/model"
|
|
)
|
|
|
|
func TestLegacyOrderCreateRejectsDistributorOutsideMerchant(t *testing.T) {
|
|
db := newServiceTestDB(t)
|
|
merchant := model.Merchant{Code: "legacy-merchant", Name: "旧后台商户", Status: model.MerchantStatusActive}
|
|
if err := db.Create(&merchant).Error; err != nil {
|
|
t.Fatalf("create merchant: %v", err)
|
|
}
|
|
inside := model.User{Username: "inside", PasswordHash: "hash", Role: model.RoleDistributor, Status: 1, InviteCode: "INSIDE"}
|
|
outside := model.User{Username: "outside", PasswordHash: "hash", Role: model.RoleDistributor, Status: 1, InviteCode: "OUTSIDE"}
|
|
if err := db.Create(&inside).Error; err != nil {
|
|
t.Fatalf("create inside user: %v", err)
|
|
}
|
|
if err := db.Create(&outside).Error; err != nil {
|
|
t.Fatalf("create outside user: %v", err)
|
|
}
|
|
if err := db.Create(&model.MerchantMember{
|
|
MerchantID: merchant.ID,
|
|
UserID: inside.ID,
|
|
Role: model.MemberRoleOperator,
|
|
Status: 1,
|
|
}).Error; err != nil {
|
|
t.Fatalf("create member: %v", err)
|
|
}
|
|
skin := model.Skin{
|
|
MerchantID: merchant.ID,
|
|
Name: "旧皮肤",
|
|
SKU: "legacy-skin",
|
|
Price: 10,
|
|
Stock: -1,
|
|
Status: 1,
|
|
}
|
|
if err := db.Create(&skin).Error; err != nil {
|
|
t.Fatalf("create skin: %v", err)
|
|
}
|
|
|
|
_, err := NewOrderService(db).Create(CreateOrderInput{
|
|
MerchantID: merchant.ID,
|
|
SkinID: skin.ID,
|
|
DistributorID: outside.ID,
|
|
BuyerName: "买家",
|
|
Status: model.OrderStatusPaid,
|
|
})
|
|
if err == nil || !strings.Contains(err.Error(), "不属于当前商户") {
|
|
t.Fatalf("expected tenant boundary error, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestUserListFiltersByMerchantMembership(t *testing.T) {
|
|
db := newServiceTestDB(t)
|
|
merchantA := model.Merchant{Code: "user-merchant-a", Name: "商户 A", Status: model.MerchantStatusActive}
|
|
merchantB := model.Merchant{Code: "user-merchant-b", Name: "商户 B", Status: model.MerchantStatusActive}
|
|
if err := db.Create(&merchantA).Error; err != nil {
|
|
t.Fatalf("create merchant a: %v", err)
|
|
}
|
|
if err := db.Create(&merchantB).Error; err != nil {
|
|
t.Fatalf("create merchant b: %v", err)
|
|
}
|
|
userA := model.User{Username: "user-a", PasswordHash: "hash", Role: model.RoleDistributor, Status: 1, InviteCode: "USERA"}
|
|
userB := model.User{Username: "user-b", PasswordHash: "hash", Role: model.RoleDistributor, Status: 1, InviteCode: "USERB"}
|
|
if err := db.Create(&userA).Error; err != nil {
|
|
t.Fatalf("create user a: %v", err)
|
|
}
|
|
if err := db.Create(&userB).Error; err != nil {
|
|
t.Fatalf("create user b: %v", err)
|
|
}
|
|
if err := db.Create(&model.MerchantMember{MerchantID: merchantA.ID, UserID: userA.ID, Role: model.MemberRoleOperator, Status: 1}).Error; err != nil {
|
|
t.Fatalf("create member a: %v", err)
|
|
}
|
|
if err := db.Create(&model.MerchantMember{MerchantID: merchantB.ID, UserID: userB.ID, Role: model.MemberRoleOperator, Status: 1}).Error; err != nil {
|
|
t.Fatalf("create member b: %v", err)
|
|
}
|
|
|
|
active := 1
|
|
list, total, err := NewUserService(db, nil).List(UserListQuery{
|
|
MerchantID: merchantA.ID,
|
|
Page: 1,
|
|
Size: 20,
|
|
Role: model.RoleDistributor,
|
|
Status: &active,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("list users: %v", err)
|
|
}
|
|
if total != 1 || len(list) != 1 || list[0].Username != "user-a" {
|
|
t.Fatalf("expected only merchant A user, total=%d list=%+v", total, list)
|
|
}
|
|
}
|