astra/owpengram-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

merged from gramsrv upstream

Browse source
This commit is contained in:
onysd 2026-09-01 12:06:31 +03:00
parent 79c64ee916
commit 21a0856587
651 changed files with 54774 additions and 4590 deletions
Download patch file Download diff file Expand all files Collapse all files

20
internal/rpc/auth_code_rate_limit_test.go
View file

@ -101,6 +101,26 @@ func TestAuthSendCodeRateLimitUsesOpaquePhoneAndRawAuthKeyKeys(t *testing.T) {
}
}
func TestAuthCodeRateLimitSharesBudgetAcrossNationalTrunkVariants(t *testing.T) {
limiter := &captureRateLimiter{}
r := New(Config{
AuthCodePhoneRateLimit: 5,
AuthCodeRateWindow: time.Minute,
}, Deps{Limiter: limiter}, zaptest.NewLogger(t), clock.System)
for _, phone := range []string{"+98 0998 167 9461", "989981679461"} {
if err := r.checkAuthCodeRateLimit(context.Background(), phone); err != nil {
t.Fatalf("checkAuthCodeRateLimit(%q): %v", phone, err)
}
}
if len(limiter.calls) != 2 {
t.Fatalf("limiter calls = %d, want 2", len(limiter.calls))
}
if limiter.calls[0].key != limiter.calls[1].key {
t.Fatalf("equivalent phone variants used different limiter keys: %q != %q", limiter.calls[0].key, limiter.calls[1].key)
}
}
func TestAuthSendCodePhoneRateLimitPrecedesBusinessLookupAndWrite(t *testing.T) {
limiter := &captureRateLimiter{block: true, retryAfter: 17}
authService := &authCodeRateTestService{captureAuthService: &captureAuthService{}}