Smail/backend/services/setup/ssl/challenge_test.go
smail f60d3b0880
Some checks are pending
Security Checks / security (push) Waiting to run
整理前后端与测试构建目录并完善配置和安全引导
统一 frontend、backend、test、test-log 和 out 目录,更新构建、部署及测试入口;纳入配置管理兼容与首次安全引导修复;排除本机配置、密钥和测试产物。
2026-09-05 19:19:52 +08:00

56 lines
1.3 KiB
Go

package ssl
import (
"fmt"
"sync"
"testing"
)
func TestHttpChallengeConcurrentAccess(t *testing.T) {
h := &HttpChallenge{authInfo: map[string]*authInfo{}}
var wg sync.WaitGroup
for i := 0; i < 100; i++ {
token := fmt.Sprintf("token-%d", i)
wg.Add(1)
go func() {
defer wg.Done()
if err := h.Present("example.com", token, "key-"+token); err != nil {
t.Errorf("Present returned an error: %v", err)
return
}
if keyAuth, ok := h.KeyAuthorization(token); !ok || keyAuth != "key-"+token {
t.Errorf("unexpected key authorization for %s", token)
}
if err := h.CleanUp("example.com", token, "key-"+token); err != nil {
t.Errorf("CleanUp returned an error: %v", err)
}
}()
}
wg.Wait()
}
func TestDNSChallengeConcurrentAccess(t *testing.T) {
h := &DNSChallenge{authInfo: map[string]*authInfo{}}
var wg sync.WaitGroup
for i := 0; i < 100; i++ {
token := fmt.Sprintf("token-%d", i)
wg.Add(1)
go func() {
defer wg.Done()
if err := h.Present("example.com", token, "key-"+token); err != nil {
t.Errorf("Present returned an error: %v", err)
return
}
_ = h.GetDNSSettings(nil)
if err := h.CleanUp("example.com", token, "key-"+token); err != nil {
t.Errorf("CleanUp returned an error: %v", err)
}
}()
}
wg.Wait()
}