package principal

import (
	"reflect"
	"testing"
)

func TestJSBlessingStore(t *testing.T) {
	s := NewJSBlessingsHandles()
	b := blessSelf(newPrincipal(), "irrelevant")

	h := s.Add(b)
	if got := s.Get(h); !reflect.DeepEqual(got, b) {
		t.Fatalf("Get after adding: got: %v, want: %v", got, b)
	}

	s.Remove(h)
	if got := s.Get(h); !got.IsZero() {
		t.Fatalf("Get after removing: got: %v, want nil", got)
	}
}