Files
gortsplib/pkg/bytecounter/bytecounter_test.go
Alessandro Ros 95d1562735 server: add methods to get read and written bytes (#147)
ServerConn.ReadBytes()
ServerConn.WrittenBytes()
ServerSession.ReadBytes()
ServerSession.WrittenBytes()
2022-11-09 13:41:48 +01:00

21 lines
341 B
Go

package bytecounter
import (
"bytes"
"testing"
"github.com/stretchr/testify/require"
)
func TestByteCounter(t *testing.T) {
bc := New(bytes.NewBuffer(nil))
bc.Write([]byte{0x01, 0x02, 0x03, 0x04})
buf := make([]byte, 2)
bc.Read(buf)
require.Equal(t, uint64(4), bc.WrittenBytes())
require.Equal(t, uint64(2), bc.ReadBytes())
}