forked from mirror/gorm
63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
|
package tests
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/jinzhu/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGroupBy(t *testing.T, db *gorm.DB) {
|
||
|
|
db.Migrator().DropTable(&User{})
|
||
|
|
db.AutoMigrate(&User{})
|
||
|
|
|
||
|
|
t.Run("GroupBy", func(t *testing.T) {
|
||
|
|
var users = []User{{
|
||
|
|
Name: "groupby",
|
||
|
|
Age: 10,
|
||
|
|
Birthday: Now(),
|
||
|
|
}, {
|
||
|
|
Name: "groupby",
|
||
|
|
Age: 20,
|
||
|
|
Birthday: Now(),
|
||
|
|
}, {
|
||
|
|
Name: "groupby",
|
||
|
|
Age: 30,
|
||
|
|
Birthday: Now(),
|
||
|
|
}, {
|
||
|
|
Name: "groupby1",
|
||
|
|
Age: 110,
|
||
|
|
Birthday: Now(),
|
||
|
|
}, {
|
||
|
|
Name: "groupby1",
|
||
|
|
Age: 220,
|
||
|
|
Birthday: Now(),
|
||
|
|
}, {
|
||
|
|
Name: "groupby1",
|
||
|
|
Age: 330,
|
||
|
|
Birthday: Now(),
|
||
|
|
}}
|
||
|
|
|
||
|
|
if err := db.Create(&users).Error; err != nil {
|
||
|
|
t.Errorf("errors happened when create: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var name string
|
||
|
|
var total int
|
||
|
|
if err := db.Model(&User{}).Select("name, sum(age)").Where("name = ?", "groupby").Group("name").Row().Scan(&name, &total); err != nil {
|
||
|
|
t.Errorf("no error should happen, but got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if name != "groupby" || total != 60 {
|
||
|
|
t.Errorf("name should be groupby, but got %v, total should be 60, but got %v", name, total)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := db.Model(&User{}).Select("name, sum(age) as total").Where("name LIKE ?", "groupby%").Group("name").Having("name = ?", "groupby1").Row().Scan(&name, &total); err != nil {
|
||
|
|
t.Errorf("no error should happen, but got %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if name != "groupby1" || total != 660 {
|
||
|
|
t.Errorf("name should be groupby, but got %v, total should be 660, but got %v", name, total)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|