mirror of
https://github.com/bluenviron/mediamtx
synced 2025-03-01 18:10:36 +00:00
24 lines
381 B
Go
24 lines
381 B
Go
package bytecounter
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestReader(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
buf.Write(bytes.Repeat([]byte{0x01}, 1024))
|
|
|
|
r := NewReader(&buf)
|
|
r.SetCount(100)
|
|
|
|
buf2 := make([]byte, 64)
|
|
n, err := r.Read(buf2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 64, n)
|
|
|
|
require.Equal(t, uint64(100+64), r.Count())
|
|
}
|