647 lines
16 KiB
Go
647 lines
16 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 (
|
|
"database/sql"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"xorm.io/xorm/contexts"
|
|
"xorm.io/xorm/schemas"
|
|
)
|
|
|
|
func TestGetVar(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type GetVar struct {
|
|
Id int64 `xorm:"autoincr pk"`
|
|
Msg string `xorm:"varchar(255)"`
|
|
Age int
|
|
Money float32
|
|
Created time.Time `xorm:"created"`
|
|
}
|
|
|
|
assert.NoError(t, testEngine.Sync2(new(GetVar)))
|
|
|
|
var data = GetVar{
|
|
Msg: "hi",
|
|
Age: 28,
|
|
Money: 1.5,
|
|
}
|
|
_, err := testEngine.InsertOne(&data)
|
|
assert.NoError(t, err)
|
|
|
|
var msg string
|
|
has, err := testEngine.Table("get_var").Cols("msg").Get(&msg)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, "hi", msg)
|
|
|
|
var age int
|
|
has, err = testEngine.Table("get_var").Cols("age").Get(&age)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, 28, age)
|
|
|
|
var ageMax int
|
|
has, err = testEngine.SQL("SELECT max(age) FROM "+testEngine.TableName("get_var", true)+" WHERE `id` = ?", data.Id).Get(&ageMax)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, 28, ageMax)
|
|
|
|
var age2 int64
|
|
has, err = testEngine.Table("get_var").Cols("age").
|
|
Where("age > ?", 20).
|
|
And("age < ?", 30).
|
|
Get(&age2)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.EqualValues(t, 28, age2)
|
|
|
|
var age3 int8
|
|
has, err = testEngine.Table("get_var").Cols("age").Get(&age3)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.EqualValues(t, 28, age3)
|
|
|
|
var age4 int16
|
|
has, err = testEngine.Table("get_var").Cols("age").
|
|
Where("age > ?", 20).
|
|
And("age < ?", 30).
|
|
Get(&age4)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.EqualValues(t, 28, age4)
|
|
|
|
var age5 int32
|
|
has, err = testEngine.Table("get_var").Cols("age").
|
|
Where("age > ?", 20).
|
|
And("age < ?", 30).
|
|
Get(&age5)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.EqualValues(t, 28, age5)
|
|
|
|
var age6 int
|
|
has, err = testEngine.Table("get_var").Cols("age").Get(&age6)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.EqualValues(t, 28, age6)
|
|
|
|
var age7 int64
|
|
has, err = testEngine.Table("get_var").Cols("age").
|
|
Where("age > ?", 20).
|
|
And("age < ?", 30).
|
|
Get(&age7)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.EqualValues(t, 28, age7)
|
|
|
|
var age8 int8
|
|
has, err = testEngine.Table("get_var").Cols("age").Get(&age8)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.EqualValues(t, 28, age8)
|
|
|
|
var age9 int16
|
|
has, err = testEngine.Table("get_var").Cols("age").
|
|
Where("age > ?", 20).
|
|
And("age < ?", 30).
|
|
Get(&age9)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.EqualValues(t, 28, age9)
|
|
|
|
var age10 int32
|
|
has, err = testEngine.Table("get_var").Cols("age").
|
|
Where("age > ?", 20).
|
|
And("age < ?", 30).
|
|
Get(&age10)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.EqualValues(t, 28, age10)
|
|
|
|
var id sql.NullInt64
|
|
has, err = testEngine.Table("get_var").Cols("id").Get(&id)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, true, id.Valid)
|
|
assert.EqualValues(t, data.Id, id.Int64)
|
|
|
|
var msgNull sql.NullString
|
|
has, err = testEngine.Table("get_var").Cols("msg").Get(&msgNull)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, true, msgNull.Valid)
|
|
assert.EqualValues(t, data.Msg, msgNull.String)
|
|
|
|
var nullMoney sql.NullFloat64
|
|
has, err = testEngine.Table("get_var").Cols("money").Get(&nullMoney)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, true, nullMoney.Valid)
|
|
assert.EqualValues(t, data.Money, nullMoney.Float64)
|
|
|
|
var money float64
|
|
has, err = testEngine.Table("get_var").Cols("money").Get(&money)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, "1.5", fmt.Sprintf("%.1f", money))
|
|
|
|
var money2 float64
|
|
if testEngine.Dialect().URI().DBType == schemas.MSSQL {
|
|
has, err = testEngine.SQL("SELECT TOP 1 money FROM " + testEngine.TableName("get_var", true)).Get(&money2)
|
|
} else {
|
|
has, err = testEngine.SQL("SELECT money FROM " + testEngine.TableName("get_var", true) + " LIMIT 1").Get(&money2)
|
|
}
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, "1.5", fmt.Sprintf("%.1f", money2))
|
|
|
|
var money3 float64
|
|
has, err = testEngine.SQL("SELECT money FROM " + testEngine.TableName("get_var", true) + " WHERE money > 20").Get(&money3)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, false, has)
|
|
|
|
var valuesString = make(map[string]string)
|
|
has, err = testEngine.Table("get_var").Get(&valuesString)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, 5, len(valuesString))
|
|
assert.Equal(t, "1", valuesString["id"])
|
|
assert.Equal(t, "hi", valuesString["msg"])
|
|
assert.Equal(t, "28", valuesString["age"])
|
|
assert.Equal(t, "1.5", valuesString["money"])
|
|
|
|
// for mymysql driver, interface{} will be []byte, so ignore it currently
|
|
if testEngine.DriverName() != "mymysql" {
|
|
var valuesInter = make(map[string]interface{})
|
|
has, err = testEngine.Table("get_var").Where("id = ?", 1).Select("*").Get(&valuesInter)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, 5, len(valuesInter))
|
|
assert.EqualValues(t, 1, valuesInter["id"])
|
|
assert.Equal(t, "hi", fmt.Sprintf("%s", valuesInter["msg"]))
|
|
assert.EqualValues(t, 28, valuesInter["age"])
|
|
assert.Equal(t, "1.5", fmt.Sprintf("%v", valuesInter["money"]))
|
|
}
|
|
|
|
var valuesSliceString = make([]string, 5)
|
|
has, err = testEngine.Table("get_var").Get(&valuesSliceString)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
assert.Equal(t, "1", valuesSliceString[0])
|
|
assert.Equal(t, "hi", valuesSliceString[1])
|
|
assert.Equal(t, "28", valuesSliceString[2])
|
|
assert.Equal(t, "1.5", valuesSliceString[3])
|
|
|
|
var valuesSliceInter = make([]interface{}, 5)
|
|
has, err = testEngine.Table("get_var").Get(&valuesSliceInter)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, true, has)
|
|
|
|
v1, err := convertInt(valuesSliceInter[0])
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, v1)
|
|
|
|
assert.Equal(t, "hi", fmt.Sprintf("%s", valuesSliceInter[1]))
|
|
|
|
v3, err := convertInt(valuesSliceInter[2])
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 28, v3)
|
|
|
|
v4, err := convertFloat(valuesSliceInter[3])
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "1.5", fmt.Sprintf("%v", v4))
|
|
}
|
|
|
|
func TestGetStruct(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type UserinfoGet struct {
|
|
Uid int `xorm:"pk autoincr"`
|
|
IsMan bool
|
|
}
|
|
|
|
assert.NoError(t, testEngine.Sync2(new(UserinfoGet)))
|
|
|
|
session := testEngine.NewSession()
|
|
defer session.Close()
|
|
|
|
var err error
|
|
if testEngine.Dialect().URI().DBType == schemas.MSSQL {
|
|
err = session.Begin()
|
|
assert.NoError(t, err)
|
|
_, err = session.Exec("SET IDENTITY_INSERT userinfo_get ON")
|
|
assert.NoError(t, err)
|
|
}
|
|
cnt, err := session.Insert(&UserinfoGet{Uid: 2})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
if testEngine.Dialect().URI().DBType == schemas.MSSQL {
|
|
err = session.Commit()
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
user := UserinfoGet{Uid: 2}
|
|
has, err := testEngine.Get(&user)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
|
|
type NoIdUser struct {
|
|
User string `xorm:"unique"`
|
|
Remain int64
|
|
Total int64
|
|
}
|
|
|
|
assert.NoError(t, testEngine.Sync2(&NoIdUser{}))
|
|
|
|
userCol := testEngine.GetColumnMapper().Obj2Table("User")
|
|
_, err = testEngine.Where("`"+userCol+"` = ?", "xlw").Delete(&NoIdUser{})
|
|
assert.NoError(t, err)
|
|
|
|
cnt, err = testEngine.Insert(&NoIdUser{"xlw", 20, 100})
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, cnt)
|
|
|
|
noIdUser := new(NoIdUser)
|
|
has, err = testEngine.Where("`"+userCol+"` = ?", "xlw").Get(noIdUser)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
}
|
|
|
|
func TestGetSlice(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type UserinfoSlice struct {
|
|
Uid int `xorm:"pk autoincr"`
|
|
IsMan bool
|
|
}
|
|
|
|
assertSync(t, new(UserinfoSlice))
|
|
|
|
var users []UserinfoSlice
|
|
has, err := testEngine.Get(&users)
|
|
assert.False(t, has)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestGetError(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type GetError struct {
|
|
Uid int `xorm:"pk autoincr"`
|
|
IsMan bool
|
|
}
|
|
|
|
assertSync(t, new(GetError))
|
|
|
|
var info = new(GetError)
|
|
has, err := testEngine.Get(&info)
|
|
assert.False(t, has)
|
|
assert.Error(t, err)
|
|
|
|
has, err = testEngine.Get(info)
|
|
assert.False(t, has)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestJSONString(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type JsonString struct {
|
|
Id int64
|
|
Content string `xorm:"json"`
|
|
}
|
|
type JsonJson struct {
|
|
Id int64
|
|
Content []string `xorm:"json"`
|
|
}
|
|
|
|
assertSync(t, new(JsonJson))
|
|
|
|
_, err := testEngine.Insert(&JsonJson{
|
|
Content: []string{"1", "2"},
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
var js JsonString
|
|
has, err := testEngine.Table("json_json").Get(&js)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 1, js.Id)
|
|
assert.True(t, `["1","2"]` == js.Content || `["1", "2"]` == js.Content)
|
|
|
|
var jss []JsonString
|
|
err = testEngine.Table("json_json").Find(&jss)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, len(jss))
|
|
assert.True(t, `["1","2"]` == jss[0].Content || `["1", "2"]` == jss[0].Content)
|
|
}
|
|
|
|
func TestGetActionMapping(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
|
|
type ActionMapping struct {
|
|
ActionId string `xorm:"pk"`
|
|
ActionName string `xorm:"index"`
|
|
ScriptId string `xorm:"unique"`
|
|
RollbackId string `xorm:"unique"`
|
|
Env string
|
|
Tags string
|
|
Description string
|
|
UpdateTime time.Time `xorm:"updated"`
|
|
DeleteTime time.Time `xorm:"deleted"`
|
|
}
|
|
|
|
assertSync(t, new(ActionMapping))
|
|
|
|
_, err := testEngine.Insert(&ActionMapping{
|
|
ActionId: "1",
|
|
ScriptId: "2",
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
var valuesSlice = make([]string, 2)
|
|
has, err := testEngine.Table(new(ActionMapping)).
|
|
Cols("script_id", "rollback_id").
|
|
ID("1").Get(&valuesSlice)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, "2", valuesSlice[0])
|
|
assert.EqualValues(t, "", valuesSlice[1])
|
|
}
|
|
|
|
func TestGetStructId(t *testing.T) {
|
|
type TestGetStruct struct {
|
|
Id int64
|
|
}
|
|
|
|
assert.NoError(t, prepareEngine())
|
|
assertSync(t, new(TestGetStruct))
|
|
|
|
_, err := testEngine.Insert(&TestGetStruct{})
|
|
assert.NoError(t, err)
|
|
_, err = testEngine.Insert(&TestGetStruct{})
|
|
assert.NoError(t, err)
|
|
|
|
type maxidst struct {
|
|
Id int64
|
|
}
|
|
|
|
//var id int64
|
|
var maxid maxidst
|
|
sql := "select max(id) as id from " + testEngine.TableName(&TestGetStruct{}, true)
|
|
has, err := testEngine.SQL(sql).Get(&maxid)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 2, maxid.Id)
|
|
}
|
|
|
|
func TestContextGet(t *testing.T) {
|
|
type ContextGetStruct struct {
|
|
Id int64
|
|
Name string
|
|
}
|
|
|
|
assert.NoError(t, prepareEngine())
|
|
assertSync(t, new(ContextGetStruct))
|
|
|
|
_, err := testEngine.Insert(&ContextGetStruct{Name: "1"})
|
|
assert.NoError(t, err)
|
|
|
|
sess := testEngine.NewSession()
|
|
defer sess.Close()
|
|
|
|
context := contexts.NewMemoryContextCache()
|
|
|
|
var c2 ContextGetStruct
|
|
has, err := sess.ID(1).NoCache().ContextCache(context).Get(&c2)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 1, c2.Id)
|
|
assert.EqualValues(t, "1", c2.Name)
|
|
sql, args := sess.LastSQL()
|
|
assert.True(t, len(sql) > 0)
|
|
assert.True(t, len(args) > 0)
|
|
|
|
var c3 ContextGetStruct
|
|
has, err = sess.ID(1).NoCache().ContextCache(context).Get(&c3)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 1, c3.Id)
|
|
assert.EqualValues(t, "1", c3.Name)
|
|
sql, args = sess.LastSQL()
|
|
assert.True(t, len(sql) == 0)
|
|
assert.True(t, len(args) == 0)
|
|
}
|
|
|
|
func TestContextGet2(t *testing.T) {
|
|
type ContextGetStruct2 struct {
|
|
Id int64
|
|
Name string
|
|
}
|
|
|
|
assert.NoError(t, prepareEngine())
|
|
assertSync(t, new(ContextGetStruct2))
|
|
|
|
_, err := testEngine.Insert(&ContextGetStruct2{Name: "1"})
|
|
assert.NoError(t, err)
|
|
|
|
context := contexts.NewMemoryContextCache()
|
|
|
|
var c2 ContextGetStruct2
|
|
has, err := testEngine.ID(1).NoCache().ContextCache(context).Get(&c2)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 1, c2.Id)
|
|
assert.EqualValues(t, "1", c2.Name)
|
|
|
|
var c3 ContextGetStruct2
|
|
has, err = testEngine.ID(1).NoCache().ContextCache(context).Get(&c3)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 1, c3.Id)
|
|
assert.EqualValues(t, "1", c3.Name)
|
|
}
|
|
|
|
type GetCustomTableInterface interface {
|
|
TableName() string
|
|
}
|
|
|
|
type MyGetCustomTableImpletation struct {
|
|
Id int64 `json:"id"`
|
|
Name string `json:"name"`
|
|
}
|
|
|
|
const getCustomTableName = "GetCustomTableInterface"
|
|
|
|
func (MyGetCustomTableImpletation) TableName() string {
|
|
return getCustomTableName
|
|
}
|
|
|
|
func TestGetCustomTableInterface(t *testing.T) {
|
|
assert.NoError(t, prepareEngine())
|
|
assert.NoError(t, testEngine.Table(getCustomTableName).Sync2(new(MyGetCustomTableImpletation)))
|
|
|
|
exist, err := testEngine.IsTableExist(getCustomTableName)
|
|
assert.NoError(t, err)
|
|
assert.True(t, exist)
|
|
|
|
_, err = testEngine.Insert(&MyGetCustomTableImpletation{
|
|
Name: "xlw",
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
var c GetCustomTableInterface = new(MyGetCustomTableImpletation)
|
|
has, err := testEngine.Get(c)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
}
|
|
|
|
func TestGetNullVar(t *testing.T) {
|
|
type TestGetNullVarStruct struct {
|
|
Id int64
|
|
Name string
|
|
Age int
|
|
}
|
|
|
|
assert.NoError(t, prepareEngine())
|
|
assertSync(t, new(TestGetNullVarStruct))
|
|
|
|
affected, err := testEngine.Exec("insert into " + testEngine.TableName(new(TestGetNullVarStruct), true) + " (name,age) values (null,null)")
|
|
assert.NoError(t, err)
|
|
a, _ := affected.RowsAffected()
|
|
assert.EqualValues(t, 1, a)
|
|
|
|
var name string
|
|
has, err := testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("name").Get(&name)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, "", name)
|
|
|
|
var age int
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age)
|
|
|
|
var age2 int8
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age2)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age2)
|
|
|
|
var age3 int16
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age3)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age3)
|
|
|
|
var age4 int32
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age4)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age4)
|
|
|
|
var age5 int64
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age5)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age5)
|
|
|
|
var age6 uint
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age6)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age6)
|
|
|
|
var age7 uint8
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age7)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age7)
|
|
|
|
var age8 int16
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age8)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age8)
|
|
|
|
var age9 int32
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age9)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age9)
|
|
|
|
var age10 int64
|
|
has, err = testEngine.Table(new(TestGetNullVarStruct)).Where("id = ?", 1).Cols("age").Get(&age10)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 0, age10)
|
|
}
|
|
|
|
func TestCustomTypes(t *testing.T) {
|
|
type MyInt int
|
|
type MyString string
|
|
|
|
type TestCustomizeStruct struct {
|
|
Id int64
|
|
Name MyString
|
|
Age MyInt
|
|
}
|
|
|
|
assert.NoError(t, prepareEngine())
|
|
assertSync(t, new(TestCustomizeStruct))
|
|
|
|
var s = TestCustomizeStruct{
|
|
Name: "test",
|
|
Age: 32,
|
|
}
|
|
_, err := testEngine.Insert(&s)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, 1, s.Id)
|
|
|
|
var name MyString
|
|
has, err := testEngine.Table(new(TestCustomizeStruct)).ID(s.Id).Cols("name").Get(&name)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, "test", name)
|
|
|
|
var age MyInt
|
|
has, err = testEngine.Table(new(TestCustomizeStruct)).ID(s.Id).Select("age").Get(&age)
|
|
assert.NoError(t, err)
|
|
assert.True(t, has)
|
|
assert.EqualValues(t, 32, age)
|
|
}
|
|
|
|
func TestGetViaMapCond(t *testing.T) {
|
|
type GetViaMapCond struct {
|
|
Id int64
|
|
Platform int
|
|
Index int
|
|
}
|
|
|
|
assert.NoError(t, prepareEngine())
|
|
assertSync(t, new(GetViaMapCond))
|
|
|
|
var (
|
|
r GetViaMapCond
|
|
platformStr = colMapper.Obj2Table("Platform")
|
|
indexStr = colMapper.Obj2Table("Index")
|
|
query = map[string]interface{}{
|
|
platformStr: 1,
|
|
indexStr: 1,
|
|
}
|
|
)
|
|
|
|
has, err := testEngine.Where(query).Get(&r)
|
|
assert.NoError(t, err)
|
|
assert.False(t, has)
|
|
}
|