mirror of
https://gitee.com/johng/gf
synced 2026-06-07 02:12:11 +08:00
786 lines
19 KiB
Go
786 lines
19 KiB
Go
// Copyright 2019 gf Author(https://github.com/gogf/gf). 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 gdb_test
|
|
|
|
import (
|
|
"database/sql"
|
|
"testing"
|
|
|
|
"github.com/gogf/gf/g"
|
|
"github.com/gogf/gf/g/os/gtime"
|
|
"github.com/gogf/gf/g/test/gtest"
|
|
)
|
|
|
|
// 基本测试
|
|
func TestModel_Insert(t *testing.T) {
|
|
result, err := db.Table("user").Filter().Data(g.Map{
|
|
"id": 1,
|
|
"uid": 1,
|
|
"passport": "t1",
|
|
"password": "25d55ad283aa400af464c76d713c07ad",
|
|
"nickname": "T1",
|
|
"create_time": gtime.Now().String(),
|
|
}).Insert()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ := result.LastInsertId()
|
|
gtest.Assert(n, 1)
|
|
|
|
result, err = db.Table("user").Filter().Data(map[interface{}]interface{}{
|
|
"id": "2",
|
|
"uid": "2",
|
|
"passport": "t2",
|
|
"password": "25d55ad283aa400af464c76d713c07ad",
|
|
"nickname": "T2",
|
|
"create_time": gtime.Now().String(),
|
|
}).Insert()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ = result.RowsAffected()
|
|
gtest.Assert(n, 1)
|
|
|
|
type User struct {
|
|
Id int `gconv:"id"`
|
|
Uid int `gconv:"uid"`
|
|
Passport string `json:"passport"`
|
|
Password string `gconv:"password"`
|
|
Nickname string `gconv:"nickname"`
|
|
CreateTime string `json:"create_time"`
|
|
}
|
|
result, err = db.Table("user").Filter().Data(User{
|
|
Id: 3,
|
|
Uid: 3,
|
|
Passport: "t3",
|
|
Password: "25d55ad283aa400af464c76d713c07ad",
|
|
Nickname: "T3",
|
|
CreateTime: gtime.Now().String(),
|
|
}).Insert()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ = result.RowsAffected()
|
|
gtest.Assert(n, 1)
|
|
value, err := db.Table("user").Fields("passport").Where("id=3").Value()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(value.String(), "t3")
|
|
|
|
result, err = db.Table("user").Filter().Data(&User{
|
|
Id: 4,
|
|
Uid: 4,
|
|
Passport: "t4",
|
|
Password: "25d55ad283aa400af464c76d713c07ad",
|
|
Nickname: "T4",
|
|
CreateTime: gtime.Now().String(),
|
|
}).Insert()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ = result.RowsAffected()
|
|
gtest.Assert(n, 1)
|
|
value, err = db.Table("user").Fields("passport").Where("id=4").Value()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(value.String(), "t4")
|
|
|
|
result, err = db.Table("user").Where("id>?", 1).Delete()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ = result.RowsAffected()
|
|
gtest.Assert(n, 3)
|
|
}
|
|
|
|
func TestModel_Batch(t *testing.T) {
|
|
// batch insert
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Filter().Data(g.List{
|
|
{
|
|
"id": 2,
|
|
"uid": 2,
|
|
"passport": "t2",
|
|
"password": "25d55ad283aa400af464c76d713c07ad",
|
|
"nickname": "T2",
|
|
"create_time": gtime.Now().String(),
|
|
},
|
|
{
|
|
"id": 3,
|
|
"uid": 3,
|
|
"passport": "t3",
|
|
"password": "25d55ad283aa400af464c76d713c07ad",
|
|
"nickname": "T3",
|
|
"create_time": gtime.Now().String(),
|
|
},
|
|
}).Batch(1).Insert()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ := result.RowsAffected()
|
|
gtest.Assert(n, 2)
|
|
})
|
|
|
|
// batch save
|
|
gtest.Case(t, func() {
|
|
table := createInitTable()
|
|
defer dropTable(table)
|
|
result, err := db.Table(table).All()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(len(result), INIT_DATA_SIZE)
|
|
for _, v := range result {
|
|
v["nickname"].Set(v["nickname"].String() + v["id"].String())
|
|
}
|
|
r, e := db.Table(table).Data(result).Save()
|
|
gtest.Assert(e, nil)
|
|
n, e := r.RowsAffected()
|
|
gtest.Assert(e, nil)
|
|
gtest.Assert(n, INIT_DATA_SIZE*2)
|
|
})
|
|
|
|
// batch replace
|
|
gtest.Case(t, func() {
|
|
table := createInitTable()
|
|
defer dropTable(table)
|
|
result, err := db.Table(table).All()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(len(result), INIT_DATA_SIZE)
|
|
for _, v := range result {
|
|
v["nickname"].Set(v["nickname"].String() + v["id"].String())
|
|
}
|
|
r, e := db.Table(table).Data(result).Replace()
|
|
gtest.Assert(e, nil)
|
|
n, e := r.RowsAffected()
|
|
gtest.Assert(e, nil)
|
|
gtest.Assert(n, INIT_DATA_SIZE*2)
|
|
})
|
|
}
|
|
|
|
func TestModel_Replace(t *testing.T) {
|
|
result, err := db.Table("user").Data(g.Map{
|
|
"id": 1,
|
|
"passport": "t11",
|
|
"password": "25d55ad283aa400af464c76d713c07ad",
|
|
"nickname": "T11",
|
|
"create_time": "2018-10-10 00:01:10",
|
|
}).Replace()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ := result.RowsAffected()
|
|
gtest.Assert(n, 2)
|
|
}
|
|
|
|
func TestModel_Save(t *testing.T) {
|
|
result, err := db.Table("user").Data(g.Map{
|
|
"id": 1,
|
|
"passport": "t111",
|
|
"password": "25d55ad283aa400af464c76d713c07ad",
|
|
"nickname": "T111",
|
|
"create_time": "2018-10-10 00:01:10",
|
|
}).Save()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ := result.RowsAffected()
|
|
gtest.Assert(n, 2)
|
|
}
|
|
|
|
func TestModel_Update(t *testing.T) {
|
|
table := createInitTable()
|
|
// UPDATE...LIMIT
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table(table).Data("nickname", "T100").OrderBy("id desc").Limit(2).Update()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ := result.RowsAffected()
|
|
gtest.Assert(n, 2)
|
|
|
|
v1, err := db.Table(table).Fields("nickname").Where("id", 10).Value()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(v1.String(), "T100")
|
|
|
|
v2, err := db.Table(table).Fields("nickname").Where("id", 8).Value()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(v2.String(), "T8")
|
|
})
|
|
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Data("passport", "t22").Where("passport=?", "t2").Update()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ := result.RowsAffected()
|
|
gtest.Assert(n, 1)
|
|
})
|
|
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Data("passport", "t2").Where("passport='t22'").Update()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ := result.RowsAffected()
|
|
gtest.Assert(n, 1)
|
|
})
|
|
}
|
|
|
|
func TestModel_Clone(t *testing.T) {
|
|
md := db.Table("user").Where("id IN(?)", g.Slice{1, 3})
|
|
count, err := md.Count()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
record, err := md.OrderBy("id DESC").One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
result, err := md.OrderBy("id ASC").All()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(count, 2)
|
|
gtest.Assert(record["id"].Int(), 3)
|
|
gtest.Assert(len(result), 2)
|
|
gtest.Assert(result[0]["id"].Int(), 1)
|
|
gtest.Assert(result[1]["id"].Int(), 3)
|
|
}
|
|
|
|
func TestModel_Safe(t *testing.T) {
|
|
gtest.Case(t, func() {
|
|
md := db.Table("user").Safe(false).Where("id IN(?)", g.Slice{1, 3})
|
|
count, err := md.Count()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(count, 2)
|
|
md.And("id = ?", 1)
|
|
count, err = md.Count()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(count, 1)
|
|
})
|
|
gtest.Case(t, func() {
|
|
md := db.Table("user").Safe(true).Where("id IN(?)", g.Slice{1, 3})
|
|
count, err := md.Count()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(count, 2)
|
|
md.And("id = ?", 1)
|
|
count, err = md.Count()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(count, 2)
|
|
})
|
|
}
|
|
|
|
func TestModel_All(t *testing.T) {
|
|
result, err := db.Table("user").All()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(result), 3)
|
|
}
|
|
|
|
func TestModel_One(t *testing.T) {
|
|
record, err := db.Table("user").Where("id", 1).One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
if record == nil {
|
|
gtest.Fatal("FAIL")
|
|
}
|
|
gtest.Assert(record["nickname"].String(), "T111")
|
|
}
|
|
|
|
func TestModel_Value(t *testing.T) {
|
|
value, err := db.Table("user").Fields("nickname").Where("id", 1).Value()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
if value == nil {
|
|
gtest.Fatal("FAIL")
|
|
}
|
|
gtest.Assert(value.String(), "T111")
|
|
}
|
|
|
|
func TestModel_Count(t *testing.T) {
|
|
count, err := db.Table("user").Count()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(count, 3)
|
|
}
|
|
|
|
func TestModel_Select(t *testing.T) {
|
|
result, err := db.Table("user").Select()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(result), 3)
|
|
}
|
|
|
|
func TestModel_Struct(t *testing.T) {
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime gtime.Time
|
|
}
|
|
user := new(User)
|
|
err := db.Table("user").Where("id=1").Struct(user)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(user.NickName, "T111")
|
|
gtest.Assert(user.CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
user := new(User)
|
|
err := db.Table("user").Where("id=1").Struct(user)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(user.NickName, "T111")
|
|
gtest.Assert(user.CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
// Auto creating struct object.
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
user := (*User)(nil)
|
|
err := db.Table("user").Where("id=1").Struct(&user)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(user.NickName, "T111")
|
|
gtest.Assert(user.CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
// Just using Scan.
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
user := (*User)(nil)
|
|
err := db.Table("user").Where("id=1").Scan(&user)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(user.NickName, "T111")
|
|
gtest.Assert(user.CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
user := new(User)
|
|
err := db.Table("user").Where("id=-1").Struct(user)
|
|
gtest.Assert(err, sql.ErrNoRows)
|
|
})
|
|
}
|
|
|
|
func TestModel_Structs(t *testing.T) {
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime gtime.Time
|
|
}
|
|
var users []User
|
|
err := db.Table("user").OrderBy("id asc").Structs(&users)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(users), 3)
|
|
gtest.Assert(users[0].Id, 1)
|
|
gtest.Assert(users[1].Id, 2)
|
|
gtest.Assert(users[2].Id, 3)
|
|
gtest.Assert(users[0].NickName, "T111")
|
|
gtest.Assert(users[1].NickName, "T2")
|
|
gtest.Assert(users[2].NickName, "T3")
|
|
gtest.Assert(users[0].CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
// Auto create struct slice.
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
var users []*User
|
|
err := db.Table("user").OrderBy("id asc").Structs(&users)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(users), 3)
|
|
gtest.Assert(users[0].Id, 1)
|
|
gtest.Assert(users[1].Id, 2)
|
|
gtest.Assert(users[2].Id, 3)
|
|
gtest.Assert(users[0].NickName, "T111")
|
|
gtest.Assert(users[1].NickName, "T2")
|
|
gtest.Assert(users[2].NickName, "T3")
|
|
gtest.Assert(users[0].CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
// Just using Scan.
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
var users []*User
|
|
err := db.Table("user").OrderBy("id asc").Scan(&users)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(users), 3)
|
|
gtest.Assert(users[0].Id, 1)
|
|
gtest.Assert(users[1].Id, 2)
|
|
gtest.Assert(users[2].Id, 3)
|
|
gtest.Assert(users[0].NickName, "T111")
|
|
gtest.Assert(users[1].NickName, "T2")
|
|
gtest.Assert(users[2].NickName, "T3")
|
|
gtest.Assert(users[0].CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
var users []*User
|
|
err := db.Table("user").Where("id<0").Structs(&users)
|
|
gtest.Assert(err, sql.ErrNoRows)
|
|
})
|
|
}
|
|
|
|
func TestModel_Scan(t *testing.T) {
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime gtime.Time
|
|
}
|
|
user := new(User)
|
|
err := db.Table("user").Where("id=1").Scan(user)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(user.NickName, "T111")
|
|
gtest.Assert(user.CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
user := new(User)
|
|
err := db.Table("user").Where("id=1").Scan(user)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(user.NickName, "T111")
|
|
gtest.Assert(user.CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime gtime.Time
|
|
}
|
|
var users []User
|
|
err := db.Table("user").OrderBy("id asc").Scan(&users)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(users), 3)
|
|
gtest.Assert(users[0].Id, 1)
|
|
gtest.Assert(users[1].Id, 2)
|
|
gtest.Assert(users[2].Id, 3)
|
|
gtest.Assert(users[0].NickName, "T111")
|
|
gtest.Assert(users[1].NickName, "T2")
|
|
gtest.Assert(users[2].NickName, "T3")
|
|
gtest.Assert(users[0].CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
var users []*User
|
|
err := db.Table("user").OrderBy("id asc").Scan(&users)
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(users), 3)
|
|
gtest.Assert(users[0].Id, 1)
|
|
gtest.Assert(users[1].Id, 2)
|
|
gtest.Assert(users[2].Id, 3)
|
|
gtest.Assert(users[0].NickName, "T111")
|
|
gtest.Assert(users[1].NickName, "T2")
|
|
gtest.Assert(users[2].NickName, "T3")
|
|
gtest.Assert(users[0].CreateTime.String(), "2018-10-10 00:01:10")
|
|
})
|
|
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int
|
|
Passport string
|
|
Password string
|
|
NickName string
|
|
CreateTime *gtime.Time
|
|
}
|
|
user := new(User)
|
|
users := new([]*User)
|
|
err1 := db.Table("user").Where("id < 0").Scan(user)
|
|
err2 := db.Table("user").Where("id < 0").Scan(users)
|
|
gtest.Assert(err1, sql.ErrNoRows)
|
|
gtest.Assert(err2, sql.ErrNoRows)
|
|
})
|
|
}
|
|
|
|
func TestModel_OrderBy(t *testing.T) {
|
|
result, err := db.Table("user").OrderBy("id DESC").Select()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(result), 3)
|
|
gtest.Assert(result[0]["nickname"].String(), "T3")
|
|
}
|
|
|
|
func TestModel_GroupBy(t *testing.T) {
|
|
result, err := db.Table("user").GroupBy("id").Select()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(result), 3)
|
|
gtest.Assert(result[0]["nickname"].String(), "T111")
|
|
}
|
|
|
|
func TestModel_Where(t *testing.T) {
|
|
// string
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where("id=? and nickname=?", 3, "T3").One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.AssertGT(len(result), 0)
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
})
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where("id", 3).One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.AssertGT(len(result), 0)
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
})
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where("id", 3).Where("nickname", "T3").One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
})
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where("id", 3).And("nickname", "T3").One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
})
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where("id", 30).Or("nickname", "T3").One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
})
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where("id", 30).Or("nickname", "T3").And("id>?", 1).One()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
})
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where("id", 30).Or("nickname", "T3").And("id>", 1).One()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
})
|
|
// map
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where(g.Map{"id": 3, "nickname": "T3"}).One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
})
|
|
// map key operator
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where(g.Map{"id>": 1, "id<": 3}).One()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(result["id"].Int(), 2)
|
|
})
|
|
// complicated where 1
|
|
gtest.Case(t, func() {
|
|
//db.SetDebug(true)
|
|
conditions := g.Map{
|
|
"nickname like ?": "%T%",
|
|
"id between ? and ?": g.Slice{1, 3},
|
|
"id > 0": nil,
|
|
"create_time > 0": nil,
|
|
"id": g.Slice{1, 2, 3},
|
|
}
|
|
result, err := db.Table("user").Where(conditions).OrderBy("id asc").All()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(len(result), 3)
|
|
gtest.Assert(result[0]["id"].Int(), 1)
|
|
})
|
|
// complicated where 2
|
|
gtest.Case(t, func() {
|
|
//db.SetDebug(true)
|
|
conditions := g.Map{
|
|
"nickname like ?": "%T%",
|
|
"id between ? and ?": g.Slice{1, 3},
|
|
"id >= ?": 1,
|
|
"create_time > ?": 0,
|
|
"id in(?)": g.Slice{1, 2, 3},
|
|
}
|
|
result, err := db.Table("user").Where(conditions).OrderBy("id asc").All()
|
|
gtest.Assert(err, nil)
|
|
gtest.Assert(len(result), 3)
|
|
gtest.Assert(result[0]["id"].Int(), 1)
|
|
})
|
|
// struct
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Id int `json:"id"`
|
|
Nickname string `gconv:"nickname"`
|
|
}
|
|
result, err := db.Table("user").Where(User{3, "T3"}).One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
|
|
result, err = db.Table("user").Where(&User{3, "T3"}).One()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(result["id"].Int(), 3)
|
|
})
|
|
// slice single
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where("id IN(?)", g.Slice{1, 3}).OrderBy("id ASC").All()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(result), 2)
|
|
gtest.Assert(result[0]["id"].Int(), 1)
|
|
gtest.Assert(result[1]["id"].Int(), 3)
|
|
})
|
|
// slice + string
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where("nickname=? AND id IN(?)", "T3", g.Slice{1, 3}).OrderBy("id ASC").All()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(result), 1)
|
|
gtest.Assert(result[0]["id"].Int(), 3)
|
|
})
|
|
// slice + map
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Where(g.Map{
|
|
"id": g.Slice{1, 3},
|
|
"nickname": "T3",
|
|
}).OrderBy("id ASC").All()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(result), 1)
|
|
gtest.Assert(result[0]["id"].Int(), 3)
|
|
})
|
|
// slice + struct
|
|
gtest.Case(t, func() {
|
|
type User struct {
|
|
Ids []int `json:"id"`
|
|
Nickname string `gconv:"nickname"`
|
|
}
|
|
result, err := db.Table("user").Where(User{
|
|
Ids: []int{1, 3},
|
|
Nickname: "T3",
|
|
}).OrderBy("id ASC").All()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
gtest.Assert(len(result), 1)
|
|
gtest.Assert(result[0]["id"].Int(), 3)
|
|
})
|
|
}
|
|
|
|
func TestModel_Delete(t *testing.T) {
|
|
// DELETE...LIMIT
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Limit(2).Delete()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ := result.RowsAffected()
|
|
gtest.Assert(n, 2)
|
|
})
|
|
|
|
gtest.Case(t, func() {
|
|
result, err := db.Table("user").Delete()
|
|
if err != nil {
|
|
gtest.Fatal(err)
|
|
}
|
|
n, _ := result.RowsAffected()
|
|
gtest.Assert(n, 1)
|
|
})
|
|
}
|