meregion_test.go

// Copyright 2019 the LinuxBoot Authors. All rights reserved
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package uefi

import (
	"reflect"
	"testing"
)

func TestMEName_MarshalText(t *testing.T) {
	var tests = []struct {
		name string
		me   MEName
	}{
		{"NAME", MEName{'N', 'A', 'M', 'E'}},
		{"NAM", MEName{'N', 'A', 'M', 0}},
		{"NA", MEName{'N', 'A', 0, 0}},
		{"N", MEName{'N', 0, 0, 0}},
	}
	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			b, err := test.me.MarshalText()
			if err != nil {
				t.Errorf("Unexpected error %v", err)
			}
			if string(b) != test.name {
				t.Errorf("error got %q want %q", b, test.name)
			}
		})
	}
}

func TestMEName_UnmarshalText(t *testing.T) {
	var tests = []struct {
		name string
		me   MEName
		msg  string
	}{
		{"NAME", MEName{'N', 'A', 'M', 'E'}, ""},
		{"NAM", MEName{'N', 'A', 'M', 0}, ""},
		{"NA", MEName{'N', 'A', 0, 0}, ""},
		{"N", MEName{'N', 0, 0, 0}, ""},
		{"NAME1", MEName{'N', 'A', 'M', 'E'}, "can’t unmarshal \"NAME1\" to MEName, 5 > 4"},
	}
	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			me := MEName{'F', 'U', 'L', 'L'}
			err := me.UnmarshalText([]byte(test.name))
			if err == nil && test.msg != "" {
				t.Errorf("Error was not returned, expected %v", test.msg)
			} else if err != nil && err.Error() != test.msg {
				t.Errorf("Mismatched Error returned, expected \n%v\n got \n%v\n", test.msg, err.Error())
			} else if !reflect.DeepEqual(me, test.me) {
				t.Errorf("error got %q want %q", me, test.me)
			}

		})
	}
}

func TestFindFPTSignature(t *testing.T) {
	var empty16 = make([]byte, 16)
	var empty12 = make([]byte, 12)
	var empty = make([]byte, 128)

	var firstRow = append(MEFPTSignature, empty12...)
	var secondRow = append(empty16, firstRow...)
	var elsewhere = append(empty, firstRow...)

	var tests = []struct {
		name string
		blob []byte
		res  int
	}{
		{"beginning", firstRow, 4},
		{"2nd row", secondRow, 20},
		{"elsewhere", elsewhere, 132},
		{"nowhere", empty, -1},
	}
	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			r, e := FindMEDescriptor(test.blob)
			if r != test.res {
				t.Errorf("got position %d want %d (%q)", r, test.res, e)
			}
			if test.res == -1 && e == nil {
				t.Errorf("expected error")
			}
		})
	}
}