feat: M14 — Observability (dashboard charts, agent fleet, stats API, metrics, structured logging, rollback)

Backend: StatsService with 5 aggregation methods, JSON metrics endpoint, slog-based structured logging middleware. Stats API: dashboard summary, certificates-by-status, expiration timeline, job trends, issuance rate. 23 new backend tests. Frontend: Recharts-powered dashboard with 4 charts (status pie, expiration heatmap, job trends line, issuance bar), agent fleet overview page with OS/arch grouping and version breakdown, deployment rollback buttons on version history. 7 new frontend tests. 78 API endpoints, 744+ total tests (658 Go + 86 Vitest). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-06-07 16:11:29 +00:00 · 2026-03-22 19:46:13 -04:00
parent 2f65dd1a61
commit ee75f149ae
21 changed files with 2125 additions and 28 deletions
@@ -0,0 +1,204 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"time"
+)
+
+// MockStatsService implements both StatsService and MetricsService.
+type MockStatsService struct {
+	GetDashboardSummaryFn     func(ctx context.Context) (interface{}, error)
+	GetCertificatesByStatusFn func(ctx context.Context) (interface{}, error)
+	GetExpirationTimelineFn   func(ctx context.Context, days int) (interface{}, error)
+	GetJobStatsFn             func(ctx context.Context, days int) (interface{}, error)
+	GetIssuanceRateFn         func(ctx context.Context, days int) (interface{}, error)
+}
+
+func (m *MockStatsService) GetDashboardSummary(ctx context.Context) (interface{}, error) {
+	if m.GetDashboardSummaryFn != nil {
+		return m.GetDashboardSummaryFn(ctx)
+	}
+	return map[string]int64{"total_certificates": 0}, nil
+}
+
+func (m *MockStatsService) GetCertificatesByStatus(ctx context.Context) (interface{}, error) {
+	if m.GetCertificatesByStatusFn != nil {
+		return m.GetCertificatesByStatusFn(ctx)
+	}
+	return []interface{}{}, nil
+}
+
+func (m *MockStatsService) GetExpirationTimeline(ctx context.Context, days int) (interface{}, error) {
+	if m.GetExpirationTimelineFn != nil {
+		return m.GetExpirationTimelineFn(ctx, days)
+	}
+	return []interface{}{}, nil
+}
+
+func (m *MockStatsService) GetJobStats(ctx context.Context, days int) (interface{}, error) {
+	if m.GetJobStatsFn != nil {
+		return m.GetJobStatsFn(ctx, days)
+	}
+	return []interface{}{}, nil
+}
+
+func (m *MockStatsService) GetIssuanceRate(ctx context.Context, days int) (interface{}, error) {
+	if m.GetIssuanceRateFn != nil {
+		return m.GetIssuanceRateFn(ctx, days)
+	}
+	return []interface{}{}, nil
+}
+
+func TestGetDashboardSummary_Success(t *testing.T) {
+	mock := &MockStatsService{}
+	h := NewStatsHandler(mock)
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/stats/summary", nil)
+	w := httptest.NewRecorder()
+	h.GetDashboardSummary(w, req)
+	if w.Code != http.StatusOK {
+		t.Errorf("expected 200, got %d", w.Code)
+	}
+}
+
+func TestGetDashboardSummary_MethodNotAllowed(t *testing.T) {
+	mock := &MockStatsService{}
+	h := NewStatsHandler(mock)
+	req := httptest.NewRequest(http.MethodPost, "/api/v1/stats/summary", nil)
+	w := httptest.NewRecorder()
+	h.GetDashboardSummary(w, req)
+	if w.Code != http.StatusMethodNotAllowed {
+		t.Errorf("expected 405, got %d", w.Code)
+	}
+}
+
+func TestGetDashboardSummary_ServiceError(t *testing.T) {
+	mock := &MockStatsService{
+		GetDashboardSummaryFn: func(ctx context.Context) (interface{}, error) {
+			return nil, fmt.Errorf("db error")
+		},
+	}
+	h := NewStatsHandler(mock)
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/stats/summary", nil)
+	w := httptest.NewRecorder()
+	h.GetDashboardSummary(w, req)
+	if w.Code != http.StatusInternalServerError {
+		t.Errorf("expected 500, got %d", w.Code)
+	}
+}
+
+func TestGetCertificatesByStatus_Success(t *testing.T) {
+	mock := &MockStatsService{}
+	h := NewStatsHandler(mock)
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/stats/certificates-by-status", nil)
+	w := httptest.NewRecorder()
+	h.GetCertificatesByStatus(w, req)
+	if w.Code != http.StatusOK {
+		t.Errorf("expected 200, got %d", w.Code)
+	}
+}
+
+func TestGetExpirationTimeline_Success(t *testing.T) {
+	mock := &MockStatsService{}
+	h := NewStatsHandler(mock)
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/stats/expiration-timeline?days=60", nil)
+	w := httptest.NewRecorder()
+	h.GetExpirationTimeline(w, req)
+	if w.Code != http.StatusOK {
+		t.Errorf("expected 200, got %d", w.Code)
+	}
+}
+
+func TestGetExpirationTimeline_DefaultDays(t *testing.T) {
+	mock := &MockStatsService{
+		GetExpirationTimelineFn: func(ctx context.Context, days int) (interface{}, error) {
+			if days != 30 {
+				t.Errorf("expected default 30 days, got %d", days)
+			}
+			return []interface{}{}, nil
+		},
+	}
+	h := NewStatsHandler(mock)
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/stats/expiration-timeline", nil)
+	w := httptest.NewRecorder()
+	h.GetExpirationTimeline(w, req)
+	if w.Code != http.StatusOK {
+		t.Errorf("expected 200, got %d", w.Code)
+	}
+}
+
+func TestGetJobTrends_Success(t *testing.T) {
+	mock := &MockStatsService{}
+	h := NewStatsHandler(mock)
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/stats/job-trends?days=14", nil)
+	w := httptest.NewRecorder()
+	h.GetJobTrends(w, req)
+	if w.Code != http.StatusOK {
+		t.Errorf("expected 200, got %d", w.Code)
+	}
+}
+
+func TestGetIssuanceRate_Success(t *testing.T) {
+	mock := &MockStatsService{}
+	h := NewStatsHandler(mock)
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/stats/issuance-rate?days=7", nil)
+	w := httptest.NewRecorder()
+	h.GetIssuanceRate(w, req)
+	if w.Code != http.StatusOK {
+		t.Errorf("expected 200, got %d", w.Code)
+	}
+}
+
+func TestGetMetrics_Success(t *testing.T) {
+	mock := &MockStatsService{
+		GetDashboardSummaryFn: func(ctx context.Context) (interface{}, error) {
+			return &DashboardSummary{
+				TotalCertificates:    10,
+				ExpiringCertificates: 2,
+				ExpiredCertificates:  1,
+				RevokedCertificates:  0,
+				ActiveAgents:         3,
+				TotalAgents:          5,
+				PendingJobs:          1,
+				FailedJobs:           0,
+				CompleteJobs:         8,
+			}, nil
+		},
+	}
+	h := NewMetricsHandler(mock, time.Now())
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/metrics", nil)
+	w := httptest.NewRecorder()
+	h.GetMetrics(w, req)
+	if w.Code != http.StatusOK {
+		t.Errorf("expected 200, got %d", w.Code)
+	}
+}
+
+func TestGetMetrics_MethodNotAllowed(t *testing.T) {
+	mock := &MockStatsService{}
+	h := NewMetricsHandler(mock, time.Now())
+	req := httptest.NewRequest(http.MethodPost, "/api/v1/metrics", nil)
+	w := httptest.NewRecorder()
+	h.GetMetrics(w, req)
+	if w.Code != http.StatusMethodNotAllowed {
+		t.Errorf("expected 405, got %d", w.Code)
+	}
+}
+
+func TestGetMetrics_ServiceError(t *testing.T) {
+	mock := &MockStatsService{
+		GetDashboardSummaryFn: func(ctx context.Context) (interface{}, error) {
+			return nil, fmt.Errorf("db error")
+		},
+	}
+	h := NewMetricsHandler(mock, time.Now())
+	req := httptest.NewRequest(http.MethodGet, "/api/v1/metrics", nil)
+	w := httptest.NewRecorder()
+	h.GetMetrics(w, req)
+	if w.Code != http.StatusInternalServerError {
+		t.Errorf("expected 500, got %d", w.Code)
+	}
+}