starainrt
09d972c7b7
- 引入 LogicalConn/TransportConn 分层，ClientConn 保留兼容适配层 - 新增 Stream、Bulk、RecordStream 三条数据面能力及对应控制路径 - 完成 transfer/file 传输内核与状态快照、诊断能力 - 补齐 reconnect、inbound dispatcher、modern psk 等基础模块 - 增加大规模回归、并发与基准测试覆盖 - 更新依赖库
86 lines
2.4 KiB
Go
86 lines
2.4 KiB
Go
package notify
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestBulkOpenDedicatedUDPRejected(t *testing.T) {
|
|
server := NewServer().(*ServerCommon)
|
|
if err := UseModernPSKServer(server, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
|
|
t.Fatalf("UseModernPSKServer failed: %v", err)
|
|
}
|
|
server.SetBulkHandler(func(info BulkAcceptInfo) error {
|
|
return nil
|
|
})
|
|
if err := server.Listen("udp", "127.0.0.1:0"); err != nil {
|
|
t.Fatalf("server Listen failed: %v", err)
|
|
}
|
|
defer func() {
|
|
_ = server.Stop()
|
|
}()
|
|
|
|
client := NewClient().(*ClientCommon)
|
|
if err := UseModernPSKClient(client, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
|
|
t.Fatalf("UseModernPSKClient failed: %v", err)
|
|
}
|
|
if err := client.Connect("udp", signalRoundTripServerAddr(server, "")); err != nil {
|
|
t.Fatalf("client Connect failed: %v", err)
|
|
}
|
|
defer func() {
|
|
_ = client.Stop()
|
|
}()
|
|
|
|
_, err := client.OpenBulk(context.Background(), BulkOpenOptions{
|
|
Range: BulkRange{
|
|
Offset: 0,
|
|
Length: 128,
|
|
},
|
|
Dedicated: true,
|
|
})
|
|
if !errors.Is(err, errBulkDedicatedStreamOnly) {
|
|
t.Fatalf("client OpenBulk dedicated over udp error = %v, want %v", err, errBulkDedicatedStreamOnly)
|
|
}
|
|
}
|
|
|
|
func TestServerOpenBulkLogicalDedicatedUDPRejected(t *testing.T) {
|
|
server := NewServer().(*ServerCommon)
|
|
if err := UseModernPSKServer(server, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
|
|
t.Fatalf("UseModernPSKServer failed: %v", err)
|
|
}
|
|
server.SetBulkHandler(func(info BulkAcceptInfo) error {
|
|
return nil
|
|
})
|
|
if err := server.Listen("udp", "127.0.0.1:0"); err != nil {
|
|
t.Fatalf("server Listen failed: %v", err)
|
|
}
|
|
defer func() {
|
|
_ = server.Stop()
|
|
}()
|
|
|
|
client := NewClient().(*ClientCommon)
|
|
if err := UseModernPSKClient(client, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
|
|
t.Fatalf("UseModernPSKClient failed: %v", err)
|
|
}
|
|
if err := client.Connect("udp", signalRoundTripServerAddr(server, "")); err != nil {
|
|
t.Fatalf("client Connect failed: %v", err)
|
|
}
|
|
defer func() {
|
|
_ = client.Stop()
|
|
}()
|
|
|
|
logical := waitForTransferControlLogicalConn(t, server, 2*time.Second)
|
|
_, err := server.OpenBulkLogical(context.Background(), logical, BulkOpenOptions{
|
|
Range: BulkRange{
|
|
Offset: 0,
|
|
Length: 128,
|
|
},
|
|
Dedicated: true,
|
|
})
|
|
if !errors.Is(err, errBulkDedicatedStreamOnly) {
|
|
t.Fatalf("server OpenBulkLogical dedicated over udp error = %v, want %v", err, errBulkDedicatedStreamOnly)
|
|
}
|
|
}
|