sharang
bb2c638fb4
internal/keycloak/ — Adapter interface with two implementations:
HTTPAdapter pgxpool-style real Admin API client with cached client-
credentials token (auto-refresh, 401 retry).
Mock in-process map for unit tests + dev convenience when
KEYCLOAK_ADMIN_URL is empty. Used by the eachStore harness.
Adapter contract (adapter.go):
CreateOrgAndInvite(ctx, InviteInput) (*InviteResult, error)
Creates a KC organization, an IT_ADMIN user, adds the user as a
member, triggers VERIFY_EMAIL + UPDATE_PASSWORD execute-actions
email. Atomic from the caller's PoV; partial failures surface as
typed errors (ErrOrgConflict, ErrUserConflict, ErrUnauthorized,
ErrUnavailable).
SyncClaims(ctx, userID, Claims) error
Pushes tenant_id / tenant_slug / org_roles / products / plan /
tenant_status into the user's KC attributes — the same shape the
realm's protocol mappers project into JWTs.
Health(ctx) error
Pings /admin/serverinfo; wired into readyz.
Wiring:
POST /v1/tenants now accepts admin_email + admin_name. When set, the
adapter creates the org and invites the user. Response wraps the
tenant with the new TenantCreated{tenant, invite_url} shape so dev
testers can use the action-token URL without waiting for the email.
KC failures DO NOT roll the tenant back — they emit a
keycloak.provision_failed audit event so the operator can resend.
Successful invites emit keycloak.invite_sent.
POST /v1/internal/keycloak/claims resolves a tenant's current claim
bundle. Lookup chain: body.tenant_id → body.tenant_slug →
body.user_attrs.tenant_id → body.user_attrs.tenant_slug. The realm's
protocol mapper calls this at token issuance, or operators on demand.
Config: KEYCLOAK_ADMIN_URL / REALM / CLIENT_ID / CLIENT_SECRET; empty
URL falls back to Mock for dev.
OpenAPI: TenantCreated + Claims schemas added; /v1/internal/keycloak/claims
documented. Contract test extended to cover the new endpoint.
Tests:
internal/keycloak/mock_test.go Mock semantics: conflict surfacing,
FailNext hook, SyncClaims persistence.
internal/server/keycloak_test.go KC provisioning end-to-end via
eachStore: invite_url returned,
mock records, invite_sent audit;
failure path emits provision_failed
but tenant still lands; claims
endpoint resolves via tenant_id /
tenant_slug / user_attrs / 404 / 400.
The real-KC integration test (against a testcontainers-spun KC 26)
lands in a follow-up — gating it behind KEYCLOAK_INTEGRATION=1 + a
slower nightly CI is cleaner than baking 30s+ of KC boot into every PR.
Refs: M4.3
85 lines
2.3 KiB
Go
85 lines
2.3 KiB
Go
package keycloak
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
)
|
|
|
|
func TestMock_createOrgAndInvite(t *testing.T) {
|
|
m := NewMock()
|
|
ctx := context.Background()
|
|
|
|
res, err := m.CreateOrgAndInvite(ctx, InviteInput{
|
|
TenantID: "t1", Slug: "acme", Name: "Acme",
|
|
AdminEmail: "a@acme.test", AdminName: "Alice",
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res.OrganizationID == "" || res.UserID == "" {
|
|
t.Errorf("ids missing: %+v", res)
|
|
}
|
|
if m.Orgs["t1"] != res.OrganizationID {
|
|
t.Errorf("Orgs map not updated")
|
|
}
|
|
if m.Users["a@acme.test"] != res.UserID {
|
|
t.Errorf("Users map not updated")
|
|
}
|
|
}
|
|
|
|
func TestMock_orgConflict(t *testing.T) {
|
|
m := NewMock()
|
|
ctx := context.Background()
|
|
_, _ = m.CreateOrgAndInvite(ctx, InviteInput{TenantID: "t1", Slug: "x", AdminEmail: "a@y.test"})
|
|
_, err := m.CreateOrgAndInvite(ctx, InviteInput{TenantID: "t1", Slug: "x", AdminEmail: "b@y.test"})
|
|
if !errors.Is(err, ErrOrgConflict) {
|
|
t.Errorf("err = %v, want ErrOrgConflict", err)
|
|
}
|
|
}
|
|
|
|
func TestMock_userConflict(t *testing.T) {
|
|
m := NewMock()
|
|
ctx := context.Background()
|
|
_, _ = m.CreateOrgAndInvite(ctx, InviteInput{TenantID: "t1", Slug: "x", AdminEmail: "a@y.test"})
|
|
_, err := m.CreateOrgAndInvite(ctx, InviteInput{TenantID: "t2", Slug: "z", AdminEmail: "a@y.test"})
|
|
if !errors.Is(err, ErrUserConflict) {
|
|
t.Errorf("err = %v, want ErrUserConflict", err)
|
|
}
|
|
}
|
|
|
|
func TestMock_failNextHook(t *testing.T) {
|
|
m := NewMock()
|
|
m.FailNext = ErrUnavailable
|
|
_, err := m.CreateOrgAndInvite(context.Background(), InviteInput{TenantID: "t1", Slug: "x", AdminEmail: "a@y.test"})
|
|
if !errors.Is(err, ErrUnavailable) {
|
|
t.Errorf("err = %v, want ErrUnavailable", err)
|
|
}
|
|
// Subsequent call recovers
|
|
_, err = m.CreateOrgAndInvite(context.Background(), InviteInput{TenantID: "t1", Slug: "x", AdminEmail: "a@y.test"})
|
|
if err != nil {
|
|
t.Errorf("FailNext should clear after one use; err=%v", err)
|
|
}
|
|
}
|
|
|
|
func TestMock_syncClaims(t *testing.T) {
|
|
m := NewMock()
|
|
err := m.SyncClaims(context.Background(), "user-1", Claims{
|
|
TenantID: "t1", Plan: "professional", Products: []string{"certifai"},
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if m.Claims["user-1"].Plan != "professional" {
|
|
t.Errorf("claims not stored")
|
|
}
|
|
}
|
|
|
|
func TestMock_syncClaimsRequiresUserID(t *testing.T) {
|
|
m := NewMock()
|
|
err := m.SyncClaims(context.Background(), "", Claims{})
|
|
if err == nil {
|
|
t.Error("expected error for empty user id")
|
|
}
|
|
}
|