package frame

import (
	"io/ioutil"
	"os"
	"testing"
)

func TestFDEForPC(t *testing.T) {
	fde1 := &FrameDescriptionEntry{AddressRange: &addrange{begin: 100, end: 200}}
	fde2 := &FrameDescriptionEntry{AddressRange: &addrange{begin: 50, end: 99}}
	fde3 := &FrameDescriptionEntry{AddressRange: &addrange{begin: 0, end: 49}}
	fde4 := &FrameDescriptionEntry{AddressRange: &addrange{begin: 201, end: 245}}

	tree := NewFrameIndex()
	tree.Put(fde1)
	tree.Put(fde2)
	tree.Put(fde3)
	tree.Put(fde4)

	node, ok := tree.Find(Addr(35))
	if !ok {
		t.Fatal("Could not find FDE")
	}

	if node != fde3 {
		t.Fatal("Got incorrect fde")
	}
}

func BenchmarkFDEForPC(b *testing.B) {
	f, err := os.Open("testdata/frame")
	if err != nil {
		b.Fatal(err)
	}
	defer f.Close()

	data, err := ioutil.ReadAll(f)
	if err != nil {
		b.Fatal(err)
	}
	fdes := Parse(data)

	for i := 0; i < b.N; i++ {
		// bench worst case, exhaustive search
		_, _ = fdes.FDEForPC(0x455555555)
	}
}