fix: emit peer:connect events from mock connection manager (#399)

Updates mock to behave similarly to the actual connection manager.
libp2p · May 10, 2023 · 836dcf3 · 836dcf3
1 parent 283e81c
commit 836dcf3
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/packages/interface-mocks/src/connection-manager.ts b/packages/interface-mocks/src/connection-manager.ts
@@ -140,6 +140,8 @@ class MockConnectionManager implements ConnectionManager, Startable {
       }
     }
 
+    this.components.events.safeDispatchEvent('peer:connect', { detail: componentsB.peerId })
+
     componentsB.events.safeDispatchEvent('connection:open', {
       detail: bToA
     })
@@ -150,6 +152,8 @@ class MockConnectionManager implements ConnectionManager, Startable {
       }
     }
 
+    componentsB.events.safeDispatchEvent('peer:connect', { detail: this.components.peerId })
+
     return aToB
   }
 
@@ -178,7 +182,15 @@ class MockConnectionManager implements ConnectionManager, Startable {
 
     this.connections = this.connections.filter(c => !c.remotePeer.equals(peerId))
 
+    if (this.connections.filter(c => !c.remotePeer.equals(peerId)).length === 0) {
+      componentsB.events.safeDispatchEvent('peer:disconnect', { detail: peerId })
+    }
+
     await componentsB.connectionManager?.closeConnections(this.components.peerId)
+
+    if (componentsB.connectionManager?.getConnectionsMap().get(this.components.peerId) == null) {
+      componentsB.events.safeDispatchEvent('peer:disconnect', { detail: this.components.peerId })
+    }
   }
 
   async acceptIncomingConnection (): Promise<boolean> {