Lunny Xiao
8f2596bf64
All checks were successful
continuous-integration/drone/push Build is passing
Fix #2134 and replace #2135 Reviewed-on: #2136
281 lines
5.8 KiB
Go
281 lines
5.8 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 integrations
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestRows(t *testing.T) {
|
|
assert.NoError(t, PrepareEngine())
|
|
|
|
type UserRows struct {
|
|
Id int64
|
|
IsMan bool
|
|
}
|
|
|
|
assert.NoError(t, testEngine.Sync(new(UserRows)))
|
|
|
|
cnt, err := testEngine.Insert(&UserRows{
|
|
IsMan: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
|
|
rows, err := testEngine.Rows(new(UserRows))
|
|
assert.NoError(t, err)
|
|
defer rows.Close()
|
|
|
|
cnt = 0
|
|
user := new(UserRows)
|
|
for rows.Next() {
|
|
err = rows.Scan(user)
|
|
assert.NoError(t, err)
|
|
cnt++
|
|
}
|
|
assert.EqualValues(t, 1, cnt)
|
|
assert.False(t, rows.Next())
|
|
assert.NoError(t, rows.Close())
|
|
|
|
rows0, err := testEngine.Where("1>1").Rows(new(UserRows))
|
|
assert.NoError(t, err)
|
|
defer rows0.Close()
|
|
|
|
cnt = 0
|
|
user0 := new(UserRows)
|
|
for rows0.Next() {
|
|
err = rows0.Scan(user0)
|
|
assert.NoError(t, err)
|
|
cnt++
|
|
}
|
|
assert.EqualValues(t, 0, cnt)
|
|
assert.NoError(t, rows0.Close())
|
|
|
|
sess := testEngine.NewSession()
|
|
defer sess.Close()
|
|
|
|
rows1, err := sess.Prepare().Rows(new(UserRows))
|
|
assert.NoError(t, err)
|
|
defer rows1.Close()
|
|
|
|
cnt = 0
|
|
for rows1.Next() {
|
|
err = rows1.Scan(user)
|
|
assert.NoError(t, err)
|
|
cnt++
|
|
}
|
|
assert.EqualValues(t, 1, cnt)
|
|
|
|
tbName := testEngine.Quote(testEngine.TableName(user, true))
|
|
rows2, err := testEngine.SQL("SELECT * FROM " + tbName).Rows(new(UserRows))
|
|
assert.NoError(t, err)
|
|
defer rows2.Close()
|
|
|
|
cnt = 0
|
|
for rows2.Next() {
|
|
err = rows2.Scan(user)
|
|
assert.NoError(t, err)
|
|
cnt++
|
|
}
|
|
assert.EqualValues(t, 1, cnt)
|
|
}
|
|
|
|
func TestRowsMyTableName(t *testing.T) {
|
|
assert.NoError(t, PrepareEngine())
|
|
|
|
type UserRowsMyTable struct {
|
|
Id int64
|
|
IsMan bool
|
|
}
|
|
|
|
tableName := "user_rows_my_table_name"
|
|
|
|
assert.NoError(t, testEngine.Table(tableName).Sync(new(UserRowsMyTable)))
|
|
|
|
cnt, err := testEngine.Table(tableName).Insert(&UserRowsMyTable{
|
|
IsMan: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
|
|
rows, err := testEngine.Table(tableName).Rows(new(UserRowsMyTable))
|
|
assert.NoError(t, err)
|
|
|
|
cnt = 0
|
|
user := new(UserRowsMyTable)
|
|
for rows.Next() {
|
|
err = rows.Scan(user)
|
|
assert.NoError(t, err)
|
|
cnt++
|
|
}
|
|
assert.EqualValues(t, 1, cnt)
|
|
|
|
rows.Close()
|
|
|
|
rows, err = testEngine.Table(tableName).Rows(&UserRowsMyTable{
|
|
Id: 2,
|
|
})
|
|
assert.NoError(t, err)
|
|
cnt = 0
|
|
user = new(UserRowsMyTable)
|
|
for rows.Next() {
|
|
err = rows.Scan(user)
|
|
assert.NoError(t, err)
|
|
cnt++
|
|
}
|
|
assert.EqualValues(t, 0, cnt)
|
|
}
|
|
|
|
type UserRowsSpecTable struct {
|
|
Id int64
|
|
IsMan bool
|
|
}
|
|
|
|
func (UserRowsSpecTable) TableName() string {
|
|
return "user_rows_my_table_name"
|
|
}
|
|
|
|
func TestRowsSpecTableName(t *testing.T) {
|
|
assert.NoError(t, PrepareEngine())
|
|
assert.NoError(t, testEngine.Sync(new(UserRowsSpecTable)))
|
|
|
|
cnt, err := testEngine.Insert(&UserRowsSpecTable{
|
|
IsMan: true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
|
|
rows, err := testEngine.Rows(new(UserRowsSpecTable))
|
|
assert.NoError(t, err)
|
|
defer rows.Close()
|
|
|
|
cnt = 0
|
|
user := new(UserRowsSpecTable)
|
|
for rows.Next() {
|
|
err = rows.Scan(user)
|
|
assert.NoError(t, err)
|
|
cnt++
|
|
}
|
|
assert.NoError(t, rows.Err())
|
|
assert.EqualValues(t, 1, cnt)
|
|
}
|
|
|
|
func TestRowsScanVars(t *testing.T) {
|
|
type RowsScanVars struct {
|
|
Id int64
|
|
Name string
|
|
Age int
|
|
}
|
|
|
|
assert.NoError(t, PrepareEngine())
|
|
assert.NoError(t, testEngine.Sync2(new(RowsScanVars)))
|
|
|
|
cnt, err := testEngine.Insert(&RowsScanVars{
|
|
Name: "xlw",
|
|
Age: 42,
|
|
}, &RowsScanVars{
|
|
Name: "xlw2",
|
|
Age: 24,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 2, cnt)
|
|
|
|
rows, err := testEngine.Cols("name", "age").Rows(new(RowsScanVars))
|
|
assert.NoError(t, err)
|
|
defer rows.Close()
|
|
|
|
cnt = 0
|
|
for rows.Next() {
|
|
var name string
|
|
var age int
|
|
err = rows.Scan(&name, &age)
|
|
assert.NoError(t, err)
|
|
if cnt == 0 {
|
|
assert.EqualValues(t, "xlw", name)
|
|
assert.EqualValues(t, 42, age)
|
|
} else if cnt == 1 {
|
|
assert.EqualValues(t, "xlw2", name)
|
|
assert.EqualValues(t, 24, age)
|
|
}
|
|
cnt++
|
|
}
|
|
assert.NoError(t, rows.Err())
|
|
assert.EqualValues(t, 2, cnt)
|
|
}
|
|
|
|
func TestRowsScanBytes(t *testing.T) {
|
|
type RowsScanBytes struct {
|
|
Id int64
|
|
Bytes1 []byte
|
|
Bytes2 []byte
|
|
}
|
|
|
|
assert.NoError(t, PrepareEngine())
|
|
assert.NoError(t, testEngine.Sync(new(RowsScanBytes)))
|
|
|
|
cnt, err := testEngine.Insert(&RowsScanBytes{
|
|
Bytes1: []byte("bytes1"),
|
|
Bytes2: []byte("bytes2"),
|
|
}, &RowsScanBytes{
|
|
Bytes1: []byte("bytes1-1"),
|
|
Bytes2: []byte("bytes2-2"),
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 2, cnt)
|
|
|
|
{
|
|
rows, err := testEngine.Cols("bytes1, bytes2").Rows(new(RowsScanBytes))
|
|
assert.NoError(t, err)
|
|
defer rows.Close()
|
|
|
|
cnt = 0
|
|
var bytes1 []byte
|
|
var bytes2 []byte
|
|
for rows.Next() {
|
|
err = rows.Scan(&bytes1, &bytes2)
|
|
assert.NoError(t, err)
|
|
if cnt == 0 {
|
|
assert.EqualValues(t, []byte("bytes1"), bytes1)
|
|
assert.EqualValues(t, []byte("bytes2"), bytes2)
|
|
} else if cnt == 1 {
|
|
// bytes1 now should be `bytes1` but will be override
|
|
assert.EqualValues(t, []byte("bytes1-1"), bytes1)
|
|
assert.EqualValues(t, []byte("bytes2-2"), bytes2)
|
|
}
|
|
cnt++
|
|
}
|
|
assert.NoError(t, rows.Err())
|
|
assert.EqualValues(t, 2, cnt)
|
|
rows.Close()
|
|
}
|
|
|
|
{
|
|
rows, err := testEngine.Cols("bytes1, bytes2").Rows(new(RowsScanBytes))
|
|
assert.NoError(t, err)
|
|
defer rows.Close()
|
|
|
|
cnt = 0
|
|
var rsb RowsScanBytes
|
|
for rows.Next() {
|
|
err = rows.Scan(&rsb)
|
|
assert.NoError(t, err)
|
|
if cnt == 0 {
|
|
assert.EqualValues(t, []byte("bytes1"), rsb.Bytes1)
|
|
assert.EqualValues(t, []byte("bytes2"), rsb.Bytes2)
|
|
} else if cnt == 1 {
|
|
// bytes1 now should be `bytes1` but will be override
|
|
assert.EqualValues(t, []byte("bytes1-1"), rsb.Bytes1)
|
|
assert.EqualValues(t, []byte("bytes2-2"), rsb.Bytes2)
|
|
}
|
|
cnt++
|
|
}
|
|
assert.NoError(t, rows.Err())
|
|
assert.EqualValues(t, 2, cnt)
|
|
rows.Close()
|
|
}
|
|
}
|