starainrt
09d972c7b7
- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
102 lines
2.9 KiB
Go
102 lines
2.9 KiB
Go
package notify
|
|
|
|
import (
|
|
"b612.me/stario"
|
|
"context"
|
|
"errors"
|
|
"math"
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func TestClientEncodeEnvelopeRequiresRuntimeQueue(t *testing.T) {
|
|
client := NewClient().(*ClientCommon)
|
|
UseLegacySecurityClient(client)
|
|
|
|
_, err := client.encodeEnvelope(newSignalAckEnvelope(1))
|
|
if !errors.Is(err, errClientSessionQueueUnavailable) {
|
|
t.Fatalf("client encodeEnvelope error = %v, want %v", err, errClientSessionQueueUnavailable)
|
|
}
|
|
}
|
|
|
|
func TestServerEncodeEnvelopeRequiresRuntimeQueue(t *testing.T) {
|
|
server := NewServer().(*ServerCommon)
|
|
UseLegacySecurityServer(server)
|
|
client := newServerCodecClientConnForTest(server)
|
|
|
|
_, err := server.encodeEnvelope(client, newSignalAckEnvelope(1))
|
|
if !errors.Is(err, errServerSessionQueueUnavailable) {
|
|
t.Fatalf("server encodeEnvelope error = %v, want %v", err, errServerSessionQueueUnavailable)
|
|
}
|
|
}
|
|
|
|
func TestClientEncodeEnvelopeUsesRuntimeQueue(t *testing.T) {
|
|
client := NewClient().(*ClientCommon)
|
|
UseLegacySecurityClient(client)
|
|
stopCtx, stopFn := context.WithCancel(context.Background())
|
|
defer stopFn()
|
|
client.setClientSessionRuntime(&clientSessionRuntime{
|
|
stopCtx: stopCtx,
|
|
stopFn: stopFn,
|
|
queue: stario.NewQueueCtx(stopCtx, 4, math.MaxUint32),
|
|
epoch: 1,
|
|
})
|
|
|
|
data, err := client.encodeEnvelope(newSignalAckEnvelope(2))
|
|
if err != nil {
|
|
t.Fatalf("client encodeEnvelope failed: %v", err)
|
|
}
|
|
if len(data) == 0 {
|
|
t.Fatal("client encodeEnvelope should return framed payload")
|
|
}
|
|
}
|
|
|
|
func TestServerEncodeEnvelopeUsesRuntimeQueue(t *testing.T) {
|
|
server := NewServer().(*ServerCommon)
|
|
UseLegacySecurityServer(server)
|
|
stopCtx, stopFn := context.WithCancel(context.Background())
|
|
defer stopFn()
|
|
server.setServerSessionRuntime(&serverSessionRuntime{
|
|
stopCtx: stopCtx,
|
|
stopFn: stopFn,
|
|
queue: stario.NewQueueCtx(stopCtx, 4, math.MaxUint32),
|
|
})
|
|
client := newServerCodecClientConnForTest(server)
|
|
|
|
data, err := server.encodeEnvelope(client, newSignalAckEnvelope(3))
|
|
if err != nil {
|
|
t.Fatalf("server encodeEnvelope failed: %v", err)
|
|
}
|
|
if len(data) == 0 {
|
|
t.Fatal("server encodeEnvelope should return framed payload")
|
|
}
|
|
}
|
|
|
|
func TestClientEncodeEnvelopeUsesPreservedQueueAfterTransportClear(t *testing.T) {
|
|
client := NewClient().(*ClientCommon)
|
|
UseLegacySecurityClient(client)
|
|
stopCtx, stopFn := context.WithCancel(context.Background())
|
|
defer stopFn()
|
|
queue := stario.NewQueueCtx(stopCtx, 4, math.MaxUint32)
|
|
left, right := net.Pipe()
|
|
defer left.Close()
|
|
defer right.Close()
|
|
client.setClientSessionRuntime(&clientSessionRuntime{
|
|
conn: left,
|
|
stopCtx: stopCtx,
|
|
stopFn: stopFn,
|
|
queue: queue,
|
|
epoch: 2,
|
|
})
|
|
|
|
client.clearClientSessionRuntimeTransport()
|
|
|
|
data, err := client.encodeEnvelope(newSignalAckEnvelope(4))
|
|
if err != nil {
|
|
t.Fatalf("client encodeEnvelope failed after transport clear: %v", err)
|
|
}
|
|
if len(data) == 0 {
|
|
t.Fatal("client encodeEnvelope should return framed payload after transport clear")
|
|
}
|
|
}
|