- add unit test to test keyToRemove and keyToKeep in normal access.

This commit is contained in:
gazebo
2018-06-05 17:05:43 +08:00
parent ad649555c4
commit 559dd82be5
2 changed files with 11 additions and 6 deletions

View File

@@ -83,7 +83,6 @@ func getJDOperator() string {
}
func getCurTimeStr() string {
// return "2018-06-04 22:35:25"
return time.Now().Format("2006-01-02 15:04:05")
}
@@ -236,12 +235,12 @@ func (j JDAPI) accessJDQueryNoPage(apiStr string, jdParams map[string]string, ke
if _, ok := jdResultInnerCodeOK[innerCode]; ok {
for _, innerDataKey := range jdResultNoPageInnerDataKeys {
if innerData, ok := data[innerDataKey]; ok {
return innerData, nil
return utils.DictKeysMan(innerData, keyToRemove, keyToKeep), nil
}
}
panic("Can not find inner data")
} else {
return utils.DictKeysMan(jsonResult, keyToRemove, keyToKeep), ErrInnerCodeIsNotOk
return jsonResult, ErrInnerCodeIsNotOk
}
}

View File

@@ -35,7 +35,7 @@ func TestAccessJDQuery(t *testing.T) {
}
func TestAccessJDQueryNoPage(t *testing.T) {
result, err := jdapi.accessJDQueryNoPage("address/allcities", nil, nil, nil)
result, err := jdapi.accessJDQueryNoPage("address/allcities", nil, []string{"yn"}, nil)
if err != nil {
t.Fatalf("accessJDQueryNoPage return error:%v", err)
}
@@ -47,21 +47,27 @@ func TestAccessJDQueryNoPage(t *testing.T) {
if _, ok := oneCity["areaName"]; !ok {
t.Fatal("no areaName key")
}
if _, ok := oneCity["yn"]; ok {
t.Fatal("yn field havn't been removed")
}
}
func TestAccessJDQueryHavePage(t *testing.T) {
result, err := jdapi.accessJDQueryHavePage("pms/querySkuInfos", nil, nil, nil, 1, 50, nil)
result, err := jdapi.accessJDQueryHavePage("pms/querySkuInfos", nil, nil, []string{"skuName", "skuId"}, 1, 50, nil)
if err != nil {
t.Fatalf("accessJDQueryHavePage return error:%v", err)
}
skuInfo := result.([]interface{})
if len(skuInfo) == 0 {
t.Fatal("city info is empty")
t.Fatal("sku info is empty")
}
oneSku := skuInfo[0].(map[string]interface{})
if _, ok := oneSku["skuName"]; !ok {
t.Fatal("no skuName key")
}
if _, ok := oneSku["outSkuId"]; ok {
t.Fatal("outSkuId key not removed")
}
}
func TestGenGetURL(t *testing.T) {