go-astikit/map_test.go

package astikit

import "testing"

func TestBiMap(t *testing.T) {
	m := NewBiMap()
	m.Set(0, 1)
	v, ok := m.Get(0)
	if !ok {
		t.Fatal("expected true, got false")
	}
	if e, g := 1, v.(int); e != g {
		t.Fatalf("expected %d, got %d", e, g)
	}
	_, ok = m.GetInverse(0)
	if ok {
		t.Fatal("expected false, got true")
	}
	v, ok = m.GetInverse(1)
	if !ok {
		t.Fatal("expected true, got false")
	}
	if e, g := 0, v.(int); e != g {
		t.Fatalf("expected %d, got %d", e, g)
	}
	m.SetInverse(0, 1)
	v, ok = m.GetInverse(0)
	if !ok {
		t.Fatal("expected true, got false")
	}
	if e, g := 1, v.(int); e != g {
		t.Fatalf("expected %d, got %d", e, g)
	}
	testPanic(t, false, func() { m.MustGet(0) })
	testPanic(t, true, func() { m.MustGet(2) })
	testPanic(t, false, func() { m.MustGetInverse(0) })
	testPanic(t, true, func() { m.MustGetInverse(2) })
}

func testPanic(t *testing.T, shouldPanic bool, fn func()) {
	defer func() {
		err := recover()
		if shouldPanic && err == nil {
			t.Fatal("should have panicked")
		} else if !shouldPanic && err != nil {
			t.Fatal("should not have panicked")
		}
	}()
	fn()
}