chore: fix tests

libp2p · Sep 25, 2024 · 3f9529c · 3f9529c
1 parent 9fbd889
commit 3f9529c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/packages/libp2p/test/connection-manager/direct.spec.ts b/packages/libp2p/test/connection-manager/direct.spec.ts
@@ -490,7 +490,7 @@ describe('libp2p.dialer (direct, WebSockets)', () => {
 
     await expect(libp2p.dial(multiaddr(`/ip4/127.0.0.1/tcp/1234/ws/p2p/${libp2p.peerId.toString()}`)))
       .to.eventually.be.rejected()
-      .and.to.have.property('name', 'DialError')
+      .and.to.have.property('name', 'InvalidPeerIdError')
   })
 
   it('should limit the maximum dial queue size', async () => {

diff --git a/packages/libp2p/test/connection-manager/index.spec.ts b/packages/libp2p/test/connection-manager/index.spec.ts
@@ -391,7 +391,9 @@ describe('Connection Manager', () => {
     })
     await connectionManager.start()
 
-    sinon.stub(connectionManager.dialQueue, 'dial').resolves(stubInterface<Connection>())
+    sinon.stub(connectionManager.dialQueue, 'dial').resolves(stubInterface<Connection>({
+      remotePeer: peerIdFromPrivateKey(await generateKeyPair('Ed25519'))
+    }))
 
     // max out the connection limit
     await connectionManager.openConnection(peerIdFromPrivateKey(await generateKeyPair('Ed25519')))
@@ -450,7 +452,9 @@ describe('Connection Manager', () => {
     })
     await connectionManager.start()
 
-    sinon.stub(connectionManager.dialQueue, 'dial').resolves(stubInterface<Connection>())
+    sinon.stub(connectionManager.dialQueue, 'dial').resolves(stubInterface<Connection>({
+      remotePeer: peerIdFromPrivateKey(await generateKeyPair('Ed25519'))
+    }))
 
     // max out the connection limit
     await connectionManager.openConnection(peerIdFromPrivateKey(await generateKeyPair('Ed25519')))
@@ -477,7 +481,9 @@ describe('Connection Manager', () => {
     })
     await connectionManager.start()
 
-    sinon.stub(connectionManager.dialQueue, 'dial').resolves(stubInterface<Connection>())
+    sinon.stub(connectionManager.dialQueue, 'dial').resolves(stubInterface<Connection>({
+      remotePeer: peerIdFromPrivateKey(await generateKeyPair('Ed25519'))
+    }))
 
     // start the upgrade
     const maConn1 = mockMultiaddrConnection({
@@ -525,7 +531,9 @@ describe('Connection Manager', () => {
         bytes: 100n
       }
     })
-    const newConnection = stubInterface<Connection>()
+    const newConnection = stubInterface<Connection>({
+      remotePeer: targetPeer
+    })
 
     sinon.stub(connectionManager.dialQueue, 'dial')
       .withArgs(addr)