feat: 支持蝦皮批量创建 PDD 采集任务 (#185)

2026-08-12 16:12:35 +08:00
parent 250998b7b9
commit 07ca1b0afb
10 changed files with 280 additions and 26 deletions
@@ -3,6 +3,7 @@ package service
 import (
 	"database/sql"
 	"errors"
+	"strings"
 	"testing"
 	"time"

@@ -120,6 +121,92 @@ func TestCreateShopeePddCollectTaskForUser_记录当前用户(t *testing.T) {
 	}
 }

+func TestCreateShopeePddCollectTasksForUser_按Pdd去重并分类跳过(t *testing.T) {
+	db := newTestDB(t)
+	admin, _, _ := insertTaskUsers(t, db)
+	for _, goodsID := range []string{"S-A", "S-B", "S-UNLINKED", "S-DELETED"} {
+		seedShopeeProduct(t, db, goodsID, goodsID)
+	}
+	for _, goodsID := range []string{"S-A", "S-B"} {
+		if _, err := AssociateShopeePdd(db, goodsID, pddURLA, false); err != nil {
+			t.Fatal(err)
+		}
+	}
+	if _, err := AssociateShopeePdd(db, "S-DELETED", pddURLB, false); err != nil {
+		t.Fatal(err)
+	}
+	if err := repository.SoftDeletePddProduct(db, "937122477375"); err != nil {
+		t.Fatal(err)
+	}
+
+	result, err := CreateShopeePddCollectTasksForUser(db, admin,
+		[]string{"S-A", "S-B", "S-A", "S-UNLINKED", "S-DELETED", "S-MISSING"}, "")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if result.Created != 1 || result.SkippedUnlinked != 1 || result.SkippedShopeeMissing != 1 || result.SkippedDeleted != 1 {
+		t.Fatalf("批量分类或 PDD 去重错误: %+v", result)
+	}
+	var taskCount int
+	if err := db.QueryRow(`SELECT COUNT(*) FROM tasks WHERE task_type='collect'`).Scan(&taskCount); err != nil || taskCount != 1 {
+		t.Fatalf("相同 PDD 商品只应创建一条任务，count=%d err=%v", taskCount, err)
+	}
+	message := FormatCollectTaskMessage(result)
+	for _, want := range []string{"已创建 1 个", "1 个蝦皮商品已不存在", "1 个未关联 PDD", "1 个已删除的"} {
+		if !strings.Contains(message, want) {
+			t.Errorf("批量结果提示 %q 缺少 %q", message, want)
+		}
+	}
+}
+
+func TestCreateShopeePddCollectTasksForUser_已采集和正常采集中分别跳过(t *testing.T) {
+	db := newTestDB(t)
+	admin, _, _ := insertTaskUsers(t, db)
+	seedShopeeProduct(t, db, "S-COLLECTED", "已采集")
+	seedShopeeProduct(t, db, "S-COLLECTING", "采集中")
+	AssociateShopeePdd(db, "S-COLLECTED", pddURLA, false)
+	AssociateShopeePdd(db, "S-COLLECTING", pddURLB, false)
+	if err := repository.SetCollectResult(db, "737116531267", "商品", "店铺", collectedThreeDimensions); err != nil {
+		t.Fatal(err)
+	}
+	if result, err := CreatePddCollectTasksForUser(db, admin, []string{"937122477375"}, ""); err != nil || result.Created != 1 {
+		t.Fatalf("准备采集中任务失败: result=%+v err=%v", result, err)
+	}
+
+	result, err := CreateShopeePddCollectTasksForUser(db, admin,
+		[]string{"S-COLLECTED", "S-COLLECTING"}, "")
+	if err != nil || result.Created != 0 || result.SkippedCollected != 1 || result.SkippedCollecting != 1 {
+		t.Fatalf("状态分类错误: result=%+v err=%v", result, err)
+	}
+}
+
+func TestCreateShopeePddCollectTasksForUser_拒绝不可见客户端且整批不创建(t *testing.T) {
+	db := newTestDB(t)
+	admin, buyerA, buyerB := prepareClientAssignmentUsers(t, db)
+	if err := RegisterClient(db, model.Client{ClientID: "client-batch-b", Name: "B 的客户端"}, true); err != nil {
+		t.Fatal(err)
+	}
+	if _, _, err := AssignClient(db, admin, "client-batch-b", buyerB.UserID, time.Now()); err != nil {
+		t.Fatal(err)
+	}
+	seedShopeeProduct(t, db, "S-PERM", "权限商品")
+	AssociateShopeePdd(db, "S-PERM", pddURLA, false)
+
+	_, err := CreateShopeePddCollectTasksForUser(db, buyerA, []string{"S-PERM"}, "client-batch-b")
+	if !IsValidationError(err) {
+		t.Fatalf("伪造他人客户端应返回校验错误，实际 %v", err)
+	}
+	var taskCount int
+	db.QueryRow(`SELECT COUNT(*) FROM tasks WHERE task_type='collect'`).Scan(&taskCount)
+	if taskCount != 0 {
+		t.Fatalf("权限失败后不应创建任务，实际 %d", taskCount)
+	}
+	product, _ := repository.GetPddProductByGoodsID(db, "737116531267")
+	if product.CollectStatus != model.CollectPending {
+		t.Fatalf("权限失败后商品状态应保持 pending，实际 %s", product.CollectStatus)
+	}
+}
+
 func TestCreateSybPddCollectTasksForUser_同一PDD商品去重(t *testing.T) {
 	db := newTestDB(t)
 	admin, _, _ := insertTaskUsers(t, db)