mirror of
https://github.com/aler9/gortsplib
synced 2025-10-05 15:16:51 +08:00
23 lines
421 B
Go
23 lines
421 B
Go
package bytecounter
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestByteCounter(t *testing.T) {
|
|
bc := New(bytes.NewBuffer(nil), nil, nil)
|
|
|
|
_, err := bc.Write([]byte{0x01, 0x02, 0x03, 0x04})
|
|
require.NoError(t, err)
|
|
|
|
buf := make([]byte, 2)
|
|
_, err = bc.Read(buf)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, uint64(4), bc.BytesSent())
|
|
require.Equal(t, uint64(2), bc.BytesReceived())
|
|
}
|