Lunny Xiao
14a0c19a0c
Merge branch 'master' into lunny/fix_buffer_iterate Exclude schema from index name (#1505) Merge branch 'master' into fix-schema-idx SetExpr support more go types (#1499) Improve tests SetExpr support more go types fix vet fix drone lint remove go1.10 test on drone Reviewed-on: xorm/xorm#1499 fix vet fix drone lint remove go1.10 test on drone Exclude schema from the index name Co-authored-by: Guillermo Prandi <guillep2k@users.noreply.github.com> Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com> Reviewed-on: xorm/xorm#1505 fix test fix bug fix bug when buffersize with iterate SetExpr support more go types (#1499) Improve tests SetExpr support more go types fix vet fix drone lint remove go1.10 test on drone Reviewed-on: xorm/xorm#1499 fix vet fix drone lint remove go1.10 test on drone Fix update with Alias (#1455) Co-authored-by: Guillermo Prandi <guillep2k@noreply.gitea.io> Reviewed-on: xorm/xorm#941
104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
// Copyright 2017 The Xorm Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package xorm
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestIterate(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type UserIterate struct {
|
|
Id int64
|
|
IsMan bool
|
|
}
|
|
|
|
assert.NoError(t, testEngine.Sync2(new(UserIterate)))
|
|
|
|
cnt, err := testEngine.Insert(&UserIterate{
|
|
IsMan: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
|
|
cnt = 0
|
|
err = testEngine.Iterate(new(UserIterate), func(i int, bean interface{}) error {
|
|
user := bean.(*UserIterate)
|
|
assert.EqualValues(t, 1, user.Id)
|
|
assert.EqualValues(t, true, user.IsMan)
|
|
cnt++
|
|
return nil
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
}
|
|
|
|
func TestBufferIterate(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type UserBufferIterate struct {
|
|
Id int64
|
|
IsMan bool
|
|
}
|
|
|
|
assert.NoError(t, testEngine.Sync2(new(UserBufferIterate)))
|
|
|
|
var size = 20
|
|
for i := 0; i < size; i++ {
|
|
cnt, err := testEngine.Insert(&UserBufferIterate{
|
|
IsMan: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
}
|
|
|
|
var cnt = 0
|
|
err := testEngine.BufferSize(9).Iterate(new(UserBufferIterate), func(i int, bean interface{}) error {
|
|
user := bean.(*UserBufferIterate)
|
|
assert.EqualValues(t, cnt+1, user.Id)
|
|
assert.EqualValues(t, true, user.IsMan)
|
|
cnt++
|
|
return nil
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, size, cnt)
|
|
|
|
cnt = 0
|
|
err = testEngine.Limit(20).BufferSize(9).Iterate(new(UserBufferIterate), func(i int, bean interface{}) error {
|
|
user := bean.(*UserBufferIterate)
|
|
assert.EqualValues(t, cnt+1, user.Id)
|
|
assert.EqualValues(t, true, user.IsMan)
|
|
cnt++
|
|
return nil
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, size, cnt)
|
|
|
|
cnt = 0
|
|
err = testEngine.Limit(7).BufferSize(9).Iterate(new(UserBufferIterate), func(i int, bean interface{}) error {
|
|
user := bean.(*UserBufferIterate)
|
|
assert.EqualValues(t, cnt+1, user.Id)
|
|
assert.EqualValues(t, true, user.IsMan)
|
|
cnt++
|
|
return nil
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 7, cnt)
|
|
|
|
cnt = 0
|
|
err = testEngine.Where("id <= 10").BufferSize(2).Iterate(new(UserBufferIterate), func(i int, bean interface{}) error {
|
|
user := bean.(*UserBufferIterate)
|
|
assert.EqualValues(t, cnt+1, user.Id)
|
|
assert.EqualValues(t, true, user.IsMan)
|
|
cnt++
|
|
return nil
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 10, cnt)
|
|
}
|