60 lines
1.5 KiB
Go
60 lines
1.5 KiB
Go
|
package ydb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strconv"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestExecAndQuery(t *testing.T) {
|
||
|
|
type UserinfoQuery struct {
|
||
|
|
Uid int64 `xorm:"pk"`
|
||
|
|
Name string
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.NoError(t, PrepareScheme(&UserinfoQuery{}))
|
||
|
|
engine, err := enginePool.GetDataQueryEngine()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
_, err = engine.
|
||
|
|
Exec("INSERT INTO "+engine.TableName("`userinfo_query`", true)+" (`uid`, `name`) VALUES (?, ?)", int64(1), "user")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
results, err := engine.
|
||
|
|
Query("select * from " + engine.Quote(engine.TableName("userinfo_query", true)))
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.EqualValues(t, 1, len(results))
|
||
|
|
|
||
|
|
id, err := strconv.Atoi(string(results[0]["uid"]))
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.EqualValues(t, 1, id)
|
||
|
|
assert.Equal(t, "user", string(results[0]["name"]))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestExecTime(t *testing.T) {
|
||
|
|
type UserinfoExecTime struct {
|
||
|
|
Uid int64 `xorm:"pk"`
|
||
|
|
Name string
|
||
|
|
Created time.Time
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.NoError(t, PrepareScheme(&UserinfoExecTime{}))
|
||
|
|
engine, err := enginePool.GetDataQueryEngine()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
now := time.Now()
|
||
|
|
_, err = engine.
|
||
|
|
Exec("INSERT INTO "+engine.TableName("`userinfo_exec_time`", true)+" (`uid`, `name`, `created`) VALUES (?, ?, ?)", int64(1), "user", now)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
|
||
|
|
var uet UserinfoExecTime
|
||
|
|
has, err := engine.Where("`uid`=?", int64(1)).Get(&uet)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.True(t, has)
|
||
|
|
assert.EqualValues(t,
|
||
|
|
now.In(engine.GetTZLocation()).Format(time.RFC3339),
|
||
|
|
uet.Created.In(engine.TZLocation).Format(time.RFC3339))
|
||
|
|
}
|