refine test

Signed-off-by: crazycs520 <[email protected]>
pingcap · Oct 18, 2023 · 8aa4cf6 · 8aa4cf6
1 parent 2f2c554
commit 8aa4cf6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/realtikvtest/sessiontest/session_fail_test.go b/tests/realtikvtest/sessiontest/session_fail_test.go
@@ -349,7 +349,12 @@ func TestBatchClientDataRace(t *testing.T) {
 					time.Sleep(time.Millisecond * time.Duration(rand.Intn(10)+1))
 					cancel()
 				}()
-				rs, _ := tk.ExecWithContext(ctx, "select * from t where a = 1")
+				sqls := []string{
+					"select * from t where a = 1",
+					"select * from t where a in ( 1, 2, 3)",
+					"select * from t",
+				}
+				rs, _ := tk.ExecWithContext(ctx, sqls[j%len(sqls)])
 				if rs != nil {
 					session.ResultSetToStringSlice(ctx, tk.Session(), rs)
 				}