From 3c5a0adb95c50436bc74c67c822c76d27024f711 Mon Sep 17 00:00:00 2001
From: gazebo <gazeboxu@gmail.com>
Date: Tue, 23 Jul 2019 18:07:15 +0800
Subject: [PATCH] - ebaiapi.SkuShopCategoryMap

---
 platformapi/ebaiapi/shop.go          |  1 +
 platformapi/ebaiapi/shop_sku.go      | 13 ++++++++++---
 platformapi/ebaiapi/shop_sku_test.go |  7 ++++++-
 3 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/platformapi/ebaiapi/shop.go b/platformapi/ebaiapi/shop.go
index 96bff74e..d4a8e4e6 100644
--- a/platformapi/ebaiapi/shop.go
+++ b/platformapi/ebaiapi/shop.go
@@ -38,6 +38,7 @@ const (
 	KeyCustomSkuID = "custom_sku_id"
 	KeySkuID       = "sku_id"
 	KeyUPC         = "upc"
+	KeyCategoryID  = "category_id"
 
 	KeyName  = "name"
 	KeyPhone = "phone"
diff --git a/platformapi/ebaiapi/shop_sku.go b/platformapi/ebaiapi/shop_sku.go
index a2207461..92860ff4 100644
--- a/platformapi/ebaiapi/shop_sku.go
+++ b/platformapi/ebaiapi/shop_sku.go
@@ -446,11 +446,18 @@ func (a *API) SkuStockUpdateOne(shopID string, stockInfo *ShopSkuInfo) (err erro
 	return err
 }
 
-func (a *API) SkuShopCategoryMap(shopID string, skuID int64, categoryIDsStr string) (err error) {
+func (a *API) SkuShopCategoryMap(shopID string, skuID int64, customSkuID string, categoryID int64, rank int) (err error) {
 	params := map[string]interface{}{
 		KeyShopID:     shopID,
-		KeySkuID:      skuID,
-		"category_id": categoryIDsStr,
+		KeyCategoryID: categoryID,
+	}
+	if skuID > 0 {
+		params[KeySkuID] = skuID
+	} else {
+		params[KeyCustomSkuID] = customSkuID
+	}
+	if rank > 0 {
+		params["rank"] = rank
 	}
 	_, err = a.AccessAPI("sku.shop.category.map", params)
 	return err
diff --git a/platformapi/ebaiapi/shop_sku_test.go b/platformapi/ebaiapi/shop_sku_test.go
index ec52a631..59fe0403 100644
--- a/platformapi/ebaiapi/shop_sku_test.go
+++ b/platformapi/ebaiapi/shop_sku_test.go
@@ -108,7 +108,7 @@ func TestSkuUpdate(t *testing.T) {
 }
 
 func TestSkuShopCategoryMap(t *testing.T) {
-	err := api.SkuShopCategoryMap(testShopID, 15378849314129969, "153760472317166")
+	err := api.SkuShopCategoryMap(testShopID, 0, "2212", 15637192270716, 100)
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -191,3 +191,8 @@ func TestGetEbaiCatIDFromName(t *testing.T) {
 	ebaiCatID := api.GetEbaiCatIDFromName("300280", "应季水果新鲜美味")
 	t.Log(ebaiCatID)
 }
+
+func TestGetEbaiSkuIDFromCustomID(t *testing.T) {
+	ebaiCatID := api.GetEbaiSkuIDFromCustomID("300001", "31554")
+	t.Log(ebaiCatID)
+}