补 wsClientCore 并发测试 + close() 防 nil conn 守卫

Result: {"status":"keep","total_issues":8,"eslint_errors":0,"eslint_problems":0,"eslint_warnings":0,"golint_canonicalheader":0,"golint_errname":0,"golint_errorlint":1,"golint_exhaustive":0,"golint_forcetypeassert":0,"golint_gosec":0,"golint_intrange":0,"golint_modernize":3,"golint_nilnil":3,"golint_perfsprint":0,"golint_prealloc":0,"golint_recvcheck":1,"golint_test_testifylint":0,"golint_test_thelper":0,"golint_test_total":0,"golint_test_usetesting":0,"golint_total":8,"golint_usestdlibvars":0,"golint_vetx_total":0,"golint_wastedassign":0,"measure_s":77,"tsc_errors":0,"vitest_failed":0,"vitest_total":126}
This commit is contained in:
ryan
2026-08-26 09:53:42 +08:00
parent efd8268a5d
commit ed1efd3d54
3 changed files with 61 additions and 1 deletions
@@ -27,7 +27,9 @@ func (c *wsClientCore) close() {
}
c.once.Do(func() {
close(c.done)
_ = c.conn.Close()
if c.conn != nil {
_ = c.conn.Close()
}
})
}
@@ -0,0 +1,57 @@
// Copyright 2026 Arctel.net
// SPDX-License-Identifier: Apache-2.0
package websocket
import (
"sync"
"testing"
)
func TestWSClientCoreCloseIsIdempotent(t *testing.T) {
core := &wsClientCore{
send: make(chan Message, 1),
done: make(chan struct{}),
}
var wg sync.WaitGroup
for range 8 {
wg.Add(1)
go func() {
defer wg.Done()
core.close()
}()
}
wg.Wait()
select {
case <-core.done:
default:
t.Fatal("close did not signal done")
}
}
func TestWSClientCoreEnqueueFailsAfterClose(t *testing.T) {
core := &wsClientCore{
send: make(chan Message, 1),
done: make(chan struct{}),
}
core.close()
if core.enqueue(Message{Type: messageTypePing}) {
t.Fatal("enqueue must fail after close")
}
}
func TestWSClientCoreEnqueueNeverBlocks(t *testing.T) {
core := &wsClientCore{
send: make(chan Message, 1), // 缓冲小于消息数,验证不阻塞
done: make(chan struct{}),
}
defer core.close()
for range 3 {
if !core.enqueue(Message{Type: messageTypePing}) && len(core.send) == 0 {
t.Fatal("enqueue failed with empty buffer")
}
}
if core.enqueue(Message{Type: messageTypePing}) {
t.Fatal("enqueue must fail when buffer full")
}
}