chore: refresh gramsrv public release

internal/store/postgres/query_stats_test.go

@@ -0,0 +1,38 @@
+package postgres
+
+import (
+	"context"
+	"errors"
+	"testing"
+
+	"github.com/jackc/pgx/v5"
+
+	"telesrv/internal/observability/dbtrace"
+)
+
+func TestQueryStatsTracerRecordsContextStats(t *testing.T) {
+	ctx, stats := dbtrace.WithStats(context.Background())
+	tracer := queryStatsTracer{}
+
+	ctx = tracer.TraceQueryStart(ctx, nil, pgx.TraceQueryStartData{})
+	tracer.TraceQueryEnd(ctx, nil, pgx.TraceQueryEndData{})
+
+	snap := stats.Snapshot()
+	if snap.Queries != 1 || snap.Errors != 0 {
+		t.Fatalf("Snapshot() = %+v, want 1 query without errors", snap)
+	}
+
+	ctx = tracer.TraceQueryStart(ctx, nil, pgx.TraceQueryStartData{})
+	tracer.TraceQueryEnd(ctx, nil, pgx.TraceQueryEndData{Err: errors.New("boom")})
+
+	snap = stats.Snapshot()
+	if snap.Queries != 2 || snap.Errors != 1 {
+		t.Fatalf("Snapshot() = %+v, want 2 queries with 1 error", snap)
+	}
+}
+
+func TestQueryStatsTracerIgnoresContextWithoutStats(t *testing.T) {
+	tracer := queryStatsTracer{}
+	ctx := tracer.TraceQueryStart(context.Background(), nil, pgx.TraceQueryStartData{})
+	tracer.TraceQueryEnd(ctx, nil, pgx.TraceQueryEndData{Err: errors.New("boom")})
+}