forked from mirror/gorm
245 lines
8.1 KiB
Go
245 lines
8.1 KiB
Go
package tests_test
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
"testing"
|
|
|
|
"gorm.io/gorm"
|
|
. "gorm.io/gorm/utils/tests"
|
|
)
|
|
|
|
func TestRow(t *testing.T) {
|
|
user1 := User{Name: "RowUser1", Age: 1}
|
|
user2 := User{Name: "RowUser2", Age: 10}
|
|
user3 := User{Name: "RowUser3", Age: 20}
|
|
DB.Save(&user1).Save(&user2).Save(&user3)
|
|
|
|
row := DB.Table("users").Where("name = ?", user2.Name).Select("age").Row()
|
|
|
|
var age int64
|
|
if err := row.Scan(&age); err != nil {
|
|
t.Fatalf("Failed to scan age, got %v", err)
|
|
}
|
|
|
|
if age != 10 {
|
|
t.Errorf("Scan with Row, age expects: %v, got %v", user2.Age, age)
|
|
}
|
|
|
|
table := "gorm.users"
|
|
if DB.Dialector.Name() != "mysql" {
|
|
table = "users" // other databases doesn't support select with `database.table`
|
|
}
|
|
|
|
DB.Table(table).Where(map[string]interface{}{"name": user2.Name}).Update("age", 20)
|
|
|
|
row = DB.Table(table+" as u").Where("u.name = ?", user2.Name).Select("age").Row()
|
|
if err := row.Scan(&age); err != nil {
|
|
t.Fatalf("Failed to scan age, got %v", err)
|
|
}
|
|
|
|
if age != 20 {
|
|
t.Errorf("Scan with Row, age expects: %v, got %v", user2.Age, age)
|
|
}
|
|
}
|
|
|
|
func TestRows(t *testing.T) {
|
|
user1 := User{Name: "RowsUser1", Age: 1}
|
|
user2 := User{Name: "RowsUser2", Age: 10}
|
|
user3 := User{Name: "RowsUser3", Age: 20}
|
|
DB.Save(&user1).Save(&user2).Save(&user3)
|
|
|
|
rows, err := DB.Table("users").Where("name = ? or name = ?", user2.Name, user3.Name).Select("name, age").Rows()
|
|
if err != nil {
|
|
t.Errorf("Not error should happen, got %v", err)
|
|
}
|
|
|
|
count := 0
|
|
for rows.Next() {
|
|
var name string
|
|
var age int64
|
|
rows.Scan(&name, &age)
|
|
count++
|
|
}
|
|
|
|
if count != 2 {
|
|
t.Errorf("Should found two records")
|
|
}
|
|
}
|
|
|
|
func TestRaw(t *testing.T) {
|
|
user1 := User{Name: "ExecRawSqlUser1", Age: 1}
|
|
user2 := User{Name: "ExecRawSqlUser2", Age: 10}
|
|
user3 := User{Name: "ExecRawSqlUser3", Age: 20}
|
|
DB.Save(&user1).Save(&user2).Save(&user3)
|
|
|
|
type result struct {
|
|
Name string
|
|
Email string
|
|
}
|
|
|
|
var results []result
|
|
DB.Raw("SELECT name, age FROM users WHERE name = ? or name = ?", user2.Name, user3.Name).Scan(&results)
|
|
if len(results) != 2 || results[0].Name != user2.Name || results[1].Name != user3.Name {
|
|
t.Errorf("Raw with scan")
|
|
}
|
|
|
|
rows, _ := DB.Raw("select name, age from users where name = ?", user3.Name).Rows()
|
|
count := 0
|
|
for rows.Next() {
|
|
count++
|
|
}
|
|
if count != 1 {
|
|
t.Errorf("Raw with Rows should find one record with name 3")
|
|
}
|
|
|
|
DB.Exec("update users set name=? where name in (?)", "jinzhu-raw", []string{user1.Name, user2.Name, user3.Name})
|
|
if DB.Where("name in (?)", []string{user1.Name, user2.Name, user3.Name}).First(&User{}).Error != gorm.ErrRecordNotFound {
|
|
t.Error("Raw sql to update records")
|
|
}
|
|
|
|
DB.Exec("update users set age=? where name = ?", gorm.Expr("age * ? + ?", 2, 10), "jinzhu-raw")
|
|
|
|
var age int
|
|
DB.Raw("select sum(age) from users where name = ?", "jinzhu-raw").Scan(&age)
|
|
|
|
if age != ((1+10+20)*2 + 30) {
|
|
t.Errorf("Invalid age, got %v", age)
|
|
}
|
|
}
|
|
|
|
func TestRowsWithGroup(t *testing.T) {
|
|
users := []User{
|
|
{Name: "having_user_1", Age: 1},
|
|
{Name: "having_user_2", Age: 10},
|
|
{Name: "having_user_1", Age: 20},
|
|
{Name: "having_user_1", Age: 30},
|
|
}
|
|
|
|
DB.Create(&users)
|
|
|
|
rows, err := DB.Select("name, count(*) as total").Table("users").Group("name").Having("name IN ?", []string{users[0].Name, users[1].Name}).Rows()
|
|
if err != nil {
|
|
t.Fatalf("got error %v", err)
|
|
}
|
|
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var name string
|
|
var total int64
|
|
rows.Scan(&name, &total)
|
|
|
|
if name == users[0].Name && total != 3 {
|
|
t.Errorf("Should have one user having name %v", users[0].Name)
|
|
} else if name == users[1].Name && total != 1 {
|
|
t.Errorf("Should have two users having name %v", users[1].Name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestQueryRaw(t *testing.T) {
|
|
users := []*User{
|
|
GetUser("row_query_user", Config{}),
|
|
GetUser("row_query_user", Config{}),
|
|
GetUser("row_query_user", Config{}),
|
|
}
|
|
DB.Create(&users)
|
|
|
|
var user User
|
|
DB.Raw("select * from users WHERE id = ?", users[1].ID).First(&user)
|
|
CheckUser(t, user, *users[1])
|
|
}
|
|
|
|
func TestDryRun(t *testing.T) {
|
|
user := *GetUser("dry-run", Config{})
|
|
|
|
dryRunDB := DB.Session(&gorm.Session{DryRun: true})
|
|
|
|
stmt := dryRunDB.Create(&user).Statement
|
|
if stmt.SQL.String() == "" || len(stmt.Vars) != 9 {
|
|
t.Errorf("Failed to generate sql, got %v", stmt.SQL.String())
|
|
}
|
|
|
|
stmt2 := dryRunDB.Find(&user, "id = ?", user.ID).Statement
|
|
if stmt2.SQL.String() == "" || len(stmt2.Vars) != 1 {
|
|
t.Errorf("Failed to generate sql, got %v", stmt2.SQL.String())
|
|
}
|
|
}
|
|
|
|
func TestGroupConditions(t *testing.T) {
|
|
type Pizza struct {
|
|
ID uint
|
|
Name string
|
|
Size string
|
|
}
|
|
dryRunDB := DB.Session(&gorm.Session{DryRun: true})
|
|
|
|
stmt := dryRunDB.Where(
|
|
DB.Where("pizza = ?", "pepperoni").Where(DB.Where("size = ?", "small").Or("size = ?", "medium")),
|
|
).Or(
|
|
DB.Where("pizza = ?", "hawaiian").Where("size = ?", "xlarge"),
|
|
).Find(&Pizza{}).Statement
|
|
|
|
execStmt := dryRunDB.Exec("WHERE (pizza = ? AND (size = ? OR size = ?)) OR (pizza = ? AND size = ?)", "pepperoni", "small", "medium", "hawaiian", "xlarge").Statement
|
|
|
|
result := DB.Dialector.Explain(stmt.SQL.String(), stmt.Vars...)
|
|
expects := DB.Dialector.Explain(execStmt.SQL.String(), execStmt.Vars...)
|
|
|
|
if !strings.HasSuffix(result, expects) {
|
|
t.Errorf("expects: %v, got %v", expects, result)
|
|
}
|
|
}
|
|
|
|
func TestCombineStringConditions(t *testing.T) {
|
|
dryRunDB := DB.Session(&gorm.Session{DryRun: true})
|
|
sql := dryRunDB.Where("a = ? or b = ?", "a", "b").Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE \(a = .+ or b = .+\) AND .users.\..deleted_at. IS NULL`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
|
|
sql = dryRunDB.Where("a = ? or b = ?", "a", "b").Or("c = ? and d = ?", "c", "d").Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE \(a = .+ or b = .+\) OR \(c = .+ and d = .+\) AND .users.\..deleted_at. IS NULL`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
|
|
sql = dryRunDB.Where("a = ? or b = ?", "a", "b").Or("c = ?", "c").Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE \(a = .+ or b = .+\) OR c = .+ AND .users.\..deleted_at. IS NULL`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
|
|
sql = dryRunDB.Where("a = ? or b = ?", "a", "b").Or("c = ? and d = ?", "c", "d").Or("e = ? and f = ?", "e", "f").Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE \(a = .+ or b = .+\) OR \(c = .+ and d = .+\) OR \(e = .+ and f = .+\) AND .users.\..deleted_at. IS NULL`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
|
|
sql = dryRunDB.Where("a = ? or b = ?", "a", "b").Where("c = ? and d = ?", "c", "d").Not("e = ? and f = ?", "e", "f").Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE \(a = .+ or b = .+\) AND \(c = .+ and d = .+\) AND NOT \(e = .+ and f = .+\) AND .users.\..deleted_at. IS NULL`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
|
|
sql = dryRunDB.Where("a = ? or b = ?", "a", "b").Where("c = ?", "c").Not("e = ? and f = ?", "e", "f").Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE \(a = .+ or b = .+\) AND c = .+ AND NOT \(e = .+ and f = .+\) AND .users.\..deleted_at. IS NULL`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
|
|
sql = dryRunDB.Where("a = ? or b = ?", "a", "b").Where("c = ? and d = ?", "c", "d").Not("e = ?", "e").Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE \(a = .+ or b = .+\) AND \(c = .+ and d = .+\) AND NOT e = .+ AND .users.\..deleted_at. IS NULL`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
|
|
sql = dryRunDB.Where("a = ? or b = ?", "a", "b").Unscoped().Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE a = .+ or b = .+$`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
|
|
sql = dryRunDB.Or("a = ? or b = ?", "a", "b").Unscoped().Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE a = .+ or b = .+$`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
|
|
sql = dryRunDB.Not("a = ? or b = ?", "a", "b").Unscoped().Find(&User{}).Statement.SQL.String()
|
|
if !regexp.MustCompile(`WHERE NOT \(a = .+ or b = .+\)$`).MatchString(sql) {
|
|
t.Fatalf("invalid sql generated, got %v", sql)
|
|
}
|
|
}
|