Add persistent fabric session client

2026-05-16 00:38:28 +03:00
parent b5a29d692e
commit ce6b9beb6b
4 changed files with 226 additions and 46 deletions
@@ -35,6 +35,104 @@ func TestClientFabricSessionFrameRoundTrip(t *testing.T) {
 	}
 }

+func TestClientFabricSessionPersistentRoundTrips(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()
+	session, _, err := client.OpenFabricSession(ctx, FabricSessionDialOptions{
+		Token:   "rap_fsn_persistent",
+		Timeout: time.Second,
+	})
+	if err != nil {
+		t.Fatalf("open fabric session: %v", err)
+	}
+	defer session.Close()
+
+	first, err := session.RoundTrip(ctx, fabricproto.Frame{
+		Type:     fabricproto.FramePing,
+		Sequence: 1,
+		Payload:  []byte("first"),
+	})
+	if err != nil {
+		t.Fatalf("first round trip: %v", err)
+	}
+	second, err := session.RoundTrip(ctx, fabricproto.Frame{
+		Type:     fabricproto.FramePing,
+		Sequence: 2,
+		Payload:  []byte("second"),
+	})
+	if err != nil {
+		t.Fatalf("second round trip: %v", err)
+	}
+	if first.Type != fabricproto.FramePong || first.Sequence != 1 || string(first.Payload) != "first" {
+		t.Fatalf("first response = %+v, want pong seq 1", first)
+	}
+	if second.Type != fabricproto.FramePong || second.Sequence != 2 || string(second.Payload) != "second" {
+		t.Fatalf("second response = %+v, want pong seq 2", second)
+	}
+}
+
+func TestClientFabricSessionPersistentDataAcks(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()
+	session, _, err := client.OpenFabricSession(ctx, FabricSessionDialOptions{
+		Token:   "rap_fsn_dataacks",
+		Timeout: time.Second,
+	})
+	if err != nil {
+		t.Fatalf("open fabric session: %v", err)
+	}
+	defer session.Close()
+
+	if err := session.WriteFrame(ctx, fabricproto.Frame{
+		Type:         fabricproto.FrameOpenStream,
+		StreamID:     77,
+		TrafficClass: fabricproto.TrafficClassInteractive,
+	}); err != nil {
+		t.Fatalf("open stream frame: %v", err)
+	}
+
+	first, err := session.RoundTrip(ctx, fabricproto.Frame{
+		Type:         fabricproto.FrameData,
+		StreamID:     77,
+		Sequence:     10,
+		TrafficClass: fabricproto.TrafficClassInteractive,
+		Payload:      []byte("first payload"),
+	})
+	if err != nil {
+		t.Fatalf("first data round trip: %v", err)
+	}
+	second, err := session.RoundTrip(ctx, fabricproto.Frame{
+		Type:         fabricproto.FrameData,
+		StreamID:     77,
+		Sequence:     11,
+		TrafficClass: fabricproto.TrafficClassInteractive,
+		Payload:      []byte("second payload"),
+	})
+	if err != nil {
+		t.Fatalf("second data round trip: %v", err)
+	}
+	if first.Type != fabricproto.FrameAck || first.StreamID != 77 || first.Sequence != 10 {
+		t.Fatalf("first ack = %+v, want stream 77 seq 10", first)
+	}
+	if second.Type != fabricproto.FrameAck || second.StreamID != 77 || second.Sequence != 11 {
+		t.Fatalf("second ack = %+v, want stream 77 seq 11", second)
+	}
+}
+
 func TestClientFabricSessionReportsRejectedStatus(t *testing.T) {
 	server := httptest.NewServer(Server{
 		Local:                PeerIdentity{ClusterID: "cluster-1", NodeID: "node-a"},