140 lines
3.8 KiB
Go
140 lines
3.8 KiB
Go
package mesh
|
|
|
|
import (
|
|
"context"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/example/remote-access-platform/agents/rap-node-agent/internal/fabricproto"
|
|
)
|
|
|
|
func TestWebSocketFabricTransportConnectsAndReusesSession(t *testing.T) {
|
|
var opened int
|
|
server := httptest.NewServer(Server{
|
|
Local: PeerIdentity{ClusterID: "cluster-1", NodeID: "node-a"},
|
|
FabricSessionEnabled: true,
|
|
FabricSessionLogger: func(entry FabricSessionEventLogEntry) {
|
|
if entry.Event == "fabric_session_websocket_opened" {
|
|
opened++
|
|
}
|
|
},
|
|
}.Handler())
|
|
defer server.Close()
|
|
|
|
transport := NewWebSocketFabricTransport(nil)
|
|
defer transport.Close()
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
|
defer cancel()
|
|
target := FabricTransportTarget{
|
|
PeerID: "node-a",
|
|
Endpoint: server.URL,
|
|
Token: "rap_fsn_transport",
|
|
Timeout: time.Second,
|
|
OutboundBuffer: 4,
|
|
InboundBuffer: 4,
|
|
ErrorBuffer: 4,
|
|
}
|
|
|
|
first, err := transport.Connect(ctx, target)
|
|
if err != nil {
|
|
t.Fatalf("first connect: %v", err)
|
|
}
|
|
second, err := transport.Connect(ctx, target)
|
|
if err != nil {
|
|
t.Fatalf("second connect: %v", err)
|
|
}
|
|
if first != second {
|
|
t.Fatal("transport did not reuse session")
|
|
}
|
|
if opened != 1 {
|
|
t.Fatalf("opened = %d, want 1", opened)
|
|
}
|
|
if err := first.Send(ctx, fabricproto.Frame{Type: fabricproto.FramePing, Sequence: 1, Payload: []byte("transport")}); err != nil {
|
|
t.Fatalf("send ping: %v", err)
|
|
}
|
|
select {
|
|
case frame := <-first.Frames():
|
|
if frame.Type != fabricproto.FramePong || frame.Sequence != 1 || string(frame.Payload) != "transport" {
|
|
t.Fatalf("frame = %+v", frame)
|
|
}
|
|
case err := <-first.Errors():
|
|
t.Fatalf("session error: %v", err)
|
|
case <-ctx.Done():
|
|
t.Fatal(ctx.Err())
|
|
}
|
|
}
|
|
|
|
func TestWebSocketFabricTransportReopensClosedSession(t *testing.T) {
|
|
var opened int
|
|
server := httptest.NewServer(Server{
|
|
Local: PeerIdentity{ClusterID: "cluster-1", NodeID: "node-a"},
|
|
FabricSessionEnabled: true,
|
|
FabricSessionLogger: func(entry FabricSessionEventLogEntry) {
|
|
if entry.Event == "fabric_session_websocket_opened" {
|
|
opened++
|
|
}
|
|
},
|
|
}.Handler())
|
|
defer server.Close()
|
|
|
|
transport := NewWebSocketFabricTransport(nil)
|
|
defer transport.Close()
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
|
defer cancel()
|
|
target := FabricTransportTarget{
|
|
PeerID: "node-a",
|
|
Endpoint: server.URL,
|
|
Token: "rap_fsn_transport_reopen",
|
|
Timeout: time.Second,
|
|
}
|
|
|
|
first, err := transport.Connect(ctx, target)
|
|
if err != nil {
|
|
t.Fatalf("first connect: %v", err)
|
|
}
|
|
if err := first.Close(); err != nil {
|
|
t.Fatalf("close first session: %v", err)
|
|
}
|
|
second, err := transport.Connect(ctx, target)
|
|
if err != nil {
|
|
t.Fatalf("second connect: %v", err)
|
|
}
|
|
if first == second {
|
|
t.Fatal("transport reused closed session")
|
|
}
|
|
if opened != 2 {
|
|
t.Fatalf("opened = %d, want 2", opened)
|
|
}
|
|
}
|
|
|
|
func TestFabricTransportForTargetSelectsQUICByScheme(t *testing.T) {
|
|
transport, target, err := FabricTransportForTarget(FabricTransportTarget{
|
|
Endpoint: "quic://127.0.0.1:4433",
|
|
}, nil, nil)
|
|
if err != nil {
|
|
t.Fatalf("select transport: %v", err)
|
|
}
|
|
if _, ok := transport.(*QUICFabricTransport); !ok {
|
|
t.Fatalf("transport = %T, want QUIC", transport)
|
|
}
|
|
if target.Endpoint != "127.0.0.1:4433" {
|
|
t.Fatalf("endpoint = %q", target.Endpoint)
|
|
}
|
|
}
|
|
|
|
func TestFabricTransportForTargetSelectsWebSocketByDefault(t *testing.T) {
|
|
transport, target, err := FabricTransportForTarget(FabricTransportTarget{
|
|
Endpoint: "https://node.example",
|
|
}, nil, nil)
|
|
if err != nil {
|
|
t.Fatalf("select transport: %v", err)
|
|
}
|
|
if _, ok := transport.(*WebSocketFabricTransport); !ok {
|
|
t.Fatalf("transport = %T, want websocket", transport)
|
|
}
|
|
if target.Endpoint != "https://node.example" {
|
|
t.Fatalf("endpoint = %q", target.Endpoint)
|
|
}
|
|
}
|