package securityevents

import (
	"context"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
	"time"
)

type metricsSnapshot struct {
	mode string
	last time.Time
}

func (m metricsSnapshot) SecurityEventMetrics(context.Context, string, string) (string, *time.Time, error) {
	return m.mode, &m.last, nil
}

func TestMetricsExposeBoundedOutcomesAndState(t *testing.T) {
	metrics := &Metrics{}
	metrics.ObserveReceipt("accepted", 20*time.Millisecond, 2)
	metrics.ObserveReceipt("duplicate", 10*time.Millisecond, 0)
	metrics.ObserveWatermarkRejection()
	metrics.ObserveHeartbeat("accepted")
	metrics.ObserveIntrospection("failed")
	metrics.ObserveJWKSRefreshFailure("ssf")
	metrics.SetAsyncWorkerCapacity(96, 128, 96, true)
	metrics.ObserveAsyncWorkerResize("success")
	metrics.ObserveConcurrencyLeaseRenewal("success")
	metrics.ObserveConcurrencyLeaseRenewal("lost")
	metrics.ObserveTaskEventSkip("duplicate")
	metrics.ObserveTaskEventSkip("budget_exceeded")

	recorder := httptest.NewRecorder()
	metrics.Handler(metricsSnapshot{mode: "push_healthy", last: time.Now().Add(-time.Minute)}, "issuer", "audience", true).
		ServeHTTP(recorder, httptest.NewRequest(http.MethodGet, "/metrics", nil))
	if recorder.Code != http.StatusOK {
		t.Fatalf("metrics status=%d", recorder.Code)
	}
	for _, expected := range []string{
		`easyai_gateway_ssf_receipts_total{outcome="accepted"} 1`,
		`easyai_gateway_ssf_receipts_total{outcome="duplicate"} 1`,
		`easyai_gateway_ssf_sessions_deleted_total 2`,
		`easyai_gateway_ssf_watermark_rejections_total 1`,
		`easyai_gateway_ssf_processing_duration_seconds_bucket{le="0.05"} 2`,
		`easyai_gateway_ssf_mode{mode="push_healthy"} 1`,
		`easyai_gateway_oidc_introspection_total{outcome="failed"} 1`,
		`easyai_gateway_jwks_refresh_failures_total{outcome="ssf"} 1`,
		`easyai_gateway_async_worker_capacity 96`,
		`easyai_gateway_async_worker_desired_capacity 128`,
		`easyai_gateway_async_worker_capacity_capped 1`,
		`easyai_gateway_async_worker_resizes_total{outcome="success"} 1`,
		`easyai_gateway_concurrency_lease_renewals_total{outcome="success"} 1`,
		`easyai_gateway_concurrency_lease_renewals_total{outcome="lost"} 1`,
		`easyai_gateway_task_events_skipped_total{outcome="duplicate"} 1`,
		`easyai_gateway_task_events_skipped_total{outcome="budget_exceeded"} 1`,
	} {
		if !strings.Contains(recorder.Body.String(), expected) {
			t.Fatalf("missing metric %q in:\n%s", expected, recorder.Body.String())
		}
	}
}
