Fix deadlock in ready test

Kubernetes-commit: 39fa78fe7d6020b508308e84c8b671273731dfd6
This commit is contained in:
Wojciech Tyczyński 2023-03-03 14:09:32 +01:00 committed by Kubernetes Publisher
parent 24494ec75b
commit ba5f84614d
1 changed files with 9 additions and 0 deletions

View File

@ -18,6 +18,7 @@ package cacher
import (
"context"
"sync"
"testing"
"time"
)
@ -81,18 +82,26 @@ func Test_newReadyRacy(t *testing.T) {
errCh := make(chan error, concurrency)
ready := newReady()
ready.set(false)
wg := sync.WaitGroup{}
wg.Add(2 * concurrency)
for i := 0; i < concurrency; i++ {
go func() {
errCh <- ready.wait(context.Background())
}()
go func() {
defer wg.Done()
ready.set(false)
}()
go func() {
defer wg.Done()
ready.set(true)
}()
}
// Last one has to be set to true.
wg.Wait()
ready.set(true)
for i := 0; i < concurrency; i++ {
if err := <-errCh; err != nil {
t.Errorf("unexpected error %v on channel %d", err, i)