@libp2p/interface-compliance-tests 1.1.17 → 1.1.20

package/src/pubsub/two-nodes.ts

@@ -11,11 +11,12 @@ import { CustomEvent } from '@libp2p/interfaces'
 import { waitForSubscriptionUpdate } from './utils.js'
 import type { TestSetup } from '../index.js'
 import type { Message } from '@libp2p/interfaces/pubsub'
-import type { EventMap, PubSubArgs } from './index.js'
+import type { PubSubArgs } from './index.js'
 import type { PeerId } from '@libp2p/interfaces/peer-id'
 import type { Registrar } from '@libp2p/interfaces/registrar'
 import type { PubSubBaseProtocol } from '@libp2p/pubsub'
 import { Components } from '@libp2p/interfaces/components'
+import { start, stop } from '../index.js'
 
 const topic = 'foo'
 
@@ -23,10 +24,10 @@ function shouldNotHappen () {
   expect.fail()
 }
 
-export default (common: TestSetup<PubSubBaseProtocol<EventMap>, PubSubArgs>) => {
+export default (common: TestSetup<PubSubBaseProtocol, PubSubArgs>) => {
   describe('pubsub with two nodes', () => {
-    let psA: PubSubBaseProtocol<EventMap>
-    let psB: PubSubBaseProtocol<EventMap>
+    let psA: PubSubBaseProtocol
+    let psB: PubSubBaseProtocol
     let peerIdA: PeerId
     let peerIdB: PeerId
     let registrarA: Registrar
@@ -60,8 +61,7 @@ export default (common: TestSetup<PubSubBaseProtocol<EventMap>, PubSubArgs>) =>
       })
 
       // Start pubsub and connect nodes
-      await psA.start()
-      await psB.start()
+      await start(psA, psB)
 
       expect(psA.getPeers()).to.be.empty()
       expect(psB.getPeers()).to.be.empty()
@@ -81,8 +81,7 @@ export default (common: TestSetup<PubSubBaseProtocol<EventMap>, PubSubArgs>) =>
     afterEach(async () => {
       sinon.restore()
 
-      await psA.stop()
-      await psB.stop()
+      await stop(psA, psB)
 
       await common.teardown()
     })