Ilia Denisov
105 lines
2.8 KiB
Go
105 lines
2.8 KiB
Go
package confirmemailcode
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"galaxy/authsession/internal/domain/challenge"
|
|
authtelemetry "galaxy/authsession/internal/telemetry"
|
|
"galaxy/authsession/internal/testkit"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
sdkmetric "go.opentelemetry.io/otel/sdk/metric"
|
|
"go.opentelemetry.io/otel/sdk/metric/metricdata"
|
|
)
|
|
|
|
func TestExecuteRecordsInvalidCodeMetricForThrottledChallenge(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
runtime, reader := newObservedConfirmTelemetryRuntime(t)
|
|
deps := newConfirmDeps(t)
|
|
record := sentChallengeFixture(t, deps.hasher, "challenge-1", "pilot@example.com", "654321", deps.now.Add(-time.Minute), deps.now.Add(time.Minute))
|
|
record.Status = challenge.StatusDeliveryThrottled
|
|
record.DeliveryState = challenge.DeliveryThrottled
|
|
require.NoError(t, record.Validate())
|
|
require.NoError(t, deps.challengeStore.Create(context.Background(), record))
|
|
|
|
service, err := NewWithTelemetry(
|
|
deps.challengeStore,
|
|
deps.sessionStore,
|
|
deps.userDirectory,
|
|
deps.configProvider,
|
|
deps.publisher,
|
|
deps.idGenerator,
|
|
deps.hasher,
|
|
testkit.FixedClock{Time: deps.now},
|
|
runtime,
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
_, err = service.Execute(context.Background(), Input{
|
|
ChallengeID: "challenge-1",
|
|
Code: "654321",
|
|
ClientPublicKey: publicKeyString(),
|
|
})
|
|
require.Error(t, err)
|
|
|
|
assertConfirmMetricCount(t, reader, map[string]string{"outcome": "invalid_code"}, 1)
|
|
}
|
|
|
|
func newObservedConfirmTelemetryRuntime(t *testing.T) (*authtelemetry.Runtime, *sdkmetric.ManualReader) {
|
|
t.Helper()
|
|
|
|
reader := sdkmetric.NewManualReader()
|
|
provider := sdkmetric.NewMeterProvider(sdkmetric.WithReader(reader))
|
|
|
|
runtime, err := authtelemetry.New(provider)
|
|
require.NoError(t, err)
|
|
|
|
return runtime, reader
|
|
}
|
|
|
|
func assertConfirmMetricCount(t *testing.T, reader *sdkmetric.ManualReader, wantAttrs map[string]string, wantValue int64) {
|
|
t.Helper()
|
|
|
|
var resourceMetrics metricdata.ResourceMetrics
|
|
require.NoError(t, reader.Collect(context.Background(), &resourceMetrics))
|
|
|
|
for _, scopeMetrics := range resourceMetrics.ScopeMetrics {
|
|
for _, metric := range scopeMetrics.Metrics {
|
|
if metric.Name != "authsession.confirm_email_code.attempts" {
|
|
continue
|
|
}
|
|
|
|
sum, ok := metric.Data.(metricdata.Sum[int64])
|
|
require.True(t, ok)
|
|
|
|
for _, point := range sum.DataPoints {
|
|
if hasConfirmMetricAttributes(point.Attributes.ToSlice(), wantAttrs) {
|
|
assert.Equal(t, wantValue, point.Value)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
require.Failf(t, "test failed", "confirm metric with attrs %v not found", wantAttrs)
|
|
}
|
|
|
|
func hasConfirmMetricAttributes(values []attribute.KeyValue, want map[string]string) bool {
|
|
if len(values) != len(want) {
|
|
return false
|
|
}
|
|
|
|
for _, value := range values {
|
|
if want[string(value.Key)] != value.Value.AsString() {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|