executor: fix data race for test TestInterruptedDuringSpilling (#50693)

close #50688
pingcap · Jan 24, 2024 · 0568385 · 0568385
1 parent eca5f20
commit 0568385
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/util/chunk/row_container_test.go b/pkg/util/chunk/row_container_test.go
@@ -533,13 +533,17 @@ func TestInterruptedDuringSpilling(t *testing.T) {
 	for i := 0; i < 102400; i++ {
 		rc.Add(chk)
 	}
+	wg := sync.WaitGroup{}
+	wg.Add(1)
 	var cancelTime time.Time
 	go func() {
 		time.Sleep(200 * time.Millisecond)
 		rootTracker.Killer.SendKillSignal(sqlkiller.QueryInterrupted)
 		cancelTime = time.Now()
+		wg.Done()
 	}()
 	rc.spillToDisk(nil)
+	wg.Wait()
 	cancelDuration := time.Since(cancelTime)
 	require.Less(t, cancelDuration, 1*time.Second)
 }