mono
63e0457886
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.
31 lines
996 B
Go
31 lines
996 B
Go
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") }
|
|
}
|