316 lines
6.8 KiB
Go
316 lines
6.8 KiB
Go
package reflect_test
|
|
|
|
import (
|
|
"net/url"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
rutil "go.unistack.org/micro/v3/util/reflect"
|
|
)
|
|
|
|
func TestStructfields(t *testing.T) {
|
|
type Config struct {
|
|
Wait time.Duration
|
|
Time time.Time
|
|
Metadata map[string]int
|
|
Broker string
|
|
Addr []string
|
|
Verbose bool
|
|
Nested *Config
|
|
}
|
|
cfg := &Config{Nested: &Config{}}
|
|
fields, err := rutil.StructFields(cfg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(fields) != 13 {
|
|
t.Fatalf("invalid fields number: %v", fields)
|
|
}
|
|
}
|
|
|
|
func TestSetFieldByPath(t *testing.T) {
|
|
type NestedStr struct {
|
|
BBB string `json:"bbb"`
|
|
CCC int `json:"ccc"`
|
|
}
|
|
type Str1 struct {
|
|
Name []string `json:"name" codec:"flatten"`
|
|
XXX string `json:"xxx"`
|
|
Nested NestedStr `json:"nested"`
|
|
}
|
|
type Str2 struct {
|
|
XXX string `json:"xxx"`
|
|
Nested *NestedStr `json:"nested"`
|
|
Name []string `json:"name" codec:"flatten"`
|
|
}
|
|
var err error
|
|
val1 := &Str1{Name: []string{"first", "second"}, XXX: "ttt", Nested: NestedStr{BBB: "ddd", CCC: 9}}
|
|
val2 := &Str2{Name: []string{"first", "second"}, XXX: "ttt", Nested: &NestedStr{BBB: "ddd", CCC: 9}}
|
|
err = rutil.SetFieldByPath(val1, "xxx", "Nested.BBB")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val1.Nested.BBB != "xxx" {
|
|
t.Fatalf("SetFieldByPath not works: %#+v", val1)
|
|
}
|
|
err = rutil.SetFieldByPath(val2, "xxx", "Nested.BBB")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val2.Nested.BBB != "xxx" {
|
|
t.Fatalf("SetFieldByPath not works: %#+v", val1)
|
|
}
|
|
}
|
|
|
|
func TestZeroFieldByPath(t *testing.T) {
|
|
type NestedStr struct {
|
|
BBB string `json:"bbb"`
|
|
CCC int `json:"ccc"`
|
|
}
|
|
type Str1 struct {
|
|
Name []string `json:"name" codec:"flatten"`
|
|
XXX string `json:"xxx"`
|
|
Nested NestedStr `json:"nested"`
|
|
}
|
|
type Str2 struct {
|
|
XXX string `json:"xxx"`
|
|
Nested *NestedStr `json:"nested"`
|
|
Name []string `json:"name" codec:"flatten"`
|
|
}
|
|
var err error
|
|
val1 := &Str1{Name: []string{"first", "second"}, XXX: "ttt", Nested: NestedStr{BBB: "ddd", CCC: 9}}
|
|
|
|
err = rutil.ZeroFieldByPath(val1, "Nested.BBB")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = rutil.ZeroFieldByPath(val1, "Nested")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val1.Nested.BBB == "ddd" {
|
|
t.Fatalf("zero field not works: %v", val1)
|
|
}
|
|
|
|
val2 := &Str2{Name: []string{"first", "second"}, XXX: "ttt", Nested: &NestedStr{BBB: "ddd", CCC: 9}}
|
|
err = rutil.ZeroFieldByPath(val2, "Nested")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val2.Nested != nil {
|
|
t.Fatalf("zero field not works: %v", val2)
|
|
}
|
|
}
|
|
|
|
func TestStructFieldsMap(t *testing.T) {
|
|
type NestedStr struct {
|
|
BBB string
|
|
CCC int
|
|
}
|
|
type Str struct {
|
|
Name []string `json:"name" codec:"flatten"`
|
|
XXX string `json:"xxx"`
|
|
Nested NestedStr
|
|
}
|
|
|
|
val := &Str{Name: []string{"first", "second"}, XXX: "ttt", Nested: NestedStr{BBB: "ddd", CCC: 9}}
|
|
fields, err := rutil.StructFieldsMap(val)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if v, ok := fields["Nested.BBB"]; !ok || v != "ddd" {
|
|
t.Fatalf("invalid field from %v", fields)
|
|
}
|
|
}
|
|
|
|
func TestStructFields(t *testing.T) {
|
|
type NestedStr struct {
|
|
BBB string
|
|
CCC int
|
|
}
|
|
type Str struct {
|
|
Name []string `json:"name" codec:"flatten"`
|
|
XXX string `json:"xxx"`
|
|
Nested NestedStr
|
|
}
|
|
|
|
val := &Str{Name: []string{"first", "second"}, XXX: "ttt", Nested: NestedStr{BBB: "ddd", CCC: 9}}
|
|
fields, err := rutil.StructFields(val)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var ok bool
|
|
for _, field := range fields {
|
|
if field.Path == "Nested.CCC" {
|
|
ok = true
|
|
}
|
|
}
|
|
if !ok {
|
|
t.Fatalf("struct fields returns invalid path: %v", fields)
|
|
}
|
|
}
|
|
|
|
func TestStructByPath(t *testing.T) {
|
|
type NestedStr struct {
|
|
BBB string
|
|
CCC int
|
|
}
|
|
type Str struct {
|
|
Name []string `json:"name" codec:"flatten"`
|
|
XXX string `json:"xxx"`
|
|
Nested NestedStr
|
|
}
|
|
|
|
val := &Str{Name: []string{"first", "second"}, XXX: "ttt", Nested: NestedStr{BBB: "ddd", CCC: 9}}
|
|
field, err := rutil.StructFieldByPath(val, "Nested.CCC")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if reflect.Indirect(reflect.ValueOf(field)).Int() != 9 {
|
|
t.Fatalf("invalid elem returned: %v", field)
|
|
}
|
|
}
|
|
|
|
func TestStructByTag(t *testing.T) {
|
|
type Str struct {
|
|
Name []string `json:"name" codec:"flatten"`
|
|
}
|
|
|
|
val := &Str{Name: []string{"first", "second"}}
|
|
|
|
iface, err := rutil.StructFieldByTag(val, "codec", "flatten")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if v, ok := iface.(*[]string); !ok {
|
|
t.Fatalf("not *[]string %v", iface)
|
|
} else if len(*v) != 2 {
|
|
t.Fatalf("invalid number %v", iface)
|
|
}
|
|
}
|
|
|
|
func TestStructByName(t *testing.T) {
|
|
type Str struct {
|
|
Name []string `json:"name" codec:"flatten"`
|
|
}
|
|
|
|
val := &Str{Name: []string{"first", "second"}}
|
|
|
|
iface, err := rutil.StructFieldByName(val, "Name")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if v, ok := iface.([]string); !ok {
|
|
t.Fatalf("not []string %v", iface)
|
|
} else if len(v) != 2 {
|
|
t.Fatalf("invalid number %v", iface)
|
|
}
|
|
}
|
|
|
|
func TestStructURLValues(t *testing.T) {
|
|
type Str struct {
|
|
Str *Str `json:"str"`
|
|
Name string `json:"name"`
|
|
Args []int `json:"args"`
|
|
}
|
|
|
|
val := &Str{Name: "test_name", Args: []int{1, 2, 3}, Str: &Str{Name: "nested_name"}}
|
|
data, err := rutil.StructURLValues(val, "", []string{"json"})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if data.Get("name") != "test_name" {
|
|
t.Fatalf("invalid data: %v", data)
|
|
}
|
|
}
|
|
|
|
func TestURLSliceVars(t *testing.T) {
|
|
u, err := url.Parse("http://localhost/v1/test/call/my_name?key=arg1&key=arg2&key=arg3")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
mp, err := rutil.URLMap(u.RawQuery)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
v, ok := mp["key"]
|
|
if !ok {
|
|
t.Fatalf("key not exists: %#+v", mp)
|
|
}
|
|
|
|
vm, ok := v.([]interface{})
|
|
if !ok {
|
|
t.Fatalf("invalid key value")
|
|
}
|
|
|
|
if len(vm) != 3 {
|
|
t.Fatalf("missing key value: %#+v", mp)
|
|
}
|
|
}
|
|
|
|
func TestURLVars(t *testing.T) {
|
|
u, err := url.Parse("http://localhost/v1/test/call/my_name?req=key&arg1=arg1&arg2=12345&nested.string_args=str1&nested.string_args=str2&arg2=54321")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
mp, err := rutil.URLMap(u.RawQuery)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_ = mp
|
|
}
|
|
|
|
func TestIsZero(t *testing.T) {
|
|
testStr1 := struct {
|
|
Name string
|
|
Value string
|
|
Nested struct {
|
|
NestedName string
|
|
}
|
|
}{
|
|
Name: "test_name",
|
|
Value: "test_value",
|
|
}
|
|
testStr1.Nested.NestedName = "nested_name"
|
|
|
|
if ok := rutil.IsZero(testStr1); ok {
|
|
t.Fatalf("zero ret on non zero struct: %#+v", testStr1)
|
|
}
|
|
|
|
testStr1.Name = ""
|
|
testStr1.Value = ""
|
|
testStr1.Nested.NestedName = ""
|
|
if ok := rutil.IsZero(testStr1); !ok {
|
|
t.Fatalf("non zero ret on zero struct: %#+v", testStr1)
|
|
}
|
|
|
|
type testStr3 struct {
|
|
Nested string
|
|
}
|
|
type testStr2 struct {
|
|
Nested *testStr3
|
|
Name string
|
|
}
|
|
vtest := &testStr2{
|
|
Name: "test_name",
|
|
Nested: &testStr3{Nested: "nested_name"},
|
|
}
|
|
if ok := rutil.IsZero(vtest); ok {
|
|
t.Fatalf("zero ret on non zero struct: %#+v", vtest)
|
|
}
|
|
vtest.Nested = nil
|
|
vtest.Name = ""
|
|
if ok := rutil.IsZero(vtest); !ok {
|
|
t.Fatalf("non zero ret on zero struct: %#+v", vtest)
|
|
}
|
|
|
|
// t.Logf("XX %#+v\n", ok)
|
|
}
|