package registry

import (
	"context"
	"errors"
	"testing"
)

func TestStatic_LookupBothWays(t *testing.T) {
	s := NewStatic([]Agent{
		{AgentID: "agent-approval-service", PocketIDClient: "agent-approval-service", K8sNamespace: "agent-trust", K8sServiceAccount: "approval-service"},
		{AgentID: "agent-testagent-transform", PocketIDClient: "agent-testagent-transform", K8sNamespace: "agent-trust", K8sServiceAccount: "testagent-transform"},
	})
	ctx := context.Background()
	got, err := s.ByPocketID(ctx, "agent-approval-service")
	if err != nil || got.AgentID != "agent-approval-service" {
		t.Fatalf("ByPocketID: got %+v, err %v", got, err)
	}
	got, err = s.ByK8sIdentity(ctx, "agent-trust", "testagent-transform")
	if err != nil || got.AgentID != "agent-testagent-transform" {
		t.Fatalf("ByK8sIdentity: got %+v, err %v", got, err)
	}
	if _, err := s.ByPocketID(ctx, "nope"); !errors.Is(err, ErrNotFound) {
		t.Fatalf("expected ErrNotFound, got %v", err)
	}
	if _, err := s.ByK8sIdentity(ctx, "agent-trust", "nope"); !errors.Is(err, ErrNotFound) {
		t.Fatalf("expected ErrNotFound, got %v", err)
	}
}