From d14ebedad59ee24c171479716f2822fa1d48bf49 Mon Sep 17 00:00:00 2001
From: black <hello.bug@foxmail.com>
Date: Fri, 17 Mar 2023 18:49:58 +0800
Subject: [PATCH] add an easy-to-understand ut

---
 tests/count_test.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/tests/count_test.go b/tests/count_test.go
index 0a24cf86..b0dfb0b5 100644
--- a/tests/count_test.go
+++ b/tests/count_test.go
@@ -11,6 +11,32 @@ import (
 	. "gorm.io/gorm/utils/tests"
 )
 
+func TestCountWithGroup(t *testing.T) {
+	DB.Create([]Company{
+		{Name: "company_count_group_a"},
+		{Name: "company_count_group_a"},
+		{Name: "company_count_group_a"},
+		{Name: "company_count_group_b"},
+		{Name: "company_count_group_c"},
+	})
+
+	var count1 int64
+	if err := DB.Model(&Company{}).Where("name = ?", "company_count_group_a").Group("name").Count(&count1).Error; err != nil {
+		t.Errorf(fmt.Sprintf("Count should work, but got err %v", err))
+	}
+	if count1 != 1 {
+		t.Errorf("Count with group should be 1, but got count: %v", count1)
+	}
+
+	var count2 int64
+	if err := DB.Debug().Model(&Company{}).Where("name in ?", []string{"company_count_group_b", "company_count_group_c"}).Group("name").Count(&count2).Error; err != nil {
+		t.Errorf(fmt.Sprintf("Count should work, but got err %v", err))
+	}
+	if count2 != 2 {
+		t.Errorf("Count with group should be 2, but got count: %v", count2)
+	}
+}
+
 func TestCount(t *testing.T) {
 	var (
 		user1                 = *GetUser("count-1", Config{})