Skip to content

Commit

Permalink
fix access request cache test race (#44652)
Browse files Browse the repository at this point in the history
  • Loading branch information
fspmarshall authored Jul 25, 2024
1 parent 3cbe2df commit 572c3f9
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 0 deletions.
7 changes: 7 additions & 0 deletions lib/services/access_request_cache.go
Original file line number Diff line number Diff line change
Expand Up @@ -344,6 +344,7 @@ func (c *AccessRequestCache) getResourcesAndUpdateCurrent(ctx context.Context) e
c.rw.Lock()
defer c.rw.Unlock()
c.primaryCache = cache
close(c.initC)
return nil
}

Expand Down Expand Up @@ -394,6 +395,12 @@ func (c *AccessRequestCache) initializationChan() <-chan struct{} {
return c.initC
}

// InitializationChan is part of the resourceCollector interface and gets the channel
// used to signal that the accessRequestCache has been initialized.
func (c *AccessRequestCache) InitializationChan() <-chan struct{} {
return c.initializationChan()
}

// Close terminates the background process that keeps the access request cache up to
// date, and terminates any inflight load operations.
func (c *AccessRequestCache) Close() error {
Expand Down
6 changes: 6 additions & 0 deletions lib/services/access_request_cache_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,12 @@ func newAccessRequestPack(t *testing.T) (accessRequestServices, *services.Access
})
require.NoError(t, err)

select {
case <-cache.InitializationChan():
case <-time.After(time.Second * 30):
require.FailNow(t, "timeout waiting for access request cache to initialize")
}

return svcs, cache
}

Expand Down

0 comments on commit 572c3f9

Please sign in to comment.