reserved_bytes_test.go

package shares

import (
	"testing"

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

func TestParseReservedBytes(t *testing.T) {
	type testCase struct {
		name      string
		input     []byte
		want      uint32
		expectErr bool
	}
	testCases := []testCase{
		{"byte index of 0", []byte{0, 0, 0, 0}, 0, false},
		{"byte index of 2", []byte{0, 0, 0, 2}, 2, false},
		{"byte index of 4", []byte{0, 0, 0, 4}, 4, false},
		{"byte index of 8", []byte{0, 0, 0, 8}, 8, false},
		{"byte index of 16", []byte{0, 0, 0, 16}, 16, false},
		{"byte index of 32", []byte{0, 0, 0, 32}, 32, false},
		{"byte index of 64", []byte{0, 0, 0, 64}, 64, false},
		{"byte index of 128", []byte{0, 0, 0, 128}, 128, false},
		{"byte index of 256", []byte{0, 0, 1, 0}, 256, false},
		{"byte index of 511", []byte{0, 0, 1, 255}, 511, false},

		// error cases
		{"empty", []byte{}, 0, true},
		{"too few reserved bytes", []byte{1}, 0, true},
		{"another case of too few reserved bytes", []byte{3, 3, 3}, 0, true},
		{"too many bytes", []byte{0, 0, 0, 0, 0}, 0, true},
		{"too high of a byte index", []byte{0, 0, 3, 232}, 0, true},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			got, err := ParseReservedBytes(tc.input)
			if tc.expectErr {
				assert.Error(t, err)
				return
			}
			assert.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}

func TestNewReservedBytes(t *testing.T) {
	type testCase struct {
		name      string
		input     uint32
		want      []byte
		expectErr bool
	}
	testCases := []testCase{
		{"byte index of 0", 0, []byte{0, 0, 0, 0}, false},
		{"byte index of 2", 2, []byte{0, 0, 0, 2}, false},
		{"byte index of 4", 4, []byte{0, 0, 0, 4}, false},
		{"byte index of 8", 8, []byte{0, 0, 0, 8}, false},
		{"byte index of 16", 16, []byte{0, 0, 0, 16}, false},
		{"byte index of 32", 32, []byte{0, 0, 0, 32}, false},
		{"byte index of 64", 64, []byte{0, 0, 0, 64}, false},
		{"byte index of 128", 128, []byte{0, 0, 0, 128}, false},
		{"byte index of 256", 256, []byte{0, 0, 1, 0}, false},
		{"byte index of 511", 511, []byte{0, 0, 1, 255}, false},

		// error cases
		{"byte index of 512 is equal to share size", 512, []byte{}, true},
		{"byte index of 1000 is greater than share size", 1000, []byte{}, true},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			got, err := NewReservedBytes(tc.input)
			if tc.expectErr {
				assert.Error(t, err)
				return
			}
			assert.NoError(t, err)
			assert.Equal(t, tc.want, got)
		})
	}
}