frames_test.go 525 B

12345678910111213141516171819202122232425262728
  1. package wire
  2. import (
  3. "bytes"
  4. "io"
  5. "testing"
  6. "github.com/stretchr/testify/assert"
  7. )
  8. func TestFLAPFrame_ReadBody(t *testing.T) {
  9. flap := FLAPFrame{
  10. PayloadLength: 4,
  11. }
  12. bufIn := bytes.NewBuffer([]byte{0, 1, 2, 3, 4, 5})
  13. buf, err := flap.ReadBody(bufIn)
  14. assert.NoError(t, err)
  15. assert.Equal(t, []byte{0, 1, 2, 3}, buf.Bytes())
  16. }
  17. func TestFLAPFrame_ReadBodyError(t *testing.T) {
  18. flap := FLAPFrame{
  19. PayloadLength: 4,
  20. }
  21. bufIn := &bytes.Buffer{}
  22. _, err := flap.ReadBody(bufIn)
  23. assert.ErrorIs(t, err, io.EOF)
  24. }