ringbuffer: consume buffer before closing

2025-10-05 07:06:58 +08:00 · 2021-05-04 12:10:39 +02:00
parent 750583341a
commit 1a2dacadd8
2 changed files with 8 additions and 5 deletions
--- a/pkg/ringbuffer/ringbuffer_test.go
+++ b/pkg/ringbuffer/ringbuffer_test.go
@@ -46,11 +46,15 @@ func TestClose(t *testing.T) {
 	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)
 	}()

-	time.Sleep(100 * time.Millisecond)
+	r.Push([]byte{0x01, 0x02, 0x03, 0x04})

 	r.Close()
 	<-done