decoders_test.go

package topicreadercommon

import (
	"bytes"
	"compress/gzip"
	"errors"
	"io"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/ydb-platform/ydb-go-sdk/v3/internal/grpcwrapper/rawtopic/rawtopiccommon"
)

func TestDecoderMap(t *testing.T) {
	decoderMap := NewDecoderMap()

	t.Run("DecodeRaw", func(t *testing.T) {
		data := []byte("test data")
		reader := bytes.NewReader(data)

		decodedReader, err := decoderMap.Decode(rawtopiccommon.CodecRaw, reader)
		require.NoError(t, err)

		result, err := io.ReadAll(decodedReader)
		require.NoError(t, err)
		require.Equal(t, data, result)
	})

	t.Run("DecodeGzip", func(t *testing.T) {
		data := []byte("test data")
		var buf bytes.Buffer
		gzipWriter := gzip.NewWriter(&buf)
		_, err := gzipWriter.Write(data)
		require.NoError(t, err)
		require.NoError(t, gzipWriter.Close())

		decodedReader, err := decoderMap.Decode(rawtopiccommon.CodecGzip, &buf)
		require.NoError(t, err)

		result, err := io.ReadAll(decodedReader)
		require.NoError(t, err)
		require.Equal(t, data, result)
	})

	t.Run("DecodeUnknownCodec", func(t *testing.T) {
		_, err := decoderMap.Decode(rawtopiccommon.Codec(999), bytes.NewReader([]byte{}))
		require.Error(t, err)
		require.True(t, errors.Is(err, ErrPublicUnexpectedCodec))
	})

	t.Run("DecodeCustomCodec", func(t *testing.T) {
		dm := NewDecoderMap()
		customCodec := rawtopiccommon.Codec(1001)
		dm.AddDecoder(customCodec, func(input io.Reader) (ReadResetter, error) {
			return gzip.NewReader(input)
		})
		require.Len(t, dm.dp, 3)

		data := []byte("custom test data")
		var buf bytes.Buffer
		gzipWriter := gzip.NewWriter(&buf)
		_, err := gzipWriter.Write(data)
		require.NoError(t, err)
		require.NoError(t, gzipWriter.Close())

		decodedReader, err := dm.Decode(customCodec, &buf)
		require.NoError(t, err)
		defer decodedReader.(io.Closer).Close()

		result, err := io.ReadAll(decodedReader)
		require.NoError(t, err)
		require.Equal(t, string(data), string(result))

		data2 := []byte("second test data")
		var buf2 bytes.Buffer
		gzipWriter2 := gzip.NewWriter(&buf2)
		_, err = gzipWriter2.Write(data2)
		require.NoError(t, err)
		require.NoError(t, gzipWriter2.Close())

		decodedReader2, err := dm.Decode(customCodec, &buf2)
		require.NoError(t, err)
		defer decodedReader2.(io.Closer).Close()

		result2, err := io.ReadAll(decodedReader2)
		require.NoError(t, err)
		require.Equal(t, string(data2), string(result2))
	})
}