@@ -4,7 +4,7 @@ import { isPeerId, start, stop } from '@libp2p/interface'
4
4
import { streamPair } from '@libp2p/interface-compliance-tests/mocks'
5
5
import { defaultLogger } from '@libp2p/logger'
6
6
import { peerIdFromString } from '@libp2p/peer-id'
7
- import { multiaddr , type Multiaddr , type Protocol } from '@multiformats/multiaddr'
7
+ import { multiaddr , type Multiaddr } from '@multiformats/multiaddr'
8
8
import { expect } from 'aegir/chai'
9
9
import { duplexPair } from 'it-pair/duplex'
10
10
import { type Libp2p } from 'libp2p'
@@ -45,20 +45,24 @@ describe('whatwg-fetch', () => {
45
45
46
46
let serverCB : StreamHandler
47
47
const serverCBRegistered = pDefer ( )
48
- serverComponents . registrar . handle . callsFake ( async ( protocol : Protocol , cb : StreamHandler ) => {
48
+ serverComponents . registrar . handle . callsFake ( async ( protocol : string , cb : StreamHandler ) => {
49
49
serverCB = cb
50
50
serverCBRegistered . resolve ( )
51
51
} )
52
52
53
53
const conn = stubInterface < Connection > ( )
54
- conn . newStream . callsFake ( async ( protos : Protocol [ ] , options ?: any ) => {
54
+ conn . newStream . callsFake ( async ( protos : string | string [ ] , options ?: any ) => {
55
55
const duplexes = duplexPair < any > ( )
56
56
const streams = streamPair ( { duplex : duplexes [ 0 ] } , { duplex : duplexes [ 1 ] } )
57
57
serverCB ( { stream : streams [ 0 ] , connection : conn } )
58
58
return streams [ 1 ]
59
59
} )
60
60
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
+
62
66
if ( isPeerId ( peer ) ? peer . equals ( serverPeerID ) : peer . getPeerId ( ) === serverMultiaddr . getPeerId ( ) ) {
63
67
await serverCBRegistered . promise
64
68
return conn
0 commit comments