Files
sub2api/backend/internal/securityaudit/prompt_metrics_test.go
T
李建琦 6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
2026-08-21 18:30:13 +08:00

53 lines
1.6 KiB
Go

package securityaudit
import (
"sync"
"testing"
"time"
"github.com/stretchr/testify/require"
)
func TestAtomicMetricsExposeCountsLatencyDistributionAndAsyncDelivery(t *testing.T) {
metrics := NewAtomicMetrics()
latencies := []time.Duration{10, 20, 30, 40, 100}
kinds := []DecisionKind{DecisionAllow, DecisionFlag, DecisionBlock, DecisionUnavailable, DecisionInvalid}
for index := range latencies {
metrics.Observe(kinds[index], latencies[index]*time.Millisecond)
}
metrics.IncTimeout()
metrics.IncFailover()
metrics.IncBulkheadFull()
metrics.IncRecordFailed()
metrics.IncEnqueued()
metrics.IncDropped()
snapshot := metrics.Snapshot()
require.Equal(t, int64(5), snapshot.Total)
require.Equal(t, int64(5), snapshot.LatencyCount)
require.Equal(t, int64(40), snapshot.LatencyAvgMS)
require.Equal(t, int64(30), snapshot.LatencyP50MS)
require.Equal(t, int64(40), snapshot.LatencyP95MS)
require.Equal(t, int64(40), snapshot.LatencyP99MS)
require.Equal(t, int64(100), snapshot.LatencyMaxMS)
require.Equal(t, AuditMetricsSnapshot{Enqueued: 1, Dropped: 1}, metrics.AuditSnapshot())
}
func TestAtomicMetricsConcurrentObservationIsBoundedAndRaceSafe(t *testing.T) {
metrics := NewAtomicMetrics()
const observations = 4096
var wg sync.WaitGroup
for index := 0; index < observations; index++ {
wg.Add(1)
go func(value int) {
defer wg.Done()
metrics.Observe(DecisionAllow, time.Duration(value%250)*time.Millisecond)
}(index)
}
wg.Wait()
require.Equal(t, int64(observations), metrics.Snapshot().Total)
metrics.latencyMu.RLock()
require.LessOrEqual(t, len(metrics.latencies), latencySampleCapacity)
metrics.latencyMu.RUnlock()
}