111 lines
3.0 KiB
Go
111 lines
3.0 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 (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
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 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 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 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"])
|
|
|
|
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))
|
|
}
|