func TestTransformArray(t *testing.T) { otherUser := User{ID: 8, FirstName: "Me"} ser := New().UseSnakeCase().Pick("ID", "FirstName") cases := []interface{}{[]User{user, otherUser}, [2]User{user, otherUser}} for _, c := range cases { arr, err := ser.TransformArray(c) assert.Nil(t, err) assert.Len(t, arr, 2) assert.Equal(t, arr[0]["id"], user.ID) assert.Equal(t, arr[1]["id"], otherUser.ID) assert.Equal(t, arr[0]["first_name"], user.FirstName) } _, err := ser.TransformArray(1) assert.NotNil(t, err) }
func TestAddFunc(t *testing.T) { m := New().AddFunc("FullName", func(u interface{}) interface{} { return u.(User).FirstName + " " + u.(User).LastName }).Transform(user) assert.Contains(t, m, "FullName") assert.Equal(t, user.FirstName+" "+user.LastName, m["FullName"]) }
func TestAddIf(t *testing.T) { m := New().AddIf(alwaysFalse, "Foo", "Bar").Transform(user) assert.NotContains(t, m, "Foo") m = New().AddIf(alwaysTrue, "Foo", "Bar").Transform(user) assert.Contains(t, m, "Foo") assert.Equal(t, "Bar", m["Foo"]) }
func TestPickFunc(t *testing.T) { m := New().PickFunc(func(t interface{}) interface{} { return t.(time.Time).Format(time.RFC3339) }, "CreatedAt", "UpdatedAt").Transform(user) assert.Contains(t, m, "CreatedAt") assert.Contains(t, m, "UpdatedAt") assert.Equal(t, m["CreatedAt"], user.CreatedAt.Format(time.RFC3339)) }
func TestAddFuncIf(t *testing.T) { m := New().AddFuncIf(alwaysTrue, "Foo", func(u interface{}) interface{} { return "Bar" }).Transform(user) assert.Contains(t, m, "Foo") assert.Equal(t, m["Foo"], "Bar") m = New().AddFuncIf(alwaysFalse, "Foo", identity).Transform(user) assert.NotContains(t, m, "Foo") }
func TestPickFuncIf(t *testing.T) { m := New().PickFuncIf(alwaysTrue, func(t interface{}) interface{} { return t.(time.Time).Format(time.RFC3339) }, "CreatedAt", "UpdatedAt").PickFuncIf(alwaysFalse, identity, "Email").Transform(user) assert.Contains(t, m, "CreatedAt") assert.Contains(t, m, "UpdatedAt") assert.NotContains(t, m, "Email") assert.Equal(t, m["CreatedAt"], user.CreatedAt.Format(time.RFC3339)) }
func TestPick(t *testing.T) { m := New().Pick("ID", "Age").Transform(user) assert.Contains(t, m, "ID") assert.Contains(t, m, "Age") assert.NotContains(t, m, "FirstName") m = New().Pick("ID").Pick("Age").Transform(user) assert.Contains(t, m, "ID") assert.Contains(t, m, "Age") assert.Equal(t, m["ID"], user.ID) }
func TestAdd(t *testing.T) { m := New().Add("Foo", "Bar").Transform(user) assert.Contains(t, m, "Foo") assert.Equal(t, "Bar", m["Foo"]) }