fix: 修复顺运宝采集任务孤儿状态 (#165)

admin/service/shopee_pdd_test.go

@@ -143,6 +143,42 @@ func TestCreateSybPddCollectTasksForUser_同一PDD商品去重(t *testing.T) {
 	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)
 	}
+	var sourceCount int
+	if err := db.QueryRow(`SELECT COUNT(*) FROM task_syb_sources`).Scan(&sourceCount); err != nil || sourceCount != 2 {
+		t.Fatalf("去重任务应保留两条顺运宝来源，count=%d err=%v", sourceCount, err)
+	}
+	for _, keyword := range []string{"ORDER-SYB-1", "ORDER-SYB-2", "SYB-1", "SYB-2"} {
+		list, err := ListTasksView(db, repository.TaskFilter{Keyword: keyword}, 1)
+		if err != nil || list.Total != 1 || len(list.Rows) != 1 {
+			t.Fatalf("按顺运宝来源 %q 搜索任务失败: total=%d rows=%d err=%v", keyword, list.Total, len(list.Rows), err)
+		}
+	}
+	if _, err := CreateSybPddCollectTasksForUser(db, admin, []string{"SYB-1"}, ""); !IsValidationError(err) {
+		t.Fatalf("未超时且有有效任务时应拒绝重复创建，实际 %v", err)
+	}
+}
+
+func TestCreateSybPddCollectTasksForUser_无任务的采集中状态可恢复(t *testing.T) {
+	db := newTestDB(t)
+	admin, _, _ := insertTaskUsers(t, db)
+	seedShopeeProduct(t, db, "1001", "商品一")
+	if _, err := AssociateShopeePdd(db, "1001", pddURLA, false); err != nil {
+		t.Fatal(err)
+	}
+	seedSybCollectOrder(t, db, "SYB-ORPHAN", "1001")
+	if _, err := db.Exec(`UPDATE pdd_products SET collect_status='collecting',updated_at=? WHERE goods_id=?`,
+		model.NowISO(), "737116531267"); err != nil {
+		t.Fatal(err)
+	}
+
+	result, err := CreateSybPddCollectTasksForUser(db, admin, []string{"SYB-ORPHAN"}, "")
+	if err != nil || result.Created != 1 {
+		t.Fatalf("孤儿采集中状态应能重建任务：result=%+v err=%v", result, err)
+	}
+	var sources int
+	if err := db.QueryRow(`SELECT COUNT(*) FROM task_syb_sources WHERE syb_id='SYB-ORPHAN'`).Scan(&sources); err != nil || sources != 1 {
+		t.Fatalf("重建任务应保存顺运宝来源：count=%d err=%v", sources, err)
+	}
 }
 
 func TestCreateSybPddCollectTasksForUser_不可采集明细整批拒绝(t *testing.T) {