more tests (#601)
This commit is contained in:
parent
0cc29f16ae
commit
244841c035
|
@ -0,0 +1,192 @@
|
||||||
|
// 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/go-xorm/core"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestTransaction(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
assertSync(t, new(Userinfo))
|
||||||
|
|
||||||
|
counter := func() {
|
||||||
|
total, err := testEngine.Count(&Userinfo{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
fmt.Printf("----now total %v records\n", total)
|
||||||
|
}
|
||||||
|
|
||||||
|
counter()
|
||||||
|
//defer counter()
|
||||||
|
|
||||||
|
session := testEngine.NewSession()
|
||||||
|
defer session.Close()
|
||||||
|
|
||||||
|
err := session.Begin()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
user1 := Userinfo{Username: "xiaoxiao", Departname: "dev", Alias: "lunny", Created: time.Now()}
|
||||||
|
_, err = session.Insert(&user1)
|
||||||
|
if err != nil {
|
||||||
|
session.Rollback()
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
user2 := Userinfo{Username: "yyy"}
|
||||||
|
_, err = session.Where("(id) = ?", 0).Update(&user2)
|
||||||
|
if err != nil {
|
||||||
|
session.Rollback()
|
||||||
|
fmt.Println(err)
|
||||||
|
//t.Error(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = session.Delete(&user2)
|
||||||
|
if err != nil {
|
||||||
|
session.Rollback()
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
err = session.Commit()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// panic(err) !nashtsai! should remove this
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCombineTransaction(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
assertSync(t, new(Userinfo))
|
||||||
|
|
||||||
|
counter := func() {
|
||||||
|
total, err := testEngine.Count(&Userinfo{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
fmt.Printf("----now total %v records\n", total)
|
||||||
|
}
|
||||||
|
|
||||||
|
counter()
|
||||||
|
//defer counter()
|
||||||
|
session := testEngine.NewSession()
|
||||||
|
defer session.Close()
|
||||||
|
|
||||||
|
err := session.Begin()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
user1 := Userinfo{Username: "xiaoxiao2", Departname: "dev", Alias: "lunny", Created: time.Now()}
|
||||||
|
_, err = session.Insert(&user1)
|
||||||
|
if err != nil {
|
||||||
|
session.Rollback()
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
user2 := Userinfo{Username: "zzz"}
|
||||||
|
_, err = session.Where("id = ?", 0).Update(&user2)
|
||||||
|
if err != nil {
|
||||||
|
session.Rollback()
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = session.Exec("delete from userinfo where username = ?", user2.Username)
|
||||||
|
if err != nil {
|
||||||
|
session.Rollback()
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = session.Commit()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCombineTransactionSameMapper(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
|
||||||
|
oldMapper := testEngine.ColumnMapper
|
||||||
|
testEngine.unMapType(rValue(new(Userinfo)).Type())
|
||||||
|
testEngine.SetMapper(core.SameMapper{})
|
||||||
|
defer func() {
|
||||||
|
testEngine.unMapType(rValue(new(Userinfo)).Type())
|
||||||
|
testEngine.SetMapper(oldMapper)
|
||||||
|
}()
|
||||||
|
|
||||||
|
assertSync(t, new(Userinfo))
|
||||||
|
|
||||||
|
counter := func() {
|
||||||
|
total, err := testEngine.Count(&Userinfo{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
fmt.Printf("----now total %v records\n", total)
|
||||||
|
}
|
||||||
|
|
||||||
|
counter()
|
||||||
|
defer counter()
|
||||||
|
session := testEngine.NewSession()
|
||||||
|
defer session.Close()
|
||||||
|
|
||||||
|
err := session.Begin()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
user1 := Userinfo{Username: "xiaoxiao2", Departname: "dev", Alias: "lunny", Created: time.Now()}
|
||||||
|
_, err = session.Insert(&user1)
|
||||||
|
if err != nil {
|
||||||
|
session.Rollback()
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
user2 := Userinfo{Username: "zzz"}
|
||||||
|
_, err = session.Where("(id) = ?", 0).Update(&user2)
|
||||||
|
if err != nil {
|
||||||
|
session.Rollback()
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = session.Exec("delete from `Userinfo` where `Username` = ?", user2.Username)
|
||||||
|
if err != nil {
|
||||||
|
session.Rollback()
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
err = session.Commit()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
|
@ -5,10 +5,13 @@
|
||||||
package xorm
|
package xorm
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
"sync"
|
"sync"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"github.com/go-xorm/core"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -82,11 +85,11 @@ type ForUpdate struct {
|
||||||
|
|
||||||
func setupForUpdate(engine *Engine) error {
|
func setupForUpdate(engine *Engine) error {
|
||||||
v := new(ForUpdate)
|
v := new(ForUpdate)
|
||||||
err := engine.DropTables(v)
|
err := testEngine.DropTables(v)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
err = engine.CreateTables(v)
|
err = testEngine.CreateTables(v)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -98,7 +101,7 @@ func setupForUpdate(engine *Engine) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, f := range list {
|
for _, f := range list {
|
||||||
_, err = engine.Insert(f)
|
_, err = testEngine.Insert(f)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -229,3 +232,880 @@ func TestWithIn(t *testing.T) {
|
||||||
assert.NoError(t, err)
|
assert.NoError(t, err)
|
||||||
assert.EqualValues(t, 3, cnt)
|
assert.EqualValues(t, 3, cnt)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type Condi map[string]interface{}
|
||||||
|
|
||||||
|
type UpdateAllCols struct {
|
||||||
|
Id int64
|
||||||
|
Bool bool
|
||||||
|
String string
|
||||||
|
Ptr *string
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpdateMustCols struct {
|
||||||
|
Id int64
|
||||||
|
Bool bool
|
||||||
|
String string
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpdateIncr struct {
|
||||||
|
Id int64
|
||||||
|
Cnt int
|
||||||
|
}
|
||||||
|
|
||||||
|
type Article struct {
|
||||||
|
Id int32 `xorm:"pk INT autoincr"`
|
||||||
|
Name string `xorm:"VARCHAR(45)"`
|
||||||
|
Img string `xorm:"VARCHAR(100)"`
|
||||||
|
Aside string `xorm:"VARCHAR(200)"`
|
||||||
|
Desc string `xorm:"VARCHAR(200)"`
|
||||||
|
Content string `xorm:"TEXT"`
|
||||||
|
Status int8 `xorm:"TINYINT(4)"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUpdateMap2(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
assertSync(t, new(UpdateMustCols))
|
||||||
|
|
||||||
|
_, err := testEngine.Table("update_must_cols").Where("id =?", 1).Update(map[string]interface{}{
|
||||||
|
"bool": true,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUpdate1(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
assertSync(t, new(Userinfo))
|
||||||
|
|
||||||
|
_, err := testEngine.Insert(&Userinfo{
|
||||||
|
Username: "user1",
|
||||||
|
})
|
||||||
|
|
||||||
|
var ori Userinfo
|
||||||
|
has, err := testEngine.Get(&ori)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
t.Error(errors.New("not exist"))
|
||||||
|
panic(errors.New("not exist"))
|
||||||
|
}
|
||||||
|
|
||||||
|
// update by id
|
||||||
|
user := Userinfo{Username: "xxx", Height: 1.2}
|
||||||
|
cnt, err := testEngine.Id(ori.Uid).Update(&user)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New("update not returned 1")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
condi := Condi{"username": "zzz", "departname": ""}
|
||||||
|
cnt, err = testEngine.Table(&user).Id(ori.Uid).Update(&condi)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New("update not returned 1")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cnt, err = testEngine.Update(&Userinfo{Username: "yyy"}, &user)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
total, err := testEngine.Count(&user)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cnt != total {
|
||||||
|
err = errors.New("insert not returned 1")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// nullable update
|
||||||
|
{
|
||||||
|
user := &Userinfo{Username: "not null data", Height: 180.5}
|
||||||
|
_, err := testEngine.Insert(user)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
userID := user.Uid
|
||||||
|
|
||||||
|
has, err := testEngine.Id(userID).
|
||||||
|
And("username = ?", user.Username).
|
||||||
|
And("height = ?", user.Height).
|
||||||
|
And("departname = ?", "").
|
||||||
|
And("detail_id = ?", 0).
|
||||||
|
And("is_man = ?", 0).
|
||||||
|
And("created IS NOT NULL").
|
||||||
|
Get(&Userinfo{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
err = errors.New("cannot insert properly")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
updatedUser := &Userinfo{Username: "null data"}
|
||||||
|
cnt, err = testEngine.Id(userID).
|
||||||
|
Nullable("height", "departname", "is_man", "created").
|
||||||
|
Update(updatedUser)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New("update not returned 1")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
has, err = testEngine.Id(userID).
|
||||||
|
And("username = ?", updatedUser.Username).
|
||||||
|
And("height IS NULL").
|
||||||
|
And("departname IS NULL").
|
||||||
|
And("is_man IS NULL").
|
||||||
|
And("created IS NULL").
|
||||||
|
And("detail_id = ?", 0).
|
||||||
|
Get(&Userinfo{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
err = errors.New("cannot update with null properly")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cnt, err = testEngine.Id(userID).Delete(&Userinfo{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New("delete not returned 1")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
err = testEngine.StoreEngine("Innodb").Sync2(&Article{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
err = testEngine.DropTables(&Article{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
a := &Article{0, "1", "2", "3", "4", "5", 2}
|
||||||
|
cnt, err = testEngine.Insert(a)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if a.Id == 0 {
|
||||||
|
err = errors.New("insert returned id is 0")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cnt, err = testEngine.Id(a.Id).Update(&Article{Name: "6"})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var s = "test"
|
||||||
|
|
||||||
|
col1 := &UpdateAllCols{Ptr: &s}
|
||||||
|
err = testEngine.Sync(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
col2 := &UpdateAllCols{col1.Id, true, "", nil}
|
||||||
|
_, err = testEngine.Id(col2.Id).AllCols().Update(col2)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
col3 := &UpdateAllCols{}
|
||||||
|
has, err = testEngine.Id(col2.Id).Get(col3)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !has {
|
||||||
|
err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if *col2 != *col3 {
|
||||||
|
err = errors.New(fmt.Sprintf("col2 should eq col3"))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
|
||||||
|
col1 := &UpdateMustCols{}
|
||||||
|
err = testEngine.Sync(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
col2 := &UpdateMustCols{col1.Id, true, ""}
|
||||||
|
boolStr := testEngine.ColumnMapper.Obj2Table("Bool")
|
||||||
|
stringStr := testEngine.ColumnMapper.Obj2Table("String")
|
||||||
|
_, err = testEngine.Id(col2.Id).MustCols(boolStr, stringStr).Update(col2)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
col3 := &UpdateMustCols{}
|
||||||
|
has, err := testEngine.Id(col2.Id).Get(col3)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !has {
|
||||||
|
err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if *col2 != *col3 {
|
||||||
|
err = errors.New(fmt.Sprintf("col2 should eq col3"))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
|
||||||
|
col1 := &UpdateIncr{}
|
||||||
|
err = testEngine.Sync(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cnt, err := testEngine.Id(col1.Id).Incr(testEngine.ColumnMapper.Obj2Table("Cnt")).Update(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New("update incr failed")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
newCol := new(UpdateIncr)
|
||||||
|
has, err := testEngine.Id(col1.Id).Get(newCol)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
err = errors.New("has incr failed")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if 1 != newCol.Cnt {
|
||||||
|
err = fmt.Errorf("incr failed %v %v %v", newCol.Cnt, newCol, col1)
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpdatedUpdate struct {
|
||||||
|
Id int64
|
||||||
|
Updated time.Time `xorm:"updated"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpdatedUpdate2 struct {
|
||||||
|
Id int64
|
||||||
|
Updated int64 `xorm:"updated"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpdatedUpdate3 struct {
|
||||||
|
Id int64
|
||||||
|
Updated int `xorm:"updated bigint"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpdatedUpdate4 struct {
|
||||||
|
Id int64
|
||||||
|
Updated int `xorm:"updated"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpdatedUpdate5 struct {
|
||||||
|
Id int64
|
||||||
|
Updated time.Time `xorm:"updated bigint"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUpdateUpdated(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
|
||||||
|
di := new(UpdatedUpdate)
|
||||||
|
err := testEngine.Sync2(di)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(&UpdatedUpdate{})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ci := &UpdatedUpdate{}
|
||||||
|
_, err = testEngine.Id(1).Update(ci)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
has, err := testEngine.Id(1).Get(di)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
t.Fatal(ErrNotExist)
|
||||||
|
}
|
||||||
|
if ci.Updated.Unix() != di.Updated.Unix() {
|
||||||
|
t.Fatal("should equal:", ci, di)
|
||||||
|
}
|
||||||
|
fmt.Println("ci:", ci, "di:", di)
|
||||||
|
|
||||||
|
di2 := new(UpdatedUpdate2)
|
||||||
|
err = testEngine.Sync2(di2)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(&UpdatedUpdate2{})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
ci2 := &UpdatedUpdate2{}
|
||||||
|
_, err = testEngine.Id(1).Update(ci2)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
has, err = testEngine.Id(1).Get(di2)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
t.Fatal(ErrNotExist)
|
||||||
|
}
|
||||||
|
if ci2.Updated != di2.Updated {
|
||||||
|
t.Fatal("should equal:", ci2, di2)
|
||||||
|
}
|
||||||
|
fmt.Println("ci2:", ci2, "di2:", di2)
|
||||||
|
|
||||||
|
di3 := new(UpdatedUpdate3)
|
||||||
|
err = testEngine.Sync2(di3)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(&UpdatedUpdate3{})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
ci3 := &UpdatedUpdate3{}
|
||||||
|
_, err = testEngine.Id(1).Update(ci3)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
has, err = testEngine.Id(1).Get(di3)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
t.Fatal(ErrNotExist)
|
||||||
|
}
|
||||||
|
if ci3.Updated != di3.Updated {
|
||||||
|
t.Fatal("should equal:", ci3, di3)
|
||||||
|
}
|
||||||
|
fmt.Println("ci3:", ci3, "di3:", di3)
|
||||||
|
|
||||||
|
di4 := new(UpdatedUpdate4)
|
||||||
|
err = testEngine.Sync2(di4)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(&UpdatedUpdate4{})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ci4 := &UpdatedUpdate4{}
|
||||||
|
_, err = testEngine.Id(1).Update(ci4)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
has, err = testEngine.Id(1).Get(di4)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
t.Fatal(ErrNotExist)
|
||||||
|
}
|
||||||
|
if ci4.Updated != di4.Updated {
|
||||||
|
t.Fatal("should equal:", ci4, di4)
|
||||||
|
}
|
||||||
|
fmt.Println("ci4:", ci4, "di4:", di4)
|
||||||
|
|
||||||
|
di5 := new(UpdatedUpdate5)
|
||||||
|
err = testEngine.Sync2(di5)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(&UpdatedUpdate5{})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
ci5 := &UpdatedUpdate5{}
|
||||||
|
_, err = testEngine.Id(1).Update(ci5)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
has, err = testEngine.Id(1).Get(di5)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
t.Fatal(ErrNotExist)
|
||||||
|
}
|
||||||
|
if ci5.Updated.Unix() != di5.Updated.Unix() {
|
||||||
|
t.Fatal("should equal:", ci5, di5)
|
||||||
|
}
|
||||||
|
fmt.Println("ci5:", ci5, "di5:", di5)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUpdateSameMapper(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
|
||||||
|
oldMapper := testEngine.ColumnMapper
|
||||||
|
testEngine.unMapType(rValue(new(Userinfo)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(Condi)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(Article)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(UpdateAllCols)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(UpdateMustCols)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(UpdateIncr)).Type())
|
||||||
|
testEngine.SetMapper(core.SameMapper{})
|
||||||
|
defer func() {
|
||||||
|
testEngine.unMapType(rValue(new(Userinfo)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(Condi)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(Article)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(UpdateAllCols)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(UpdateMustCols)).Type())
|
||||||
|
testEngine.unMapType(rValue(new(UpdateIncr)).Type())
|
||||||
|
testEngine.SetMapper(oldMapper)
|
||||||
|
}()
|
||||||
|
|
||||||
|
assertSync(t, new(Userinfo))
|
||||||
|
|
||||||
|
_, err := testEngine.Insert(&Userinfo{
|
||||||
|
Username: "user1",
|
||||||
|
})
|
||||||
|
assert.NoError(t, err)
|
||||||
|
|
||||||
|
var ori Userinfo
|
||||||
|
has, err := testEngine.Get(&ori)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
t.Error(errors.New("not exist"))
|
||||||
|
panic(errors.New("not exist"))
|
||||||
|
}
|
||||||
|
// update by id
|
||||||
|
user := Userinfo{Username: "xxx", Height: 1.2}
|
||||||
|
cnt, err := testEngine.Id(ori.Uid).Update(&user)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New("update not returned 1")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
condi := Condi{"Username": "zzz", "Departname": ""}
|
||||||
|
cnt, err = testEngine.Table(&user).Id(ori.Uid).Update(&condi)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New("update not returned 1")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cnt, err = testEngine.Update(&Userinfo{Username: "yyy"}, &user)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
total, err := testEngine.Count(&user)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cnt != total {
|
||||||
|
err = errors.New("insert not returned 1")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
err = testEngine.Sync(&Article{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
err = testEngine.DropTables(&Article{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
a := &Article{0, "1", "2", "3", "4", "5", 2}
|
||||||
|
cnt, err = testEngine.Insert(a)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if a.Id == 0 {
|
||||||
|
err = errors.New("insert returned id is 0")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cnt, err = testEngine.Id(a.Id).Update(&Article{Name: "6"})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
col1 := &UpdateAllCols{}
|
||||||
|
err = testEngine.Sync(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
col2 := &UpdateAllCols{col1.Id, true, "", nil}
|
||||||
|
_, err = testEngine.Id(col2.Id).AllCols().Update(col2)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
col3 := &UpdateAllCols{}
|
||||||
|
has, err = testEngine.Id(col2.Id).Get(col3)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !has {
|
||||||
|
err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if *col2 != *col3 {
|
||||||
|
err = errors.New(fmt.Sprintf("col2 should eq col3"))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
col1 := &UpdateMustCols{}
|
||||||
|
err = testEngine.Sync(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
col2 := &UpdateMustCols{col1.Id, true, ""}
|
||||||
|
boolStr := testEngine.ColumnMapper.Obj2Table("Bool")
|
||||||
|
stringStr := testEngine.ColumnMapper.Obj2Table("String")
|
||||||
|
_, err = testEngine.Id(col2.Id).MustCols(boolStr, stringStr).Update(col2)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
col3 := &UpdateMustCols{}
|
||||||
|
has, err := testEngine.Id(col2.Id).Get(col3)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !has {
|
||||||
|
err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if *col2 != *col3 {
|
||||||
|
err = errors.New(fmt.Sprintf("col2 should eq col3"))
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
|
||||||
|
col1 := &UpdateIncr{}
|
||||||
|
err = testEngine.Sync(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Insert(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cnt, err := testEngine.Id(col1.Id).Incr("`Cnt`").Update(col1)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if cnt != 1 {
|
||||||
|
err = errors.New("update incr failed")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
newCol := new(UpdateIncr)
|
||||||
|
has, err := testEngine.Id(col1.Id).Get(newCol)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if !has {
|
||||||
|
err = errors.New("has incr failed")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if 1 != newCol.Cnt {
|
||||||
|
err = errors.New("incr failed")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUseBool(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
assertSync(t, new(Userinfo))
|
||||||
|
|
||||||
|
cnt1, err := testEngine.Count(&Userinfo{})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
users := make([]Userinfo, 0)
|
||||||
|
err = testEngine.Find(&users)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
var fNumber int64
|
||||||
|
for _, u := range users {
|
||||||
|
if u.IsMan == false {
|
||||||
|
fNumber += 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
cnt2, err := testEngine.UseBool().Update(&Userinfo{IsMan: true})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
if fNumber != cnt2 {
|
||||||
|
fmt.Println("cnt1", cnt1, "fNumber", fNumber, "cnt2", cnt2)
|
||||||
|
/*err = errors.New("Updated number is not corrected.")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)*/
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.Update(&Userinfo{IsMan: true})
|
||||||
|
if err == nil {
|
||||||
|
err = errors.New("error condition")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBool(t *testing.T) {
|
||||||
|
assert.NoError(t, prepareEngine())
|
||||||
|
assertSync(t, new(Userinfo))
|
||||||
|
|
||||||
|
_, err := testEngine.UseBool().Update(&Userinfo{IsMan: true})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
users := make([]Userinfo, 0)
|
||||||
|
err = testEngine.Find(&users)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
for _, user := range users {
|
||||||
|
if !user.IsMan {
|
||||||
|
err = errors.New("update bool or find bool error")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = testEngine.UseBool().Update(&Userinfo{IsMan: false})
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
users = make([]Userinfo, 0)
|
||||||
|
err = testEngine.Find(&users)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
for _, user := range users {
|
||||||
|
if user.IsMan {
|
||||||
|
err = errors.New("update bool or find bool error")
|
||||||
|
t.Error(err)
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue