fix test

Signed-off-by: lhy1024 <admin@liudos.us>

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

@@ -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())
 }