From 8794628623e59a2640861bccb692ffe17935041a Mon Sep 17 00:00:00 2001
From: Monet Lee <monet_lee@163.com>
Date: Fri, 25 Apr 2025 15:31:20 +0800
Subject: [PATCH] improve stress_test-v2

---
 test/stress-test-v2/main.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/test/stress-test-v2/main.go b/test/stress-test-v2/main.go
index 0dd069547..0e4609964 100644
--- a/test/stress-test-v2/main.go
+++ b/test/stress-test-v2/main.go
@@ -56,6 +56,7 @@ var (
 
 const (
 	MaxUser            = 100000
+	Max1kUser          = 1000
 	Max100KGroup       = 100
 	Max999Group        = 1000
 	MaxInviteUserLimit = 999
@@ -553,7 +554,7 @@ func main() {
 						// continue
 					}
 
-					for i := 0; i < MaxUser/MaxInviteUserLimit; i++ {
+					for i := 0; i <= MaxUser/MaxInviteUserLimit; i++ {
 						InviteUserIDs := make([]string, 0)
 						// ensure TargetUserList is in group
 						InviteUserIDs = append(InviteUserIDs, TestTargetUserList...)
@@ -635,13 +636,13 @@ func main() {
 						log.ZError(st.Ctx, "Create group failed.", err)
 						// continue
 					}
-					for i := 0; i < MaxUser/MaxInviteUserLimit; i++ {
+					for i := 0; i <= Max1kUser/MaxInviteUserLimit; i++ {
 						InviteUserIDs := make([]string, 0)
 						// ensure TargetUserList is in group
 						InviteUserIDs = append(InviteUserIDs, TestTargetUserList...)
 
 						startIdx := max(i*MaxInviteUserLimit, 1)
-						endIdx := min((i+1)*MaxInviteUserLimit, MaxUser)
+						endIdx := min((i+1)*MaxInviteUserLimit, Max1kUser)
 
 						for j := startIdx; j < endIdx; j++ {
 							userCreatedID := fmt.Sprintf("v2_StressTest_User_%d", j)