fast_node_test.go

package iavl

import (
	"bytes"
	"encoding/hex"
	"testing"

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

func TestFastNode_encodedSize(t *testing.T) {
	fastNode := &FastNode{
		key:                  randBytes(10),
		versionLastUpdatedAt: 1,
		value:                randBytes(20),
	}

	expectedSize := 1 + len(fastNode.value) + 1

	require.Equal(t, expectedSize, fastNode.encodedSize())
}

func TestFastNode_encode_decode(t *testing.T) {
	testcases := map[string]struct {
		node        *FastNode
		expectHex   string
		expectError bool
	}{
		"nil":   {nil, "", true},
		"empty": {&FastNode{}, "0000", false},
		"inner": {&FastNode{
			key:                  []byte{0x4},
			versionLastUpdatedAt: 1,
			value:                []byte{0x2},
		}, "020102", false},
	}
	for name, tc := range testcases {
		tc := tc
		t.Run(name, func(t *testing.T) {
			var buf bytes.Buffer
			err := tc.node.writeBytes(&buf)
			if tc.expectError {
				require.Error(t, err)
				return
			}
			require.NoError(t, err)
			require.Equal(t, tc.expectHex, hex.EncodeToString(buf.Bytes()))

			node, err := DeserializeFastNode(tc.node.key, buf.Bytes())
			require.NoError(t, err)
			// since value and leafHash are always decoded to []byte{} we augment the expected struct here
			if tc.node.value == nil {
				tc.node.value = []byte{}
			}
			require.Equal(t, tc.node, node)
		})
	}
}