mirror of https://github.com/go-gorm/gorm.git
61 lines
1.8 KiB
Go
61 lines
1.8 KiB
Go
|
package tests_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
. "gorm.io/gorm/utils/tests"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDistinct(t *testing.T) {
|
||
|
|
var users = []User{
|
||
|
|
*GetUser("distinct", Config{}),
|
||
|
|
*GetUser("distinct", Config{}),
|
||
|
|
*GetUser("distinct", Config{}),
|
||
|
|
*GetUser("distinct-2", Config{}),
|
||
|
|
*GetUser("distinct-3", Config{}),
|
||
|
|
}
|
||
|
|
users[0].Age = 20
|
||
|
|
|
||
|
|
if err := DB.Create(&users).Error; err != nil {
|
||
|
|
t.Fatalf("errors happened when create users: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var names []string
|
||
|
|
DB.Model(&User{}).Where("name like ?", "distinct%").Order("name").Pluck("Name", &names)
|
||
|
|
AssertEqual(t, names, []string{"distinct", "distinct", "distinct", "distinct-2", "distinct-3"})
|
||
|
|
|
||
|
|
var names1 []string
|
||
|
|
DB.Model(&User{}).Where("name like ?", "distinct%").Distinct().Order("name").Pluck("Name", &names1)
|
||
|
|
|
||
|
|
AssertEqual(t, names1, []string{"distinct", "distinct-2", "distinct-3"})
|
||
|
|
|
||
|
|
var results []User
|
||
|
|
if err := DB.Distinct("name", "age").Where("name like ?", "distinct%").Order("name, age desc").Find(&results).Error; err != nil {
|
||
|
|
t.Errorf("failed to query users, got error: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
expects := []User{
|
||
|
|
{Name: "distinct", Age: 20},
|
||
|
|
{Name: "distinct", Age: 18},
|
||
|
|
{Name: "distinct-2", Age: 18},
|
||
|
|
{Name: "distinct-3", Age: 18},
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(results) != 4 {
|
||
|
|
t.Fatalf("invalid results length found, expects: %v, got %v", len(expects), len(results))
|
||
|
|
}
|
||
|
|
|
||
|
|
for idx, expect := range expects {
|
||
|
|
AssertObjEqual(t, results[idx], expect, "Name", "Age")
|
||
|
|
}
|
||
|
|
|
||
|
|
var count int64
|
||
|
|
if err := DB.Model(&User{}).Where("name like ?", "distinct%").Count(&count).Error; err != nil || count != 5 {
|
||
|
|
t.Errorf("failed to query users count, got error: %v, count: %v", err, count)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := DB.Model(&User{}).Distinct("name").Where("name like ?", "distinct%").Count(&count).Error; err != nil || count != 3 {
|
||
|
|
t.Errorf("failed to query users count, got error: %v, count %v", err, count)
|
||
|
|
}
|
||
|
|
}
|