starainrt
09d972c7b7
- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
39 lines
1.5 KiB
Go
39 lines
1.5 KiB
Go
package notify
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func newStartedClientConnForTest(t *testing.T, id string, server Server, conn net.Conn, stopCtx context.Context, stopFn context.CancelFunc) (*ClientConn, context.Context, context.CancelFunc) {
|
|
t.Helper()
|
|
client := &ClientConn{
|
|
ClientID: id,
|
|
server: server,
|
|
}
|
|
stopCtx, stopFn = client.startClientConnSession(conn, stopCtx, stopFn)
|
|
return client, stopCtx, stopFn
|
|
}
|
|
|
|
func newRegisteredServerClientForTest(t *testing.T, server *ServerCommon, id string, conn net.Conn, stopCtx context.Context, stopFn context.CancelFunc) (*ClientConn, context.Context, context.CancelFunc) {
|
|
t.Helper()
|
|
client, stopCtx, stopFn := newStartedClientConnForTest(t, id, server, conn, stopCtx, stopFn)
|
|
server.getPeerRegistry().registerClient(client)
|
|
return client, stopCtx, stopFn
|
|
}
|
|
|
|
func newRegisteredServerLogicalForTest(t *testing.T, server *ServerCommon, id string, conn net.Conn, stopCtx context.Context, stopFn context.CancelFunc) (*LogicalConn, context.Context, context.CancelFunc) {
|
|
t.Helper()
|
|
client, stopCtx, stopFn := newStartedClientConnForTest(t, id, server, conn, stopCtx, stopFn)
|
|
logical := logicalConnFromClient(client)
|
|
server.getPeerRegistry().registerLogical(logical)
|
|
return logical, stopCtx, stopFn
|
|
}
|
|
|
|
func newServerCodecClientConnForTest(server *ServerCommon) *ClientConn {
|
|
client := &ClientConn{server: server}
|
|
client.applyClientConnAttachmentProfile(0, 0, server.defaultMsgEn, server.defaultMsgDe, server.handshakeRsaKey, server.SecretKey)
|
|
return client
|
|
}
|