package service import ( "database/sql" "path/filepath" "strings" "testing" "github.com/xuri/excelize/v2" "cmautobuy/admin/model" ) // ── ParseSpec ────────────────────────────────────────── func TestParseSpec(t *testing.T) { cases := []struct { name string raw string wantColor string wantSize string wantAdvice string wantOK bool }{ { name: "带【】的建议", raw: "黑色,M【建議40-50公斤】", wantColor: "黑色", wantSize: "M", wantAdvice: "40-50公斤", wantOK: true, }, { name: "空格分隔的建议", raw: "卡其色拼黑色,L 建議50-57.5kg", wantColor: "卡其色拼黑色", wantSize: "L", wantAdvice: "50-57.5kg", wantOK: true, }, { name: "只有颜色和尺码没有建议", raw: "黑色,M", wantColor: "黑色", wantSize: "M", wantAdvice: "", wantOK: true, }, { name: "颜色自己也带【】", raw: "黑色 【夏裝單件T恤】,L 【57.5/70公斤】", wantColor: "黑色 【夏裝單件T恤】", wantSize: "L", wantAdvice: "57.5/70公斤", wantOK: true, }, { // 真实样本里的脏数据:缺左括号。见 ParseSpec 注释里"21 行括号 // 不配对怎么处理"一节——整体判定失败,不猜。 name: "括号不配对_缺左括号", raw: "紅色,3XL建議80-90公斤】", wantColor: "", wantSize: "", wantAdvice: "", wantOK: false, }, { // 真实样本里的另一种脏数据:缺右括号,而且用"寬鬆版"代替"建議"。 name: "括号不配对_缺右括号", raw: "粉色,3XL【寬鬆版 82.5-92.5kg", wantColor: "", wantSize: "", wantAdvice: "", wantOK: false, }, { name: "完全认不出的格式_没有逗号", raw: "莫名其妙的格式", wantColor: "", wantSize: "", wantAdvice: "", wantOK: false, }, { name: "两个逗号_超出已知范围", raw: "黑色,M,多余", wantColor: "", wantSize: "", wantAdvice: "", wantOK: false, }, { name: "空字符串", raw: "", wantColor: "", wantSize: "", wantAdvice: "", wantOK: false, }, { // raw 本身非空(trim 前有内容),但 trim 后为空——用来跟"空字符串" // 那条区分开:如果 trimmed=="" 分支直接把 raw 当返回值, // 这条能抓到(raw!=""),"空字符串"那条抓不到(raw 本身就是 "")。 name: "只有空白字符", raw: " ", wantColor: "", wantSize: "", wantAdvice: "", wantOK: false, }, { // 逗号右侧为空——只有颜色,没有尺码信息。 name: "逗号右侧为空", raw: "黑色,", wantColor: "", wantSize: "", wantAdvice: "", wantOK: false, }, { // 逗号左侧为空——只有尺码,没有颜色信息。 name: "逗号左侧为空", raw: ",M", wantColor: "", wantSize: "", wantAdvice: "", wantOK: false, }, { // 真实样本第 5454 行:右半整个被【】包住,括号外没有尺码文字, // 提取后 size 为空。这是真实数据里 23 行失败中的 2 行之一, // 直接用真实原文而不是自造用例,证明的是"这个真实格式我们 // 确实认不出来,而且认不出来时不猜",不只是代码逻辑自洽。 name: "右半整个被【】包住_size提取后为空", raw: "面膜安全褲3條膚色,【2xl 70-85公斤可穿】", wantColor: "", wantSize: "", wantAdvice: "", wantOK: false, }, } for _, tc := range cases { t.Run(tc.name, func(t *testing.T) { color, size, advice, ok := ParseSpec(tc.raw) if ok != tc.wantOK { t.Fatalf("ok = %v,想要 %v(raw=%q)", ok, tc.wantOK, tc.raw) } if !ok { // [必须] ok=false 时前三个必须全为空,不许出现"猜出来的一半"。 if color != "" || size != "" || advice != "" { t.Fatalf("ok=false 时 color/size/advice 必须全为空,得到 %q/%q/%q", color, size, advice) } return } if color != tc.wantColor || size != tc.wantSize || advice != tc.wantAdvice { t.Fatalf("got %q/%q/%q,want %q/%q/%q", color, size, advice, tc.wantColor, tc.wantSize, tc.wantAdvice) } }) } } // ── 测试固件 ────────────────────────────────────────── // shopeeHeader 是真实报表的表头,只留导入用得到的 8 个业务列—— // 和真实报表一样按名字定位,列的相对顺序特意打乱一下(把 8 列之外的 // 顺序无所谓,这里保持简单,只放业务列)。 var shopeeHeader = []string{ "商品ID", "商品名稱", "商品當前狀態", "商品規格ID", "商品規格", "規格當前狀態", "商品選項貨號", "主商品貨號", } // buildShopeeWorkbook 现造一个 xlsx 固件,覆盖工单要求的场景: // 正常行、三种规格格式、括号不配对、以及无关的广告报表 sheet // (用来验证"按名字取 sheet,不用下标 0")。 // // [必须] 测试不依赖 raw_data/,固件必须是现造或者提交进 admin/testdata/ // 的小文件,见工单 #38。这里选择现造:文件是二进制,提交进仓库不好 review。 func buildShopeeWorkbook(t *testing.T, header []string, rows [][]string) string { t.Helper() f := excelize.NewFile() defer f.Close() // 默认 sheet 改名成一个无关的广告报表,模拟真实工作簿里 // "最佳表現商品"不是第 0 个 sheet 这种情况(虽然样本里恰好是, // 但不能依赖这个巧合,见 ImportShopeeExcel 的实现和它的注释)。 f.SetSheetName("Sheet1", "追蹤商品廣告成效") f.NewSheet("最佳表現商品") if header != nil { for col, name := range header { cellRef, _ := excelize.CoordinatesToCellName(col+1, 1) f.SetCellStr("最佳表現商品", cellRef, name) } } for r, row := range rows { for col, val := range row { cellRef, _ := excelize.CoordinatesToCellName(col+1, r+2) f.SetCellStr("最佳表現商品", cellRef, val) } } path := filepath.Join(t.TempDir(), "shopee_test.xlsx") if err := f.SaveAs(path); err != nil { t.Fatalf("保存测试固件失败: %v", err) } return path } // row 按 shopeeHeader 的列顺序拼一行,减少每个测试用例里的列对齐负担。 func row(goodsID, title, status, specID, spec, specStatus, skuCode, mainSKUCode string) []string { return []string{goodsID, title, status, specID, spec, specStatus, skuCode, mainSKUCode} } // ── ImportShopeeExcel ────────────────────────────────── func TestImportShopeeExcel_基本导入(t *testing.T) { db := newTestDB(t) rows := [][]string{ // 商品 A:汇总行 + 3 种规格格式的 SKU 行 row("1001", "测试商品A", "正常", "-", "-", "", "", "货号A"), row("1001", "测试商品A", "正常", "9001", "黑色,M【建議40-50公斤】", "正常", "sku-9001", "货号A"), row("1001", "测试商品A", "正常", "9002", "卡其色拼黑色,L 建議50-57.5kg", "正常", "sku-9002", "货号A"), row("1001", "测试商品A", "正常", "9003", "黑色,M", "正常", "sku-9003", "货号A"), // 商品 B:汇总行 + 1 个解析失败的 SKU(括号不配对) row("1002", "测试商品B", "正常", "-", "-", "", "", "货号B"), row("1002", "测试商品B", "正常", "9004", "紅色,3XL建議80-90公斤】", "正常", "sku-9004", "货号B"), } path := buildShopeeWorkbook(t, shopeeHeader, rows) result, err := ImportShopeeExcel(db, path) if err != nil { t.Fatalf("导入失败: %v", err) } if result.ProductCount != 2 { t.Errorf("ProductCount = %d,想要 2", result.ProductCount) } if result.SKUCount != 4 { t.Errorf("SKUCount = %d,想要 4", result.SKUCount) } if len(result.Failures) != 1 { t.Fatalf("Failures 数量 = %d,想要 1(括号不配对那一行)", len(result.Failures)) } if result.Failures[0].Raw != "紅色,3XL建議80-90公斤】" { t.Errorf("失败行原文 = %q,不对", result.Failures[0].Raw) } if result.Failures[0].Row != 7 { // 表头第 1 行 + 6 条数据行,"紅色..."是第 6 条数据 -> Excel 行号 7 t.Errorf("失败行行号 = %d,想要 7", result.Failures[0].Row) } // 解析失败的 SKU 仍然要写进库:spec_raw 照存,color/size/advice 为空,parse_ok=0。 var color, size, advice sql.NullString var parseOK int err = db.QueryRow( `SELECT color, size, advice, parse_ok FROM shopee_skus WHERE sku_id = ?`, "9004", ).Scan(&color, &size, &advice, &parseOK) if err != nil { t.Fatalf("查询失败 SKU 出错: %v", err) } if color.Valid && color.String != "" { t.Errorf("解析失败的行 color 应为空,得到 %q", color.String) } if parseOK != 0 { t.Errorf("解析失败的行 parse_ok 应为 0,得到 %d", parseOK) } } func TestImportShopeeExcel_错误sheet名(t *testing.T) { db := newTestDB(t) f := excelize.NewFile() f.SetSheetName("Sheet1", "新上架商品") f.NewSheet("高潛力廣告商品") path := filepath.Join(t.TempDir(), "no_target_sheet.xlsx") if err := f.SaveAs(path); err != nil { t.Fatalf("保存测试固件失败: %v", err) } f.Close() _, err := ImportShopeeExcel(db, path) if err == nil { t.Fatal("缺少「最佳表現商品」sheet 应该报错") } msg := err.Error() if !strings.Contains(msg, "最佳表現商品") || !strings.Contains(msg, "新上架商品") || !strings.Contains(msg, "高潛力廣告商品") { t.Errorf("错误信息应列出实际有哪些 sheet,得到: %s", msg) } } func TestImportShopeeExcel_缺列(t *testing.T) { db := newTestDB(t) // 表头缺了「商品規格ID」,模拟蝦皮加/删列导致列错位的场景。 badHeader := []string{"商品ID", "商品名稱", "商品當前狀態", "商品規格", "規格當前狀態", "商品選項貨號", "主商品貨號"} path := buildShopeeWorkbook(t, badHeader, nil) _, err := ImportShopeeExcel(db, path) if err == nil { t.Fatal("缺少必需列应该报错") } if !strings.Contains(err.Error(), "商品規格ID") { t.Errorf("错误信息应说明缺的是哪一列,得到: %s", err.Error()) } } // TestImportShopeeExcel_不覆盖人工字段 是工单 #38 里最要命的一条: // 报表里没有 pdd_goods_url / pdd_goods_id 这两列,upsert 绝不能把它们 // 写成空值,否则操作员攒的 PDD 链接会被一次导入洗光,而且不报错。 func TestImportShopeeExcel_不覆盖人工字段(t *testing.T) { db := newTestDB(t) rows := [][]string{ row("2001", "测试商品C", "正常", "-", "-", "", "", "货号C"), row("2001", "测试商品C", "正常", "8001", "黑色,M", "正常", "sku-8001", "货号C"), } path := buildShopeeWorkbook(t, shopeeHeader, rows) // 先导入一次,让商品存在。 if _, err := ImportShopeeExcel(db, path); err != nil { t.Fatalf("第一次导入失败: %v", err) } // 模拟操作员手工填了 PDD 链接。 const manualURL = "https://mobile.yangkeduo.com/goods.html?goods_id=999999" if _, err := db.Exec( `UPDATE shopee_products SET pdd_goods_url = ?, pdd_goods_id = ? WHERE goods_id = ?`, manualURL, "999999", "2001", ); err != nil { t.Fatalf("模拟人工填链接失败: %v", err) } // 再导入同一个文件(同一个商品名字也可能变了,模拟蝦皮改了标题)。 rows[0][1] = "测试商品C改名后" rows[1][1] = "测试商品C改名后" path2 := buildShopeeWorkbook(t, shopeeHeader, rows) if _, err := ImportShopeeExcel(db, path2); err != nil { t.Fatalf("第二次导入失败: %v", err) } var gotURL, gotGoodsID, gotTitle sql.NullString err := db.QueryRow( `SELECT pdd_goods_url, pdd_goods_id, title FROM shopee_products WHERE goods_id = ?`, "2001", ).Scan(&gotURL, &gotGoodsID, &gotTitle) if err != nil { t.Fatalf("查询商品失败: %v", err) } if gotURL.String != manualURL { t.Errorf("pdd_goods_url 被导入洗掉了:got %q, want %q", gotURL.String, manualURL) } if gotGoodsID.String != "999999" { t.Errorf("pdd_goods_id 被导入洗掉了:got %q, want %q", gotGoodsID.String, "999999") } // 标题这类报表字段应该正常刷新,证明"不覆盖"只针对人工字段,不是整行不更新。 if gotTitle.String != "测试商品C改名后" { t.Errorf("title 应该被报表里的新值刷新:got %q", gotTitle.String) } } // TestImportShopeeExcel_不删除人工新增SKU 对应「05」§4.4 提到的人工新增 SKU // (本工单不实现新增入口,但导入必须留出这个口子,不能做成 // "先 DELETE 再 INSERT" 的全量替换)。 func TestImportShopeeExcel_不删除人工新增SKU(t *testing.T) { db := newTestDB(t) rows := [][]string{ row("3001", "测试商品D", "正常", "-", "-", "", "", "货号D"), row("3001", "测试商品D", "正常", "7001", "黑色,M", "正常", "sku-7001", "货号D"), } path := buildShopeeWorkbook(t, shopeeHeader, rows) if _, err := ImportShopeeExcel(db, path); err != nil { t.Fatalf("导入失败: %v", err) } // 模拟操作员手动新增一个报表里没有的 SKU。 now := model.NowISO() if _, err := db.Exec(` INSERT INTO shopee_skus (sku_id, goods_id, spec_raw, color, size, advice, parse_ok, sku_code, is_manual, created_at, updated_at) VALUES ('manual-1', '3001', '手动新增-均码', '均码', '', '', 1, '', 1, ?, ?)`, now, now); err != nil { t.Fatalf("插入人工新增 SKU 失败: %v", err) } // 再导入同一个文件。 if _, err := ImportShopeeExcel(db, path); err != nil { t.Fatalf("第二次导入失败: %v", err) } var isManual int err := db.QueryRow(`SELECT is_manual FROM shopee_skus WHERE sku_id = 'manual-1'`).Scan(&isManual) if err != nil { t.Fatalf("人工新增的 SKU 被导入删掉了: %v", err) } if isManual != 1 { t.Errorf("is_manual 应该还是 1,得到 %d", isManual) } } func TestImportShopeeExcel_二次导入不翻倍(t *testing.T) { db := newTestDB(t) rows := [][]string{ row("4001", "测试商品E", "正常", "-", "-", "", "", "货号E"), row("4001", "测试商品E", "正常", "6001", "黑色,M", "正常", "sku-6001", "货号E"), row("4001", "测试商品E", "正常", "6002", "白色,L", "正常", "sku-6002", "货号E"), } path := buildShopeeWorkbook(t, shopeeHeader, rows) first, err := ImportShopeeExcel(db, path) if err != nil { t.Fatalf("第一次导入失败: %v", err) } second, err := ImportShopeeExcel(db, path) if err != nil { t.Fatalf("第二次导入失败: %v", err) } if first.ProductCount != second.ProductCount || first.SKUCount != second.SKUCount { t.Fatalf("两次导入统计不一致: %+v vs %+v", first, second) } var productCount, skuCount int db.QueryRow(`SELECT COUNT(*) FROM shopee_products WHERE goods_id = '4001'`).Scan(&productCount) db.QueryRow(`SELECT COUNT(*) FROM shopee_skus WHERE goods_id = '4001'`).Scan(&skuCount) if productCount != 1 { t.Errorf("shopee_products 应该只有 1 行,得到 %d(重复导入产生了重复行)", productCount) } if skuCount != 2 { t.Errorf("shopee_skus 应该只有 2 行,得到 %d(重复导入产生了重复行)", skuCount) } }