diff --git a/pkg/util/lock/metrics_mutex_test.go b/pkg/util/lock/metrics_mutex_test.go index 8db946c966..3f050b713a 100644 --- a/pkg/util/lock/metrics_mutex_test.go +++ b/pkg/util/lock/metrics_mutex_test.go @@ -23,15 +23,15 @@ func TestMetricsLockLock(t *testing.T) { testRWLock := lManager.applyRWLock(lName) wg := sync.WaitGroup{} testRWLock.Lock("main_thread") + wg.Add(1) go func() { - wg.Add(1) + defer wg.Done() before := time.Now() testRWLock.Lock("sub_thread") lkDuration := time.Since(before) assert.True(t, lkDuration >= lockDuration) testRWLock.UnLock("sub_threadXX") testRWLock.UnLock("sub_thread") - wg.Done() }() time.Sleep(lockDuration) testRWLock.UnLock("main_thread") @@ -52,14 +52,14 @@ func TestMetricsLockRLock(t *testing.T) { testRWLock := lManager.applyRWLock(lName) wg := sync.WaitGroup{} testRWLock.RLock("main_thread") + wg.Add(1) go func() { - wg.Add(1) + defer wg.Done() before := time.Now() testRWLock.Lock("sub_thread") lkDuration := time.Since(before) assert.True(t, lkDuration >= lockDuration) testRWLock.UnLock("sub_thread") - wg.Done() }() time.Sleep(lockDuration) assert.Equal(t, 1, len(testRWLock.acquireTimeMap))