@@ -4,7 +4,7 @@ import { isPeerId, start, stop } from '@libp2p/interface'
44import { streamPair } from '@libp2p/interface-compliance-tests/mocks'
55import { defaultLogger } from '@libp2p/logger'
66import { peerIdFromString } from '@libp2p/peer-id'
7- import { multiaddr , type Multiaddr , type Protocol } from '@multiformats/multiaddr'
7+ import { multiaddr , type Multiaddr } from '@multiformats/multiaddr'
88import { expect } from 'aegir/chai'
99import { duplexPair } from 'it-pair/duplex'
1010import { type Libp2p } from 'libp2p'
@@ -45,20 +45,24 @@ describe('whatwg-fetch', () => {
4545
4646 let serverCB : StreamHandler
4747 const serverCBRegistered = pDefer ( )
48- serverComponents . registrar . handle . callsFake ( async ( protocol : Protocol , cb : StreamHandler ) => {
48+ serverComponents . registrar . handle . callsFake ( async ( protocol : string , cb : StreamHandler ) => {
4949 serverCB = cb
5050 serverCBRegistered . resolve ( )
5151 } )
5252
5353 const conn = stubInterface < Connection > ( )
54- conn . newStream . callsFake ( async ( protos : Protocol [ ] , options ?: any ) => {
54+ conn . newStream . callsFake ( async ( protos : string | string [ ] , options ?: any ) => {
5555 const duplexes = duplexPair < any > ( )
5656 const streams = streamPair ( { duplex : duplexes [ 0 ] } , { duplex : duplexes [ 1 ] } )
5757 serverCB ( { stream : streams [ 0 ] , connection : conn } )
5858 return streams [ 1 ]
5959 } )
6060
61- clientComponents . connectionManager . openConnection . callsFake ( async ( peer : PeerId | Multiaddr , options ?: any ) => {
61+ clientComponents . connectionManager . openConnection . callsFake ( async ( peer : PeerId | Multiaddr | Multiaddr [ ] , options ?: any ) => {
62+ if ( Array . isArray ( peer ) ) {
63+ peer = peer [ 0 ]
64+ }
65+
6266 if ( isPeerId ( peer ) ? peer . equals ( serverPeerID ) : peer . getPeerId ( ) === serverMultiaddr . getPeerId ( ) ) {
6367 await serverCBRegistered . promise
6468 return conn
0 commit comments