sharang
fd5f8ae36f
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
122 lines
3.3 KiB
Go
122 lines
3.3 KiB
Go
package server_test
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"gitea.meghsakha.com/platform/tenant-registry/internal/store"
|
|
)
|
|
|
|
func TestHealthz(t *testing.T) {
|
|
eachStore(t, func(t *testing.T, h *testHarness) {
|
|
resp, _ := h.do("GET", "/healthz", nil)
|
|
if resp.StatusCode != 200 {
|
|
t.Fatalf("status = %d", resp.StatusCode)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestCreateTenant(t *testing.T) {
|
|
eachStore(t, func(t *testing.T, h *testHarness) {
|
|
resp, body := h.do("POST", "/v1/tenants", map[string]any{
|
|
"slug": "beta-co", "name": "Beta Co.", "plan": "starter",
|
|
})
|
|
if resp.StatusCode != http.StatusCreated {
|
|
t.Fatalf("status = %d, body=%s", resp.StatusCode, body)
|
|
}
|
|
out := decode[struct {
|
|
Tenant *store.Tenant `json:"tenant"`
|
|
InviteURL string `json:"invite_url"`
|
|
}](t, body)
|
|
if out.Tenant.Slug != "beta-co" || out.Tenant.Status != "trial" || out.Tenant.Plan != "starter" {
|
|
t.Errorf("unexpected: %+v", out.Tenant)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestCreateTenant_invalidSlug(t *testing.T) {
|
|
eachStore(t, func(t *testing.T, h *testHarness) {
|
|
resp, _ := h.do("POST", "/v1/tenants", map[string]any{
|
|
"slug": "X", "name": "Bad",
|
|
})
|
|
if resp.StatusCode != http.StatusBadRequest {
|
|
t.Fatalf("status = %d", resp.StatusCode)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestCreateTenant_duplicateSlugConflict(t *testing.T) {
|
|
eachStore(t, func(t *testing.T, h *testHarness) {
|
|
// 'acme' is pre-seeded
|
|
resp, _ := h.do("POST", "/v1/tenants", map[string]any{
|
|
"slug": "acme", "name": "Dup",
|
|
})
|
|
if resp.StatusCode != http.StatusConflict {
|
|
t.Fatalf("status = %d", resp.StatusCode)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestGetTenantBySlug(t *testing.T) {
|
|
eachStore(t, func(t *testing.T, h *testHarness) {
|
|
resp, body := h.do("GET", "/v1/tenants/by-slug/acme", nil)
|
|
if resp.StatusCode != 200 {
|
|
t.Fatalf("status = %d", resp.StatusCode)
|
|
}
|
|
got := decode[store.Tenant](t, body)
|
|
if got.Slug != "acme" {
|
|
t.Errorf("slug = %q", got.Slug)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestGetTenantBySlug_notFound(t *testing.T) {
|
|
eachStore(t, func(t *testing.T, h *testHarness) {
|
|
resp, _ := h.do("GET", "/v1/tenants/by-slug/nope-nope", nil)
|
|
if resp.StatusCode != http.StatusNotFound {
|
|
t.Fatalf("status = %d", resp.StatusCode)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestActivateTenant(t *testing.T) {
|
|
eachStore(t, func(t *testing.T, h *testHarness) {
|
|
_, body := h.do("POST", "/v1/tenants", map[string]any{
|
|
"slug": "trial-co", "name": "Trial Co.",
|
|
})
|
|
createdWrap := decode[struct {
|
|
Tenant *store.Tenant `json:"tenant"`
|
|
}](t, body)
|
|
created := createdWrap.Tenant
|
|
if created.Status != "trial" {
|
|
t.Fatalf("precondition: %q", created.Status)
|
|
}
|
|
|
|
resp, body := h.do("POST", "/v1/tenants/"+created.ID+"/activate", map[string]any{
|
|
"plan": "professional", "erp_customer_id": "ERP-001",
|
|
})
|
|
if resp.StatusCode != 200 {
|
|
t.Fatalf("status = %d, body=%s", resp.StatusCode, body)
|
|
}
|
|
got := decode[store.Tenant](t, body)
|
|
if got.Status != "active" || got.Plan != "professional" || got.ErpCustomerID != "ERP-001" {
|
|
t.Errorf("unexpected: %+v", got)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestCancelTenant(t *testing.T) {
|
|
eachStore(t, func(t *testing.T, h *testHarness) {
|
|
resp, body := h.do("POST", "/v1/tenants/"+h.tenant.ID+"/cancel", map[string]any{
|
|
"reason": "test", "at_period_end": true,
|
|
})
|
|
if resp.StatusCode != 200 {
|
|
t.Fatalf("status = %d, body=%s", resp.StatusCode, body)
|
|
}
|
|
got := decode[store.Tenant](t, body)
|
|
if got.Status != "frozen" {
|
|
t.Errorf("status = %q", got.Status)
|
|
}
|
|
})
|
|
}
|