fix: harden identity, live access, drafts and setup (review only)
Source-reviewed hardening changes and regression tests. UNBUILT / UNTESTED: not a deployable release. Go formatting, compilation, automated regression tests, full frontend validation and staging acceptance remain required. See docs/hardening/VALIDATION.md. Main remains the original clean snapshot; no upstream Git history imported.
This commit is contained in:
30
internal/services/ws_session_concurrency_test.go
Normal file
30
internal/services/ws_session_concurrency_test.go
Normal file
@@ -0,0 +1,30 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// Run with -race: close may be triggered independently by revocation, a reader,
|
||||
// a writer, or slow-client eviction while broadcasts still hold a session pointer.
|
||||
func TestWsConcurrentEnqueueCloseAndSubscribe(t *testing.T) {
|
||||
ws:=newWsService()
|
||||
session:=&ClientSession{ID:"concurrent",Topics:map[string]struct{}{},Send:make(chan []byte,64)}
|
||||
ws.manager.Register(session,[]string{"notification:1"})
|
||||
var work sync.WaitGroup
|
||||
for i:=0;i<8;i++ {
|
||||
work.Add(1)
|
||||
go func() {
|
||||
defer work.Done()
|
||||
for j:=0;j<100;j++ {
|
||||
session.enqueue([]byte("test"))
|
||||
ws.manager.Subscribe(session,[]string{"notification:1"})
|
||||
ws.manager.TopicList(session)
|
||||
}
|
||||
}()
|
||||
}
|
||||
ws.closeSession(session)
|
||||
work.Wait()
|
||||
if !session.Closed.Load() || ws.manager.HasTopic("notification:1") || len(ws.manager.TopicList(session))!=0 { t.Fatal("closed session was re-registered") }
|
||||
if session.enqueue([]byte("test")) { t.Fatal("closed session accepted a message") }
|
||||
}
|
||||
Reference in New Issue
Block a user