3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-05-31 09:57:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

storage/inmemory: add tests for close behavior (#2336)

Browse source
This commit is contained in:
Caleb Doxsey 2021-07-06 09:36:04 -06:00 • committed by GitHub
parent 0bca5c9556
commit 031fa1a91b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 43 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
pkg/storage/inmemory/stream.go
View file

@ -99,11 +99,20 @@ func (stream *recordStream) Err() error {
select {
case <-stream.ctx.Done():
return stream.ctx.Err()
case <-stream.closed:
return storage.ErrStreamClosed
default:
}
select {
case <-stream.backend.closed:
return storage.ErrStreamClosed
default:
return nil
}
select {
case <-stream.closed:
return storage.ErrStreamClosed
default:
}
return nil
}