@@ -70,14 +70,16 @@ describe("Peer Cache Discovery", function () {
7070
7171 it ( "should discover peers from provided peer cache" , async function ( ) {
7272 const mockCache = new MockPeerCache ( ) ;
73+ const peerId1 = ( await nwaku1 . getPeerId ( ) ) . toString ( ) ;
74+ const peerId2 = ( await nwaku2 . getPeerId ( ) ) . toString ( ) ;
7375
7476 mockCache . set ( [
7577 {
76- id : ( await nwaku1 . getPeerId ( ) ) . toString ( ) ,
78+ id : peerId1 ,
7779 multiaddrs : [ ( await nwaku1 . getMultiaddrWithId ( ) ) . toString ( ) ]
7880 } ,
7981 {
80- id : ( await nwaku2 . getPeerId ( ) ) . toString ( ) ,
82+ id : peerId2 ,
8183 multiaddrs : [ ( await nwaku2 . getMultiaddrWithId ( ) ) . toString ( ) ]
8284 }
8385 ] ) ;
@@ -96,21 +98,22 @@ describe("Peer Cache Discovery", function () {
9698 const discoveredPeers = new Set < string > ( ) ;
9799 await new Promise < void > ( ( resolve ) => {
98100 waku . libp2p . addEventListener ( "peer:identify" , ( evt ) => {
99- const peerId = evt . detail . peerId ;
100- discoveredPeers . add ( peerId . toString ( ) ) ;
101+ discoveredPeers . add ( evt . detail . peerId . toString ( ) ) ;
101102
102- if ( discoveredPeers . size === 2 ) {
103+ if ( discoveredPeers . has ( peerId1 ) && discoveredPeers . has ( peerId2 ) ) {
103104 resolve ( ) ;
104105 }
105106 } ) ;
106107 } ) ;
107108
108- expect ( dialPeerSpy . callCount ) . to . equal ( 2 ) ;
109- expect ( discoveredPeers . size ) . to . equal ( 2 ) ;
109+ expect ( dialPeerSpy . callCount ) . to . be . greaterThanOrEqual ( 2 ) ;
110+ expect ( discoveredPeers ) . to . include ( peerId1 ) ;
111+ expect ( discoveredPeers ) . to . include ( peerId2 ) ;
110112 } ) ;
111113
112114 it ( "should monitor connected peers and store them into cache" , async function ( ) {
113115 const mockCache = new MockPeerCache ( ) ;
116+ const targetPeerId = ( await nwaku2 . getPeerId ( ) ) . toString ( ) ;
114117
115118 waku = await createLightNode ( {
116119 networkConfig : DefaultTestNetworkConfig ,
@@ -126,19 +129,18 @@ describe("Peer Cache Discovery", function () {
126129
127130 await new Promise < void > ( ( resolve ) => {
128131 waku . libp2p . addEventListener ( "peer:identify" , ( evt ) => {
129- const peerId = evt . detail . peerId ;
130- discoveredPeers . add ( peerId . toString ( ) ) ;
132+ discoveredPeers . add ( evt . detail . peerId . toString ( ) ) ;
131133
132- if ( discoveredPeers . size === 1 ) {
134+ if ( discoveredPeers . has ( targetPeerId ) ) {
133135 resolve ( ) ;
134136 }
135137 } ) ;
136138 } ) ;
137139
138- expect ( discoveredPeers . size ) . to . equal ( 1 ) ;
140+ expect ( discoveredPeers ) . to . include ( targetPeerId ) ;
139141
140142 const cachedPeers = mockCache . get ( ) ;
141- expect ( cachedPeers . length ) . to . equal ( 1 ) ;
142- expect ( discoveredPeers . has ( cachedPeers [ 0 ] . id ) ) . to . be . true ;
143+ const isTargetCached = cachedPeers . some ( ( p ) => p . id === targetPeerId ) ;
144+ expect ( isTargetCached ) . to . be . true ;
143145 } ) ;
144146} ) ;
0 commit comments