session: fix goleak in the sendBatchRequest (pingcap#31991)

close pingcap#31990
denimyftiu · Jan 26, 2022 · b83dc00 · b83dc00
1 parent 93454da
commit b83dc00
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/session/pessimistic_test.go b/session/pessimistic_test.go
@@ -673,7 +673,9 @@ func (s *testPessimisticSuite) TestKillStopTTLManager(c *C) {
 
 	// This query should success rather than returning a ResolveLock error.
 	tk2.MustExec("update test_kill set c = c + 1 where id = 1")
-
+	succ = atomic.CompareAndSwapUint32(&sessVars.Killed, 1, 0)
+	c.Assert(succ, IsTrue)
+	tk.MustExec("rollback")
 	tk2.MustExec("rollback")
 }