实现多商户履约平台基础

2026-07-30 12:02:32 +08:00
parent dde6f2c269
commit 83429456a6
41 changed files with 5487 additions and 504 deletions
@@ -0,0 +1,95 @@
+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)
+	}
+}