test: fix unstable TestMemoryLeak

pingcap · Sep 17, 2021 · a9982b8 · a9982b8
1 parent b839744
commit a9982b8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/session/session_test.go b/session/session_test.go
@@ -5677,13 +5677,14 @@ func (s *testTiDBAsLibrary) TestMemoryLeak(c *C) {
 	runtime.ReadMemStats(&memStat)
 	oldHeapInUse := memStat.HeapInuse
 
-	for i := 0; i < 10; i++ {
+	for i := 0; i < 20; i++ {
 		initAndCloseTiDB()
 	}
 
 	runtime.GC()
 	runtime.ReadMemStats(&memStat)
-	c.Assert(memStat.HeapInuse-oldHeapInUse, Less, uint64(150*units.MiB))
+	// before the fix, initAndCloseTiDB for 20 times will cost 900 MB memory, so we test for a quite loose upper bound.
+	c.Assert(memStat.HeapInuse-oldHeapInUse, Less, uint64(300*units.MiB))
 }
 
 func (s *testSessionSuite) TestTiDBReadStaleness(c *C) {