package connectsrv

import (
	"context"
	"testing"
	"time"

	"go.opentelemetry.io/otel/attribute"
	sdkmetric "go.opentelemetry.io/otel/sdk/metric"
	"go.opentelemetry.io/otel/sdk/metric/metricdata"
)

// TestEdgeMetric records Execute outcomes through a manual reader and asserts the
// edge_request_duration histogram splits by message_type and result.
func TestEdgeMetric(t *testing.T) {
	ctx := context.Background()
	reader := sdkmetric.NewManualReader()
	meter := sdkmetric.NewMeterProvider(sdkmetric.WithReader(reader)).Meter("test")
	m := newServerMetrics(meter)

	m.recordEdge(ctx, "game.submit_play", "ok", time.Now().Add(-time.Millisecond))
	m.recordEdge(ctx, "game.submit_play", "ok", time.Now().Add(-time.Millisecond))
	m.recordEdge(ctx, "auth.guest", "domain", time.Now().Add(-time.Millisecond))

	var rm metricdata.ResourceMetrics
	if err := reader.Collect(ctx, &rm); err != nil {
		t.Fatalf("collect: %v", err)
	}

	type key struct{ messageType, result string }
	counts := map[key]uint64{}
	for _, sm := range rm.ScopeMetrics {
		for _, md := range sm.Metrics {
			if md.Name != "edge_request_duration" {
				continue
			}
			h, ok := md.Data.(metricdata.Histogram[float64])
			if !ok {
				t.Fatalf("edge_request_duration is not a float64 histogram")
			}
			for _, dp := range h.DataPoints {
				mt, _ := dp.Attributes.Value(attribute.Key("message_type"))
				res, _ := dp.Attributes.Value(attribute.Key("result"))
				counts[key{mt.AsString(), res.AsString()}] += dp.Count
			}
		}
	}
	if got := counts[key{"game.submit_play", "ok"}]; got != 2 {
		t.Errorf("edge game.submit_play/ok = %d, want 2", got)
	}
	if got := counts[key{"auth.guest", "domain"}]; got != 1 {
		t.Errorf("edge auth.guest/domain = %d, want 1", got)
	}
}