mirror of
https://github.com/shankar0123/certctl.git
synced 2026-06-14 13:48:59 +00:00
feat(M48): continuous TLS health monitoring — endpoint state machine, shared tlsprobe, 8 API endpoints, GUI
Adds continuous TLS endpoint health monitoring that closes the deploy→verify→monitor loop. After M25 verifies a deployment succeeded once, M48 continuously confirms it stays healthy. Key components: - Shared `internal/tlsprobe/` package extracted from network scanner for reuse - Health status state machine: healthy → degraded (2 failures) → down (5 failures), plus cert_mismatch when served fingerprint differs from expected - 8th scheduler loop (60s tick, per-endpoint configurable intervals) - PostgreSQL migration 000011: endpoint_health_checks + endpoint_health_history tables - 8 REST API endpoints (CRUD, history, acknowledge, summary) - Health Monitor GUI page with summary bar, status table, create modal, auto-refresh - 38 new tests (5 tlsprobe + 11 domain + 10 service + 8 handler + 4 frontend) - All coverage thresholds maintained (service 68%, handler 83%, domain 87%, middleware 63%) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,350 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"log/slog"
|
||||
"os"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/shankar0123/certctl/internal/domain"
|
||||
"github.com/shankar0123/certctl/internal/repository"
|
||||
)
|
||||
|
||||
// mockHealthCheckRepo implements the HealthCheckRepository interface for testing.
|
||||
type mockHealthCheckRepo struct {
|
||||
checks map[string]*domain.EndpointHealthCheck
|
||||
history []*domain.HealthHistoryEntry
|
||||
createErr error
|
||||
getErr error
|
||||
updateErr error
|
||||
deleteErr error
|
||||
listErr error
|
||||
listDueErr error
|
||||
getHistoryErr error
|
||||
recordHistoryErr error
|
||||
purgeHistoryErr error
|
||||
getSummaryErr error
|
||||
getSummaryResult *domain.HealthCheckSummary
|
||||
}
|
||||
|
||||
func newMockHealthCheckRepo() *mockHealthCheckRepo {
|
||||
return &mockHealthCheckRepo{
|
||||
checks: make(map[string]*domain.EndpointHealthCheck),
|
||||
history: []*domain.HealthHistoryEntry{},
|
||||
getSummaryResult: &domain.HealthCheckSummary{
|
||||
Healthy: 0,
|
||||
Degraded: 0,
|
||||
Down: 0,
|
||||
CertMismatch: 0,
|
||||
Unknown: 0,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) Create(ctx context.Context, check *domain.EndpointHealthCheck) error {
|
||||
if m.createErr != nil {
|
||||
return m.createErr
|
||||
}
|
||||
m.checks[check.ID] = check
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) Get(ctx context.Context, id string) (*domain.EndpointHealthCheck, error) {
|
||||
if m.getErr != nil {
|
||||
return nil, m.getErr
|
||||
}
|
||||
if check, ok := m.checks[id]; ok {
|
||||
return check, nil
|
||||
}
|
||||
return nil, errors.New("not found")
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) GetByEndpoint(ctx context.Context, endpoint string) (*domain.EndpointHealthCheck, error) {
|
||||
for _, check := range m.checks {
|
||||
if check.Endpoint == endpoint {
|
||||
return check, nil
|
||||
}
|
||||
}
|
||||
return nil, errors.New("not found")
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) Update(ctx context.Context, check *domain.EndpointHealthCheck) error {
|
||||
if m.updateErr != nil {
|
||||
return m.updateErr
|
||||
}
|
||||
m.checks[check.ID] = check
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) Delete(ctx context.Context, id string) error {
|
||||
if m.deleteErr != nil {
|
||||
return m.deleteErr
|
||||
}
|
||||
delete(m.checks, id)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) List(ctx context.Context, filter *repository.HealthCheckFilter) ([]*domain.EndpointHealthCheck, int, error) {
|
||||
if m.listErr != nil {
|
||||
return nil, 0, m.listErr
|
||||
}
|
||||
checks := make([]*domain.EndpointHealthCheck, 0, len(m.checks))
|
||||
for _, check := range m.checks {
|
||||
checks = append(checks, check)
|
||||
}
|
||||
return checks, len(checks), nil
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) ListDueForCheck(ctx context.Context) ([]*domain.EndpointHealthCheck, error) {
|
||||
if m.listDueErr != nil {
|
||||
return nil, m.listDueErr
|
||||
}
|
||||
checks := make([]*domain.EndpointHealthCheck, 0, len(m.checks))
|
||||
for _, check := range m.checks {
|
||||
if check.Enabled {
|
||||
checks = append(checks, check)
|
||||
}
|
||||
}
|
||||
return checks, nil
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) GetHistory(ctx context.Context, healthCheckID string, limit int) ([]*domain.HealthHistoryEntry, error) {
|
||||
if m.getHistoryErr != nil {
|
||||
return nil, m.getHistoryErr
|
||||
}
|
||||
return m.history, nil
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) RecordHistory(ctx context.Context, entry *domain.HealthHistoryEntry) error {
|
||||
if m.recordHistoryErr != nil {
|
||||
return m.recordHistoryErr
|
||||
}
|
||||
m.history = append(m.history, entry)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) PurgeHistory(ctx context.Context, before time.Time) (int64, error) {
|
||||
if m.purgeHistoryErr != nil {
|
||||
return 0, m.purgeHistoryErr
|
||||
}
|
||||
return 0, nil
|
||||
}
|
||||
|
||||
func (m *mockHealthCheckRepo) GetSummary(ctx context.Context) (*domain.HealthCheckSummary, error) {
|
||||
if m.getSummaryErr != nil {
|
||||
return nil, m.getSummaryErr
|
||||
}
|
||||
return m.getSummaryResult, nil
|
||||
}
|
||||
|
||||
// Tests
|
||||
|
||||
func newTestLogger() *slog.Logger {
|
||||
return slog.New(slog.NewTextHandler(os.Stderr, &slog.HandlerOptions{Level: slog.LevelDebug}))
|
||||
}
|
||||
|
||||
func TestHealthCheckService_Create_Success(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
check := &domain.EndpointHealthCheck{
|
||||
Endpoint: "example.com:443",
|
||||
Status: domain.HealthStatusUnknown,
|
||||
Enabled: true,
|
||||
CheckIntervalSecs: 300,
|
||||
}
|
||||
|
||||
err := svc.Create(context.Background(), check)
|
||||
if err != nil {
|
||||
t.Fatalf("Create failed: %v", err)
|
||||
}
|
||||
|
||||
if check.ID == "" {
|
||||
t.Fatal("Expected ID to be set")
|
||||
}
|
||||
|
||||
retrieved, _ := repo.Get(context.Background(), check.ID)
|
||||
if retrieved == nil {
|
||||
t.Fatal("Expected check to be in repo")
|
||||
}
|
||||
if retrieved.Endpoint != "example.com:443" {
|
||||
t.Errorf("Expected endpoint example.com:443, got %s", retrieved.Endpoint)
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthCheckService_Create_RepoError(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
repo.createErr = errors.New("db error")
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
check := &domain.EndpointHealthCheck{
|
||||
Endpoint: "example.com:443",
|
||||
Enabled: true,
|
||||
}
|
||||
|
||||
err := svc.Create(context.Background(), check)
|
||||
if err == nil {
|
||||
t.Fatal("Expected error, got nil")
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthCheckService_Get_Success(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
check := &domain.EndpointHealthCheck{
|
||||
ID: "hc-test-1",
|
||||
Endpoint: "example.com:443",
|
||||
Status: domain.HealthStatusHealthy,
|
||||
}
|
||||
repo.checks["hc-test-1"] = check
|
||||
|
||||
retrieved, err := svc.Get(context.Background(), "hc-test-1")
|
||||
if err != nil {
|
||||
t.Fatalf("Get failed: %v", err)
|
||||
}
|
||||
if retrieved.Endpoint != "example.com:443" {
|
||||
t.Errorf("Expected endpoint example.com:443, got %s", retrieved.Endpoint)
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthCheckService_Get_NotFound(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
_, err := svc.Get(context.Background(), "nonexistent")
|
||||
if err == nil {
|
||||
t.Fatal("Expected error for nonexistent check")
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthCheckService_List_Success(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
check1 := &domain.EndpointHealthCheck{
|
||||
ID: "hc-1",
|
||||
Endpoint: "api.example.com:443",
|
||||
Status: domain.HealthStatusHealthy,
|
||||
}
|
||||
check2 := &domain.EndpointHealthCheck{
|
||||
ID: "hc-2",
|
||||
Endpoint: "web.example.com:443",
|
||||
Status: domain.HealthStatusDegraded,
|
||||
}
|
||||
repo.checks["hc-1"] = check1
|
||||
repo.checks["hc-2"] = check2
|
||||
|
||||
checks, total, err := svc.List(context.Background(), nil)
|
||||
if err != nil {
|
||||
t.Fatalf("List failed: %v", err)
|
||||
}
|
||||
if len(checks) != 2 {
|
||||
t.Errorf("Expected 2 checks, got %d", len(checks))
|
||||
}
|
||||
if total != 2 {
|
||||
t.Errorf("Expected total 2, got %d", total)
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthCheckService_Delete_Success(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
check := &domain.EndpointHealthCheck{
|
||||
ID: "hc-test-1",
|
||||
Endpoint: "example.com:443",
|
||||
}
|
||||
repo.checks["hc-test-1"] = check
|
||||
|
||||
err := svc.Delete(context.Background(), "hc-test-1")
|
||||
if err != nil {
|
||||
t.Fatalf("Delete failed: %v", err)
|
||||
}
|
||||
|
||||
if _, ok := repo.checks["hc-test-1"]; ok {
|
||||
t.Fatal("Expected check to be deleted")
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthCheckService_AcknowledgeIncident_Success(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
check := &domain.EndpointHealthCheck{
|
||||
ID: "hc-test-1",
|
||||
Endpoint: "example.com:443",
|
||||
Status: domain.HealthStatusDown,
|
||||
Acknowledged: false,
|
||||
}
|
||||
repo.checks["hc-test-1"] = check
|
||||
|
||||
err := svc.AcknowledgeIncident(context.Background(), "hc-test-1", "user@example.com")
|
||||
if err != nil {
|
||||
t.Fatalf("AcknowledgeIncident failed: %v", err)
|
||||
}
|
||||
|
||||
retrieved := repo.checks["hc-test-1"]
|
||||
if !retrieved.Acknowledged {
|
||||
t.Fatal("Expected Acknowledged to be true")
|
||||
}
|
||||
if retrieved.AcknowledgedBy != "user@example.com" {
|
||||
t.Errorf("Expected AcknowledgedBy to be user@example.com, got %s", retrieved.AcknowledgedBy)
|
||||
}
|
||||
if retrieved.AcknowledgedAt == nil {
|
||||
t.Fatal("Expected AcknowledgedAt to be set")
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthCheckService_GetSummary_Success(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
repo.getSummaryResult = &domain.HealthCheckSummary{
|
||||
Healthy: 5,
|
||||
Degraded: 2,
|
||||
Down: 1,
|
||||
CertMismatch: 1,
|
||||
Unknown: 0,
|
||||
}
|
||||
|
||||
summary, err := svc.GetSummary(context.Background())
|
||||
if err != nil {
|
||||
t.Fatalf("GetSummary failed: %v", err)
|
||||
}
|
||||
if summary.Healthy != 5 {
|
||||
t.Errorf("Expected 5 healthy, got %d", summary.Healthy)
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthCheckService_RunHealthChecks_NoEndpoints(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
err := svc.RunHealthChecks(context.Background())
|
||||
if err != nil {
|
||||
t.Fatalf("RunHealthChecks failed: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestHealthCheckService_PurgeOldHistory_Success(t *testing.T) {
|
||||
repo := newMockHealthCheckRepo()
|
||||
logger := newTestLogger()
|
||||
svc := NewHealthCheckService(repo, nil, logger, 10, 5*time.Second, 30*24*time.Hour, false)
|
||||
|
||||
err := svc.PurgeOldHistory(context.Background())
|
||||
if err != nil {
|
||||
t.Fatalf("PurgeOldHistory failed: %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user