147 lines
2.4 KiB
Go
147 lines
2.4 KiB
Go
package helper
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
type x struct {
|
|
Id uint64
|
|
}
|
|
|
|
func c(x []*x) (r []uint64) {
|
|
for i := 0; i < len(x); i++ {
|
|
r = append(r, x[i].Id)
|
|
}
|
|
return
|
|
}
|
|
|
|
func getX() (r []*x) {
|
|
for i := 0; i < 10; i++ {
|
|
r = append(r, &x{
|
|
uint64(i),
|
|
})
|
|
}
|
|
return
|
|
}
|
|
|
|
func BenchmarkOr(b *testing.B) {
|
|
y := getX()
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
c(y)
|
|
}
|
|
}
|
|
|
|
func BenchmarkStructColumnToSlice(b *testing.B) {
|
|
y := getX()
|
|
fmt.Println(y)
|
|
b.ResetTimer()
|
|
//b.N = 2
|
|
for i := 0; i < 1; i++ {
|
|
StructColumnToSlice[int, *x](y, "Id")
|
|
}
|
|
}
|
|
|
|
func TestStructColumnToSlice(t *testing.T) {
|
|
type args struct {
|
|
arr []x
|
|
field string
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantR []uint64
|
|
}{
|
|
{name: "test1", args: args{
|
|
arr: []x{
|
|
{Id: 1},
|
|
{2},
|
|
{4},
|
|
{6},
|
|
},
|
|
field: "Id",
|
|
}, wantR: []uint64{1, 2, 4, 6}},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if gotR := StructColumnToSlice[uint64, x](tt.args.arr, tt.args.field); !reflect.DeepEqual(gotR, tt.wantR) {
|
|
t.Errorf("StructColumnToSlice() = %v, want %v", gotR, tt.wantR)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestToAny(t *testing.T) {
|
|
type args struct {
|
|
v int
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want any
|
|
}{
|
|
{
|
|
name: "t1",
|
|
args: args{v: 1},
|
|
want: any(1),
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := ToAny(tt.args.v); !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("ToAny() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestCutUrlHost(t *testing.T) {
|
|
type args struct {
|
|
u string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want string
|
|
}{
|
|
{
|
|
name: "http",
|
|
args: args{"http://xx.yy/xxoo?ss=fff"},
|
|
want: "/xxoo?ss=fff",
|
|
}, {
|
|
name: "https",
|
|
args: args{"https://xx.yy/xxoo?ff=fff"},
|
|
want: "/xxoo?ff=fff",
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got := CutUrlHost(tt.args.u); got != tt.want {
|
|
t.Errorf("CutUrlHost() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDefaults(t *testing.T) {
|
|
v := 0
|
|
want := 1
|
|
t.Run("int", func(t *testing.T) {
|
|
if got := Defaults(v, want); !reflect.DeepEqual(got, want) {
|
|
t.Errorf("Defaults() = %v, want %v", got, want)
|
|
}
|
|
})
|
|
{
|
|
v := ""
|
|
want := "a"
|
|
t.Run("string", func(t *testing.T) {
|
|
if got := Defaults(v, want); !reflect.DeepEqual(got, want) {
|
|
t.Errorf("Defaults() = %v, want %v", got, want)
|
|
}
|
|
})
|
|
}
|
|
}
|