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)
|
||
|
|
}
|
||
|
|
}
|