gorm/tests/distinct_test.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)
}
}