server/tests/cursor: fix cursor fetch error flacky test (pingcap#50969)

close pingcap#47029
xzhangxian1008 · Feb 4, 2024 · e528b68 · e528b68
1 parent 9a0b6b6
commit e528b68
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pkg/server/tests/cursor/cursor_test.go b/pkg/server/tests/cursor/cursor_test.go
@@ -67,15 +67,16 @@ func TestCursorFetchErrorInFetch(t *testing.T) {
 
 	tk.MustExec(fmt.Sprintf("set tidb_mem_quota_query=%d", 1))
 
+	require.NoError(t, failpoint.Enable("github.com/pingcap/tidb/pkg/util/chunk/get-chunk-error", "return(true)"))
+	defer func() {
+		require.NoError(t, failpoint.Disable("github.com/pingcap/tidb/pkg/util/chunk/get-chunk-error"))
+	}()
+
 	require.NoError(t, c.Dispatch(ctx, append(
 		appendUint32([]byte{tmysql.ComStmtExecute}, uint32(stmt.ID())),
 		tmysql.CursorTypeReadOnly, 0x1, 0x0, 0x0, 0x0,
 	)))
 
-	require.NoError(t, failpoint.Enable("github.com/pingcap/tidb/pkg/util/chunk/get-chunk-error", "return(true)"))
-	defer func() {
-		require.NoError(t, failpoint.Disable("github.com/pingcap/tidb/pkg/util/chunk/get-chunk-error"))
-	}()
 	require.ErrorContains(t, c.Dispatch(ctx, appendUint32(appendUint32([]byte{tmysql.ComStmtFetch}, uint32(stmt.ID())), 1024)), "fail to get chunk for test")
 	// after getting a failed FETCH, the cursor should have been reseted
 	require.False(t, stmt.GetCursorActive())