package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-05 23:26:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

ringbuffer: discard pending data when buffer is closed (#387)

Browse Source
This commit is contained in:
Alessandro Ros
2023-08-26 17:23:54 +02:00
committed by GitHub
parent bf3de06b3c
commit 04a3c45f60
2 changed files with 26 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
pkg/ringbuffer/ringbuffer_test.go
View File

@@ -51,30 +51,28 @@ func TestClose(t *testing.T) {
r, err := New(1024)
require.NoError(t, err)
done := make(chan struct{})
go func() {
defer close(done)
_, ok := r.Pull()
require.Equal(t, true, ok)
_, ok = r.Pull()
require.Equal(t, false, ok)
}()
ok := r.Push([]byte{1, 2, 3, 4})
require.Equal(t, true, ok)
r.Close()
<-done
r.Reset()
ok = r.Push([]byte{5, 6, 7, 8})
require.Equal(t, true, ok)
_, ok = r.Pull()
require.Equal(t, true, ok)
ok = r.Push([]byte{5, 6, 7, 8})
require.Equal(t, true, ok)
r.Close()
_, ok = r.Pull()
require.Equal(t, false, ok)
r.Reset()
ok = r.Push([]byte{9, 10, 11, 12})
require.Equal(t, true, ok)
data, ok := r.Pull()
require.Equal(t, true, ok)
require.Equal(t, []byte{9, 10, 11, 12}, data)
}
func TestOverflow(t *testing.T) {