36 lines
1.1 KiB
Go
36 lines
1.1 KiB
Go
package helper
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/yaoapp/kun/maps"
|
|
)
|
|
|
|
func TestArrayPluck(t *testing.T) {
|
|
columns := []string{"城市", "行业", "计费"}
|
|
pluck := map[string]interface{}{
|
|
"行业": map[string]interface{}{"key": "city", "value": "数量", "items": []map[string]interface{}{{"city": "北京", "数量": 32}, {"city": "上海", "数量": 20}}},
|
|
"计费": map[string]interface{}{"key": "city", "value": "计费种类", "items": []map[string]interface{}{{"city": "北京", "计费种类": 6}, {"city": "西安", "计费种类": 3}}},
|
|
}
|
|
items := ArrayPluck(columns, pluck)
|
|
assert.Equal(t, 3, len(items))
|
|
for _, item := range items {
|
|
maps.Of(item).Has("城市")
|
|
maps.Of(item).Has("行业")
|
|
maps.Of(item).Has("计费")
|
|
}
|
|
}
|
|
|
|
func TestArraySplit(t *testing.T) {
|
|
records := []map[string]interface{}{
|
|
{"name": "阿里云计算有限公司", "short_name": "阿里云"},
|
|
{"name": "世纪互联蓝云", "short_name": "上海蓝云"},
|
|
}
|
|
columns, values := ArraySplit(records)
|
|
assert.Equal(t, 2, len(columns))
|
|
assert.Equal(t, 2, len(values))
|
|
for _, value := range values {
|
|
assert.Equal(t, 2, len(value))
|
|
}
|
|
}
|