From 354fe9c9d2641501de87781c654840b77c3c5de2 Mon Sep 17 00:00:00 2001
From: wei liu <wei.liu@zilliz.com>
Date: Fri, 5 Dec 2025 14:01:12 +0800
Subject: [PATCH] fix: unstable test case TestTask_VarCharPrimaryKey (#46106)

issue: #46105

Signed-off-by: Wei Liu <wei.liu@zilliz.com>
---
 internal/proxy/task_test.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/internal/proxy/task_test.go b/internal/proxy/task_test.go
index fbf0a499d7..188acc28aa 100644
--- a/internal/proxy/task_test.go
+++ b/internal/proxy/task_test.go
@@ -2593,7 +2593,11 @@ func TestTask_VarCharPrimaryKey(t *testing.T) {
 	})
 
 	t.Run("upsert", func(t *testing.T) {
-		hash := testutils.GenerateHashKeys(nb)
+		// upsert require pk unique in same batch
+		hash := make([]uint32, nb)
+		for i := 0; i < nb; i++ {
+			hash[i] = uint32(i)
+		}
 		task := &upsertTask{
 			upsertMsg: &msgstream.UpsertMsg{
 				InsertMsg: &BaseInsertTask{