byte_encoding_test.go

package encoding

import (
	"bytes"
	"errors"
	"io"

	"github.com/stretchr/testify/suite"
)

func NewByteEncodingSuite(e ByteEncoding) *ByteEncodingSuite {
	return &ByteEncodingSuite{
		byteArrayEncoding:  e,
		byteStreamEncoding: e,
	}
}

type ByteEncodingSuite struct {
	suite.Suite
	byteArrayEncoding  ByteArrayEncoding
	byteStreamEncoding ByteStreamEncoding
}

func (suite ByteEncodingSuite) TestByteArray() {
	if suite.byteArrayEncoding == nil {
		suite.T().Skip()
	}

	tests := [][]byte{
		{},
		{0},
		{0, 0},
		[]byte("123"),
		[]byte(`"123"`),
	}

	for _, input := range tests {
		suite.Run(string(input), func() {
			enc, err := suite.byteArrayEncoding.Encode(input)
			suite.Require().NoError(err)

			dec, err := suite.byteArrayEncoding.Decode(enc)
			suite.Require().NoError(err)

			suite.Require().Equal(input, dec)
		})
	}
}

var _ io.ReadWriteCloser = (*closeableBuffer)(nil)

type closeableBuffer struct {
	bytes.Buffer
	closed bool
}

func (c *closeableBuffer) Close() error {
	if c.closed {
		return errors.New("already closed")
	}
	c.closed = true
	return nil
}

func (suite ByteEncodingSuite) TestByteStream() {
	if suite.byteStreamEncoding == nil {
		suite.T().Skip()
	}

	tests := [][]byte{
		{},
		{0},
		{0, 0},
		[]byte("123"),
		[]byte(`"123"`),
	}

	for _, input := range tests {
		suite.Run(string(input), func() {
			var buf closeableBuffer

			w, err := suite.byteStreamEncoding.StreamEncode(&buf)
			suite.Require().NoError(err)

			_, err = w.Write(input)
			suite.Require().NoError(err)

			err = w.Close()
			suite.Require().NoError(err)

			r, err := suite.byteStreamEncoding.StreamDecode(&buf)
			suite.Require().NoError(err)

			dec, err := readAllClose(r)
			suite.Require().NoError(err)

			suite.Require().Equal(input, dec)

			suite.Require().False(buf.closed)

			suite.Require().NoError(buf.Close())
			suite.Require().True(buf.closed)
		})
	}
}