mediamtx/internal/rtmp/bytecounter/writer_test.go
Alessandro Ros 8bee4af86a
api, metrics: add number of bytes received and sent from/to all entities (#1235)
* API: number of bytes received/sent from/to RTSP connections
* API: number of bytes received/sent from/to RTSP sessions
* API: number of bytes received/sent from/to RTMP connections
* API: number of bytes sent to HLS connections
* API: number of bytes received from paths
* metrics of all the above
2022-11-11 11:59:52 +01:00

19 lines
276 B
Go

package bytecounter
import (
"bytes"
"testing"
"github.com/stretchr/testify/require"
)
func TestWriter(t *testing.T) {
var buf bytes.Buffer
w := NewWriter(&buf)
w.SetCount(100)
w.Write(bytes.Repeat([]byte{0x01}, 64))
require.Equal(t, uint64(100+64), w.Count())
}