package handler import ( "context" "encoding/json" "log/slog" "net/http" "net/http/httptest" "os" "path/filepath" "strings" "testing" "codeberg.org/VARASYS/ZDDC/zddc/internal/config" "codeberg.org/VARASYS/ZDDC/zddc/internal/zddc" ) // profileTestRoot creates a temp dir, writes a .zddc with the given admins // list, and returns a Config pointing at it. func profileTestRoot(t *testing.T, admins []string) (config.Config, *LogRing) { t.Helper() root := t.TempDir() if len(admins) > 0 { var b strings.Builder b.WriteString("admins:\n") for _, a := range admins { b.WriteString(" - \"") b.WriteString(a) b.WriteString("\"\n") } if err := os.WriteFile(filepath.Join(root, ".zddc"), []byte(b.String()), 0o644); err != nil { t.Fatalf("write .zddc: %v", err) } zddc.InvalidateCache(root) } return config.Config{ Root: root, Addr: ":8443", EmailHeader: "X-Auth-Request-Email", }, NewLogRing(50) } // requestWithEmail builds a request whose context already carries email (as // the real ACLMiddleware would inject) and whose path is path. func requestWithEmail(method, path, email string) *http.Request { r := httptest.NewRequest(method, path, nil) if email != "" { r.Header.Set("X-Auth-Request-Email", email) ctx := context.WithValue(r.Context(), EmailKey, email) r = r.WithContext(ctx) } return r } // TestServeProfileGateMatrix checks the authorization decisions for every // sub-route. The page itself (/.profile/) is reachable to anyone (anonymous // included); admin-only sub-resources stay 404 for non-eligible callers, // preserving the existence-leakage policy on a per-resource basis. func TestServeProfileGateMatrix(t *testing.T) { cfg, ring := profileTestRoot(t, []string{"alice@example.com"}) cases := []struct { name string path string email string wantStatus int }{ // /.profile/ itself — public landing for everyone. {"anonymous /.profile/", "/.profile/", "", http.StatusOK}, {"non-admin /.profile/", "/.profile/", "bob@example.com", http.StatusOK}, {"admin /.profile/", "/.profile/", "alice@example.com", http.StatusOK}, // /.profile/access — JSON, also public. {"anonymous /.profile/access", "/.profile/access", "", http.StatusOK}, {"admin /.profile/access", "/.profile/access", "alice@example.com", http.StatusOK}, // Admin-only sub-resources — 404 for non-eligible callers. {"anonymous /.profile/whoami", "/.profile/whoami", "", http.StatusNotFound}, {"anonymous /.profile/config", "/.profile/config", "", http.StatusNotFound}, {"anonymous /.profile/logs", "/.profile/logs", "", http.StatusNotFound}, {"non-admin /.profile/whoami", "/.profile/whoami", "bob@example.com", http.StatusNotFound}, {"non-admin /.profile/config", "/.profile/config", "bob@example.com", http.StatusNotFound}, {"non-admin /.profile/logs", "/.profile/logs", "bob@example.com", http.StatusNotFound}, {"admin /.profile/whoami", "/.profile/whoami", "alice@example.com", http.StatusOK}, {"admin /.profile/config", "/.profile/config", "alice@example.com", http.StatusOK}, {"admin /.profile/logs", "/.profile/logs", "alice@example.com", http.StatusOK}, // effective-policy is admin-only too. With no params an admin // gets 400 (bad request), confirming the gate cleared. Same // 404 for non-admins as the other admin-only routes. {"anonymous /.profile/effective-policy", "/.profile/effective-policy", "", http.StatusNotFound}, {"non-admin /.profile/effective-policy", "/.profile/effective-policy", "bob@example.com", http.StatusNotFound}, {"admin /.profile/effective-policy without params", "/.profile/effective-policy", "alice@example.com", http.StatusBadRequest}, // Unknown sub-route still 404. {"admin unknown subroute", "/.profile/nope", "alice@example.com", http.StatusNotFound}, } for _, tc := range cases { t.Run(tc.name, func(t *testing.T) { rec := httptest.NewRecorder() ServeProfile(cfg, ring, rec, requestWithEmail(http.MethodGet, tc.path, tc.email)) if rec.Code != tc.wantStatus { t.Errorf("status = %d, want %d (body: %s)", rec.Code, tc.wantStatus, rec.Body.String()) } }) } } func TestServeProfileWhoamiPayload(t *testing.T) { cfg, ring := profileTestRoot(t, []string{"alice@example.com"}) rec := httptest.NewRecorder() r := requestWithEmail(http.MethodGet, "/.profile/whoami", "alice@example.com") r.Header.Set("X-Other-Header", "hi there") ServeProfile(cfg, ring, rec, r) if rec.Code != 200 { t.Fatalf("status = %d", rec.Code) } if ct := rec.Header().Get("Content-Type"); !strings.HasPrefix(ct, "application/json") { t.Errorf("Content-Type = %q, want application/json", ct) } var got map[string]any if err := json.Unmarshal(rec.Body.Bytes(), &got); err != nil { t.Fatalf("invalid JSON: %v\n%s", err, rec.Body.String()) } if got["configured_email_header"] != "X-Auth-Request-Email" { t.Errorf("configured_email_header = %v", got["configured_email_header"]) } if got["observed_email"] != "alice@example.com" { t.Errorf("observed_email = %v", got["observed_email"]) } if got["resolved_email"] != "alice@example.com" { t.Errorf("resolved_email = %v", got["resolved_email"]) } headers, _ := got["headers"].(map[string]any) if _, ok := headers["X-Auth-Request-Email"]; !ok { t.Errorf("headers map missing X-Auth-Request-Email: %+v", headers) } if _, ok := headers["X-Other-Header"]; !ok { t.Errorf("headers map missing X-Other-Header: %+v", headers) } } func TestServeProfileConfigPayload(t *testing.T) { cfg, ring := profileTestRoot(t, []string{"alice@example.com"}) cfg.LogLevel = "info" cfg.IndexPath = ".archive" cfg.CORSOrigins = []string{"https://zddc.varasys.io"} rec := httptest.NewRecorder() ServeProfile(cfg, ring, rec, requestWithEmail(http.MethodGet, "/.profile/config", "alice@example.com")) if rec.Code != 200 { t.Fatalf("status = %d", rec.Code) } var got map[string]any if err := json.Unmarshal(rec.Body.Bytes(), &got); err != nil { t.Fatalf("invalid JSON: %v", err) } for _, want := range []string{"root", "addr", "email_header", "log_level", "cors_origins"} { if _, ok := got[want]; !ok { t.Errorf("config payload missing key %q: %+v", want, got) } } if got["email_header"] != "X-Auth-Request-Email" { t.Errorf("email_header = %v", got["email_header"]) } } func TestServeProfileLogsPayload(t *testing.T) { cfg, ring := profileTestRoot(t, []string{"alice@example.com"}) rh := NewRingHandler(ring, slog.LevelDebug) logger := slog.New(rh) logger.Info("first") logger.Warn("second", "code", 42) rec := httptest.NewRecorder() ServeProfile(cfg, ring, rec, requestWithEmail(http.MethodGet, "/.profile/logs", "alice@example.com")) if rec.Code != 200 { t.Fatalf("status = %d", rec.Code) } var got []map[string]any if err := json.Unmarshal(rec.Body.Bytes(), &got); err != nil { t.Fatalf("invalid JSON: %v\n%s", err, rec.Body.String()) } if len(got) != 2 { t.Fatalf("entries = %d, want 2", len(got)) } if got[0]["message"] != "first" || got[1]["message"] != "second" { t.Errorf("ordering wrong: %v / %v", got[0]["message"], got[1]["message"]) } } func TestServeProfileLogsLevelFilter(t *testing.T) { cfg, ring := profileTestRoot(t, []string{"alice@example.com"}) rh := NewRingHandler(ring, slog.LevelDebug) logger := slog.New(rh) logger.Debug("d") logger.Info("i") logger.Warn("w") rec := httptest.NewRecorder() ServeProfile(cfg, ring, rec, requestWithEmail(http.MethodGet, "/.profile/logs?level=warn", "alice@example.com")) var got []map[string]any _ = json.Unmarshal(rec.Body.Bytes(), &got) if len(got) != 1 || got[0]["message"] != "w" { t.Errorf("level=warn filter failed: %+v", got) } } // stripTemplates removes every block from the // HTML body so substring assertions check only ACTIVE markup — i.e. live // DOM content the user (and their browser) actually sees, as opposed to // inert content that JS may clone in based on a later access fetch. // // Naive but sufficient for the controlled output of profileTemplate (the // template tags are unnested and well-formed). If the page ever grows // nested templates, swap this for an html.Tokenizer-based pass. func stripTemplates(body string) string { var b strings.Builder for { i := strings.Index(body, "") if j < 0 { // Unterminated "):] } } // TestServeProfileHTMLLayered pins the page-render contract after the // lazy-load refactor: // // - The shell is the same byte-stream for every caller modulo the // identity card and the super-admin diagnostics scaffold (gated by the // cheap IsSuperAdmin check on the root .zddc). // - Subtree-admin scaffolds (Editable .zddc files / Create new project) // live ONLY inside