package api_test import ( "context" "fmt" "net/http" "net/http/httptest" "path/filepath" "strings" "testing" "github.com/go-chi/chi/v5" "vetting/internal/api" "vetting/internal/db" "vetting/internal/events" "vetting/internal/model" "vetting/internal/orchestrator" "vetting/internal/store" ) func setupDetail(t *testing.T) (*api.UI, *store.Hosts, *store.Runs) { t.Helper() conn, err := db.Open(filepath.Join(t.TempDir(), "vetting.db")) if err != nil { t.Fatalf("open db: %v", err) } t.Cleanup(func() { _ = conn.Close() }) hosts := &store.Hosts{DB: conn} runs := &store.Runs{DB: conn} stages := &store.Stages{DB: conn} diffs := &store.SpecDiffs{DB: conn} arts := &store.Artifacts{DB: conn} hub := events.NewHub() runner := &orchestrator.Runner{Runs: runs, Hosts: hosts, Stages: stages, EventHub: hub} tiles := &api.TileEnricher{Runs: runs, Artifacts: arts, SpecDiffs: diffs} ui := &api.UI{ Hosts: hosts, Runs: runs, Stages: stages, SpecDiffs: diffs, Artifacts: arts, EventHub: hub, Runner: runner, Tiles: tiles, } return ui, hosts, runs } func detailReq(id int64) *http.Request { req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/hosts/%d", id), nil) rctx := chi.NewRouteContext() rctx.URLParams.Add("id", fmt.Sprintf("%d", id)) return req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx)) } func TestHostDetail_OK(t *testing.T) { ui, hosts, runs := setupDetail(t) ctx := context.Background() id, err := hosts.Create(ctx, model.Host{ Name: "detail-host", MAC: "aa:bb:cc:dd:ee:30", WoLBroadcastIP: "10.0.0.255", WoLPort: 9, ExpectedSpecYAML: "memory:\n total_gib: 16\n", }) if err != nil { t.Fatalf("create host: %v", err) } runID, err := runs.Create(ctx, id, "deadbeef") if err != nil { t.Fatalf("create run: %v", err) } rr := httptest.NewRecorder() ui.HostDetail(rr, detailReq(id)) if rr.Code != http.StatusOK { t.Fatalf("status = %d, body = %q", rr.Code, rr.Body.String()) } body := rr.Body.String() if !strings.Contains(body, "detail-host") { t.Fatalf("body missing host name: %s", body) } wantPipelineID := fmt.Sprintf(`id="pipeline-%d"`, runID) if !strings.Contains(body, wantPipelineID) { t.Fatalf("body missing %s", wantPipelineID) } wantLogID := fmt.Sprintf(`id="log-%d"`, runID) if !strings.Contains(body, wantLogID) { t.Fatalf("body missing %s", wantLogID) } } func TestHostDetail_NeverRun(t *testing.T) { ui, hosts, _ := setupDetail(t) id, err := hosts.Create(context.Background(), model.Host{ Name: "never-run", MAC: "aa:bb:cc:dd:ee:31", WoLBroadcastIP: "10.0.0.255", WoLPort: 9, ExpectedSpecYAML: "memory:\n total_gib: 8\n", }) if err != nil { t.Fatalf("create host: %v", err) } rr := httptest.NewRecorder() ui.HostDetail(rr, detailReq(id)) if rr.Code != http.StatusOK { t.Fatalf("status = %d, body = %q", rr.Code, rr.Body.String()) } body := rr.Body.String() if !strings.Contains(body, "Start vetting") { t.Fatalf("never-run page missing Start vetting: %s", body) } // Ghost pipeline: all nodes rendered but none are running/failed/passed. if strings.Contains(body, "stage-dot-running") || strings.Contains(body, "stage-dot-failed") { t.Fatalf("ghost pipeline should have no running/failed dots: %s", body) } if !strings.Contains(body, "stage-dot-pending") { t.Fatalf("expected pending stage dots in ghost pipeline: %s", body) } } func TestHostDetail_UnknownID(t *testing.T) { ui, _, _ := setupDetail(t) rr := httptest.NewRecorder() ui.HostDetail(rr, detailReq(9999)) if rr.Code != http.StatusNotFound { t.Fatalf("status = %d, want 404", rr.Code) } } func TestHostDetail_BadID(t *testing.T) { ui, _, _ := setupDetail(t) req := httptest.NewRequest(http.MethodGet, "/hosts/not-a-number", nil) rctx := chi.NewRouteContext() rctx.URLParams.Add("id", "not-a-number") req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx)) rr := httptest.NewRecorder() ui.HostDetail(rr, req) if rr.Code != http.StatusBadRequest { t.Fatalf("status = %d, want 400", rr.Code) } }