mirror of
https://github.com/gogf/gf.git
synced 2025-04-05 03:05:05 +08:00
test(drivers/mssql): add unit testing cases of transaction (#3818)
This commit is contained in:
oldme
GitHub
parent
7cd672d22c
commit
77cb7fb412
@ -231,7 +231,6 @@ func Test_DB_Insert(t *testing.T) {
|
||||
|
||||
one, err := db.Model(table).Where("id", 3).One()
|
||||
t.AssertNil(err)
|
||||
fmt.Println(one)
|
||||
t.Assert(one["ID"].Int(), 3)
|
||||
t.Assert(one["PASSPORT"].String(), "user_3")
|
||||
t.Assert(one["PASSWORD"].String(), "25d55ad283aa400af464c76d713c07ad")
|
||||
|
||||
@ -126,7 +126,7 @@ func createInitTable(table ...string) (name string) {
|
||||
"passport": fmt.Sprintf(`user_%d`, i),
|
||||
"password": fmt.Sprintf(`pass_%d`, i),
|
||||
"nickname": fmt.Sprintf(`name_%d`, i),
|
||||
"create_time": gtime.Now(),
|
||||
"create_time": "2018-10-24 10:00:00",
|
||||
})
|
||||
}
|
||||
result, err := db.Insert(context.Background(), name, array.Slice())
|
||||
|
||||
983
contrib/drivers/mssql/mssql_z_unit_transaction_test.go
Normal file
983
contrib/drivers/mssql/mssql_z_unit_transaction_test.go
Normal file
@ -0,0 +1,983 @@
|
||||
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
||||
//
|
||||
// This Source Code Form is subject to the terms of the MIT License.
|
||||
// If a copy of the MIT was not distributed with this file,
|
||||
// You can obtain one at https://github.com/gogf/gf.
|
||||
|
||||
package mssql_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/gogf/gf/v2/database/gdb"
|
||||
"github.com/gogf/gf/v2/errors/gerror"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"github.com/gogf/gf/v2/os/gctx"
|
||||
"github.com/gogf/gf/v2/os/gtime"
|
||||
"github.com/gogf/gf/v2/test/gtest"
|
||||
)
|
||||
|
||||
func Test_TX_Query(t *testing.T) {
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Query("SELECT ?", 1)
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Query("SELECT ?+?", 1, 2)
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Query("SELECT ?+?", g.Slice{1, 2})
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Query("ERROR")
|
||||
t.AssertNE(err, nil)
|
||||
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_Exec(t *testing.T) {
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Exec("SELECT ?", 1)
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Exec("SELECT ?+?", 1, 2)
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Exec("SELECT ?+?", g.Slice{1, 2})
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Exec("ERROR")
|
||||
t.AssertNE(err, nil)
|
||||
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_Commit(t *testing.T) {
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_Rollback(t *testing.T) {
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
err = tx.Rollback()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_Prepare(t *testing.T) {
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
st, err := tx.Prepare("SELECT 100")
|
||||
t.AssertNil(err)
|
||||
|
||||
rows, err := st.Query()
|
||||
t.AssertNil(err)
|
||||
|
||||
var value int
|
||||
if rows.Next() {
|
||||
err = rows.Scan(&value)
|
||||
t.AssertNil(err)
|
||||
|
||||
}
|
||||
t.Assert(value, 100)
|
||||
|
||||
err = rows.Close()
|
||||
t.AssertNil(err)
|
||||
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_Insert(t *testing.T) {
|
||||
table := createTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
user := tx.Model(table)
|
||||
_, err = user.Data(g.Map{
|
||||
"id": 1,
|
||||
"passport": "t1",
|
||||
"password": "25d55ad283aa400af464c76d713c07ad",
|
||||
"nickname": "T1",
|
||||
"create_time": gtime.Now().String(),
|
||||
}).Insert()
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Insert(table, g.Map{
|
||||
"id": 2,
|
||||
"passport": "t1",
|
||||
"password": "25d55ad283aa400af464c76d713c07ad",
|
||||
"nickname": "T1",
|
||||
"create_time": gtime.Now().String(),
|
||||
})
|
||||
t.AssertNil(err)
|
||||
|
||||
n, err := tx.Model(table).Count()
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(n, int64(2))
|
||||
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_BatchInsert(t *testing.T) {
|
||||
table := createTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Insert(table, g.List{
|
||||
{
|
||||
"id": 2,
|
||||
"passport": "t",
|
||||
"password": "25d55ad283aa400af464c76d713c07ad",
|
||||
"nickname": "T2",
|
||||
"create_time": gtime.Now().String(),
|
||||
},
|
||||
{
|
||||
"id": 3,
|
||||
"passport": "t3",
|
||||
"password": "25d55ad283aa400af464c76d713c07ad",
|
||||
"nickname": "T3",
|
||||
"create_time": gtime.Now().String(),
|
||||
},
|
||||
}, 10)
|
||||
t.AssertNil(err)
|
||||
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
|
||||
n, err := db.Model(table).Count()
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(n, int64(2))
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_Update(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
result, err := tx.Update(table, "create_time='2019-10-24 10:00:00'", "id=3")
|
||||
t.AssertNil(err)
|
||||
|
||||
n, _ := result.RowsAffected()
|
||||
t.Assert(n, 1)
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Model(table).Fields("create_time").Where("id", 3).Value()
|
||||
t.AssertNE(err, nil)
|
||||
|
||||
value, err := db.Model(table).Fields("create_time").Where("id", 3).Value()
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(value.String(), "2019-10-24 10:00:00")
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_GetAll(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
result, err := tx.GetAll(fmt.Sprintf("SELECT * FROM %s WHERE id=?", table), 1)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(len(result), 1)
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_GetOne(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
record, err := tx.GetOne(fmt.Sprintf("SELECT * FROM %s WHERE passport=?", table), "user_2")
|
||||
t.AssertNil(err)
|
||||
|
||||
t.AssertNE(record, nil)
|
||||
t.Assert(record["NICKNAME"].String(), "name_2")
|
||||
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_GetValue(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
value, err := tx.GetValue(fmt.Sprintf("SELECT id FROM %s WHERE passport=?", table), "user_3")
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(value.Int(), 3)
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_GetCount(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
count, err := tx.GetCount("SELECT * FROM " + table)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(count, int64(TableSize))
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_GetStruct(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
type User struct {
|
||||
Id int
|
||||
Passport string
|
||||
Password string
|
||||
NickName string
|
||||
CreateTime gtime.Time
|
||||
}
|
||||
user := new(User)
|
||||
err = tx.GetStruct(user, fmt.Sprintf("SELECT * FROM %s WHERE id=?", table), 3)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(user.NickName, "name_3")
|
||||
t.Assert(user.CreateTime.String(), "2018-10-24 10:00:00")
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
type User struct {
|
||||
Id int
|
||||
Passport string
|
||||
Password string
|
||||
NickName string
|
||||
CreateTime *gtime.Time
|
||||
}
|
||||
user := new(User)
|
||||
err = tx.GetStruct(user, fmt.Sprintf("SELECT * FROM %s WHERE id=?", table), 3)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(user.NickName, "name_3")
|
||||
t.Assert(user.CreateTime.String(), "2018-10-24 10:00:00")
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_GetStructs(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
type User struct {
|
||||
Id int
|
||||
Passport string
|
||||
Password string
|
||||
NickName string
|
||||
CreateTime gtime.Time
|
||||
}
|
||||
var users []User
|
||||
err = tx.GetStructs(&users, fmt.Sprintf("SELECT * FROM %s WHERE id>=?", table), 1)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(len(users), TableSize)
|
||||
t.Assert(users[0].Id, 1)
|
||||
t.Assert(users[1].Id, 2)
|
||||
t.Assert(users[2].Id, 3)
|
||||
t.Assert(users[0].NickName, "name_1")
|
||||
t.Assert(users[1].NickName, "name_2")
|
||||
t.Assert(users[2].NickName, "name_3")
|
||||
t.Assert(users[2].CreateTime.String(), "2018-10-24 10:00:00")
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
type User struct {
|
||||
Id int
|
||||
Passport string
|
||||
Password string
|
||||
NickName string
|
||||
CreateTime *gtime.Time
|
||||
}
|
||||
var users []User
|
||||
err = tx.GetStructs(&users, fmt.Sprintf("SELECT * FROM %s WHERE id>=?", table), 1)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(len(users), TableSize)
|
||||
t.Assert(users[0].Id, 1)
|
||||
t.Assert(users[1].Id, 2)
|
||||
t.Assert(users[2].Id, 3)
|
||||
t.Assert(users[0].NickName, "name_1")
|
||||
t.Assert(users[1].NickName, "name_2")
|
||||
t.Assert(users[2].NickName, "name_3")
|
||||
t.Assert(users[2].CreateTime.String(), "2018-10-24 10:00:00")
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_GetScan(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
type User struct {
|
||||
Id int
|
||||
Passport string
|
||||
Password string
|
||||
NickName string
|
||||
CreateTime gtime.Time
|
||||
}
|
||||
user := new(User)
|
||||
err = tx.GetScan(user, fmt.Sprintf("SELECT * FROM %s WHERE id=?", table), 3)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(user.NickName, "name_3")
|
||||
t.Assert(user.CreateTime.String(), "2018-10-24 10:00:00")
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
type User struct {
|
||||
Id int
|
||||
Passport string
|
||||
Password string
|
||||
NickName string
|
||||
CreateTime *gtime.Time
|
||||
}
|
||||
user := new(User)
|
||||
err = tx.GetScan(user, fmt.Sprintf("SELECT * FROM %s WHERE id=?", table), 3)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(user.NickName, "name_3")
|
||||
t.Assert(user.CreateTime.String(), "2018-10-24 10:00:00")
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
type User struct {
|
||||
Id int
|
||||
Passport string
|
||||
Password string
|
||||
NickName string
|
||||
CreateTime gtime.Time
|
||||
}
|
||||
var users []User
|
||||
err = tx.GetScan(&users, fmt.Sprintf("SELECT * FROM %s WHERE id>=?", table), 1)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(len(users), TableSize)
|
||||
t.Assert(users[0].Id, 1)
|
||||
t.Assert(users[1].Id, 2)
|
||||
t.Assert(users[2].Id, 3)
|
||||
t.Assert(users[0].NickName, "name_1")
|
||||
t.Assert(users[1].NickName, "name_2")
|
||||
t.Assert(users[2].NickName, "name_3")
|
||||
t.Assert(users[2].CreateTime.String(), "2018-10-24 10:00:00")
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
type User struct {
|
||||
Id int
|
||||
Passport string
|
||||
Password string
|
||||
NickName string
|
||||
CreateTime *gtime.Time
|
||||
}
|
||||
var users []User
|
||||
err = tx.GetScan(&users, fmt.Sprintf("SELECT * FROM %s WHERE id>=?", table), 1)
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(len(users), TableSize)
|
||||
t.Assert(users[0].Id, 1)
|
||||
t.Assert(users[1].Id, 2)
|
||||
t.Assert(users[2].Id, 3)
|
||||
t.Assert(users[0].NickName, "name_1")
|
||||
t.Assert(users[1].NickName, "name_2")
|
||||
t.Assert(users[2].NickName, "name_3")
|
||||
t.Assert(users[2].CreateTime.String(), "2018-10-24 10:00:00")
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_TX_Delete(t *testing.T) {
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
tx, err := db.Begin(ctx)
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = tx.Delete(table, "1=1")
|
||||
t.AssertNil(err)
|
||||
|
||||
err = tx.Commit()
|
||||
t.AssertNil(err)
|
||||
|
||||
n, err := db.Model(table).Count()
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(n, int64(0))
|
||||
t.Assert(tx.IsClosed(), true)
|
||||
})
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
tx, err := db.Begin(ctx)
|
||||
if err != nil {
|
||||
gtest.Error(err)
|
||||
}
|
||||
_, err = tx.Delete(table, "1=1")
|
||||
t.AssertNil(err)
|
||||
|
||||
n, err := tx.Model(table).Count()
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(n, int64(0))
|
||||
err = tx.Rollback()
|
||||
t.AssertNil(err)
|
||||
|
||||
n, err = db.Model(table).Count()
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(n, int64(TableSize))
|
||||
t.AssertNE(n, int64(0))
|
||||
t.Assert(tx.IsClosed(), true)
|
||||
})
|
||||
}
|
||||
|
||||
func Test_Transaction(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
ctx := context.TODO()
|
||||
err := db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
_, err := tx.Ctx(ctx).Replace(table, g.Map{
|
||||
"id": 1,
|
||||
"passport": "USER_1",
|
||||
"password": "PASS_1",
|
||||
"nickname": "NAME_1",
|
||||
"create_time": gtime.Now().String(),
|
||||
})
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(tx.IsClosed(), false)
|
||||
return gerror.New("error")
|
||||
})
|
||||
t.AssertNE(err, nil)
|
||||
|
||||
value, err := db.Model(table).Ctx(ctx).Fields("nickname").Where("id", 1).Value()
|
||||
t.AssertNil(err)
|
||||
t.Assert(value.String(), "name_1")
|
||||
})
|
||||
}
|
||||
|
||||
func Test_Transaction_Panic(t *testing.T) {
|
||||
table := createInitTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
ctx := context.TODO()
|
||||
err := db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
_, err := tx.Replace(table, g.Map{
|
||||
"id": 1,
|
||||
"passport": "USER_1",
|
||||
"password": "PASS_1",
|
||||
"nickname": "NAME_1",
|
||||
"create_time": gtime.Now().String(),
|
||||
})
|
||||
t.AssertNil(err)
|
||||
|
||||
panic("error")
|
||||
return nil
|
||||
})
|
||||
t.AssertNE(err, nil)
|
||||
|
||||
value, err := db.Model(table).Fields("nickname").Where("id", 1).Value()
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(value.String(), "name_1")
|
||||
})
|
||||
}
|
||||
|
||||
// TODO
|
||||
// MSSQL does not support nested transaction.
|
||||
// So the following test cases are not supported.
|
||||
// If the problem is solved in the future, the test cases will be enabled.
|
||||
|
||||
// func Test_Transaction_Nested_Begin_Rollback_Commit(t *testing.T) {
|
||||
// table := createTable()
|
||||
// defer dropTable(table)
|
||||
//
|
||||
// gtest.C(t, func(t *gtest.T) {
|
||||
// tx, err := db.Begin(ctx)
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // tx begin.
|
||||
// err = tx.Begin()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // tx rollback.
|
||||
// _, err = tx.Model(table).Data(g.Map{
|
||||
// "id": 1,
|
||||
// "passport": "user_1",
|
||||
// "password": "pass_1",
|
||||
// "nickname": "name_1",
|
||||
// }).Insert()
|
||||
// err = tx.Rollback()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // tx commit.
|
||||
// _, err = tx.Model(table).Data(g.Map{
|
||||
// "id": 2,
|
||||
// "passport": "user_2",
|
||||
// "password": "pass_2",
|
||||
// "nickname": "name_2",
|
||||
// }).Insert()
|
||||
// err = tx.Commit()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // check data.
|
||||
// all, err := db.Model(table).All()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// t.Assert(len(all), 1)
|
||||
// t.Assert(all[0]["id"], 2)
|
||||
// })
|
||||
// }
|
||||
//
|
||||
// func Test_Transaction_Nested_TX_Transaction_UseTX(t *testing.T) {
|
||||
// table := createTable()
|
||||
// defer dropTable(table)
|
||||
//
|
||||
// db.SetDebug(true)
|
||||
// defer db.SetDebug(false)
|
||||
//
|
||||
// gtest.C(t, func(t *gtest.T) {
|
||||
// var (
|
||||
// err error
|
||||
// ctx = context.TODO()
|
||||
// )
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// // commit
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// _, err = tx.Model(table).Data(g.Map{
|
||||
// "id": 1,
|
||||
// "passport": "USER_1",
|
||||
// "password": "PASS_1",
|
||||
// "nickname": "NAME_1",
|
||||
// "create_time": gtime.Now().String(),
|
||||
// }).Insert()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // rollback
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// _, err = tx.Model(table).Data(g.Map{
|
||||
// "id": 2,
|
||||
// "passport": "USER_2",
|
||||
// "password": "PASS_2",
|
||||
// "nickname": "NAME_2",
|
||||
// "create_time": gtime.Now().String(),
|
||||
// }).Insert()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// panic("error")
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNE(err, nil)
|
||||
// return nil
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// all, err := db.Ctx(ctx).Model(table).All()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// t.Assert(len(all), 1)
|
||||
// t.Assert(all[0]["id"], 1)
|
||||
//
|
||||
// // another record.
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// // commit
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// _, err = tx.Model(table).Data(g.Map{
|
||||
// "id": 3,
|
||||
// "passport": "USER_1",
|
||||
// "password": "PASS_1",
|
||||
// "nickname": "NAME_1",
|
||||
// "create_time": gtime.Now().String(),
|
||||
// }).Insert()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // rollback
|
||||
// err = tx.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// _, err = tx.Model(table).Data(g.Map{
|
||||
// "id": 4,
|
||||
// "passport": "USER_2",
|
||||
// "password": "PASS_2",
|
||||
// "nickname": "NAME_2",
|
||||
// "create_time": gtime.Now().String(),
|
||||
// }).Insert()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// panic("error")
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNE(err, nil)
|
||||
// return nil
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// all, err = db.Ctx(ctx).Model(table).All()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// t.Assert(len(all), 2)
|
||||
// t.Assert(all[0]["id"], 1)
|
||||
// t.Assert(all[1]["id"], 3)
|
||||
// })
|
||||
// }
|
||||
//
|
||||
// func Test_Transaction_Nested_TX_Transaction_UseDB(t *testing.T) {
|
||||
// table := createTable()
|
||||
// defer dropTable(table)
|
||||
//
|
||||
// // db.SetDebug(true)
|
||||
// // defer db.SetDebug(false)
|
||||
//
|
||||
// gtest.C(t, func(t *gtest.T) {
|
||||
// var (
|
||||
// err error
|
||||
// ctx = context.TODO()
|
||||
// )
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// // commit
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// _, err = db.Model(table).Ctx(ctx).Data(g.Map{
|
||||
// "id": 1,
|
||||
// "passport": "USER_1",
|
||||
// "password": "PASS_1",
|
||||
// "nickname": "NAME_1",
|
||||
// "create_time": gtime.Now().String(),
|
||||
// }).Insert()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // rollback
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// _, err = tx.Model(table).Ctx(ctx).Data(g.Map{
|
||||
// "id": 2,
|
||||
// "passport": "USER_2",
|
||||
// "password": "PASS_2",
|
||||
// "nickname": "NAME_2",
|
||||
// "create_time": gtime.Now().String(),
|
||||
// }).Insert()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // panic makes this transaction rollback.
|
||||
// panic("error")
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNE(err, nil)
|
||||
// return nil
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// all, err := db.Model(table).All()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// t.Assert(len(all), 1)
|
||||
// t.Assert(all[0]["id"], 1)
|
||||
//
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// // commit
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// _, err = db.Model(table).Ctx(ctx).Data(g.Map{
|
||||
// "id": 3,
|
||||
// "passport": "USER_1",
|
||||
// "password": "PASS_1",
|
||||
// "nickname": "NAME_1",
|
||||
// "create_time": gtime.Now().String(),
|
||||
// }).Insert()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // rollback
|
||||
// err = db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||||
// _, err = tx.Model(table).Ctx(ctx).Data(g.Map{
|
||||
// "id": 4,
|
||||
// "passport": "USER_2",
|
||||
// "password": "PASS_2",
|
||||
// "nickname": "NAME_2",
|
||||
// "create_time": gtime.Now().String(),
|
||||
// }).Insert()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // panic makes this transaction rollback.
|
||||
// panic("error")
|
||||
// return err
|
||||
// })
|
||||
// t.AssertNE(err, nil)
|
||||
// return nil
|
||||
// })
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// all, err = db.Model(table).All()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// t.Assert(len(all), 2)
|
||||
// t.Assert(all[0]["id"], 1)
|
||||
// t.Assert(all[1]["id"], 3)
|
||||
// })
|
||||
// }
|
||||
//
|
||||
// func Test_Transaction_Nested_SavePoint_RollbackTo(t *testing.T) {
|
||||
// table := createTable()
|
||||
// defer dropTable(table)
|
||||
//
|
||||
// gtest.C(t, func(t *gtest.T) {
|
||||
// tx, err := db.Begin(ctx)
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // tx save point.
|
||||
// _, err = tx.Model(table).Data(g.Map{
|
||||
// "id": 1,
|
||||
// "passport": "user_1",
|
||||
// "password": "pass_1",
|
||||
// "nickname": "name_1",
|
||||
// }).Insert()
|
||||
// err = tx.SavePoint("MyPoint")
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// _, err = tx.Model(table).Data(g.Map{
|
||||
// "id": 2,
|
||||
// "passport": "user_2",
|
||||
// "password": "pass_2",
|
||||
// "nickname": "name_2",
|
||||
// }).Insert()
|
||||
// // tx rollback to.
|
||||
// err = tx.RollbackTo("MyPoint")
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // tx commit.
|
||||
// err = tx.Commit()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// // check data.
|
||||
// all, err := db.Model(table).All()
|
||||
// t.AssertNil(err)
|
||||
//
|
||||
// t.Assert(len(all), 1)
|
||||
// t.Assert(all[0]["id"], 1)
|
||||
// })
|
||||
// }
|
||||
|
||||
func Test_Transaction_Method(t *testing.T) {
|
||||
table := createTable()
|
||||
defer dropTable(table)
|
||||
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
var err error
|
||||
err = db.Transaction(gctx.New(), func(ctx context.Context, tx gdb.TX) error {
|
||||
_, err = db.Model(table).Ctx(ctx).Data(g.Map{
|
||||
"id": 1,
|
||||
"passport": "t1",
|
||||
"password": "25d55ad283aa400af464c76d713c07ad",
|
||||
"nickname": "T1",
|
||||
"create_time": gtime.Now().String(),
|
||||
}).Insert()
|
||||
t.AssertNil(err)
|
||||
|
||||
_, err = db.Ctx(ctx).Exec(ctx, fmt.Sprintf(
|
||||
"insert into %s(passport , password , nickname , create_time , id ) "+
|
||||
"VALUES('t2','25d55ad283aa400af464c76d713c07ad','T2','2021-08-25 21:53:00',2) ",
|
||||
table))
|
||||
t.AssertNil(err)
|
||||
|
||||
return gerror.New("rollback")
|
||||
})
|
||||
t.AssertNE(err, nil)
|
||||
|
||||
count, err := db.Model(table).Count()
|
||||
t.AssertNil(err)
|
||||
|
||||
t.Assert(count, int64(0))
|
||||
})
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user