Reopen closed fabric peer sessions
This commit is contained in:
@@ -117,6 +117,54 @@ func TestFabricSessionPeerManagerClosePeerReopens(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestFabricSessionPeerManagerReopensClosedPump(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()
|
||||
|
||||
manager := NewFabricSessionPeerManager()
|
||||
defer manager.Close()
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
||||
defer cancel()
|
||||
target := FabricSessionPeerTarget{
|
||||
PeerID: "node-a",
|
||||
BaseURL: server.URL,
|
||||
Options: FabricSessionDialOptions{
|
||||
Token: "rap_fsn_manager_closed",
|
||||
Timeout: time.Second,
|
||||
},
|
||||
}
|
||||
|
||||
first, err := manager.Get(ctx, target)
|
||||
if err != nil {
|
||||
t.Fatalf("first get: %v", err)
|
||||
}
|
||||
if err := first.Close(); err != nil {
|
||||
t.Fatalf("close first pump: %v", err)
|
||||
}
|
||||
if !first.Closed() {
|
||||
t.Fatal("first pump should report closed")
|
||||
}
|
||||
second, err := manager.Get(ctx, target)
|
||||
if err != nil {
|
||||
t.Fatalf("second get: %v", err)
|
||||
}
|
||||
if first == second {
|
||||
t.Fatal("manager reused closed pump")
|
||||
}
|
||||
if opened != 2 {
|
||||
t.Fatalf("opened sessions = %d, want 2", opened)
|
||||
}
|
||||
}
|
||||
|
||||
func TestFabricSessionPeerManagerRejectsIncompleteTarget(t *testing.T) {
|
||||
manager := NewFabricSessionPeerManager()
|
||||
_, err := manager.Get(context.Background(), FabricSessionPeerTarget{PeerID: "node-a"})
|
||||
|
||||
Reference in New Issue
Block a user