fix test

Signed-off-by: lhy1024 <[email protected]>

pkg/utils/syncutil/flexible_wait_group.go

@@ -60,3 +60,11 @@ func (fwg *FlexibleWaitGroup) Wait() {
 	}
 	fwg.Unlock()
 }
+
+// getCount returns the current count of the FlexibleWaitGroup.
+// It is only used for testing.
+func (fwg *FlexibleWaitGroup) getCount() int {
+	fwg.Lock()
+	defer fwg.Unlock()
+	return fwg.count
+}

pkg/utils/syncutil/flexible_wait_group_test.go

@@ -1,4 +1,4 @@
-// Copyright 2022 TiKV Project Authors.
+// Copyright 2023 TiKV Project Authors.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -24,14 +24,14 @@ import (
 func TestFlexibleWaitGroup(t *testing.T) {
 	re := require.New(t)
 	fwg := NewFlexibleWaitGroup()
+	now := time.Now()
 	for i := 20; i >= 0; i-- {
 		fwg.Add(1)
 		go func(i int) {
 			defer fwg.Done()
 			time.Sleep(time.Millisecond * time.Duration(i*50))
 		}(i)
 	}
-	now := time.Now()
 	fwg.Wait()
 	re.GreaterOrEqual(time.Since(now).Milliseconds(), int64(1000))
 }
@@ -83,7 +83,7 @@ func TestNegativeDelta(t *testing.T) {
 		fwg.Done()
 	}()
 	fwg.Wait()
-	require.Equal(0, fwg.count)
+	require.Equal(0, fwg.getCount())
 }
 
 // TestMultipleWait tests the case where Wait is called multiple times concurrently.
@@ -108,7 +108,7 @@ func TestMultipleWait(t *testing.T) {
 	}()
 	<-done
 	<-done
-	require.Equal(0, fwg.count)
+	require.Equal(0, fwg.getCount())
 }
 
 // TestAddAfterWaitFinished tests the case where Add is called after Wait has finished.
@@ -126,7 +126,7 @@ func TestAddAfterWaitFinished(t *testing.T) {
 	}()
 	<-done
 	fwg.Add(1)
-	require.Equal(1, fwg.count)
+	require.Equal(1, fwg.getCount())
 	fwg.Done()
-	require.Equal(0, fwg.count)
+	require.Equal(0, fwg.getCount())
 }