Test Order/GroupBy

This commit is contained in:
Jinzhu 2020-07-05 22:12:52 +08:00
parent 4e066c9590
commit 9a4941ba70
4 changed files with 40 additions and 5 deletions

View File

@ -14,7 +14,7 @@ func (s Select) Name() string {
func (s Select) Build(builder Builder) { func (s Select) Build(builder Builder) {
if len(s.Columns) > 0 { if len(s.Columns) > 0 {
if s.Distinct { if s.Distinct {
builder.WriteString(" DISTINCT ") builder.WriteString("DISTINCT ")
} }
for idx, column := range s.Columns { for idx, column := range s.Columns {

View File

@ -67,6 +67,27 @@ func TestGroupBy(t *testing.T) {
t.Errorf("name should be groupby, but got %v, total should be 660, but got %v", name, total) t.Errorf("name should be groupby, but got %v, total should be 660, but got %v", name, total)
} }
var result = struct {
Name string
Total int64
}{}
if err := DB.Model(&User{}).Select("name, sum(age) as total").Where("name LIKE ?", "groupby%").Group("name").Having("name = ?", "groupby1").Find(&result).Error; err != nil {
t.Errorf("no error should happen, but got %v", err)
}
if result.Name != "groupby1" || result.Total != 660 {
t.Errorf("name should be groupby, total should be 660, but got %+v", result)
}
if err := DB.Model(&User{}).Select("name, sum(age) as total").Where("name LIKE ?", "groupby%").Group("name").Having("name = ?", "groupby1").Scan(&result).Error; err != nil {
t.Errorf("no error should happen, but got %v", err)
}
if result.Name != "groupby1" || result.Total != 660 {
t.Errorf("name should be groupby, total should be 660, but got %+v", result)
}
var active bool var active bool
if err := DB.Model(&User{}).Select("name, active, sum(age)").Where("name = ? and active = ?", "groupby", true).Group("name").Group("active").Row().Scan(&name, &active, &total); err != nil { if err := DB.Model(&User{}).Select("name, active, sum(age)").Where("name = ? and active = ?", "groupby", true).Group("name").Group("active").Row().Scan(&name, &active, &total); err != nil {
t.Errorf("no error should happen, but got %v", err) t.Errorf("no error should happen, but got %v", err)

View File

@ -101,6 +101,7 @@ func TestJoinsWithSelect(t *testing.T) {
DB.Save(&user) DB.Save(&user)
var results []result var results []result
DB.Table("users").Select("users.id, pets.id as pet_id, pets.name").Joins("left join pets on pets.user_id = users.id").Where("users.name = ?", "joins_with_select").Scan(&results) DB.Table("users").Select("users.id, pets.id as pet_id, pets.name").Joins("left join pets on pets.user_id = users.id").Where("users.name = ?", "joins_with_select").Scan(&results)
sort.Slice(results, func(i, j int) bool { sort.Slice(results, func(i, j int) bool {

View File

@ -223,17 +223,17 @@ func TestOr(t *testing.T) {
result := dryDB.Where("role = ?", "admin").Or("role = ?", "super_admin").Find(&User{}) result := dryDB.Where("role = ?", "admin").Or("role = ?", "super_admin").Find(&User{})
if !regexp.MustCompile("SELECT \\* FROM .*users.* WHERE .*role.* = .+ OR .*role.* = .+").MatchString(result.Statement.SQL.String()) { if !regexp.MustCompile("SELECT \\* FROM .*users.* WHERE .*role.* = .+ OR .*role.* = .+").MatchString(result.Statement.SQL.String()) {
t.Fatalf("Build NOT condition, but got %v", result.Statement.SQL.String()) t.Fatalf("Build OR condition, but got %v", result.Statement.SQL.String())
} }
result = dryDB.Where("name = ?", "jinzhu").Or(User{Name: "jinzhu 2", Age: 18}).Find(&User{}) result = dryDB.Where("name = ?", "jinzhu").Or(User{Name: "jinzhu 2", Age: 18}).Find(&User{})
if !regexp.MustCompile("SELECT \\* FROM .*users.* WHERE .*name.* = .+ OR \\(.*name.* AND .*age.*\\)").MatchString(result.Statement.SQL.String()) { if !regexp.MustCompile("SELECT \\* FROM .*users.* WHERE .*name.* = .+ OR \\(.*name.* AND .*age.*\\)").MatchString(result.Statement.SQL.String()) {
t.Fatalf("Build NOT condition, but got %v", result.Statement.SQL.String()) t.Fatalf("Build OR condition, but got %v", result.Statement.SQL.String())
} }
result = dryDB.Where("name = ?", "jinzhu").Or(map[string]interface{}{"name": "jinzhu 2", "age": 18}).Find(&User{}) result = dryDB.Where("name = ?", "jinzhu").Or(map[string]interface{}{"name": "jinzhu 2", "age": 18}).Find(&User{})
if !regexp.MustCompile("SELECT \\* FROM .*users.* WHERE .*name.* = .+ OR \\(.*age.* AND .*name.*\\)").MatchString(result.Statement.SQL.String()) { if !regexp.MustCompile("SELECT \\* FROM .*users.* WHERE .*name.* = .+ OR \\(.*age.* AND .*name.*\\)").MatchString(result.Statement.SQL.String()) {
t.Fatalf("Build NOT condition, but got %v", result.Statement.SQL.String()) t.Fatalf("Build OR condition, but got %v", result.Statement.SQL.String())
} }
} }
@ -426,6 +426,20 @@ func TestSearchWithEmptyChain(t *testing.T) {
} }
} }
func TestOrder(t *testing.T) {
dryDB := DB.Session(&gorm.Session{DryRun: true})
result := dryDB.Order("age desc, name").Find(&User{})
if !regexp.MustCompile("SELECT \\* FROM .*users.* ORDER BY age desc, name").MatchString(result.Statement.SQL.String()) {
t.Fatalf("Build Order condition, but got %v", result.Statement.SQL.String())
}
result = dryDB.Order("age desc").Order("name").Find(&User{})
if !regexp.MustCompile("SELECT \\* FROM .*users.* ORDER BY age desc,name").MatchString(result.Statement.SQL.String()) {
t.Fatalf("Build Order condition, but got %v", result.Statement.SQL.String())
}
}
func TestLimit(t *testing.T) { func TestLimit(t *testing.T) {
users := []User{ users := []User{
{Name: "LimitUser1", Age: 1}, {Name: "LimitUser1", Age: 1},
@ -461,7 +475,6 @@ func TestOffset(t *testing.T) {
if (len(users1) != len(users4)) || (len(users1)-len(users2) != 3) || (len(users1)-len(users3) != 5) { if (len(users1) != len(users4)) || (len(users1)-len(users2) != 3) || (len(users1)-len(users3) != 5) {
t.Errorf("Offset should work without limit.") t.Errorf("Offset should work without limit.")
} }
} }
func TestSearchWithMap(t *testing.T) { func TestSearchWithMap(t *testing.T) {