diff --git a/internal/rtmp/bytecounter/reader_test.go b/internal/rtmp/bytecounter/reader_test.go new file mode 100644 index 00000000..6778f375 --- /dev/null +++ b/internal/rtmp/bytecounter/reader_test.go @@ -0,0 +1,21 @@ +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) + buf2 := make([]byte, 64) + n, err := r.Read(buf2) + require.NoError(t, err) + require.Equal(t, 64, n) + + require.Equal(t, uint32(1024), r.Count()) +} diff --git a/internal/rtmp/bytecounter/writer_test.go b/internal/rtmp/bytecounter/writer_test.go new file mode 100644 index 00000000..b2ecae4d --- /dev/null +++ b/internal/rtmp/bytecounter/writer_test.go @@ -0,0 +1,15 @@ +package bytecounter + +import ( + "bytes" + "testing" + + "github.com/stretchr/testify/require" +) + +func TestWriter(t *testing.T) { + var buf bytes.Buffer + w := NewWriter(&buf) + w.Write(bytes.Repeat([]byte{0x01}, 64)) + require.Equal(t, uint32(64), w.Count()) +}