tenant-registry/internal/server/tenants_test.go

package server_test

import (
	"net/http"
	"testing"
	"time"

	"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)
		}
	})
}

func TestCreateTenant_setsTrialEndsAt(t *testing.T) {
	eachStore(t, func(t *testing.T, h *testHarness) {
		_, body := h.do("POST", "/v1/tenants", map[string]any{
			"slug": "trial-ends-co", "name": "Trial Ends Co.",
		})
		out := decode[struct {
			Tenant *store.Tenant `json:"tenant"`
		}](t, body)
		if out.Tenant.Status != "trial" {
			t.Fatalf("status = %q, want trial", out.Tenant.Status)
		}
		if out.Tenant.TrialEndsAt == nil {
			t.Fatal("trial_ends_at is nil; should be ~14 days from now")
		}
		// Sanity-check: ends_at is in the future, within 13.5-14.5 days.
		delta := time.Until(*out.Tenant.TrialEndsAt)
		if delta < 13*24*time.Hour || delta > 15*24*time.Hour {
			t.Errorf("trial_ends_at offset = %v, want ~14d", delta)
		}
	})
}

func TestCreateTenant_demoKindHasNoTrialEnd(t *testing.T) {
	eachStore(t, func(t *testing.T, h *testHarness) {
		_, body := h.do("POST", "/v1/tenants", map[string]any{
			"slug": "demo-co", "name": "Demo", "kind": "demo",
		})
		out := decode[struct {
			Tenant *store.Tenant `json:"tenant"`
		}](t, body)
		if out.Tenant.Status != "demo" {
			t.Errorf("status = %q, want demo", out.Tenant.Status)
		}
		if out.Tenant.TrialEndsAt != nil {
			t.Errorf("trial_ends_at = %v, want nil for demo kind", out.Tenant.TrialEndsAt)
		}
	})
}