util_test.go

// Copyright 2015, David Howden
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package tag

import (
	"bytes"
	"testing"
)

func TestGetBit(t *testing.T) {
	for i := uint(0); i < 8; i++ {
		b := byte(1 << i)
		got := getBit(b, i)
		if !got {
			t.Errorf("getBit(%v, %v) = %v, expected %v", b, i, got, true)
		}
	}
}

func TestGet7BitChunkedInt(t *testing.T) {
	tests := []struct {
		input  []byte
		output int
	}{
		{
			[]byte{},
			0,
		},
		{
			[]byte{0x01},
			1,
		},
		{
			[]byte{0x7F, 0x7F},
			0x3FFF,
		},
	}

	for ii, tt := range tests {
		got := get7BitChunkedInt(tt.input)
		if got != tt.output {
			t.Errorf("[%d] get7BitChunkedInt(%v) = %v, expected %v", ii, tt.input, got, tt.output)
		}
	}
}

func TestGetInt(t *testing.T) {
	tests := []struct {
		input  []byte
		output int
	}{
		{
			[]byte{},
			0,
		},
		{
			[]byte{0x01},
			1,
		},
		{
			[]byte{0xF1, 0xF2},
			0xF1F2,
		},
		{
			[]byte{0xF1, 0xF2, 0xF3},
			0xF1F2F3,
		},
		{
			[]byte{0xF1, 0xF2, 0xF3, 0xF4},
			0xF1F2F3F4,
		},
	}

	for ii, tt := range tests {
		got := getInt(tt.input)
		if got != tt.output {
			t.Errorf("[%d] getInt(%v) = %v, expected %v", ii, tt.input, got, tt.output)
		}
	}
}

func TestGetUintLittleEndian(t *testing.T) {
	tests := []struct {
		input  []byte
		output uint64
	}{
		{
			[]byte{0, 0, 0, 0, 0, 0, 0, 0},
			0,
		},
		{
			[]byte{0x01, 0, 0, 0, 0, 0, 0, 0},
			1,
		},
		{
			[]byte{0xF1, 0xF2, 0, 0, 0, 0, 0, 0},
			0xF2F1,
		},
		{
			[]byte{0xF1, 0xF2, 0xF3, 0, 0, 0, 0, 0},
			0xF3F2F1,
		},
		{
			[]byte{0xF1, 0xF2, 0xF3, 0xF4, 0, 0, 0, 0},
			0xF4F3F2F1,
		},
	}

	for ii, tt := range tests {
		got, err := readUint64LittleEndian(bytes.NewReader(tt.input))
		if err != nil {
			t.Errorf("unexpected error: %v", err)
		}
		if got != tt.output {
			t.Errorf("[%d] getInt(%v) = %v, expected %v", ii, tt.input, got, tt.output)
		}
	}
}