wp-go/helper/maps/function_test.go
2023-02-10 21:23:30 +08:00

126 lines
2.3 KiB
Go

package maps
import (
"reflect"
"testing"
)
func TestGetStrMapAnyVal(t *testing.T) {
type args struct {
key string
v map[string]any
}
type testCase[T any] struct {
name string
args args
wantR T
wantO bool
}
tests := []testCase[int]{
{name: "t1", args: args{
key: "k1",
v: map[string]any{
"k1": 1,
"k2": 2,
},
}, wantR: 1, wantO: true},
{name: "t2", args: args{
key: "k2.kk",
v: map[string]any{
"k1": 1,
"k2": map[string]any{
"kk": 10,
},
},
}, wantR: 10, wantO: true},
{name: "t3", args: args{
key: "k2.vv",
v: map[string]any{
"k1": 1,
"k2": map[string]any{
"kk": 10,
},
},
}, wantR: 0, wantO: false},
{name: "t4", args: args{
key: "k3",
v: map[string]any{
"k1": 1,
"k2": map[string]any{
"kk": 10,
},
},
}, wantR: 0, wantO: false},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
gotR, gotO := GetStrMapAnyVal[int](tt.args.v, tt.args.key)
if !reflect.DeepEqual(gotR, tt.wantR) {
t.Errorf("GetStrMapAnyVal() gotR = %v, want %v", gotR, tt.wantR)
}
if gotO != tt.wantO {
t.Errorf("GetStrMapAnyVal() gotO = %v, want %v", gotO, tt.wantO)
}
})
}
}
func TestGetStrMapAnyValWithAny(t *testing.T) {
type args struct {
key string
v map[string]any
}
tests := []struct {
name string
args args
wantR any
wantO bool
}{
{name: "t1", args: args{
key: "k1",
v: map[string]any{
"k1": 1,
"k2": 2,
},
}, wantR: any(1), wantO: true},
{name: "t2", args: args{
key: "k2.kk",
v: map[string]any{
"k1": 1,
"k2": map[string]any{
"kk": 10,
},
},
}, wantR: any(10), wantO: true},
{name: "t3", args: args{
key: "k2.vv",
v: map[string]any{
"k1": 1,
"k2": map[string]any{
"kk": 10,
},
},
}, wantR: nil, wantO: false},
{name: "t4", args: args{
key: "k3",
v: map[string]any{
"k1": 1,
"k2": map[string]any{
"kk": 10,
},
},
}, wantR: nil, wantO: false},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
gotR, gotO := GetStrMapAnyValWithAny(tt.args.key, tt.args.v)
if !reflect.DeepEqual(gotR, tt.wantR) {
t.Errorf("GetStrMapAnyValWithAny() gotR = %v, want %v", gotR, tt.wantR)
}
if gotO != tt.wantO {
t.Errorf("GetStrMapAnyValWithAny() gotO = %v, want %v", gotO, tt.wantO)
}
})
}
}