fix race

Signed-off-by: nolouch <[email protected]>
nolouch · Jul 5, 2024 · 503b4e0 · 503b4e0
1 parent 7e8945e
commit 503b4e0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/client/resource_group/controller/controller_test.go b/client/resource_group/controller/controller_test.go
@@ -261,7 +261,9 @@ func TestControllerWithTwoGroupRequestConcurrency(t *testing.T) {
 	// wait default group request token by PeriodicReport.
 	time.Sleep(2 * time.Second)
 	counter := c2.run.requestUnitTokens[0]
+	counter.limiter.mu.Lock()
 	counter.limiter.notify()
+	counter.limiter.mu.Unlock()
 	select {
 	case res := <-recTestGroupAcquireTokenRequest:
 		re.True(res)