mirror of
https://gitee.com/johng/gf
synced 2026-06-06 16:21:40 +08:00
f565a347c4
Change the return type of the GetArray method to Array for consistency
in data structure. Introduce a new Bools method to convert Vars to a
slice of bools, along with corresponding test cases to validate the
functionality.
```go
// 改进前
res, _ := db.Model(table).Fields("id").Array()
ids := make([]int64, 0, len(res))
for _, v := range res {
ids = append(ids, v.Int64())
}
g.Dump(ids)
// 改进后
res, _ := db.Model(table).Fields("id").Array()
ids := res.Int64s()
g.Dump(ids)
```
102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package gvar_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/gogf/gf/v2/container/gvar"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/test/gtest"
|
|
)
|
|
|
|
func TestVars(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
var vs = gvar.Vars{
|
|
gvar.New(1),
|
|
gvar.New(2),
|
|
gvar.New(3),
|
|
}
|
|
t.AssertEQ(vs.Strings(), []string{"1", "2", "3"})
|
|
t.AssertEQ(vs.Bools(), []bool{true, true, true})
|
|
t.AssertEQ(vs.Interfaces(), []any{1, 2, 3})
|
|
t.AssertEQ(vs.Float32s(), []float32{1, 2, 3})
|
|
t.AssertEQ(vs.Float64s(), []float64{1, 2, 3})
|
|
t.AssertEQ(vs.Ints(), []int{1, 2, 3})
|
|
t.AssertEQ(vs.Int8s(), []int8{1, 2, 3})
|
|
t.AssertEQ(vs.Int16s(), []int16{1, 2, 3})
|
|
t.AssertEQ(vs.Int32s(), []int32{1, 2, 3})
|
|
t.AssertEQ(vs.Int64s(), []int64{1, 2, 3})
|
|
t.AssertEQ(vs.Uints(), []uint{1, 2, 3})
|
|
t.AssertEQ(vs.Uint8s(), []uint8{1, 2, 3})
|
|
t.AssertEQ(vs.Uint16s(), []uint16{1, 2, 3})
|
|
t.AssertEQ(vs.Uint32s(), []uint32{1, 2, 3})
|
|
t.AssertEQ(vs.Uint64s(), []uint64{1, 2, 3})
|
|
})
|
|
}
|
|
|
|
func TestVars_Bools(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
// Test with various boolean-like values
|
|
var vs = gvar.Vars{
|
|
gvar.New(true),
|
|
gvar.New(false),
|
|
gvar.New(1),
|
|
gvar.New(0),
|
|
gvar.New("true"),
|
|
gvar.New("false"),
|
|
gvar.New("1"),
|
|
gvar.New("0"),
|
|
}
|
|
expected := []bool{true, false, true, false, true, false, true, false}
|
|
t.AssertEQ(vs.Bools(), expected)
|
|
})
|
|
}
|
|
|
|
func TestVars_Empty(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
// Test with empty Vars
|
|
var vs = gvar.Vars{}
|
|
t.AssertEQ(vs.Strings(), []string{})
|
|
t.AssertEQ(vs.Bools(), []bool{})
|
|
t.AssertEQ(vs.Interfaces(), []any{})
|
|
t.AssertEQ(vs.Ints(), []int{})
|
|
t.AssertEQ(vs.Float64s(), []float64{})
|
|
})
|
|
}
|
|
|
|
func TestVars_SingleElement(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
// Test with single element
|
|
var vs = gvar.Vars{gvar.New(42)}
|
|
t.AssertEQ(vs.Strings(), []string{"42"})
|
|
t.AssertEQ(vs.Bools(), []bool{true})
|
|
t.AssertEQ(vs.Ints(), []int{42})
|
|
})
|
|
}
|
|
|
|
func TestVars_Scan(t *testing.T) {
|
|
gtest.C(t, func(t *gtest.T) {
|
|
type User struct {
|
|
Id int
|
|
Name string
|
|
}
|
|
var vs = gvar.Vars{
|
|
gvar.New(g.Map{"id": 1, "name": "john"}),
|
|
gvar.New(g.Map{"id": 2, "name": "smith"}),
|
|
}
|
|
var users []User
|
|
err := vs.Scan(&users)
|
|
t.AssertNil(err)
|
|
t.Assert(len(users), 2)
|
|
t.Assert(users[0].Id, 1)
|
|
t.Assert(users[0].Name, "john")
|
|
t.Assert(users[1].Id, 2)
|
|
t.Assert(users[1].Name, "smith")
|
|
})
|
|
}
|