30 lines
1.1 KiB
Go
30 lines
1.1 KiB
Go
|
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)
|
||
|
|
}
|
||
|
|
}
|