binary_decoder_positioning_test.go

package avro

import (
	"encoding/hex"
	"fmt"
	"math/rand"
	"testing"
	"time"
)

//this tests whether the decoder is able to sequentially read values and keep track of his position normally
var primitives = []string{type_boolean, type_int, type_long, type_float, type_double, type_bytes, type_string}

func TestPositioning(t *testing.T) {
	bytes, types, expected := getTestData()
	bd := NewBinaryDecoder(bytes)
	for i := 0; i < len(types); i++ {
		currentType := types[i]
		currentExpected := expected[i]

		switch currentType {
		case type_boolean:
			{
				value, _ := bd.ReadBoolean()
				if value != currentExpected.(bool) {
					t.Fatalf("Unexpected boolean: expected %v, actual %v\n", currentExpected, value)
				}
			}
		case type_int:
			{
				value, _ := bd.ReadInt()
				if value != currentExpected.(int32) {
					t.Fatalf("Unexpected int: expected %v, actual %v\n", currentExpected, value)
				}
			}
		case type_long:
			{
				value, _ := bd.ReadLong()
				if value != currentExpected.(int64) {
					t.Fatalf("Unexpected long: expected %v, actual %v\n", currentExpected, value)
				}
			}
		case type_float:
			{
				value, _ := bd.ReadFloat()
				if value != currentExpected.(float32) {
					t.Fatalf("Unexpected float: expected %v, actual %v\n", currentExpected, value)
				}
			}
		case type_double:
			{
				value, _ := bd.ReadDouble()
				if value != currentExpected.(float64) {
					t.Fatalf("Unexpected double: expected %v, actual %v\n", currentExpected, value)
				}
			}
		case type_bytes:
			{
				position := bd.Tell()
				value, err := bd.ReadBytes()
				if err != nil {
					t.Fatal("Unexpected error during decoding bytes: %v", err)
				}
				for i := 0; i < len(value); i++ {
					if value[i] != bytes[position+int64(i)+int64(1)] {
						t.Fatalf("Unexpected byte at index %d: expected 0x%v, actual 0x%v\n", i, hex.EncodeToString([]byte{bytes[i+1]}), hex.EncodeToString([]byte{value[i]}))
					}
				}
			}
		case type_string:
			{
				value, _ := bd.ReadString()
				if value != currentExpected.(string) {
					t.Fatalf("Unexpected string: expected %v, actual %v\n", currentExpected, value)
				}
			}
		}
	}
}

func getTestData() ([]byte, []string, []interface{}) {
	rand.Seed(time.Now().Unix())
	testSize := rand.Intn(10000) + 1
	fmt.Printf("Testing positioning on %d sequential values\n", testSize)

	var bytes []byte
	var types []string
	var expected []interface{}

	for i := 0; i < testSize; i++ {
		currentType := primitives[rand.Intn(len(primitives))]
		types = append(types, currentType)
		k, v := getRandomFromMap(currentType)
		bytes = append(bytes, k...)
		expected = append(expected, v)
	}

	return bytes, types, expected
}

func getRandomFromMap(mapType string) ([]byte, interface{}) {
	i := 0
	switch mapType {
	case type_boolean:
		{
			random := rand.Intn(len(goodBooleans))
			for value, bytes := range goodBooleans {
				if i == random {
					return bytes, value
				}
				i++
			}
		}
	case type_int:
		{
			random := rand.Intn(len(goodInts))
			for value, bytes := range goodInts {
				if i == random {
					return bytes, value
				}
				i++
			}
		}
	case type_long:
		{
			random := rand.Intn(len(goodLongs))
			for value, bytes := range goodLongs {
				if i == random {
					return bytes, value
				}
				i++
			}
		}
	case type_float:
		{
			random := rand.Intn(len(goodFloats))
			for value, bytes := range goodFloats {
				if i == random {
					return bytes, value
				}
				i++
			}
		}
	case type_double:
		{
			random := rand.Intn(len(goodDoubles))
			for value, bytes := range goodDoubles {
				if i == random {
					return bytes, value
				}
				i++
			}
		}
	case type_bytes:
		{
			return goodBytes[rand.Intn(len(goodBytes))], "1"
		}
	case type_string:
		{
			random := rand.Intn(len(goodStrings))
			for value, bytes := range goodStrings {
				if i == random {
					return bytes, value
				}
				i++
			}
		}
	}
	panic("cant get random from map")
}