mirror of
https://github.com/aler9/rtsp-simple-server
synced 2025-10-05 15:46:58 +08:00
24 lines
465 B
Go
24 lines
465 B
Go
package message
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/bluenviron/mediamtx/internal/protocols/rtmp/bytecounter"
|
|
)
|
|
|
|
func TestWriter(t *testing.T) {
|
|
for _, ca := range readWriterCases {
|
|
t.Run(ca.name, func(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
bc := bytecounter.NewWriter(&buf)
|
|
r := NewWriter(bc, bc, true)
|
|
err := r.Write(ca.dec)
|
|
require.NoError(t, err)
|
|
require.Equal(t, ca.enc, buf.Bytes())
|
|
})
|
|
}
|
|
}
|