74 lines
2.2 KiB
Go
74 lines
2.2 KiB
Go
package mesh
|
|
|
|
import (
|
|
"context"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/example/remote-access-platform/agents/rap-node-agent/internal/fabricproto"
|
|
)
|
|
|
|
func TestClientFabricSessionFrameRoundTrip(t *testing.T) {
|
|
server := httptest.NewServer(Server{
|
|
Local: PeerIdentity{ClusterID: "cluster-1", NodeID: "node-a"},
|
|
FabricSessionEnabled: true,
|
|
}.Handler())
|
|
defer server.Close()
|
|
|
|
client := NewClient(server.URL)
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
|
defer cancel()
|
|
response, err := client.SendFabricSessionFrame(ctx, FabricSessionDialOptions{
|
|
Token: "rap_fsn_clienttest",
|
|
Timeout: time.Second,
|
|
}, fabricproto.Frame{
|
|
Type: fabricproto.FramePing,
|
|
Sequence: 12,
|
|
Payload: []byte("probe"),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("send fabric session frame: %v", err)
|
|
}
|
|
if response.Type != fabricproto.FramePong || response.Sequence != 12 || string(response.Payload) != "probe" {
|
|
t.Fatalf("response = %+v, want pong seq 12", response)
|
|
}
|
|
}
|
|
|
|
func TestClientFabricSessionReportsRejectedStatus(t *testing.T) {
|
|
server := httptest.NewServer(Server{
|
|
Local: PeerIdentity{ClusterID: "cluster-1", NodeID: "node-a"},
|
|
FabricSessionEnabled: true,
|
|
}.Handler())
|
|
defer server.Close()
|
|
|
|
client := NewClient(server.URL)
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
|
defer cancel()
|
|
_, err := client.SendFabricSessionFrame(ctx, FabricSessionDialOptions{}, fabricproto.Frame{Type: fabricproto.FramePing})
|
|
if err == nil {
|
|
t.Fatal("send fabric session without token unexpectedly succeeded")
|
|
}
|
|
}
|
|
|
|
func TestClientFabricSessionWebSocketURL(t *testing.T) {
|
|
cases := []struct {
|
|
base string
|
|
want string
|
|
}{
|
|
{base: "http://node.example", want: "ws://node.example/mesh/v1/fabric/session/ws"},
|
|
{base: "https://node.example/base/", want: "wss://node.example/base/mesh/v1/fabric/session/ws"},
|
|
{base: "ws://node.example", want: "ws://node.example/mesh/v1/fabric/session/ws"},
|
|
}
|
|
for _, tc := range cases {
|
|
client := NewClient(tc.base)
|
|
got, err := client.fabricSessionWebSocketURL()
|
|
if err != nil {
|
|
t.Fatalf("fabricSessionWebSocketURL(%q): %v", tc.base, err)
|
|
}
|
|
if got != tc.want {
|
|
t.Fatalf("fabricSessionWebSocketURL(%q) = %q, want %q", tc.base, got, tc.want)
|
|
}
|
|
}
|
|
}
|