fix race in test

Signed-off-by: crazycs520 <[email protected]>
tikv · Apr 1, 2024 · a6e4f8b · a6e4f8b
1 parent 2a07af4
commit a6e4f8b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/internal/client/client_test.go b/internal/client/client_test.go
@@ -904,6 +904,8 @@ func TestRandomRestartStoreAndForwarding(t *testing.T) {
 	wg := sync.WaitGroup{}
 	done := int64(0)
 	concurrency := 500
+	addr1 := store1.Addr()
+	addr2 := store2.Addr()
 	wg.Add(1)
 	go func() {
 		defer wg.Done()
@@ -927,7 +929,7 @@ func TestRandomRestartStoreAndForwarding(t *testing.T) {
 		}
 	}()
 
-	conn, err := client1.getConnArray(store1.Addr(), true)
+	conn, err := client1.getConnArray(addr1, true)
 	assert.Nil(t, err)
 	for j := 0; j < concurrency; j++ {
 		wg.Add(1)
@@ -940,8 +942,9 @@ func TestRandomRestartStoreAndForwarding(t *testing.T) {
 				req := &tikvpb.BatchCommandsRequest_Request{Cmd: &tikvpb.BatchCommandsRequest_Request_Coprocessor{Coprocessor: &coprocessor.Request{}}}
 				forwardedHost := ""
 				if i%2 != 0 {
+					forwardedHost = addr2
 				}
-				_, err := sendBatchRequest(context.Background(), store1.Addr(), forwardedHost, conn.batchConn, req, time.Millisecond*50, 0)
+				_, err := sendBatchRequest(context.Background(), addr1, forwardedHost, conn.batchConn, req, time.Millisecond*50, 0)
 				if err == nil ||
 					err.Error() == "EOF" ||
 					err.Error() == "rpc error: code = Unavailable desc = error reading from server: EOF" ||