Add fabric session websocket client smoke API
This commit is contained in:
@@ -0,0 +1,73 @@
|
||||
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)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user